diff --git a/include/nvcc_detail/hip_runtime_api.h b/include/nvcc_detail/hip_runtime_api.h index d7f5d521a4..9ca3542569 100644 --- a/include/nvcc_detail/hip_runtime_api.h +++ b/include/nvcc_detail/hip_runtime_api.h @@ -148,6 +148,11 @@ default: } } +inline static hipError_t hipInit(unsigned int flags) +{ + return hipCUResultTohipError(cuInit(flags)); +} + inline static hipError_t hipDeviceReset() { return hipCUDAErrorTohipError(cudaDeviceReset()); } @@ -219,6 +224,12 @@ inline static hipError_t hipMemcpyDtoH(void* dst, return hipCUResultTohipError(cuMemcpyDtoH(dst, src, size)); } +inline static hipError_t hipMemcpyDtoD(hipDeviceptr_t dst, + hipDeviceptr_t src, size_t size) +{ + return hipCUResultTohipError(cuMemcpyDtoD(dst, src, size)); +} + inline static hipError_t hipMemcpy(void* dst, const void* src, size_t sizeBytes, hipMemcpyKind copyKind) { return hipCUDAErrorTohipError(cudaMemcpy(dst, src, sizeBytes, hipMemcpyKindToCudaMemcpyKind(copyKind))); } diff --git a/tests/src/sampleModule.cpp b/tests/src/sampleModule.cpp index bef7345b4f..606b19717d 100644 --- a/tests/src/sampleModule.cpp +++ b/tests/src/sampleModule.cpp @@ -50,7 +50,7 @@ int main(){ #ifdef __HIP_PLATFORM_NVCC__ - cuInit(0); + hipInit(0); hipDevice_t device; hipCtx_t context; hipDeviceGet(&device, 0);