Skip to content

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

Open
andrewwhitecdw wants to merge 1 commit into
NVIDIA:mainfrom
andrewwhitecdw:bugfix/test-distributed-fused-attn-sbhd-reorder-skip-uses-original-shape
Open

[JAX]: SBHD reorder skip uses original shape instead of swapped tensor#3373
andrewwhitecdw wants to merge 1 commit into
NVIDIA:mainfrom
andrewwhitecdw:bugfix/test-distributed-fused-attn-sbhd-reorder-skip-uses-original-shape

Commits

Commits on Aug 18, 2026