Merge pull request #31 from hzliangbin/master

add cuGetProcAddress to cuda_hook_entry list
tkestack · Nov 15, 2022 · 72e0115 · 72e0115
2 parents 5a2600c + a3542ed
commit 72e0115
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/hijack_call.c b/src/hijack_call.c
@@ -94,6 +94,8 @@ static int delta(int, int, int);
 /** export function definition */
 CUresult cuDriverGetVersion(int *driverVersion);
 CUresult cuInit(unsigned int flag);
+CUresult cuGetProcAddress(const char *symbol, void **pfn, int cudaVersion,
+                          cuuint64_t flags); 
 CUresult cuMemAllocManaged(CUdeviceptr *dptr, size_t bytesize,
                            unsigned int flags);
 CUresult cuMemAlloc_v2(CUdeviceptr *dptr, size_t bytesize);
@@ -151,6 +153,7 @@ CUresult cuFuncSetBlockShape(CUfunction hfunc, int x, int y, int z);
 entry_t cuda_hooks_entry[] = {
     {.name = "cuDriverGetVersion", .fn_ptr = cuDriverGetVersion},
     {.name = "cuInit", .fn_ptr = cuInit},
+    {.name = "cuGetProcAddress", .fn_ptr = cuGetProcAddress},
     {.name = "cuMemAllocManaged", .fn_ptr = cuMemAllocManaged},
     {.name = "cuMemAlloc_v2", .fn_ptr = cuMemAlloc_v2},
     {.name = "cuMemAlloc", .fn_ptr = cuMemAlloc},