Searched refs:memcpy_and_sync (Results 1 – 6 of 6) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | CUDAScalar.cu | 45 …at::cuda::memcpy_and_sync((void *)value.const_data_ptr<scalar_t>(), self.const_data_ptr<scalar_t>(… in _local_scalar_dense_cuda()
|
H A D | Nonzero.cu | 82 …at::cuda::memcpy_and_sync((void *)pinned_num_nonzeros_h.const_data_ptr<int>(), num_nonzeros.get(),… in nonzero_cuda_out_impl()
|
H A D | TensorModeKernel.cu | 175 at::cuda::memcpy_and_sync(indices_data, &index, sizeof(int64_t), cudaMemcpyHostToDevice, stream); in calculate_mode()
|
H A D | Copy.cu | 384 at::cuda::memcpy_and_sync(dst, src, nbytes, kind, stream); in copy_kernel_cuda()
|
H A D | UniqueCub.cu | 293 at::cuda::memcpy_and_sync(&num_true, tmp_num_true.get(), sizeof(int), in operator ()()
|
/aosp_15_r20/external/pytorch/c10/cuda/ |
H A D | CUDAFunctions.h | 77 C10_CUDA_API void __inline__ memcpy_and_sync( in memcpy_and_sync() function
|