From dfd8c4e3f2cefb81433e67a1eeebf417f4a8fd62 Mon Sep 17 00:00:00 2001 From: Aditya Atluri Date: Fri, 9 Sep 2016 10:25:57 -0500 Subject: [PATCH] Added async memcpy driver api for nvcc backend Change-Id: If401fc949b834c883cff9de74adb0d532a0fd942 [ROCm/hip commit: 1cb90bdd6ae7f7fd772653a870fe141e0de8cb30] --- .../hip/include/nvcc_detail/hip_runtime_api.h | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/projects/hip/include/nvcc_detail/hip_runtime_api.h b/projects/hip/include/nvcc_detail/hip_runtime_api.h index 8e9b0d92a7..1ad1af2267 100644 --- a/projects/hip/include/nvcc_detail/hip_runtime_api.h +++ b/projects/hip/include/nvcc_detail/hip_runtime_api.h @@ -232,6 +232,24 @@ inline static hipError_t hipMemcpyDtoD(hipDeviceptr_t dst, return hipCUResultTohipError(cuMemcpyDtoD(dst, src, size)); } +inline static hipError_t hipMemcpyHtoDAsync(hipDeviceptr_t dst, + void* src, size_t size, hipStream_t stream) +{ + return hipCUResultTohipError(cuMemcpyHtoDAsync(dst, src, size, stream)); +} + +inline static hipError_t hipMemcpyDtoHAsync(void* dst, + hipDeviceptr_t src, size_t size, hipStream_t stream) +{ + return hipCUResultTohipError(cuMemcpyDtoH(dst, src, size, stream)); +} + +inline static hipError_t hipMemcpyDtoDAsync(hipDeviceptr_t dst, + hipDeviceptr_t src, size_t size, hipStream_t stream) +{ + return hipCUResultTohipError(cuMemcpyDtoD(dst, src, size, stream)); +} + inline static hipError_t hipMemcpy(void* dst, const void* src, size_t sizeBytes, hipMemcpyKind copyKind) { return hipCUDAErrorTohipError(cudaMemcpy(dst, src, sizeBytes, hipMemcpyKindToCudaMemcpyKind(copyKind))); }