update build

[ROCm/composable_kernel commit: a414e3fdf8]
This commit is contained in:
Chao Liu
2019-02-15 02:06:34 -06:00
parent e7f6b820cd
commit 153629655f
6 changed files with 17 additions and 11 deletions

View File

@@ -11,7 +11,7 @@ set_target_properties(tensor PROPERTIES POSITION_INDEPENDENT_CODE ON)
if(DEVICE_BACKEND STREQUAL "CUDA")
target_link_libraries(device nvToolsExt cudart)
target_link_libraries(tensor nvToolsExt cudart)
endif()
install(TARGETS tensor LIBRARY DESTINATION lib)

View File

@@ -43,7 +43,7 @@ float launch_kernel(F kernel, dim3 grid_dim, dim3 block_dim, Args... args)
hipGetErrorString(hipGetLastError());
#elif DEVICE_BACKEND_CUDA
const void* f = reinterpret_cast<const void*>(kernel);
void* p_args = {&args...};
void* p_args[] = {&args...};
timer.Start();