diff --git a/src/backend/cuda/CMakeLists.txt b/src/backend/cuda/CMakeLists.txt index 40ac455411..8fedfb5b4d 100644 --- a/src/backend/cuda/CMakeLists.txt +++ b/src/backend/cuda/CMakeLists.txt @@ -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) @@ -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 diff --git a/src/backend/metal/CMakeLists.txt b/src/backend/metal/CMakeLists.txt index 330e3f728b..98498e1e31 100644 --- a/src/backend/metal/CMakeLists.txt +++ b/src/backend/metal/CMakeLists.txt @@ -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. diff --git a/src/backend/rocm/CMakeLists.txt b/src/backend/rocm/CMakeLists.txt index b8b557eaca..24b2a83245 100644 --- a/src/backend/rocm/CMakeLists.txt +++ b/src/backend/rocm/CMakeLists.txt @@ -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}) @@ -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()