---
id: pr-pytorch-153104
repo: pytorch/pytorch
pr: 153104
title: '[FlexAttention] Remove Old Constraint on lastdim strides'
author: drisspg
date: '2025-05-15'
url: https://github.com/pytorch/pytorch/pull/153104
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: 5beafa5faacc93dacb09b6f7fb10987d947a5ced
inclusion_reason: kernel file changes
changed_paths:
- test/inductor/test_flex_attention.py
- torch/_inductor/kernel/flex_attention.py
head_ref_oid: 55eb43fbeccc8c6d3f419fce5c8864c88cfe0539
artifact_dir: evidence/pull-bundles/pytorch/gh-153104
---
# [FlexAttention] Remove Old Constraint on lastdim strides

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

## Changed Paths

- `test/inductor/test_flex_attention.py`
- `torch/_inductor/kernel/flex_attention.py`

## Evidence Bundle

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