hd128

210aa44d · danyao12 · 9809f5d4 · 210aa44d
Commit 210aa44d authored Aug 29, 2023 by danyao12
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_backward_v3_protro.cpp ...x_gemm/batched_multihead_attention_backward_v3_protro.cpp +1 -1

No files found.
--- a/example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_backward_v3_protro.cpp
+++ b/example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_backward_v3_protro.cpp
@@ -25,7 +25,7 @@ Kernel outputs:
 #define PRINT_HOST 0
 #define USING_MASK 0
-#define DIM 64 // DIM should be a multiple of 8.
+#define DIM 128 // DIM should be a multiple of 8.
 #include <iostream>
 #include <numeric>