update HIP libs path (hipfft, hiprand, hipcub) for compatibilit…

…y with ROCm > 5.7
3dem · Dec 9, 2024 · e0553df · e0553df
1 parent cafe3b0
commit e0553df
Show file tree

Hide file tree

Showing 8 changed files with 14 additions and 14 deletions.
diff --git a/src/acc/hip/hip_fft.h b/src/acc/hip/hip_fft.h
@@ -7,7 +7,7 @@
 #include "src/acc/hip/hip_settings.h"
 #include "src/acc/hip/hip_mem_utils.h"
 #include <hip/hip_runtime.h>
-#include <hipfft.h>
+#include <hipfft/hipfft.h>
 
 #ifdef DEBUG_HIP
 #define HANDLE_HIPFFT_ERROR( err ) (HipfftHandleError( err, __FILE__, __LINE__ ))

diff --git a/src/acc/hip/hip_kernels/helper.h b/src/acc/hip/hip_kernels/helper.h
@@ -5,8 +5,8 @@
 #define HIP_HELPER_KERNELS_H_
 
 #include <hip/hip_runtime.h>
-#include <hiprand.h>
-#include <hiprand_kernel.h>
+#include <hiprand/hiprand.h>
+#include <hiprand/hiprand_kernel.h>
 #include <vector>
 #include <iostream>
 #include <fstream>

diff --git a/src/acc/hip/hip_kernels/helper.hip.cpp b/src/acc/hip/hip_kernels/helper.hip.cpp
@@ -7,8 +7,8 @@
 #include "src/acc/hip/hip_kernels/helper.h"
 #include "src/acc/hip/hip_settings.h"
 
-#include <hiprand.h>
-#include <hiprand_kernel.h>
+#include <hiprand/hiprand.h>
+#include <hiprand/hiprand_kernel.h>
 
 /// Needed explicit template instantiations
 template __global__ void hip_kernel_make_eulers_2D<true>(XFLOAT *,

diff --git a/src/acc/hip/hip_mem_utils.h b/src/acc/hip/hip_mem_utils.h
@@ -6,7 +6,7 @@
 
 #ifdef _HIP_ENABLED
 #include <hip/hip_runtime.h>
-#include <hiprand.h>
+#include <hiprand/hiprand.h>
 #include "src/acc/hip/hip_settings.h"
 #include "src/acc/hip/custom_allocator.h"
 #endif

diff --git a/src/acc/hip/hip_ml_optimiser.hip.cpp b/src/acc/hip/hip_ml_optimiser.hip.cpp
@@ -11,8 +11,8 @@
 #include <iostream>
 #include "src/ml_optimiser.h"
 #include <hip/hip_runtime.h>
-#include <hiprand.h>
-#include <hiprand_kernel.h>
+#include <hiprand/hiprand.h>
+#include <hiprand/hiprand_kernel.h>
 
 #include "src/acc/acc_ptr.h"
 #include "src/acc/acc_projector.h"

diff --git a/src/acc/hip/hip_settings.h b/src/acc/hip/hip_settings.h
@@ -15,7 +15,7 @@
 #include "src/macros.h"
 #include "src/error.h"
 
-#include <hiprand.h>
+#include <hiprand/hiprand.h>
 
 // Required compute capability
 #define HIP_CC_MAJOR 5

diff --git a/src/acc/hip/hip_utils_cub.h b/src/acc/hip/hip_utils_cub.h
@@ -39,14 +39,14 @@ if (ptr.getAllocator() == NULL)
 	max_pair.deviceAlloc();
 	size_t temp_storage_size = 0;
 
-	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMax( NULL, temp_storage_size, ~ptr, ~max_pair, ptr.getSize()));
+	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMax( NULL, temp_storage_size, ~ptr, ~max_pair, static_cast<int>(ptr.getSize())));
 
 	if(temp_storage_size==0)
 		temp_storage_size=1;
 
 	HipCustomAllocator::Alloc* alloc = ptr.getAllocator()->alloc(temp_storage_size);
 
-	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMax( alloc->getPtr(), temp_storage_size, ~ptr, ~max_pair, ptr.getSize(), ptr.getStream()));
+	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMax( alloc->getPtr(), temp_storage_size, ~ptr, ~max_pair, static_cast<int>(ptr.getSize()), ptr.getStream()));
 
 	max_pair.cpToHost();
 	ptr.streamSync();
@@ -75,14 +75,14 @@ if (ptr.getAllocator() == NULL)
 	min_pair.deviceAlloc();
 	size_t temp_storage_size = 0;
 
-	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMin( NULL, temp_storage_size, ~ptr, ~min_pair, ptr.getSize()));
+	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMin( NULL, temp_storage_size, ~ptr, ~min_pair, static_cast<int>(ptr.getSize())));
 
 	if(temp_storage_size==0)
 		temp_storage_size=1;
 
 	HipCustomAllocator::Alloc* alloc = ptr.getAllocator()->alloc(temp_storage_size);
 
-	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMin( alloc->getPtr(), temp_storage_size, ~ptr, ~min_pair, ptr.getSize(), ptr.getStream()));
+	DEBUG_HANDLE_ERROR(hipcub::DeviceReduce::ArgMin( alloc->getPtr(), temp_storage_size, ~ptr, ~min_pair, static_cast<int>(ptr.getSize()), ptr.getStream()));
 
 	min_pair.cpToHost();
 	ptr.streamSync();

diff --git a/src/projector.cpp b/src/projector.cpp
@@ -24,7 +24,7 @@
 #ifdef _CUDA_ENABLED
 	#include <cufft.h>
 #elif _HIP_ENABLED
-	#include <hipfft.h>
+	#include <hipfft/hipfft.h>
 #endif
 #if defined _CUDA_ENABLED || defined _HIP_ENABLED
 	#include <sys/types.h>