Skip to content

[Build] Fix cross platform CMake and add messages when enabling backends #2183

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

Merged
SiriusNEO merged 1 commit into from
May 11, 2026
Merged

[Build] Fix cross platform CMake and add messages when enabling backends #2183

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
6 changes: 4 additions & 2 deletions src/backend/cuda/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@ if(NOT USE_CUDA)
return()
endif()

message(STATUS "CUDA Backend is enabled")

set(CMAKE_CUDA_STANDARD 17)
find_package(CUDAToolkit REQUIRED)
if(CUDAToolkit_NVCC_EXECUTABLE)
Expand Down Expand Up @@ -156,8 +158,8 @@ link_directories(${CUDAToolkit_LIBRARY_DIR} ${CUDAToolkit_LIBRARY_DIR}/stubs)

# Register stubs for linking and install
if(TILELANG_USE_CUDA_STUBS)
set(TILELANG_ACTIVE_BACKEND_STUB_LINK cuda_stub)
set(TILELANG_ACTIVE_BACKEND_STUB_TARGETS cuda_stub cudart_stub nvrtc_stub)
list(APPEND TILELANG_ACTIVE_BACKEND_STUB_LINK cuda_stub)
list(APPEND TILELANG_ACTIVE_BACKEND_STUB_TARGETS cuda_stub cudart_stub nvrtc_stub)
endif()

# Register additional RPATH for CUDA toolkit lib directory
Expand Down
2 changes: 2 additions & 0 deletions src/backend/metal/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@ if(NOT USE_METAL)
return()
endif()

message(STATUS "METAL Backend is enabled")

if(NOT APPLE)
# On non-Apple platforms USE_METAL=ON enables only codegen (Metal source
# generation) without requiring the Metal/Foundation frameworks.
Expand Down
6 changes: 4 additions & 2 deletions src/backend/rocm/CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,8 @@ if(NOT USE_ROCM)
return()
endif()

message(STATUS "ROCM Backend is enabled")

set(CMAKE_HIP_STANDARD 17)
include(${TVM_SOURCE}/cmake/utils/FindROCM.cmake)
find_rocm(${USE_ROCM})
Expand Down Expand Up @@ -75,6 +77,6 @@ list(APPEND TILE_LANG_INCLUDES ${ROCM_INCLUDE_DIRS})

# Register stubs for linking and install
if(TILELANG_USE_HIP_STUBS)
set(TILELANG_ACTIVE_BACKEND_STUB_LINK hip_stub)
set(TILELANG_ACTIVE_BACKEND_STUB_TARGETS hip_stub hiprtc_stub)
list(APPEND TILELANG_ACTIVE_BACKEND_STUB_LINK hip_stub)
list(APPEND TILELANG_ACTIVE_BACKEND_STUB_TARGETS hip_stub hiprtc_stub)
endif()
Loading