From 09d1a461393de5e7dce8fd17ea7e3508f52ba03b Mon Sep 17 00:00:00 2001 From: Rahul Garg Date: Tue, 14 Mar 2017 23:49:21 +0530 Subject: [PATCH] hipMemsetD8 support for HIP/NVCC path Change-Id: I48eee8266afd7b45a12d5ce2c4849b687a006c0f --- hipamd/include/hip/nvcc_detail/hip_runtime_api.h | 5 +++++ hipamd/tests/src/context/hipMemsetD8.cpp | 6 +++--- 2 files changed, 8 insertions(+), 3 deletions(-) diff --git a/hipamd/include/hip/nvcc_detail/hip_runtime_api.h b/hipamd/include/hip/nvcc_detail/hip_runtime_api.h index dc51290167..8c3e0da639 100644 --- a/hipamd/include/hip/nvcc_detail/hip_runtime_api.h +++ b/hipamd/include/hip/nvcc_detail/hip_runtime_api.h @@ -397,6 +397,11 @@ inline static hipError_t hipMemsetAsync(void* devPtr,int value, size_t count, hi return hipCUDAErrorTohipError(cudaMemsetAsync(devPtr, value, count, stream)); } +inline static hipError_t hipMemsetD8(hipDeviceptr_t dest, unsigned char value, size_t sizeBytes ) +{ + return hipCUResultTohipError(cuMemsetD8(dest, value, sizeBytes)); +} + inline static hipError_t hipGetDeviceProperties(hipDeviceProp_t *p_prop, int device) { cudaDeviceProp cdprop; diff --git a/hipamd/tests/src/context/hipMemsetD8.cpp b/hipamd/tests/src/context/hipMemsetD8.cpp index 1cd43696aa..3730fcb70b 100644 --- a/hipamd/tests/src/context/hipMemsetD8.cpp +++ b/hipamd/tests/src/context/hipMemsetD8.cpp @@ -45,13 +45,13 @@ int main(int argc, char *argv[]) hipDeviceptr_t A_d; A_h = new char[Nbytes]; - HIPCHECK ( hipMalloc(&A_d, Nbytes) ); + HIPCHECK ( hipMalloc((void **) &A_d, Nbytes) ); A_h = (char*)malloc(Nbytes); printf ("Size=%zu memsetval=%2x \n", Nbytes, memsetval); HIPCHECK ( hipMemsetD8(A_d, memsetval, Nbytes) ); - HIPCHECK ( hipMemcpy(A_h, A_d, Nbytes, hipMemcpyDeviceToHost)); + HIPCHECK ( hipMemcpy(A_h, (void *) A_d, Nbytes, hipMemcpyDeviceToHost)); for (int i=0; i