fix: add missing NVTE_BHSD to qkv format to_string

[andrewwhitecdw]

This PR addresses the following issue in transformer_engine/common/fused_attn/fused_attn.cpp: add missing NVTE_BHSD to qkv format to_string.

Changes

• transformer_engine/common/fused_attn/fused_attn.cpp: add missing NVTE_BHSD to qkv format to_string.

Details

--- a/transformer_engine/common/fused_attn/fused_attn.cpp
+++ b/transformer_engine/common/fused_attn/fused_attn.cpp
@@ -1,6 +1,8 @@
-    case NVTE_BSHD:
-      return "NVTE_BSHD";
-    case NVTE_THD:
-      return "NVTE_THD";
-    case NVTE_BSHD_2SBHD:
-      return "NVTE_BSHD_2SBHD";
+    case NVTE_BSHD:
+      return "NVTE_BSHD";
+    case NVTE_BHSD:
+      return "NVTE_BHSD";
+    case NVTE_THD:
+      return "NVTE_THD";
+    case NVTE_BSHD_2SBHD:
+      return "NVTE_BSHD_2SBHD";

Tests

• tests/test_qkv_format_to_string.cpp

diff --git a/tests/test_qkv_format_to_string.cpp b/tests/test_qkv_format_to_string.cpp
new file mode 100644
--- /dev/null
+++ b/tests/test_qkv_format_to_string.cpp
@@ -0,0 +1,15 @@
+/*************************************************************************
+ * Copyright (c) 2022-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+ *
+ * See LICENSE for license information.
+ ************************************************************************/
+
+#include <gtest/gtest.h>
+#include <string>
+
+#include "transformer_engine/fused_attn.h"
+
+TEST(QKVFormatToString, BHSD) {
+  EXPECT_EQ(transformer_engine::to_string(NVTE_QKV_Format::NVTE_BHSD), "NVTE_BHSD");
+}
+
+TEST(QKVFormatToString, LayoutBHSD_BHSD_BHSD) {
+  EXPECT_EQ(nvte_get_qkv_format(NVTE_QKV_Layout::NVTE_BHSD_BHSD_BHSD),
+            NVTE_QKV_Format::NVTE_BHSD);
+}

[greptile-apps]

Greptile Summary

The PR adds the missing NVTE_BHSD string representation for fused-attention QKV formats.

• Maps NVTE_BHSD to "NVTE_BHSD" in the format conversion switch.
• Adds a focused GoogleTest assertion for the mapping.

Confidence Score: 5/5

The PR appears safe to merge.

No blocking failure remains.

Important Files Changed

Filename	Overview
transformer_engine/common/fused_attn/fused_attn.cpp	Adds the missing NVTE_BHSD case to the QKV-format string conversion.
tests/test_qkv_format_to_string.cpp	Adds a focused assertion covering the new enum-to-string mapping.

_{Reviews (3): Last reviewed commit: "[pre-commit.ci] auto fixes from pre-comm..." | Re-trigger Greptile}

[cyanguwa]

Thanks @andrewwhitecdw for submitting PRs #3361 to #3379. I've reviewed them and I think I should make a few suggestions regarding the PR styling and CI testing.

• Some of the PRs are one-liners, or fairly small, and they touch the same files as others. So could you please consolidate the relevant ones to one PR, and this will save rebase churns and some CI runs.
• Not all of these PRs require adding a test. For example, the [()] -> [] change is probably safe with just some local testing, instead of bloating the CI infra.
• Some PRs have code diffs in their descriptions. This is unnecessary because we have the "Files changed" tab, and can be misleading if they don't match what's really changed in the code.

Thanks for the contribution!