From 926f410b79a23754fbdcca3ddc687dc00701abe7 Mon Sep 17 00:00:00 2001 From: dan_the_3rd <43445237+danthe3rd@users.noreply.github.com> Date: Wed, 7 Aug 2024 14:57:10 +0000 Subject: [PATCH] Fix build with cuda 12.6 (fairinternal/xformers#1177) Co-authored-by: danthe3rd __original_commit__ = fairinternal/xformers@de08b3fbef7db22c7dc0657a625e2d0cacbeb7fd --- xformers/csrc/sequence_parallel_fused/synchronization_kernels.cu | 1 + 1 file changed, 1 insertion(+) diff --git a/xformers/csrc/sequence_parallel_fused/synchronization_kernels.cu b/xformers/csrc/sequence_parallel_fused/synchronization_kernels.cu index afc1bb5bd1..239d99edb7 100644 --- 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 #include #include // cuda::atomic_ref is only available in libcudacxx 1.7.0+ (which corresponds to