fix return type for double complex atomics

ginkgo-project · Nov 27, 2019 · 9081b31 · 9081b31
1 parent 035e23e
commit 9081b31
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cuda/components/atomic.cuh b/cuda/components/atomic.cuh
@@ -63,7 +63,7 @@ __forceinline__ __device__ thrust::complex<float> atomic_add(
  *
  * @note It is not 'real' complex<double> atomic add operation
  */
-__forceinline__ __device__ thrust::complex<float> atomic_add(
+__forceinline__ __device__ thrust::complex<double> atomic_add(
     thrust::complex<double> *__restrict__ address, thrust::complex<double> val)
 {
     cuDoubleComplex *cuaddr = reinterpret_cast<cuDoubleComplex *>(address);

diff --git a/hip/components/atomic.hip.hpp b/hip/components/atomic.hip.hpp
@@ -63,7 +63,7 @@ __forceinline__ __device__ thrust::complex<float> atomic_add(
  *
  * @note It is not 'real' complex<double> atomic add operation
  */
-__forceinline__ __device__ thrust::complex<float> atomic_add(
+__forceinline__ __device__ thrust::complex<double> atomic_add(
     thrust::complex<double> *__restrict__ address, thrust::complex<double> val)
 {
     hipDoubleComplex *cuaddr = reinterpret_cast<hipDoubleComplex *>(address);