fix and/or logic in get_backend

Signed-off-by: Charlene Yang <[email protected]>
NVIDIA · Nov 14, 2023 · a30f49e · a30f49e
1 parent d05c7ef
commit a30f49e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transformer_engine/common/fused_attn/fused_attn.cpp b/transformer_engine/common/fused_attn/fused_attn.cpp
@@ -120,7 +120,7 @@ NVTE_Fused_Attn_Backend nvte_get_fused_attn_backend(
       flag_m512 = true;
     }
     if ((cudnn_runtime_version >= 8903 && sm_arch_ >= 80)
-            && (cudnn_runtime_version < 8903 && sm_arch_ == 80 || sm_arch_ == 90)
+            && (cudnn_runtime_version < 8903 && (sm_arch_ == 80 || sm_arch_ == 90))
             && (max_seqlen_q % 64 == 0)
             && (max_seqlen_kv % 64 == 0)
             && (num_attn_heads == num_gqa_groups)