Fix build with cuda 12.6 (fairinternal/xformers#1177)

Co-authored-by: danthe3rd <danthe3rd> __original_commit__ = fairinternal/xformers@de08b3f
facebookresearch · Aug 7, 2024 · 926f410 · 926f410
1 parent 567cb66
commit 926f410
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xformers/csrc/sequence_parallel_fused/synchronization_kernels.cu b/xformers/csrc/sequence_parallel_fused/synchronization_kernels.cu
@@ -12,6 +12,7 @@
   (!defined(__CUDA_ARCH__) || __CUDA_ARCH__ >= 700 || \
    (!defined(_MSC_VER) && __CUDA_ARCH__ >= 600))
 #if CUDA_ARCH_SUPPORTS_ATOMICS
+#include <cuda/atomic>
 #include <cuda/std/atomic>
 #include <cuda/std/version>
 // cuda::atomic_ref is only available in libcudacxx 1.7.0+ (which corresponds to