From ca6e2e9062aabda722b3b7675ce1a2fefe2562ad Mon Sep 17 00:00:00 2001 From: Harshit Monish <143435143+hmonishN@users.noreply.github.com> Date: Tue, 11 Jun 2024 11:53:27 -0700 Subject: [PATCH] Update test-pax.sh --- .github/container/test-pax.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/container/test-pax.sh b/.github/container/test-pax.sh index 6651364f7..0dc3ef9e3 100755 --- 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"