Fix minor compilations issues with rocm 6 and spack (#143)

eth-cscs · Oct 24, 2024 · ac34bd2 · ac34bd2
1 parent e64dae2
commit ac34bd2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/cosma/communicator.hpp b/src/cosma/communicator.hpp
@@ -16,7 +16,7 @@
 #endif
 
 #if defined(COSMA_WITH_NCCL) && defined(TILED_MM_ROCM)
-#include <rccl.h>
+#include <rccl/rccl.h>
 #endif
 
 namespace cosma {

diff --git a/src/cosma/gpu/nccl_mapper.hpp b/src/cosma/gpu/nccl_mapper.hpp
@@ -6,7 +6,7 @@
 #include <nccl.h>
 
 #elif defined(TILED_MM_ROCM)
-#include <rccl.h>
+#include <rccl/rccl.h>
 
 #else
 #error Either TILED_MM_CUDA or TILED_MM_ROCM must be defined!

diff --git a/src/cosma/gpu/nccl_utils.hpp b/src/cosma/gpu/nccl_utils.hpp
@@ -10,7 +10,7 @@
 #include <nccl.h>
 
 #elif defined(TILED_MM_ROCM)
-#include <rccl.h>
+#include <rccl/rccl.h>
 
 #else
 #error Either TILED_MM_CUDA or TILED_MM_ROCM must be defined!