Graph Safe Current Scaling Support for GroupedLinear Module/Ops + Fix CUBLAS GGEMM heuristics - #3143

Merged

vthumbe1503 merged 18 commits into

NVIDIA:mainfrom

vthumbe1503:nvfp4_and_fp8_current_scaling

Jul 1, 2026

Commits on Jun 25, 2026

support in grouped linear and relevant tests
vthumbe1503
committed
bad0b2c
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
9af6df6
Unecessary details remove
vthumbe1503
authored
bd0832c

Commits on Jun 26, 2026

fix grouped linear module's grouped tensor path
vthumbe1503
committed
a163af3
allow more current scaling use-cases.. block nvfp4+rht+single grouped weight being cuda graphable
vthumbe1503
committed
971160f
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
cb57694
some minor comment fixing
vthumbe1503
committed
f97ef20
Merge branch 'nvfp4_and_fp8_current_scaling' of github.com:vthumbe1503/TransformerEngine into nvfp4_and_fp8_current_scaling
vthumbe1503
committed
b195152
Merge branch 'main' into nvfp4_and_fp8_current_scaling
vthumbe1503
authored
2770984

Commits on Jun 27, 2026

fix heuristics
vthumbe1503
committed
018e6e6
Merge branch 'main' into nvfp4_and_fp8_current_scaling
vthumbe1503
authored
4b9a2a0

Commits on Jun 29, 2026

Commits on Jun 30, 2026

fix for other 2 nvte APIs
vthumbe1503
committed
c54bd22
Merge branch 'nvfp4_and_fp8_current_scaling' of github.com:vthumbe1503/TransformerEngine into nvfp4_and_fp8_current_scaling
vthumbe1503
committed
95b5b7d
fix m and n
vthumbe1503
committed
e7fdec2