Aaryaman Vasishta 62843c14bb [ROCm/Windows] Support aotriton for scaled_dot_product_attention on Windows. (#162330) ...

Enables flash attention and/or memory efficient attention on Windows with scaled_dot_product_attention via. aotriton.
Already tested to be working on Windows with TheRock.

Steps to enable: simply set `USE_FLASH_ATTENTION=1` and `USE_MEM_EFF_ATTENTION=1` as usual. See https://github.com/ROCm/TheRock/blob/main/external-builds/pytorch/build_prod_wheels.py#L578-L604

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162330
Approved by: https://github.com/xinyazhang, https://github.com/ScottTodd, https://github.com/jeffdaily

Co-authored-by: Scott Todd <scott.todd0@gmail.com>

2025-09-11 22:35:09 +00:00

..

External

[ROCm/Windows] Support aotriton for scaled_dot_product_attention on Windows. (#162330)

2025-09-11 22:35:09 +00:00

Modules

[Intel GPU] Upgrade OneDNN XPU Tag to v3.9.1 (#161932)

2025-09-04 11:05:10 +00:00

Modules_CUDA_fix

[2/N] Remove FindPackageHandleStandardArgs.cmake (#156559)

2025-07-24 02:34:10 +00:00

public

Revert "Move prioritized text linker optimization code from setup.py to cmake (#160078)"

2025-09-11 15:29:29 +00:00

Allowlist.cmake

Replace whitelist with allowlist (#42067)

2020-07-28 08:01:16 -07:00

BLAS_ABI.cmake

[submodule] Bump fbgemm to latest (#158210)

2025-08-11 13:48:02 +00:00

BuildVariables.cmake

Remove Caffe2_MAIN_LIBS (#38408)

2020-05-15 12:27:15 -07:00

Caffe2Config.cmake.in

xpu: improve error handling and reporting in XPU cmake files (#149353)

2025-03-20 02:00:39 +00:00

cmake_uninstall.cmake.in

Formatting cmake (to lowercase without space for if/elseif/else/endif) (#35521)

2020-03-27 14:25:17 -07:00

Codegen.cmake

Remove outdated CMAKE_CUDA_COMPILER_VERSION branch (#160075)

2025-08-09 14:23:17 +00:00

DebugHelper.cmake

[BE] fix typos in cmake/ (#156079)

2025-06-17 19:25:43 +00:00

Dependencies.cmake

Revert "Make distributed modules importable even when backend not built (#159889)" (#162568)

2025-09-10 04:29:42 +00:00

FlatBuffers.cmake

[pytorch][PR] Add ability for a mobile::Module to save as flatbuffer (#70201)

2022-01-12 16:30:39 -08:00

IncludeSource.cpp.in

CMake: Include instead of copying cpu kernel files (#67656)

2021-11-30 19:13:53 -08:00

iOS.cmake

[BE] fix typos in cmake/ (#156079)

2025-06-17 19:25:43 +00:00

Metal.cmake

[Build] Allow metal shaders to include ATen headers (#156256)

2025-06-18 01:03:25 +00:00

MiscCheck.cmake

[submodule] Bump fbgemm to latest (#158210)

2025-08-11 13:48:02 +00:00

prioritized_text.txt

[Build] Add linker script optimization (#121975)

2024-04-09 20:22:25 +00:00

ProtoBuf.cmake

[Reland] Use 3.27 as the minimum CMake version (#154783)

2025-06-14 16:37:51 +00:00

ProtoBufPatch.cmake

Migrate PyTorch to C++17 (#85969)

2022-12-08 02:27:48 +00:00

Summary.cmake

Revert "Move prioritized text linker optimization code from setup.py to cmake (#160078)"

2025-09-11 15:29:29 +00:00

TorchConfig.cmake.in

Revert "Simplify nvtx3 CMake handling, always use nvtx3 (#153784)"

2025-06-24 20:02:07 +00:00

TorchConfigVersion.cmake.in

Formatting cmake (to lowercase without space for if/elseif/else/endif) (#35521)

2020-03-27 14:25:17 -07:00

VulkanCodegen.cmake

[BE][CMake] Use FindPython module (#124613)

2024-05-29 13:17:35 +00:00

VulkanDependencies.cmake

[Vulkan] Remove GLSL Code Gen (#91912)

2023-01-10 20:29:47 +00:00