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

Method half GpuAtomicMax

tensorflow/core/util/gpu_device_functions.h:784–788  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

782#endif
783
784__device__ inline Eigen::half GpuAtomicMax(Eigen::half* ptr,
785 Eigen::half value) {
786 return detail::GpuAtomicCasHelper(
787 ptr, [value](Eigen::half a) { return max(a, value); });
788}
789
790#if __CUDA_ARCH__ < 320
791__device__ inline tensorflow::uint64 GpuAtomicMax(tensorflow::uint64* ptr,

Callers

nothing calls this directly

Calls 2

GpuAtomicCasHelperFunction · 0.85
maxFunction · 0.50

Tested by

no test coverage detected