update build

CMakeLists.txt

-cmake_minimum_required(VERSION 2.8.3)
+#cmake_minimum_required(VERSION 2.8.3)
+cmake_minimum_required(VERSION 3.9)
 project(modular_convolution)
 
 #c++

driver/CMakeLists.txt

-set(DRIVER_SOURCE driver.cpp)
+if(DEVICE_BACKEND STREQUAL "HIP")
+    set(DRIVER_SOURCE driver.hip.cpp)
+elseif(DEVICE_BACKEND STREQUAL "CUDA")
+    set(DRIVER_SOURCE driver.cu)
+endif()
 
 add_executable(driver ${DRIVER_SOURCE}) 
 target_link_libraries(driver PRIVATE tensor)

driver/driver.cu

+driver.hip.cpp
\ No newline at end of file

driver/driver.cpp → driver/driver.hip.cpp

@@ -50,7 +50,7 @@ struct GeneratorTensor_3
         std::initializer_list<std::size_t> ids = {static_cast<std::size_t>(is)...};
         std::vector<std::size_t> lens(sizeof...(Is), 100);
         std::vector<std::size_t> strides(sizeof...(Is), 1);
-        std::partial_sum(lens.rbegin(), lens.rbegin() + (sizeof...(Is)-1), strides.rbegin() + 1);
+        std::partial_sum(lens.rbegin(), lens.rbegin() + (sizeof...(Is) - 1), strides.rbegin() + 1);
         return std::inner_product(ids.begin(), ids.end(), strides.begin(), std::size_t(0)) + 1;
 #endif
     }

src/CMakeLists.txt

@@ -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) 

src/include/device.hpp

@@ -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();