Adding arch detection in xccl_ucs

Signed-off-by: Tomislavj Janjusic <[email protected]> Co-authored-by: Artem Polyakov <[email protected]> Co-authored-by: Sergey Lebedev <[email protected]>
openucx · Jan 7, 2021 · 2e158cb · 2e158cb
1 parent 434f3dc
commit 2e158cb
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/core/xccl_ucs.h b/src/core/xccl_ucs.h
@@ -8,7 +8,16 @@
 #define XCCL_UCS_H_
 
 #include <inttypes.h>
-#include <ucs/arch/x86_64/bitops.h>
+
+#if defined(__x86_64__)
+#   include <ucs/arch/x86_64/bitops.h>
+#elif defined(__aarch64__)
+#   include <ucs/arch/aarch64/bitops.h>
+#elif defined(__powerpc64__)
+#   include <ucs/arch/ppc64/bitops.h>
+#else
+#  error "Unsupported architecture"
+#endif
 
 #define ucs_ilog2(_n)                   \
 (                                       \