Update grouped_multihead_attention_backward.cpp

a2dce6fe · Junhao Zhang · GitHub · 0286b9bf · a2dce6fe
Unverified Commit a2dce6fe authored Jul 11, 2023 by Junhao Zhang Committed by GitHub Jul 11, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_backward.cpp ...ale_softmax_gemm/grouped_multihead_attention_backward.cpp +1 -1

No files found.
--- a/example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_backward.cpp
+++ b/example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_backward.cpp
@@ -23,7 +23,7 @@ Kernel outputs:

 */

-#define USING_MASK 0
+#define USING_MASK 1
 #define DIM 64 // DIM should be a multiple of 8.

 #include <iostream>