Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
CMakeLists.txt 237 Bytes
Newer Older
Chao Liu's avatar
change build  
Chao Liu committed
1 
if(DEVICE_BACKEND STREQUAL "AMD")
Chao Liu's avatar
fixed build issue  
Chao Liu committed
2 
    set(DRIVER_SOURCE driver.cpp)
Chao Liu's avatar
change build  
Chao Liu committed
3 
elseif(DEVICE_BACKEND STREQUAL "NVIDIA")
Chao Liu's avatar
update build  
Chao Liu committed
4
5 
    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)