---
id: pr-sglang-4178
repo: sgl-project/sglang
pr: 4178
title: 'ROCm: Flex Attention Enablement with custom backends'
author: HaiShaw
date: '2025-03-07'
url: https://github.com/sgl-project/sglang/pull/4178
source_category: upstream-code
architectures:
- sm100
tags: []
techniques: []
hardware_features: []
kernel_types: []
languages: []
captured_at: '2026-05-16'
status: merged
merge_sha: 0beea4503f4f23f6f4348748c610af4c4233d2fc
inclusion_reason: defer
changed_paths:
- docker/Dockerfile.rocm
- python/sglang/srt/layers/attention/aiter_backend.py
- python/sglang/srt/layers/attention/aiter_decode_backend.py
- python/sglang/srt/model_executor/model_runner.py
- python/sglang/srt/server_args.py
- sgl-kernel/src/sgl-kernel/csrc/moe/moe_align_kernel.hip
- sgl-kernel/src/sgl-kernel/include/utils_hip.h
artifact_dir: evidence/pull-bundles/sglang/gh-4178
head_ref_oid: 307ac09f45dd280fe614295becfcf9cc1f8e16d8
---
# ROCm: Flex Attention Enablement with custom backends

- Repository: `sgl-project/sglang`
- PR: [4178](https://github.com/sgl-project/sglang/pull/4178)
- Decision: `defer`
- Reason: defer

## Changed Paths

- `docker/Dockerfile.rocm`
- `python/sglang/srt/layers/attention/aiter_backend.py`
- `python/sglang/srt/layers/attention/aiter_decode_backend.py`
- `python/sglang/srt/model_executor/model_runner.py`
- `python/sglang/srt/server_args.py`
- `sgl-kernel/src/sgl-kernel/csrc/moe/moe_align_kernel.hip`
- `sgl-kernel/src/sgl-kernel/include/utils_hip.h`

## Evidence Bundle

`evidence/pull-bundles/sglang/gh-4178`
