diff options
| author | Sai Praveen Bangaru <31557731+saipraveenb25@users.noreply.github.com> | 2023-08-02 23:04:21 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-02 23:04:21 -0400 |
| commit | 0dcae1c64c821920a22022a5930121b5f84019f3 (patch) | |
| tree | 6393718078b9db93d5923e705cf6fd2bd8d0eed9 /prelude | |
| parent | e80cd4f1de5b4d0ceaf61b7a48bd6b8b5b3a0097 (diff) | |
Only define atomics for `float2` and `float4` when CUDA arch<900 (#3041)
From Ada onwards, these definitions are already available in CUDA's stdlib and will cause a compiler error.
Diffstat (limited to 'prelude')
| -rw-r--r-- | prelude/slang-cuda-prelude.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/prelude/slang-cuda-prelude.h b/prelude/slang-cuda-prelude.h index 0b56b8a71..1447be05b 100644 --- a/prelude/slang-cuda-prelude.h +++ b/prelude/slang-cuda-prelude.h @@ -423,9 +423,11 @@ SLANG_MAKE_VECTOR_FROM_SCALAR(__half) return result; \ }\ +#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ < 900 SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(atomicAdd, float, 2) -SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(atomicAdd, float, 3) SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(atomicAdd, float, 4) +#endif +SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(atomicAdd, float, 3) SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(atomicAdd, int, 2) SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(atomicAdd, int, 3) SLANG_CUDA_VECTOR_ATOMIC_BINARY_IMPL(atomicAdd, int, 4) |