Skip to content

Commit ddf9765

Browse files
source/kernel.cu: cosetics
1 parent c4129a4 commit ddf9765

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

source/kernel.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -741,7 +741,7 @@ static void bm3d(
741741
wdst_val = (wdst_val + extractor) - extractor;
742742
weight_val = (weight_val + extractor) - extractor;
743743

744-
#if defined(CUDART_VERSION) && CUDART_VERSION >= 12080 && __CUDA_ARCH__ >= 700
744+
#if defined(__CUDACC_DEVICE_ATOMIC_BUILTINS__)
745745
__nv_atomic_add(&wdstp[j * stride], wdst_val, __NV_ATOMIC_RELAXED, __NV_THREAD_SCOPE_DEVICE);
746746
__nv_atomic_add(&weightp[j * stride], weight_val, __NV_ATOMIC_RELAXED, __NV_THREAD_SCOPE_DEVICE);
747747
#else // defined(CUDART_VERSION) && CUDART_VERSION >= 12080 && __CUDA_ARCH__ >= 700

0 commit comments

Comments
 (0)