Skip to content

Commit 72e0115

Browse files
authored
Merge pull request #31 from hzliangbin/master
add cuGetProcAddress to cuda_hook_entry list
2 parents 5a2600c + a3542ed commit 72e0115

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/hijack_call.c

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,8 @@ static int delta(int, int, int);
9494
/** export function definition */
9595
CUresult cuDriverGetVersion(int *driverVersion);
9696
CUresult cuInit(unsigned int flag);
97+
CUresult cuGetProcAddress(const char *symbol, void **pfn, int cudaVersion,
98+
cuuint64_t flags);
9799
CUresult cuMemAllocManaged(CUdeviceptr *dptr, size_t bytesize,
98100
unsigned int flags);
99101
CUresult cuMemAlloc_v2(CUdeviceptr *dptr, size_t bytesize);
@@ -151,6 +153,7 @@ CUresult cuFuncSetBlockShape(CUfunction hfunc, int x, int y, int z);
151153
entry_t cuda_hooks_entry[] = {
152154
{.name = "cuDriverGetVersion", .fn_ptr = cuDriverGetVersion},
153155
{.name = "cuInit", .fn_ptr = cuInit},
156+
{.name = "cuGetProcAddress", .fn_ptr = cuGetProcAddress},
154157
{.name = "cuMemAllocManaged", .fn_ptr = cuMemAllocManaged},
155158
{.name = "cuMemAlloc_v2", .fn_ptr = cuMemAlloc_v2},
156159
{.name = "cuMemAlloc", .fn_ptr = cuMemAlloc},

0 commit comments

Comments
 (0)