Repository navigation

[PyTorch] Deprecate grouped linear and MLP environment gates - #3584

Open

zhongbozhu wants to merge 15 commits into

NVIDIA:mainfrom

zhongbozhu:cleanup/grouped-mlp-env-gates

Open

[PyTorch] Deprecate grouped linear and MLP environment gates#3584
zhongbozhu wants to merge 15 commits into
NVIDIA:mainfrom
zhongbozhu:cleanup/grouped-mlp-env-gates

Commits on Oct 13, 2026
- [PyTorch] Deprecate grouped linear and MLP environment gates
  zhongbozhu
  committed
  36eec71
- resolve comments
  zhongbozhu
  committed
  7e85ad4
- no need to check if cuda exists
  zhongbozhu
  committed
  d5739dd
- fix torch compile CI
  zhongbozhu
  committed
  927f1e9
- fix alignment config
  zhongbozhu
  committed
  53b35ae
- add more tests
  zhongbozhu
  committed
  7846a6c
- attempt to fix ci issue
  zhongbozhu
  committed
  429c9b7
- split alignment for being safe, deprecated env var for grouped mlp falls back to split alignment of 256
  
  vthumbe1503
  authored and
  zhongbozhu
  committed
  880c88d
- warn if fusion doesnt happen only because alignment requirement
  
  vthumbe1503
  authored and
  zhongbozhu
  committed
  de65c64
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  
  pre-commit-ci[bot]
  authored and
  zhongbozhu
  committed
  d235b75
- address greptile review comments
  vthumbe1503
  committed
  7d4e8d3
- Merge branch 'main' into cleanup/grouped-mlp-env-gates
  vthumbe1503
  authored
  2ef8cb4
Commits on Oct 14, 2026
- Fix grouped MLP fusion test alignments
  vthumbe1503
  committed
  e243478
- fix merge conflict
  vthumbe1503
  committed
  4d568c8
- Merge branch 'main' into cleanup/grouped-mlp-env-gates
  vthumbe1503
  authored
  3ae2185