frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] Add specific compile options for CK SDPA (#161759)

Browse Source 
Updates CK version and adds CK specific compilation options

Pull Request resolved: https://github.com/pytorch/pytorch/pull/161759
Approved by: https://github.com/jeffdaily
This commit is contained in:
Andy Lugo
2025-09-09 20:04:19 +00:00
committed by PyTorch MergeBot
parent 86d34a43f5
commit d22d916719
4 changed files with 94 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
caffe2/CMakeLists.txt
View File

@ -1762,6 +1762,10 @@ if(USE_ROCM)
target_link_libraries(torch_hip PUBLIC torch_cpu_library ${Caffe2_PUBLIC_HIP_DEPENDENCY_LIBS})
target_link_libraries(torch_hip PRIVATE ${Caffe2_HIP_DEPENDENCY_LIBS})
if(USE_ROCM_CK_SDPA)
target_link_libraries(torch_hip PRIVATE ck_sdpa)
endif()
if(USE_FBGEMM_GENAI)
if(USE_ROCM)
target_link_libraries(torch_hip PRIVATE fbgemm_genai)