Searched defs:GPU_ATOMIC_INTEGER (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | Atomic.cuh | 161 # define GPU_ATOMIC_INTEGER(NAME, OP, DTYPE) … macro
351 GPU_ATOMIC_INTEGER(Mul, a * b, uint8_t) in GPU_ATOMIC_INTEGER() function
412 GPU_ATOMIC_INTEGER(Max, safe_max(a, b), uint8_t) in GPU_ATOMIC_INTEGER() function
472 GPU_ATOMIC_INTEGER(Min, safe_min(a, b), uint8_t) in GPU_ATOMIC_INTEGER() function
|