Searched defs:gpuAtomicMax (Results 1 – 1 of 1) sorted by relevance
425 inline __device__ at::BFloat16 gpuAtomicMax(at::BFloat16 * address, at::BFloat16 val) { in gpuAtomicMax() function432 inline __device__ double gpuAtomicMax(double * address, double val) { in gpuAtomicMax() function440 inline __device__ float gpuAtomicMax(float * address, float val) { in gpuAtomicMax() function