Skip to content

Commit 166d220

Browse files
author
zhouwg
committed
ggml-qnn: AI-assisted ggml_qnn_mul_mat_4d by Grok 3 --- narrow down to make AI happy
1 parent 8ee7295 commit 166d220

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

tests/ggml-qnn-ut.cpp

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -439,10 +439,13 @@ int main(int argc, char * argv[]) {
439439
//src0 = ggml_new_tensor_3d(ctx, qtype, 128, 64, 8);
440440
//src1 = ggml_new_tensor_3d(ctx, GGML_TYPE_F32, 128, 2, 8);
441441
//verify 4D matrix
442+
#if 1 //failure
442443
src0 = ggml_new_tensor_4d(ctx, GGML_TYPE_F32, 256, 16, 3, 2);
443444
src1 = ggml_new_tensor_4d(ctx, GGML_TYPE_F32, 256, 1, 6, 4);
444-
//src0 = ggml_new_tensor_4d(ctx, GGML_TYPE_F32, 256, 16, 3, 2);
445-
//src1 = ggml_new_tensor_4d(ctx, GGML_TYPE_F32, 256, 16, 3, 2);
445+
#else //ok
446+
src0 = ggml_new_tensor_4d(ctx, GGML_TYPE_F32, 256, 16, 3, 2);
447+
src1 = ggml_new_tensor_4d(ctx, GGML_TYPE_F32, 256, 16, 3, 2);
448+
#endif
446449
}
447450

448451
ggml_set_input(src0);

0 commit comments

Comments
 (0)