---
id: pr-pytorch-164368
repo: pytorch/pytorch
pr: 164368
title: '[Flex attention] Fix flex attention head broadcast'
author: pytorchbot
date: '2025-10-01'
url: https://github.com/pytorch/pytorch/pull/164368
source_category: upstream-code
architectures:
- sm100
tags:
- attention
techniques: []
hardware_features: []
kernel_types:
- attention
languages:
- python
captured_at: '2026-04-17'
status: merged
merge_sha: 1cd83de3151786164eebcb591e919a424cfad60e
inclusion_reason: kernel file changes
changed_paths:
- test/inductor/test_flex_attention.py
- torch/nn/attention/flex_attention.py
head_ref_oid: 04425d26d6cbbbea2195af6d0e71bb088308e4f0
artifact_dir: evidence/pull-bundles/pytorch/gh-164368
---
# [Flex attention] Fix flex attention head broadcast

- Repository: `pytorch/pytorch`
- PR: [164368](https://github.com/pytorch/pytorch/pull/164368)
- Decision: `kernel file changes`
- Reason: kernel file changes

## Changed Paths

- `test/inductor/test_flex_attention.py`
- `torch/nn/attention/flex_attention.py`

## Evidence Bundle

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