From ce42aaf85240b439e9f00c37bed48223ca97f4f9 Mon Sep 17 00:00:00 2001 From: Andrew White Date: Mon, 17 Aug 2026 19:17:05 -0500 Subject: [PATCH] fix: add missing NVTE_BHSD to qkv format to_string Squashed to single commit for review. Original PR: https://github.com/andrewwhitecdw/TransformerEngine/pull/16 Signed-off-by: Andrew White --- transformer_engine/common/fused_attn/fused_attn.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/transformer_engine/common/fused_attn/fused_attn.cpp b/transformer_engine/common/fused_attn/fused_attn.cpp index 1ac7a36383..307b3c5ec6 100644 --- a/transformer_engine/common/fused_attn/fused_attn.cpp +++ b/transformer_engine/common/fused_attn/fused_attn.cpp @@ -79,6 +79,8 @@ std::string to_string(NVTE_QKV_Format format) { return "NVTE_SBHD"; case NVTE_BSHD: return "NVTE_BSHD"; + case NVTE_BHSD: + return "NVTE_BHSD"; case NVTE_THD: return "NVTE_THD"; case NVTE_BSHD_2SBHD: