Skip to content

Commit 681ffbe

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 8572c1f commit 681ffbe

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

transformer_engine/common/fused_attn/fused_attn.cpp

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -178,9 +178,8 @@ NVTE_Fused_Attn_Backend nvte_get_fused_attn_backend(
178178
max_seqlen_q <= max_seqlen_kv && dropout == 0.0) ||
179179
((cudnn_runtime_version >= 90500) &&
180180
(attn_mask_type == NVTE_Mask_Type::NVTE_CAUSAL_BOTTOM_RIGHT_MASK ||
181-
attn_mask_type == NVTE_Mask_Type::NVTE_PADDING_CAUSAL_BOTTOM_RIGHT_MASK) &&
182-
bias_type == NVTE_Bias_Type::NVTE_NO_BIAS &&
183-
dropout == 0.0)) &&
181+
attn_mask_type == NVTE_Mask_Type::NVTE_PADDING_CAUSAL_BOTTOM_RIGHT_MASK) &&
182+
bias_type == NVTE_Bias_Type::NVTE_NO_BIAS && dropout == 0.0)) &&
184183
// bias + mask combination
185184
(!(cudnn_runtime_version >= 8906 &&
186185
(attn_mask_type == NVTE_Mask_Type::NVTE_PADDING_MASK ||

0 commit comments

Comments
 (0)