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

Merge branch 'main' into bugfix/test-distributed-fused-attn-sbhd-reor…

59962c2
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 18, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO