|
2 | 2 | # |
3 | 3 | # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE |
4 | 4 | # |
5 | | -# This code was automatically generated with version 12.9.0. Do not modify it directly. |
| 5 | +# This code was automatically generated across versions from 12.9.0 to 12.9.1. Do not modify it directly. |
6 | 6 |
|
7 | 7 | from libc.stdint cimport intptr_t, uintptr_t |
8 | 8 | import threading |
@@ -71,6 +71,7 @@ cdef void* __cuFileGetParameterString = NULL |
71 | 71 | cdef void* __cuFileSetParameterSizeT = NULL |
72 | 72 | cdef void* __cuFileSetParameterBool = NULL |
73 | 73 | cdef void* __cuFileSetParameterString = NULL |
| 74 | +cdef void* __cuFileDriverClose = NULL |
74 | 75 |
|
75 | 76 |
|
76 | 77 | cdef void* load_library(const int driver_ver) except* with gil: |
@@ -314,6 +315,13 @@ cdef int _check_or_init_cufile() except -1 nogil: |
314 | 315 | handle = load_library(driver_ver) |
315 | 316 | __cuFileSetParameterString = dlsym(handle, 'cuFileSetParameterString') |
316 | 317 |
|
| 318 | + global __cuFileDriverClose |
| 319 | + __cuFileDriverClose = dlsym(RTLD_DEFAULT, 'cuFileDriverClose') |
| 320 | + if __cuFileDriverClose == NULL: |
| 321 | + if handle == NULL: |
| 322 | + handle = load_library(driver_ver) |
| 323 | + __cuFileDriverClose = dlsym(handle, 'cuFileDriverClose') |
| 324 | + |
317 | 325 | __py_cufile_init = True |
318 | 326 | return 0 |
319 | 327 |
|
@@ -419,6 +427,9 @@ cpdef dict _inspect_function_pointers(): |
419 | 427 | global __cuFileSetParameterString |
420 | 428 | data["__cuFileSetParameterString"] = <intptr_t>__cuFileSetParameterString |
421 | 429 |
|
| 430 | + global __cuFileDriverClose |
| 431 | + data["__cuFileDriverClose"] = <intptr_t>__cuFileDriverClose |
| 432 | + |
422 | 433 | func_ptrs = data |
423 | 434 | return data |
424 | 435 |
|
@@ -734,3 +745,13 @@ cdef CUfileError_t _cuFileSetParameterString(CUFileStringConfigParameter_t param |
734 | 745 | raise FunctionNotFoundError("function cuFileSetParameterString is not found") |
735 | 746 | return (<CUfileError_t (*)(CUFileStringConfigParameter_t, const char*) noexcept nogil>__cuFileSetParameterString)( |
736 | 747 | param, desc_str) |
| 748 | + |
| 749 | + |
| 750 | +cdef CUfileError_t _cuFileDriverClose() except?<CUfileError_t>CUFILE_LOADING_ERROR nogil: |
| 751 | + global __cuFileDriverClose |
| 752 | + _check_or_init_cufile() |
| 753 | + if __cuFileDriverClose == NULL: |
| 754 | + with gil: |
| 755 | + raise FunctionNotFoundError("function cuFileDriverClose is not found") |
| 756 | + return (<CUfileError_t (*)() noexcept nogil>__cuFileDriverClose)( |
| 757 | + ) |
0 commit comments