CMakeLists.txt 239 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
5
if(DEVICE_BACKEND STREQUAL "HIP")
    set(DRIVER_SOURCE driver.hip.cpp)
elseif(DEVICE_BACKEND STREQUAL "CUDA")
    set(DRIVER_SOURCE driver.cu)
endif()
Chao Liu's avatar
Chao Liu committed
6
7
8

add_executable(driver ${DRIVER_SOURCE}) 
target_link_libraries(driver PRIVATE tensor)