From fc0d9521f250a1fb3c246129aedf2bc50454d62d 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


[ROCm/clr commit: fefb9ae466015b57ea7a473c33293b093062e706]
---
 .../hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/projects/clr/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h b/projects/clr/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
index 3ae9bc8150..d76e9812a2 100644
--- a/projects/clr/hipamd/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
+++ b/projects/clr/hipamd/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) {