Searched refs:cudaMemcpy (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/tests/ut/cpp/stub/runtime/ |
| D | cuda_runtime_api.cc | 22 cudaError_t cudaMemcpy(void *dst, const void *src, size_t count, enum cudaMemcpyKind kind) { return… in cudaMemcpy() function
|
| D | cuda_runtime_api.h | 39 cudaError_t cudaMemcpy(void *dst, const void *src, size_t count, enum cudaMemcpyKind kind);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | cuda_driver.cc | 77 auto ret = cudaMemcpy(dst, src, size, cudaMemcpyHostToDevice); in CopyHostMemToDevice()
86 auto ret = cudaMemcpy(dst, src, size, cudaMemcpyDeviceToHost); in CopyDeviceMemToHost()
|
| /third_party/mindspore/mindspore/lite/src/delegate/tensorrt/ |
| D | tensorrt_allocator.cc | 116 auto cuda_ret = cudaMemcpy(dst_ptr, src_ptr, data_size, kind); in SyncMemInHostAndDevice()
|