diff options
| author | Georgi Gerganov <ggerganov@gmail.com> | 2024-05-29 20:17:31 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-29 20:17:31 +0300 |
| commit | fb76ec31a9914b7761c1727303ab30380fd4f05c (patch) | |
| tree | a0bcc5041d8cf3373ad853bea4befd0b96e098d4 /ggml-kompute.cpp | |
| parent | cce3dcffc5695bd24835f04e6080070a2a119873 (diff) | |
ggml : fix YARN + add tests + add asserts (#7617)
* tests : add rope tests
ggml-ci
* ggml : fixes (hopefully)
ggml-ci
* tests : add non-cont tests
ggml-ci
* cuda : add asserts for rope/norm + fix DS2
ggml-ci
* ggml : assert contiguousness
* tests : reduce RoPE tests
ggml-ci
Diffstat (limited to 'ggml-kompute.cpp')
| -rw-r--r-- | ggml-kompute.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/ggml-kompute.cpp b/ggml-kompute.cpp index 6c6058b2..ed59d2be 100644 --- a/ggml-kompute.cpp +++ b/ggml-kompute.cpp @@ -1597,7 +1597,9 @@ static void ggml_vk_graph_compute(struct ggml_kompute_context * ctx, struct ggml { GGML_ASSERT(ne00 == ne10); - // TODO: assert that dim2 and dim3 are contiguous + ggml_is_contiguous_2(src0); + ggml_is_contiguous_2(src1); + GGML_ASSERT(ne12 % ne02 == 0); GGML_ASSERT(ne13 % ne03 == 0); |