Skip to content

[PyTorch] Build the attention padding mask without host syncs - #3268

Merged
pggPL merged 2 commits into
NVIDIA:mainfrom
pggPL:padding_mask_no_host_sync
Jul 29, 2026
Merged

[PyTorch] Build the attention padding mask without host syncs#3268
pggPL merged 2 commits into
NVIDIA:mainfrom
pggPL:padding_mask_no_host_sync

Keep the padding mask on the attention device

e82b29d
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jul 28, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO