fixed build issue

[ROCm/composable_kernel commit: f2b92ba945]
This commit is contained in:
Chao Liu
2019-06-12 09:59:28 -05:00
parent 14451df1fd
commit 42e9613bb8
2 changed files with 3 additions and 11 deletions

View File

@@ -1,5 +1,5 @@
if(DEVICE_BACKEND STREQUAL "HIP")
set(DRIVER_SOURCE driver.hip.cpp)
set(DRIVER_SOURCE driver.cpp)
elseif(DEVICE_BACKEND STREQUAL "CUDA")
set(DRIVER_SOURCE driver.cu)
endif()