Repository navigation

perf(pt): fused Triton zonal scatter for the GIE head (DP_GIE_SCATTER_INFER) - #6072

Open

long-yi-2019 wants to merge 7 commits into

deepmodeling:masterfrom

long-yi-2019:pr/gie-scatter

Open

perf(pt): fused Triton zonal scatter for the GIE head (DP_GIE_SCATTER_INFER)#6072
long-yi-2019 wants to merge 7 commits into
deepmodeling:masterfrom
long-yi-2019:pr/gie-scatter

Commits on Oct 18, 2026
- perf(pt): fused Triton zonal scatter for the GIE head (DP_GIE_SCATTER_INFER)
  longjunhao
  committed
  27beb6c
- test(pt): cover the fused GIE zonal scatter operator
  longjunhao
  committed
  629eeba
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  
  pre-commit-ci[bot]
  authored and
  longjunhao
  committed
  603c596
- fix(pt): padded channel blocks and a training gate for the fused GIE scatter
  longjunhao
  committed
  f2251c2
- style(pt): explicit index dtypes for the GIE scatter path
  longjunhao
  committed
  fa4e1df
- fix(pt): channel-masked radial loads, level-1 routing and a unified scatter boundary for the GIE head
  longjunhao
  committed
  43e6437
- fix(pt): guard the optional Triton import and defer CSR acquisition to a fused backend
  longjunhao
  committed
  7103523