From dd55ba8b5d0c1fa97c9ac4f7f4b77f91fcdf6746 Mon Sep 17 00:00:00 2001
From: Aditya Atluri <Aditya.Atluri@amd.com>
Date: Mon, 3 Oct 2016 12:38:41 -0500
Subject: [PATCH] Stream fix on nvcc hipMemsetAsync

Change-Id: Ia0eb81dff0f422af55d93b4635d42e9aa6921377


[ROCm/hip commit: 9f05eca0a01b9e4d54228e9853d602d4560b89f8]
---
 projects/hip/include/nvcc_detail/hip_runtime_api.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/projects/hip/include/nvcc_detail/hip_runtime_api.h b/projects/hip/include/nvcc_detail/hip_runtime_api.h
index a36b951400..b5dee3898f 100644
--- a/projects/hip/include/nvcc_detail/hip_runtime_api.h
+++ b/projects/hip/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)