---
id: pr-tensorrt-llm-11181
repo: NVIDIA/TensorRT-LLM
pr: 11181
title: '[https://nvbugs/5854860][fix] Fix cutedsl argmax on sm120'
author: dongfengy
date: '2026-02-04'
url: https://github.com/NVIDIA/TensorRT-LLM/pull/11181
source_category: upstream-code
architectures:
- sm100
tags:
- arch_pipeline
- benchmark_test
- compiler_runtime
- cute dsl
techniques:
- pipeline-stages
hardware_features: []
kernel_types:
- compiler
- runtime
languages:
- cuda-cpp
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: 0bd4630cd167dcb4efea240ca8340e287caa0354
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:
- tensorrt_llm/_torch/cute_dsl_kernels/argmax.py
- tests/integration/test_lists/waives.txt
artifact_dir: evidence/pull-bundles/tensorrt-llm/gh-11181
head_ref_oid: b68760fb399e9657aa1360b3bfb88672bce1b364
---
# [https://nvbugs/5854860][fix] Fix cutedsl argmax on sm120

- Repository: `NVIDIA/TensorRT-LLM`
- PR: [11181](https://github.com/NVIDIA/TensorRT-LLM/pull/11181)
- 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

- `tensorrt_llm/_torch/cute_dsl_kernels/argmax.py`
- `tests/integration/test_lists/waives.txt`

## Evidence Bundle

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