---
id: pr-pytorch-135869
repo: pytorch/pytorch
pr: 135869
title: '[ROCm] Update to AOTriton 0.7b (Cherry-picked)'
author: xinyazhang
date: '2024-09-30'
url: https://github.com/pytorch/pytorch/pull/135869
source_category: upstream-code
architectures:
- sm100
tags: []
techniques: []
hardware_features: []
kernel_types: []
languages: []
captured_at: '2026-05-16'
status: merged
merge_sha: 051df20ac2360862a2830108eb10ba8e374a4455
inclusion_reason: defer
changed_paths:
- .ci/docker/aotriton_version.txt
- .ci/docker/common/install_aotriton.sh
- aten/src/ATen/native/transformers/cuda/attention.cu
- aten/src/ATen/native/transformers/cuda/attention_backward.cu
- aten/src/ATen/native/transformers/cuda/sdp_utils.cpp
- aten/src/ATen/native/transformers/hip/aotriton_adapter.h
- aten/src/ATen/native/transformers/hip/flash_attn/flash_api.hip
- test/functorch/test_control_flow.py
- test/inductor/test_flex_attention.py
- test/inductor/test_flex_decoding.py
- test/nn/test_multihead_attention.py
- test/run_test.py
- test/test_native_mha.py
- test/test_nn.py
- test/test_transformers.py
- torch/testing/_internal/common_methods_invocations.py
artifact_dir: evidence/pull-bundles/pytorch/gh-135869
head_ref_oid: 63c83e9a589c7147cc46feb9320fd52008b8c263
---
# [ROCm] Update to AOTriton 0.7b (Cherry-picked)

- Repository: `pytorch/pytorch`
- PR: [135869](https://github.com/pytorch/pytorch/pull/135869)
- Decision: `defer`
- Reason: defer

## Changed Paths

- `.ci/docker/aotriton_version.txt`
- `.ci/docker/common/install_aotriton.sh`
- `aten/src/ATen/native/transformers/cuda/attention.cu`
- `aten/src/ATen/native/transformers/cuda/attention_backward.cu`
- `aten/src/ATen/native/transformers/cuda/sdp_utils.cpp`
- `aten/src/ATen/native/transformers/hip/aotriton_adapter.h`
- `aten/src/ATen/native/transformers/hip/flash_attn/flash_api.hip`
- `test/functorch/test_control_flow.py`
- `test/inductor/test_flex_attention.py`
- `test/inductor/test_flex_decoding.py`
- `test/nn/test_multihead_attention.py`
- `test/run_test.py`
- `test/test_native_mha.py`
- `test/test_nn.py`
- `test/test_transformers.py`
- `torch/testing/_internal/common_methods_invocations.py`

## Evidence Bundle

`evidence/pull-bundles/pytorch/gh-135869`
