Update grouped_multihead_attention_backward_v2.cpp

f837a870 · Junhao Zhang · GitHub · f5c70413 · f837a870
Unverified Commit f837a870 authored Jul 12, 2023 by Junhao Zhang Committed by GitHub Jul 12, 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_v2.cpp ..._softmax_gemm/grouped_multihead_attention_backward_v2.cpp +1 -1

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

 */

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

 #include <iostream>