---
id: pr-pytorch-170555
repo: pytorch/pytorch
pr: 170555
title: '[cherry-pick] Fix vllm issue for flex (#170499)'
author: zou3519
date: '2025-12-17'
url: https://github.com/pytorch/pytorch/pull/170555
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: 054b1104c08fe2a7943f892332926b901181b533
inclusion_reason: kernel file changes
changed_paths:
- test/inductor/test_flex_attention.py
- torch/_dynamo/variables/higher_order_ops.py
head_ref_oid: acfa4f46bd14d09d75465b0ef6024168c4edc4dd
artifact_dir: evidence/pull-bundles/pytorch/gh-170555
---
# [cherry-pick] Fix vllm issue for flex (#170499)

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

## Changed Paths

- `test/inductor/test_flex_attention.py`
- `torch/_dynamo/variables/higher_order_ops.py`

## Evidence Bundle

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