---
id: pr-tilelang-1627
repo: tile-ai/tilelang
pr: 1627
title: '[BugFix] Correct index_map selection for transposed A matrix in MFMA Layout with `k_dim==4` and open rocm-ci for gemmsr'
author: benenzhu
date: '2026-01-08'
url: https://github.com/tile-ai/tilelang/pull/1627
source_category: upstream-code
architectures:
- sm100
tags:
- cutlass
techniques: []
hardware_features: []
kernel_types:
- gemm
languages:
- cute-dsl
captured_at: '2026-05-16'
status: merged
merge_sha: b914318b460ca37ff03235a33ee6de42e8ffc990
inclusion_reason: Search-matched CUDA kernel optimization PR; materialize diff/source before synthesis.
changed_paths:
- testing/python/tilelibrary/test_tilelang_tilelibrary_gemm.py
- tilelang/intrinsics/mfma_macro_generator.py
artifact_dir: evidence/pull-bundles/tilelang/gh-1627
head_ref_oid: 2c4bce43bffddd64dd25aaa8271b2e5aaa16fe63
---
# [BugFix] Correct index_map selection for transposed A matrix in MFMA Layout with `k_dim==4` and open rocm-ci for gemmsr

- Repository: `tile-ai/tilelang`
- PR: [1627](https://github.com/tile-ai/tilelang/pull/1627)
- 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

- `testing/python/tilelibrary/test_tilelang_tilelibrary_gemm.py`
- `tilelang/intrinsics/mfma_macro_generator.py`

## Evidence Bundle

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