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: