From 87f262a56faea6733a4cfd62c093091993b35ac7 Mon Sep 17 00:00:00 2001
From: "Li, Todd tiantuo" <Toddtiantuo.Li@amd.com>
Date: Fri, 15 Aug 2025 22:38:34 -0700
Subject: [PATCH] SWDEV-546285 - add hipGetDriverEntryPoint (#35)

---
 .../hip/nvidia_detail/nvidia_hip_runtime_api.h   | 16 ++++++++++++++++
 1 file changed, 16 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 01dc583edc..047538fab1 100644
--- a/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
+++ b/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h
@@ -1000,6 +1000,15 @@ typedef enum cudaGraphInstantiateFlags hipGraphInstantiateFlags;
 #define hipGraphInstantiateFlagDeviceLaunch cudaGraphInstantiateFlagDeviceLaunch
 #define hipGraphInstantiateFlagUseNodePriority cudaGraphInstantiateFlagUseNodePriority
 
+typedef enum cudaDriverEntryPointQueryResult hipDriverEntryPointQueryResult;
+#define hipDriverEntryPointSuccess cudaDriverEntryPointSuccess
+#define hipDriverEntryPointSymbolNotFound cudaDriverEntryPointSymbolNotFound
+#define hipDriverEntryPointVersionNotSufficent cudaDriverEntryPointVersionNotSufficent
+
+#define hipEnableDefault cudaEnableDefault
+#define hipEnableLegacyStream cudaEnableLegacyStream
+#define hipEnablePerThreadDefaultStream cudaEnablePerThreadDefaultStream
+
 inline static hipError_t hipCUDAErrorTohipError(cudaError_t cuError) {
     switch (cuError) {
         case cudaSuccess:
@@ -3722,6 +3731,13 @@ inline static hipError_t hipGetChannelDesc(hipChannelFormatDesc* desc, hipArray_
     return hipCUDAErrorTohipError(cudaGetChannelDesc(desc,array));
 }
 
+inline static hipError_t hipGetDriverEntryPoint(const char* symbol, void** funcPtr,
+                                                unsigned long long flags,
+                                                hipDriverEntryPointQueryResult* driverStatus)
+{
+    return hipCUDAErrorTohipError(cudaGetDriverEntryPoint(symbol, funcPtr, flags, driverStatus));
+}
+
 inline static hipError_t hipLaunchCooperativeKernel(const void* f, dim3 gridDim, dim3 blockDim,
                                       void** kernelParams, unsigned int sharedMemBytes,
                                       hipStream_t stream) {