HIP

WeiqunZhang · Dec 8, 2024 · a3e2667 · a3e2667
1 parent 610b40f
commit a3e2667
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 9 deletions.
diff --git a/Src/LinearSolvers/AMReX_SpMV.H b/Src/LinearSolvers/AMReX_SpMV.H
@@ -9,7 +9,7 @@
 #if defined(AMREX_USE_CUDA)
 #  include <cusparse.h>
 #elif defined(AMREX_USE_HIP)
-#  include <hipsparse.h>
+#  include <hipsparse/hipsparse.h>
 #elif defined(AMREX_USE_DPCPP)
 #  include <oneapi/mkl/spblas.hpp>
 #endif

diff --git a/Tools/GNUMake/comps/hip.mak b/Tools/GNUMake/comps/hip.mak
@@ -122,16 +122,10 @@ ifeq ($(HIP_COMPILER),clang)
   ROC_PATH=$(realpath $(HIP_PATH))
   SYSTEM_INCLUDE_LOCATIONS += $(ROC_PATH)/include
 
-  # rocRand
-  SYSTEM_INCLUDE_LOCATIONS += $(ROC_PATH)/include/hiprand $(ROC_PATH)/include/rocrand
   LIBRARY_LOCATIONS += $(ROC_PATH)/lib
-  LIBRARIES += -Wl,--rpath=$(ROC_PATH)/lib -lhiprand -lrocrand
 
-  # rocPrim - Header only
-  SYSTEM_INCLUDE_LOCATIONS += $(ROC_PATH)/include/rocprim
-
-  # rocThrust - Header only
-  # SYSTEM_INCLUDE_LOCATIONS += $(ROC_PATH)/include/rocthrust
+  # hiprand & hipsparse
+  LIBRARIES += -Wl,--rpath=$(ROC_PATH)/lib -lhiprand -lrocrand -lhipsparse -lrocsparse
 
   # rocTracer
   ifeq ($(USE_ROCTX),TRUE)