[PyTorch] Enable e5m3 fused GEMM kernels from cuDNN - #2

Merged

timmoon10 merged 6 commits into

timmoon10:nvfp4-ue5m3-prototypefrom

kainzhong:nvfp4-ue5m3-prototype-cudnn

Aug 13, 2026

Commits on Aug 11, 2026

[PyTorch] Enable e5m3 fused GEMM kernels from cuDNN
kainzhong
committed
abde002

Commits on Aug 12, 2026

have to pad to 256 to use cuDNN
kainzhong
committed
ccb900c
fix: need to pass scale_dtype
kainzhong
committed
ee42aae

Commits on Aug 13, 2026

route wgrad to cuDNN's wgrad API
kainzhong
committed
4197910
Support grouped linear with NVFP4-UE5M3
timmoon10
committed
403d218
Fix typos

timmoon10
and
codex
committed
67903fa