summaryrefslogtreecommitdiff
path: root/ggml.h
diff options
context:
space:
mode:
authorGeorgi Gerganov <ggerganov@gmail.com>2024-05-14 19:09:30 +0300
committerGitHub <noreply@github.com>2024-05-14 19:09:30 +0300
commite8a7fd4fb06d82f663850c21fcf86c0fb98ad9b4 (patch)
treecf3e07d88d47f14717ae7ae6923b653950d87911 /ggml.h
parenta5e3fde8578d54b98d941344a4da150669af200d (diff)
metal : support FA without mask + add asserts (#7278)
* ggml : fa without mask + add asserts ggml-ci * metal : support non-contiguous KV ggml-ci
Diffstat (limited to 'ggml.h')
-rw-r--r--ggml.h3
1 files changed, 2 insertions, 1 deletions
diff --git a/ggml.h b/ggml.h
index 3fe95ed5..25f4f73a 100644
--- a/ggml.h
+++ b/ggml.h
@@ -766,7 +766,8 @@ extern "C" {
GGML_API bool ggml_is_3d (const struct ggml_tensor * tensor);
GGML_API int ggml_n_dims (const struct ggml_tensor * tensor); // returns 1 for scalars
- GGML_API bool ggml_are_same_shape(const struct ggml_tensor * t0, const struct ggml_tensor * t1);
+ GGML_API bool ggml_are_same_shape (const struct ggml_tensor * t0, const struct ggml_tensor * t1);
+ GGML_API bool ggml_are_same_stride(const struct ggml_tensor * t0, const struct ggml_tensor * t1);
// use this to compute the memory overhead of a tensor
GGML_API size_t ggml_tensor_overhead(void);