Searched refs:CudaAtomicCasHelper (Results 1 – 1 of 1) sorted by relevance
407 __device__ T CudaAtomicCasHelper(T* ptr, F accumulate) { in CudaAtomicCasHelper() function420 __device__ float CudaAtomicCasHelper(float* ptr, F accumulate) { in CudaAtomicCasHelper() function422 CudaAtomicCasHelper(reinterpret_cast<int32*>(ptr), [accumulate](int32 a) { in CudaAtomicCasHelper()427 __device__ double CudaAtomicCasHelper(double* ptr, F accumulate) { in CudaAtomicCasHelper() function428 return __longlong_as_double(CudaAtomicCasHelper( in CudaAtomicCasHelper()448 __device__ Eigen::half CudaAtomicCasHelper(Eigen::half* ptr, F accumulate) { in CudaAtomicCasHelper() function458 uint32 result = CudaAtomicCasHelper(address, [accumulate](uint32 arg) { in CudaAtomicCasHelper()467 uint32 result = CudaAtomicCasHelper(address, [accumulate](uint32 arg) { in CudaAtomicCasHelper()492 return detail::CudaAtomicCasHelper( in CudaAtomicAdd()499 return detail::CudaAtomicCasHelper(ptr, in CudaAtomicAdd()[all …]