Searched refs:GpuAtomicCasHelper (Results 1 – 1 of 1) sorted by relevance
585 __device__ T GpuAtomicCasHelper(T* ptr, F accumulate) { in GpuAtomicCasHelper() function594 CREATE_CUDA_DEVICE_FUNCTION_ALIAS(GpuAtomicCasHelper, CudaAtomicCasHelper);599 __device__ float GpuAtomicCasHelper(float* ptr, F accumulate) { in GpuAtomicCasHelper() function601 GpuAtomicCasHelper(reinterpret_cast<int32*>(ptr), [accumulate](int32 a) { in GpuAtomicCasHelper()606 __device__ double GpuAtomicCasHelper(double* ptr, F accumulate) { in GpuAtomicCasHelper() function612 GpuAtomicCasHelper(reinterpret_cast<unsigned long long*>(ptr), in GpuAtomicCasHelper()619 return __longlong_as_double(GpuAtomicCasHelper( in GpuAtomicCasHelper()640 __device__ Eigen::half GpuAtomicCasHelper(Eigen::half* ptr, F accumulate) { in GpuAtomicCasHelper() function650 uint32 result = GpuAtomicCasHelper(address, [accumulate](uint32 arg) { in GpuAtomicCasHelper()659 uint32 result = GpuAtomicCasHelper(address, [accumulate](uint32 arg) { in GpuAtomicCasHelper()[all …]