Skip to content

[JAX] SBHD reorder skip uses original shape instead of swapped tensor - #3373

Merged
KshitijLakhani merged 2 commits into
NVIDIA:mainfrom
andrewwhitecdw:bugfix/test-distributed-fused-attn-sbhd-reorder-skip-uses-original-shape
Aug 19, 2026
Merged

[JAX] SBHD reorder skip uses original shape instead of swapped tensor#3373
KshitijLakhani merged 2 commits into
NVIDIA:mainfrom
andrewwhitecdw:bugfix/test-distributed-fused-attn-sbhd-reorder-skip-uses-original-shape

Commits

Commits on Aug 18, 2026