Fix: storage_gpu for HIPCC-AMDGPU (#1540)

GridTools · Aug 6, 2020 · 83240c0 · 83240c0
1 parent 77f82d1
commit 83240c0
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/cmake/public/gridtools_setup_targets.cmake b/cmake/public/gridtools_setup_targets.cmake
@@ -249,9 +249,12 @@ macro(_gt_setup_targets _config_mode clang_cuda_mode)
     endif()
 
     find_package(CUDAToolkit)
-    if(CUDAToolkit_FOUND)
+    if(CUDAToolkit_FOUND OR GT_CUDA_TYPE STREQUAL HIPCC-AMDGPU)
         _gt_add_library(${_config_mode} storage_gpu)
-        target_link_libraries(${_gt_namespace}storage_gpu INTERFACE ${_gt_namespace}gridtools CUDA::cudart)
+        target_link_libraries(${_gt_namespace}storage_gpu INTERFACE ${_gt_namespace}gridtools)
+        if(CUDAToolkit_FOUND)
+            target_link_libraries(${_gt_namespace}storage_gpu INTERFACE CUDA::cudart)
+        endif()
         list(APPEND GT_STORAGES gpu)
     endif()