From 62aa537628f64e45de1f831ad4a4231790b815ec Mon Sep 17 00:00:00 2001 From: andrewwhitecdw Date: Wed, 12 Aug 2026 15:03:14 -0500 Subject: [PATCH] fix: add missing NVTE_BHSD_BHSD_BHSD to layout to_string Squashed to single commit for review. Original PR: https://github.com/andrewwhitecdw/TransformerEngine/pull/17 --- transformer_engine/common/fused_attn/fused_attn.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/transformer_engine/common/fused_attn/fused_attn.cpp b/transformer_engine/common/fused_attn/fused_attn.cpp index 1ac7a36383..3a61c3c787 100644 --- a/transformer_engine/common/fused_attn/fused_attn.cpp +++ b/transformer_engine/common/fused_attn/fused_attn.cpp @@ -68,6 +68,8 @@ std::string to_string(NVTE_QKV_Layout layout) { return "NVTE_Paged_KV_THD_BSHD_BSHD"; case NVTE_Paged_KV_THD_SBHD_SBHD: return "NVTE_Paged_KV_THD_SBHD_SBHD"; + case NVTE_BHSD_BHSD_BHSD: + return "NVTE_BHSD_BHSD_BHSD"; default: return "UNKNOWN_QKV_LAYOUT(" + std::to_string(static_cast(layout)) + ")"; } @@ -89,6 +91,8 @@ std::string to_string(NVTE_QKV_Format format) { return "NVTE_THD_2BSHD"; case NVTE_THD_2SBHD: return "NVTE_THD_2SBHD"; + case NVTE_BHSD: + return "NVTE_BHSD"; default: return "UNKNOWN_QKV_FORMAT(" + std::to_string(static_cast(format)) + ")"; }