---
id: pr-tensorrt-llm-12163
repo: NVIDIA/TensorRT-LLM
pr: 12163
title: '[None][feat] Minimax RMS norm optimization'
author: jmydurant
date: '2026-04-20'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/12163
source_category: upstream-code
architectures:
- sm100
tags:
- allreduce
- benchmark kernel
- benchmark_test
- memory_primitives
- moe_routing
- norm_elementwise
- userbuffers
techniques: []
hardware_features: []
kernel_types:
- attention
- memory
- moe
- norm
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: a56a8d2435b631b667672ac4994e6e424fb80440
inclusion_reason: Look for fusion, vectorization, dtype conversion, and store-traffic reductions before changing math.
changed_paths:
- cpp/tensorrt_llm/kernels/communicationKernels/MiniMaxReduceRMSKernel.cu
- cpp/tensorrt_llm/kernels/communicationKernels/MiniMaxReduceRMSKernel.h
- cpp/tensorrt_llm/thop/allreduceOp.cpp
- tensorrt_llm/_torch/custom_ops/cpp_custom_ops.py
- tensorrt_llm/_torch/distributed/__init__.py
- tensorrt_llm/_torch/distributed/ops.py
- tensorrt_llm/_torch/models/modeling_minimaxm2.py
- tests/microbenchmarks/minimax_all_reduce.py
- tests/unittest/_torch/multi_gpu/test_allreduce.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-12163
head_ref_oid: d912d1c7fa8d31a918ca74754b750f81c4b79776
---
# [None][feat] Minimax RMS norm optimization

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [12163](https://github.com/NVIDIA/TensorRT-LLM/pull/12163)
- Decision: `Look for fusion, vectorization, dtype conversion, and store-traffic reductions before changing math.`
- Reason: Look for fusion, vectorization, dtype conversion, and store-traffic reductions before changing math.

## Changed Paths

- `cpp/tensorrt_llm/kernels/communicationKernels/MiniMaxReduceRMSKernel.cu`
- `cpp/tensorrt_llm/kernels/communicationKernels/MiniMaxReduceRMSKernel.h`
- `cpp/tensorrt_llm/thop/allreduceOp.cpp`
- `tensorrt_llm/_torch/custom_ops/cpp_custom_ops.py`
- `tensorrt_llm/_torch/distributed/__init__.py`
- `tensorrt_llm/_torch/distributed/ops.py`
- `tensorrt_llm/_torch/models/modeling_minimaxm2.py`
- `tests/microbenchmarks/minimax_all_reduce.py`
- `tests/unittest/_torch/multi_gpu/test_allreduce.py`

## Evidence Bundle

`evidence/pull-bundles/tensorrt-llm/gh-12163`
