Searched defs:gpuAtomicAddNoReturn (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | Atomic.cuh | 324 inline __device__ void gpuAtomicAddNoReturn(c10::complex<T> *address, c10::complex<T> val) { gpuAto… in gpuAtomicAddNoReturn() function
325 inline __device__ void gpuAtomicAddNoReturn(uint8_t *address, uint8_t val) { gpuAtomicAdd(address, … in gpuAtomicAddNoReturn() function
326 inline __device__ void gpuAtomicAddNoReturn(int8_t *address, int8_t val) { gpuAtomicAdd(address, va… in gpuAtomicAddNoReturn() function
327 inline __device__ void gpuAtomicAddNoReturn(int16_t *address, int16_t val) { gpuAtomicAdd(address, … in gpuAtomicAddNoReturn() function
328 inline __device__ void gpuAtomicAddNoReturn(int32_t *address, int32_t val) { gpuAtomicAdd(address, … in gpuAtomicAddNoReturn() function
329 inline __device__ void gpuAtomicAddNoReturn(int64_t *address, int64_t val) { gpuAtomicAdd(address, … in gpuAtomicAddNoReturn() function
330 inline __device__ void gpuAtomicAddNoReturn(bool *address, bool val) { gpuAtomicAdd(address, val); } in gpuAtomicAddNoReturn() function
331 inline __device__ void gpuAtomicAddNoReturn(at::Half *address, at::Half val) { gpuAtomicAdd(address… in gpuAtomicAddNoReturn() function
332 inline __device__ void gpuAtomicAddNoReturn(at::BFloat16 *address, at::BFloat16 val) { gpuAtomicAdd… in gpuAtomicAddNoReturn() function
333 inline __device__ void gpuAtomicAddNoReturn(double *address, double val) { gpuAtomicAdd(address, va… in gpuAtomicAddNoReturn() function
[all …]
|