Update test-pax.sh

NVIDIA · Jun 11, 2024 · ca6e2e9 · ca6e2e9
1 parent 0ef811a
commit ca6e2e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/container/test-pax.sh b/.github/container/test-pax.sh
@@ -57,7 +57,7 @@ NVTE_FUSED_ATTN=1
 DROPOUT=0
 EVALUATE=0
 ADDITIONAL_ARGS=""
-ENABLE_FMHA=${ENABLE_FMHA:-0}
+ENABLE_FMHA=${ENABLE_FMHA:-1}
 SAVE_HLO=${SAVE_HLO:-0}
 
 eval set -- "$args"