Function SafeAdd3D

oneflow/user/kernels/grid_sample_kernel_util.h:309–319

template<typename scalar_t, typename index_t>
static OF_DEVICE_FUNC void SafeAdd3D(scalar_t* data, int d, int h, int w, int sD, int sH, int sW,
                                     int D, int H, int W, scalar_t delta, const index_t NC_offset,
                                     const index_t memory_span) {
  if (WithinBounds3D(d, h, w, D, H, W)) {
#if defined(__CUDACC__)
    cuda::atomic::Add(data + NC_offset + d * sD + h * sH + w * sW, delta);
#else
    data[NC_offset + d * sD + h * sH + w * sW] += delta;
#endif
  }
}

template<typename scalar_t, typename index_t>
static OF_DEVICE_FUNC void AddValueBounded(scalar_t* data, scalar_t x, scalar_t y, int W, int H,