mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-11 17:00:18 +00:00
9 lines
239 B
CMake
9 lines
239 B
CMake
if(DEVICE_BACKEND STREQUAL "HIP")
|
|
set(DRIVER_SOURCE driver.hip.cpp)
|
|
elseif(DEVICE_BACKEND STREQUAL "CUDA")
|
|
set(DRIVER_SOURCE driver.cu)
|
|
endif()
|
|
|
|
add_executable(driver ${DRIVER_SOURCE})
|
|
target_link_libraries(driver PRIVATE tensor)
|