Support ScaledSReLU in TE grouped MLP fuser - #4859

Merged

sraman-rgb merged 14 commits into

NVIDIA:mainfrom

sraman-rgb:srelu-fused-grouped-mlp

May 29, 2026

Commits on May 19, 2026

Support ScaledSReLU in TE grouped MLP fuser
sraman
committed
dc1fa88
Add tests for ScaledSReLU grouped MLP fuser
sraman
committed
882e27b

Commits on May 20, 2026

Merge branch 'main' into srelu-fused-grouped-mlp
sraman-rgb
authored
68349b0
Fix grouped MLP lint formatting
Siddhartha Raman S
committed
b079324

Commits on May 21, 2026

Address grouped SReLU fusion review comments
Siddhartha Raman S
committed
82f4d74
Cover grouped SReLU fusion edge cases
Siddhartha Raman S
committed
e7dbc32

Commits on May 22, 2026

Pass activation recompute to TE fused MoE ops
a
committed
9ffcd47
Merge branch 'main' into srelu-fused-grouped-mlp
sraman-rgb
authored
6f6b1ab
Fix grouped SReLU lint formatting
sraman-rgb
committed
a7c5feb

Commits on May 28, 2026

Commits on May 29, 2026

Guard TE activation recompute argument
sraman-rgb
committed
1292dcc
Fix grouped MLP fuser formatting
sraman-rgb
committed
092f812
Merge branch 'main' into srelu-fused-grouped-mlp
sraman-rgb
authored
dcf3a7e