---
id: pr-tensorrt-llm-10476
repo: NVIDIA/TensorRT-LLM
pr: 10476
title: '[TRTLLM-10276][feat] Integrate cutedsl argmax kernel'
author: ameynaik-hub
date: '2026-01-27'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/10476
source_category: upstream-code
architectures:
- sm100
tags:
- benchmark_test
- compiler_runtime
- cute dsl
techniques: []
hardware_features: []
kernel_types:
- compiler
- decode
- sampling
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: df8be0c50c5058b7b31878e87d8f03b60ed9b791
inclusion_reason: 'Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.'
changed_paths:
- LICENSE
- tensorrt_llm/_torch/cute_dsl_kernels/argmax.py
- tensorrt_llm/_torch/cute_dsl_kernels/test_argmax.py
- tensorrt_llm/_torch/speculative/interface.py
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-10476
head_ref_oid: 2e30e1c0624ea093044a3ca08b847e7f5e49e511
---
# [TRTLLM-10276][feat] Integrate cutedsl argmax kernel

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [10476](https://github.com/NVIDIA/TensorRT-LLM/pull/10476)
- Decision: `Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.`
- Reason: Treat compiler/runtime integration as part of the kernel: launch wrapper, cache/JIT behavior, generated code, and build flags need tests.

## Changed Paths

- `LICENSE`
- `tensorrt_llm/_torch/cute_dsl_kernels/argmax.py`
- `tensorrt_llm/_torch/cute_dsl_kernels/test_argmax.py`
- `tensorrt_llm/_torch/speculative/interface.py`

## Evidence Bundle

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