MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / half GpuAtomicAdd

Method half GpuAtomicAdd

tensorflow/core/util/gpu_device_functions.h:674–678  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

672}
673
674__device__ inline Eigen::half GpuAtomicAdd(Eigen::half* ptr,
675 Eigen::half value) {
676 return detail::GpuAtomicCasHelper(
677 ptr, [value](Eigen::half a) { return a + value; });
678}
679
680#if (__CUDA_ARCH__ < 600) || TENSORFLOW_USE_ROCM
681__device__ inline double GpuAtomicAdd(double* ptr, double value) {

Callers

nothing calls this directly

Calls 1

GpuAtomicCasHelperFunction · 0.85

Tested by

no test coverage detected