From f59b58bb50100197de2341262b0b95f33ec3d897 Mon Sep 17 00:00:00 2001 From: andrewwhitecdw Date: Wed, 12 Aug 2026 15:03:11 -0500 Subject: [PATCH] fix: simplify redundant softmax version condition Squashed to single commit for review. Original PR: https://github.com/andrewwhitecdw/TransformerEngine/pull/15 --- transformer_engine/common/fused_attn/fused_attn.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/transformer_engine/common/fused_attn/fused_attn.cpp b/transformer_engine/common/fused_attn/fused_attn.cpp index 1ac7a36383..6f11af9dbb 100644 --- a/transformer_engine/common/fused_attn/fused_attn.cpp +++ b/transformer_engine/common/fused_attn/fused_attn.cpp @@ -471,8 +471,7 @@ NVTE_Fused_Attn_Backend nvte_get_fused_attn_backend( // pre-9.13.1: vanilla // 9.13.1+: vanilla, off-by-one, learnable (cudnn_runtime_version >= 91301 || - (cudnn_runtime_version < 91301 && - softmax_type == NVTE_Softmax_Type::NVTE_VANILLA_SOFTMAX)) && + softmax_type == NVTE_Softmax_Type::NVTE_VANILLA_SOFTMAX) && // max_logit // pre-9.21: no (the composite softmax node rejects the Stats + Max output combination) // 9.21+: yes (Stats + Max via the unified softmax node)