Repository navigation

[Common] Fuse RMSNorm forward with MXFP8 quantization - #3657

Merged

phu0ngng merged 9 commits into

NVIDIA:mainfrom

sraman-rgb:sraman/rmsnorm-mxfp8-fused-kernel

Oct 9, 2026

Commits on Oct 8, 2026
- [Common] Fuse RMSNorm forward with MXFP8 quantization
  
  sraman-rgb
  and
  claude
  committed
  5b3653a
- Update transformer_engine/common/include/transformer_engine/normalization.h
  
  sraman-rgb
  and
  timmoon10
  authored
  1d07cc3
- Update docs/envvars.rst
  
  sraman-rgb
  and
  timmoon10
  authored
  e1991d0
- [Common] Select the RMSNorm + MXFP8 kernel with NVTE_NORM_FWD_USE_CUDNN
  
  sraman-rgb
  and
  claude
  committed
  59b913b
- Merge branch 'main' into sraman/rmsnorm-mxfp8-fused-kernel
  sraman-rgb
  authored
  aed1413
Commits on Oct 9, 2026
- Apply suggestion from @timmoon10
  timmoon10
  authored
  0aef241
- Apply suggestion from @timmoon10
  timmoon10
  authored
  e27db7f
- [pre-commit.ci] auto fixes from pre-commit.com hooks
  pre-commit-ci[bot]
  committed
  06d34c6
- Merge branch 'main' into sraman/rmsnorm-mxfp8-fused-kernel
  vthumbe1503
  authored
  917c90c