---
id: pr-tensorrt-llm-9729
repo: NVIDIA/TensorRT-LLM
pr: 9729
title: '[None][feat] add fp4 gemm + allreduce'
author: benzh-2025
date: '2026-01-13'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/9729
source_category: upstream-code
architectures:
- sm100
tags:
- allreduce
- benchmark_test
- gemm_quant
- memory_primitives
- moe_routing
techniques:
- fine-grained-quantization
- kernel-fusion
hardware_features:
- blackwell
- fp4
- nvfp4
kernel_types:
- gemm
- memory
- moe
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: 6df2c8a074bbf8324211f4fa48bf1e14f9022cc4
inclusion_reason: Inspect scale layout, accumulator type, tile/schedule choice, epilogue fusion, and partial-tile guards before deriving a candidate.
changed_paths:
- cpp/tensorrt_llm/kernels/cutlass_kernels/allreduce_gemm/allreduce_gemm_impl_sm100.h
- cpp/tensorrt_llm/kernels/cutlass_kernels/allreduce_gemm/allreduce_gemm_runner.cu
- cpp/tensorrt_llm/thop/CMakeLists.txt
- cpp/tensorrt_llm/thop/fusedGemmAllreduceOp.cpp
- tensorrt_llm/_torch/custom_ops/torch_custom_ops.py
- tensorrt_llm/_torch/models/modeling_llama.py
- tensorrt_llm/_torch/modules/linear.py
- tests/unittest/_torch/multi_gpu/test_linear.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-9729
head_ref_oid: dc5db6deaa703eb3edf033b284888244e1421e90
---
# [None][feat] add fp4 gemm + allreduce

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [9729](https://github.com/NVIDIA/TensorRT-LLM/pull/9729)
- Decision: `Inspect scale layout, accumulator type, tile/schedule choice, epilogue fusion, and partial-tile guards before deriving a candidate.`
- Reason: Inspect scale layout, accumulator type, tile/schedule choice, epilogue fusion, and partial-tile guards before deriving a candidate.

## Changed Paths

- `cpp/tensorrt_llm/kernels/cutlass_kernels/allreduce_gemm/allreduce_gemm_impl_sm100.h`
- `cpp/tensorrt_llm/kernels/cutlass_kernels/allreduce_gemm/allreduce_gemm_runner.cu`
- `cpp/tensorrt_llm/thop/CMakeLists.txt`
- `cpp/tensorrt_llm/thop/fusedGemmAllreduceOp.cpp`
- `tensorrt_llm/_torch/custom_ops/torch_custom_ops.py`
- `tensorrt_llm/_torch/models/modeling_llama.py`
- `tensorrt_llm/_torch/modules/linear.py`
- `tests/unittest/_torch/multi_gpu/test_linear.py`

## Evidence Bundle

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