From 9466ab4fcafdd727465f426b9e4073d1becefa12 Mon Sep 17 00:00:00 2001
From: Ben Sander <ben.sander@amd.com>
Date: Fri, 2 Sep 2016 17:24:16 -0500
Subject: [PATCH] add hipStreamQuery on nvcc, fix hipStreamWaitEvent for nvcc

---
 hipamd/include/nvcc_detail/hip_runtime_api.h              | 5 +++++
 hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp | 1 +
 2 files changed, 6 insertions(+)

diff --git a/hipamd/include/nvcc_detail/hip_runtime_api.h b/hipamd/include/nvcc_detail/hip_runtime_api.h
index e2bcf41dd9..8e9b0d92a7 100644
--- a/hipamd/include/nvcc_detail/hip_runtime_api.h
+++ b/hipamd/include/nvcc_detail/hip_runtime_api.h
@@ -489,6 +489,11 @@ inline static hipError_t hipStreamWaitEvent(hipStream_t stream, hipEvent_t event
     return hipCUDAErrorTohipError(cudaStreamWaitEvent(stream, event, flags));
 }
 
+inline static hipError_t hipStreamQuery(hipStream_t stream)
+{
+    return hipCUDAErrorTohipError(cudaStreamQuery(stream));
+}
+
 
 inline static hipError_t hipDriverGetVersion(int *driverVersion)
 {
diff --git a/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp b/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp
index 7bdae6a96f..c5a74b2bc0 100644
--- a/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp
+++ b/hipamd/tests/src/runtimeApi/stream/hipStreamWaitEvent.cpp
@@ -21,6 +21,7 @@ THE SOFTWARE.
 
 #include "hip_runtime.h"
 #include "test_common.h"
+#include <vector>
 unsigned p_streams = 6;