From 3edff7dd3bd9bd59dfbf2951911591fd24161970 Mon Sep 17 00:00:00 2001 From: Aditya Atluri Date: Mon, 3 Oct 2016 12:38:41 -0500 Subject: [PATCH] Stream fix on nvcc hipMemsetAsync Change-Id: Ia0eb81dff0f422af55d93b4635d42e9aa6921377 --- hipamd/include/nvcc_detail/hip_runtime_api.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hipamd/include/nvcc_detail/hip_runtime_api.h b/hipamd/include/nvcc_detail/hip_runtime_api.h index a36b951400..b5dee3898f 100644 --- a/hipamd/include/nvcc_detail/hip_runtime_api.h +++ b/hipamd/include/nvcc_detail/hip_runtime_api.h @@ -329,8 +329,8 @@ inline static hipError_t hipMemset(void* devPtr,int value, size_t count) { return hipCUDAErrorTohipError(cudaMemset(devPtr, value, count)); } -inline static hipError_t hipMemsetAsync(void* devPtr,int value, size_t count) { - return hipCUDAErrorTohipError(cudaMemsetAsync(devPtr, value, count)); +inline static hipError_t hipMemsetAsync(void* devPtr,int value, size_t count, hipStream stream = 0) { + return hipCUDAErrorTohipError(cudaMemsetAsync(devPtr, value, count, stream)); } inline static hipError_t hipGetDeviceProperties(hipDeviceProp_t *p_prop, int device)