Skip to content

[PyT] Plumbing correct bias dims from TE to cudnn, while adding support for additional bias shapes#2537

Merged
KshitijLakhani merged 25 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/fix/bias-shape
Feb 18, 2026
Merged

[PyT] Plumbing correct bias dims from TE to cudnn, while adding support for additional bias shapes#2537
KshitijLakhani merged 25 commits into
NVIDIA:mainfrom
KshitijLakhani:klakhani/fix/bias-shape

nit: Reinstate the original test for right side swa

ff174a8
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Feb 18, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO