Skip to content

Commit f6c4c4c

Browse files
committed
tests : add FA test with non-padded KV size
1 parent 28baac9 commit f6c4c4c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test-backend-ops.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6535,7 +6535,7 @@ static std::vector<std::unique_ptr<test_case>> make_test_cases_eval() {
65356535
if (hsk > 64 && nr3 > 1) continue; // skip broadcast for large head sizes
65366536
for (int nr2 : { 1, 4, 16 }) {
65376537
if (nr2 == 16 && hsk != 128) continue;
6538-
for (int kv : { 512, 1024, }) {
6538+
for (int kv : { 113, 512, 1024, }) {
65396539
if (nr2 != 1 && kv != 512) continue;
65406540
for (int nb : { 1, 3, 32, 35, }) {
65416541
for (ggml_prec prec : {GGML_PREC_F32, GGML_PREC_DEFAULT}) {

0 commit comments

Comments
 (0)