---
id: pr-tilelang-1740
repo: tile-ai/tilelang
pr: 1740
title: '[fix]: fix deepseek_mla amd example and add aiter mla compare test'
author: ZiguanWang
date: '2026-01-28'
url: https://github.com/tile-ai/tilelang/pull/1740
source_category: upstream-code
architectures:
- sm100
tags:
- mla
- triton kernel
techniques: []
hardware_features: []
kernel_types:
- mla
languages:
- cuda-cpp
- triton
captured_at: '2026-05-16'
status: merged
merge_sha: 413ecbb4ce62107131fa8f218483facc88876183
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- examples/deepseek_mla/amd/benchmark_mla_decode_amd_aiter.py
- examples/deepseek_mla/amd/benchmark_mla_decode_amd_tilelang.py
- examples/deepseek_mla/amd/benchmark_mla_decode_amd_triton.py
artifact_dir: evidence/pull-bundles/tilelang/gh-1740
head_ref_oid: ee3a42948354395e9836b15358e0b33f0a158d9f
---
# [fix]: fix deepseek_mla amd example and add aiter mla compare test

- Repository: `tile-ai/tilelang`
- PR: [1740](https://github.com/tile-ai/tilelang/pull/1740)
- Decision: `Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.`
- Reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.

## Changed Paths

- `examples/deepseek_mla/amd/benchmark_mla_decode_amd_aiter.py`
- `examples/deepseek_mla/amd/benchmark_mla_decode_amd_tilelang.py`
- `examples/deepseek_mla/amd/benchmark_mla_decode_amd_triton.py`

## Evidence Bundle

`evidence/pull-bundles/tilelang/gh-1740`
