fix: add missing NVTE_BHSD labels to fused_attn to_string

[andrewwhitecdw]

Adds the two missing NVTE_BHSD entries to the fused_attn enum-to-string helpers, consolidating the two one-liner PRs (#3361, #3362) as requested in review.

Changes

• transformer_engine/common/fused_attn/fused_attn.cpp — add case labels for NVTE_BHSD_BHSD_BHSD in to_string(NVTE_QKV_Layout) and NVTE_BHSD in to_string(NVTE_QKV_Format).

Details

NVTE_BHSD_BHSD_BHSD (= 25) and NVTE_BHSD (= 7) exist in the NVTE_QKV_Layout / NVTE_QKV_Format enums in transformer_engine/common/include/transformer_engine/fused_attn.h, and other helpers in the same file already special-case them, but the two to_string switches never got case labels. Code logging those layouts/formats currently falls through to the UNKNOWN_QKV_LAYOUT(...) / UNKNOWN_QKV_FORMAT(...) fallback. Each switch gets one new case returning its own enum name, placed in the same order as the enum declarations and mirroring the formatting of the neighboring cases. No behavior change for any other value.

Tests

Not run — a full compile requires the CUDA/cuDNN toolchain, which is not available in the contributor's environment. Lightweight verification only: grep confirmed both labels match the enum declarations and the surrounding switch style, and no duplicate case label was introduced.

Per the maintainer's feedback on #3362, no test file is included — the addition of BHSD to these enums is a straightforward label addition, and the removed test in #3362 was not appropriate here.

[greptile-apps]

Confidence Score: 5/5

The PR appears safe to merge because both added switch cases correctly map existing enum values without affecting other behavior.

Summary

Adds the missing string conversions for the existing BHSD fused-attention enum values.

• Maps NVTE_BHSD_BHSD_BHSD to its enum name in to_string(NVTE_QKV_Layout).
• Maps NVTE_BHSD to its enum name in to_string(NVTE_QKV_Format).
• Preserves the existing unknown-value fallbacks and behavior for all other enum values.

_{Reviews (2) · Last reviewed commit: "Merge branch 'main' into bugfix/fused-at..."}