Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 235 Bytes
Newer Older
Chao Liu's avatar
update build  
Chao Liu committed
1 
if(DEVICE_BACKEND STREQUAL "HIP")
Chao Liu's avatar
fixed build issue  
Chao Liu committed
2 
    set(DRIVER_SOURCE driver.cpp)
Chao Liu's avatar
update build  
Chao Liu committed
3
4
5 
elseif(DEVICE_BACKEND STREQUAL "CUDA")
    set(DRIVER_SOURCE driver.cu)
endif()
Chao Liu's avatar
hip build  
Chao Liu committed
6
7
8 

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