From bd1ffabfc1aaa9f50507599d49ef9bc3fb26fa39 Mon Sep 17 00:00:00 2001
From: sdashmiz <shadi.dashmiz@amd.com>
Date: Wed, 20 Jul 2022 13:33:03 -0400
Subject: [PATCH] SWDEV-347345 - add missing func for nvdia

Signed-off-by: sdashmiz <shadi.dashmiz@amd.com>
Change-Id: Ie2a70778d01948083d724a9d20e1f17541178ec5
---
 hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h b/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
index 3ae9bc8150..d76e9812a2 100644
--- a/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
+++ b/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
@@ -2826,6 +2826,10 @@ inline static hipError_t hipGraphMemsetNodeSetParams(hipGraphNode_t node,
     return hipCUDAErrorTohipError(cudaGraphMemsetNodeSetParams(node, pNodeParams));
 }
 
+inline static hipError_t hipThreadExchangeStreamCaptureMode(hipStreamCaptureMode* mode) {
+    return hipCUDAErrorTohipError(cudaThreadExchangeStreamCaptureMode(mode));
+}
+
 inline static hipError_t hipGraphExecKernelNodeSetParams(hipGraphExec_t hGraphExec,
                                                          hipGraphNode_t node,
                                                          const hipKernelNodeParams* pNodeParams) {