Skip to content

[release/2.12] Disable CK GEMM/SDPA when gfx1250 is the only arch #3458

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
ethanwee1 wants to merge 2 commits into
base: release/2.12
Choose a base branch
from
Draft

[release/2.12] Disable CK GEMM/SDPA when gfx1250 is the only arch #3458

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
3731b21
[ROCm] Disable CK GEMM when gfx1250 is the only arch
ethanwee1 Jul 21, 2026
e7167bb
[ROCm] Move gfx1250 CK GEMM disable before -DUSE_ROCM_CK_GEMM flag
ethanwee1 Jul 22, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 10 additions & 0 deletions cmake/Dependencies.cmake
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1025,6 +1025,16 @@ if(USE_ROCM)
if(HIPBLASLT_VEC_EXT)
list(APPEND HIP_CXX_FLAGS -DHIPBLASLT_VEC_EXT)
endif()
# composable_kernel has no gfx1250 support yet. If gfx1250 is the only arch,
# disable CK GEMM here (before -DUSE_ROCM_CK_GEMM is added) so that both the
# CK GEMM sources and their call sites (guarded by USE_ROCM_CK_GEMM in
# CUDABlas.cpp/GroupedBlas.cpp) are compiled out consistently. Doing this in
# aten/src/ATen/CMakeLists.txt instead excludes the sources but still defines
# USE_ROCM_CK_GEMM, leaving undefined references to gemm_internal_ck at link.
if("${PYTORCH_ROCM_ARCH}" STREQUAL "gfx1250")
message(WARNING "gfx1250 is the only arch in PYTORCH_ROCM_ARCH: disabling USE_ROCM_CK_GEMM (composable_kernel lacks gfx1250 support)")
caffe2_update_option(USE_ROCM_CK_GEMM OFF)
endif()
if(USE_ROCM_CK_GEMM)
list(APPEND HIP_CXX_FLAGS -DUSE_ROCM_CK_GEMM)
endif()
Expand Down