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

mtaillefumier · web-flow · commit ac34bd279759 · 2024-10-24T17:58:37.000+02:00
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!
