---
id: pr-tensorrt-llm-9924
repo: NVIDIA/TensorRT-LLM
pr: 9924
title: '[TRTLLM-9493][feat] Add helixPostProcessNative kernel for cp_dim=2'
author: brb-nv
date: '2025-12-13'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/9924
source_category: upstream-code
architectures:
- sm90
tags:
- benchmark_test
- moe_routing
- sm90
techniques:
- kernel-fusion
hardware_features:
- hopper
kernel_types:
- moe
languages:
- cuda-cpp
captured_at: '2026-05-16'
status: merged
merge_sha: 461446045e62028ff6bc6516f8e5858e26890742
inclusion_reason: Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.
changed_paths:
- cpp/tensorrt_llm/kernels/helixKernels.cu
- cpp/tensorrt_llm/kernels/helixKernels.h
- cpp/tensorrt_llm/thop/helixPostProcessOp.cpp
- tensorrt_llm/_torch/custom_ops/cpp_custom_ops.py
- tests/unittest/_torch/thop/parallel/test_helix_postprocess.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-9924
head_ref_oid: 89221f1ddff07bb27a91be849ed581b3ff5becda
---
# [TRTLLM-9493][feat] Add helixPostProcessNative kernel for cp_dim=2

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [9924](https://github.com/NVIDIA/TensorRT-LLM/pull/9924)
- Decision: `Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.`
- Reason: Mine shape sets, tolerance rules, warmup logic, and profile commands before mutating code.

## Changed Paths

- `cpp/tensorrt_llm/kernels/helixKernels.cu`
- `cpp/tensorrt_llm/kernels/helixKernels.h`
- `cpp/tensorrt_llm/thop/helixPostProcessOp.cpp`
- `tensorrt_llm/_torch/custom_ops/cpp_custom_ops.py`
- `tests/unittest/_torch/thop/parallel/test_helix_postprocess.py`

## Evidence Bundle

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