Revert "[FlexAttention] Fix device test instantation (#151846)"

This reverts commit b37fa20771.

Reverted https://github.com/pytorch/pytorch/pull/151846 on behalf of https://github.com/jithunnair-amd due to PR broke rocm workflow ([comment](https://github.com/pytorch/pytorch/pull/151846#issuecomment-2824607429))
This commit is contained in:
PyTorch MergeBot 2025-04-23 15:01:36 +00:00
parent 21b0ef520d
commit 7310049c42

File diff suppressed because it is too large Load Diff