diff options
author | Georgi Gerganov <ggerganov@gmail.com> | 2024-05-14 19:09:30 +0300 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-05-14 19:09:30 +0300 |
commit | e8a7fd4fb06d82f663850c21fcf86c0fb98ad9b4 (patch) | |
tree | cf3e07d88d47f14717ae7ae6923b653950d87911 /ggml.h | |
parent | a5e3fde8578d54b98d941344a4da150669af200d (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.h | 3 |
1 files changed, 2 insertions, 1 deletions
@@ -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); |