|
2 | 2 | # |
3 | 3 | # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE |
4 | 4 | # |
5 | | -# This code was automatically generated across versions from 12.9.1 to 13.2.0, generator version 0.3.1.dev1406+gd8426ea19.d20260316. Do not modify it directly. |
| 5 | +# This code was automatically generated across versions from 12.9.1 to 13.2.0, generator version 0.3.1.dev1422+gf4812259e.d20260318. Do not modify it directly. |
6 | 6 |
|
7 | 7 | from libc.stdint cimport intptr_t, uintptr_t |
8 | 8 |
|
@@ -188,6 +188,7 @@ cdef void* __nvmlDeviceGetDriverModel_v2 = NULL |
188 | 188 | cdef void* __nvmlDeviceGetVbiosVersion = NULL |
189 | 189 | cdef void* __nvmlDeviceGetBridgeChipInfo = NULL |
190 | 190 | cdef void* __nvmlDeviceGetComputeRunningProcesses_v3 = NULL |
| 191 | +cdef void* __nvmlDeviceGetGraphicsRunningProcesses_v3 = NULL |
191 | 192 | cdef void* __nvmlDeviceGetMPSComputeRunningProcesses_v3 = NULL |
192 | 193 | cdef void* __nvmlDeviceGetRunningProcessDetailList = NULL |
193 | 194 | cdef void* __nvmlDeviceOnSameBoard = NULL |
@@ -403,6 +404,13 @@ cdef void* __nvmlDeviceGetSramUniqueUncorrectedEccErrorCounts = NULL |
403 | 404 | cdef void* __nvmlDeviceGetUnrepairableMemoryFlag_v1 = NULL |
404 | 405 | cdef void* __nvmlDeviceReadPRMCounters_v1 = NULL |
405 | 406 | cdef void* __nvmlDeviceSetRusdSettings_v1 = NULL |
| 407 | +cdef void* __nvmlDeviceVgpuForceGspUnload = NULL |
| 408 | +cdef void* __nvmlDeviceGetVgpuSchedulerState_v2 = NULL |
| 409 | +cdef void* __nvmlGpuInstanceGetVgpuSchedulerState_v2 = NULL |
| 410 | +cdef void* __nvmlDeviceGetVgpuSchedulerLog_v2 = NULL |
| 411 | +cdef void* __nvmlGpuInstanceGetVgpuSchedulerLog_v2 = NULL |
| 412 | +cdef void* __nvmlDeviceSetVgpuSchedulerState_v2 = NULL |
| 413 | +cdef void* __nvmlGpuInstanceSetVgpuSchedulerState_v2 = NULL |
406 | 414 |
|
407 | 415 |
|
408 | 416 | cdef void* load_library() except* with gil: |
@@ -1316,6 +1324,13 @@ cdef int _init_nvml() except -1 nogil: |
1316 | 1324 | handle = load_library() |
1317 | 1325 | __nvmlDeviceGetComputeRunningProcesses_v3 = dlsym(handle, 'nvmlDeviceGetComputeRunningProcesses_v3') |
1318 | 1326 |
|
| 1327 | + global __nvmlDeviceGetGraphicsRunningProcesses_v3 |
| 1328 | + __nvmlDeviceGetGraphicsRunningProcesses_v3 = dlsym(RTLD_DEFAULT, 'nvmlDeviceGetGraphicsRunningProcesses_v3') |
| 1329 | + if __nvmlDeviceGetGraphicsRunningProcesses_v3 == NULL: |
| 1330 | + if handle == NULL: |
| 1331 | + handle = load_library() |
| 1332 | + __nvmlDeviceGetGraphicsRunningProcesses_v3 = dlsym(handle, 'nvmlDeviceGetGraphicsRunningProcesses_v3') |
| 1333 | + |
1319 | 1334 | global __nvmlDeviceGetMPSComputeRunningProcesses_v3 |
1320 | 1335 | __nvmlDeviceGetMPSComputeRunningProcesses_v3 = dlsym(RTLD_DEFAULT, 'nvmlDeviceGetMPSComputeRunningProcesses_v3') |
1321 | 1336 | if __nvmlDeviceGetMPSComputeRunningProcesses_v3 == NULL: |
@@ -2821,6 +2836,55 @@ cdef int _init_nvml() except -1 nogil: |
2821 | 2836 | handle = load_library() |
2822 | 2837 | __nvmlDeviceSetRusdSettings_v1 = dlsym(handle, 'nvmlDeviceSetRusdSettings_v1') |
2823 | 2838 |
|
| 2839 | + global __nvmlDeviceVgpuForceGspUnload |
| 2840 | + __nvmlDeviceVgpuForceGspUnload = dlsym(RTLD_DEFAULT, 'nvmlDeviceVgpuForceGspUnload') |
| 2841 | + if __nvmlDeviceVgpuForceGspUnload == NULL: |
| 2842 | + if handle == NULL: |
| 2843 | + handle = load_library() |
| 2844 | + __nvmlDeviceVgpuForceGspUnload = dlsym(handle, 'nvmlDeviceVgpuForceGspUnload') |
| 2845 | + |
| 2846 | + global __nvmlDeviceGetVgpuSchedulerState_v2 |
| 2847 | + __nvmlDeviceGetVgpuSchedulerState_v2 = dlsym(RTLD_DEFAULT, 'nvmlDeviceGetVgpuSchedulerState_v2') |
| 2848 | + if __nvmlDeviceGetVgpuSchedulerState_v2 == NULL: |
| 2849 | + if handle == NULL: |
| 2850 | + handle = load_library() |
| 2851 | + __nvmlDeviceGetVgpuSchedulerState_v2 = dlsym(handle, 'nvmlDeviceGetVgpuSchedulerState_v2') |
| 2852 | + |
| 2853 | + global __nvmlGpuInstanceGetVgpuSchedulerState_v2 |
| 2854 | + __nvmlGpuInstanceGetVgpuSchedulerState_v2 = dlsym(RTLD_DEFAULT, 'nvmlGpuInstanceGetVgpuSchedulerState_v2') |
| 2855 | + if __nvmlGpuInstanceGetVgpuSchedulerState_v2 == NULL: |
| 2856 | + if handle == NULL: |
| 2857 | + handle = load_library() |
| 2858 | + __nvmlGpuInstanceGetVgpuSchedulerState_v2 = dlsym(handle, 'nvmlGpuInstanceGetVgpuSchedulerState_v2') |
| 2859 | + |
| 2860 | + global __nvmlDeviceGetVgpuSchedulerLog_v2 |
| 2861 | + __nvmlDeviceGetVgpuSchedulerLog_v2 = dlsym(RTLD_DEFAULT, 'nvmlDeviceGetVgpuSchedulerLog_v2') |
| 2862 | + if __nvmlDeviceGetVgpuSchedulerLog_v2 == NULL: |
| 2863 | + if handle == NULL: |
| 2864 | + handle = load_library() |
| 2865 | + __nvmlDeviceGetVgpuSchedulerLog_v2 = dlsym(handle, 'nvmlDeviceGetVgpuSchedulerLog_v2') |
| 2866 | + |
| 2867 | + global __nvmlGpuInstanceGetVgpuSchedulerLog_v2 |
| 2868 | + __nvmlGpuInstanceGetVgpuSchedulerLog_v2 = dlsym(RTLD_DEFAULT, 'nvmlGpuInstanceGetVgpuSchedulerLog_v2') |
| 2869 | + if __nvmlGpuInstanceGetVgpuSchedulerLog_v2 == NULL: |
| 2870 | + if handle == NULL: |
| 2871 | + handle = load_library() |
| 2872 | + __nvmlGpuInstanceGetVgpuSchedulerLog_v2 = dlsym(handle, 'nvmlGpuInstanceGetVgpuSchedulerLog_v2') |
| 2873 | + |
| 2874 | + global __nvmlDeviceSetVgpuSchedulerState_v2 |
| 2875 | + __nvmlDeviceSetVgpuSchedulerState_v2 = dlsym(RTLD_DEFAULT, 'nvmlDeviceSetVgpuSchedulerState_v2') |
| 2876 | + if __nvmlDeviceSetVgpuSchedulerState_v2 == NULL: |
| 2877 | + if handle == NULL: |
| 2878 | + handle = load_library() |
| 2879 | + __nvmlDeviceSetVgpuSchedulerState_v2 = dlsym(handle, 'nvmlDeviceSetVgpuSchedulerState_v2') |
| 2880 | + |
| 2881 | + global __nvmlGpuInstanceSetVgpuSchedulerState_v2 |
| 2882 | + __nvmlGpuInstanceSetVgpuSchedulerState_v2 = dlsym(RTLD_DEFAULT, 'nvmlGpuInstanceSetVgpuSchedulerState_v2') |
| 2883 | + if __nvmlGpuInstanceSetVgpuSchedulerState_v2 == NULL: |
| 2884 | + if handle == NULL: |
| 2885 | + handle = load_library() |
| 2886 | + __nvmlGpuInstanceSetVgpuSchedulerState_v2 = dlsym(handle, 'nvmlGpuInstanceSetVgpuSchedulerState_v2') |
| 2887 | + |
2824 | 2888 | __py_nvml_init = True |
2825 | 2889 | return 0 |
2826 | 2890 |
|
@@ -3227,6 +3291,9 @@ cpdef dict _inspect_function_pointers(): |
3227 | 3291 | global __nvmlDeviceGetComputeRunningProcesses_v3 |
3228 | 3292 | data["__nvmlDeviceGetComputeRunningProcesses_v3"] = <intptr_t>__nvmlDeviceGetComputeRunningProcesses_v3 |
3229 | 3293 |
|
| 3294 | + global __nvmlDeviceGetGraphicsRunningProcesses_v3 |
| 3295 | + data["__nvmlDeviceGetGraphicsRunningProcesses_v3"] = <intptr_t>__nvmlDeviceGetGraphicsRunningProcesses_v3 |
| 3296 | + |
3230 | 3297 | global __nvmlDeviceGetMPSComputeRunningProcesses_v3 |
3231 | 3298 | data["__nvmlDeviceGetMPSComputeRunningProcesses_v3"] = <intptr_t>__nvmlDeviceGetMPSComputeRunningProcesses_v3 |
3232 | 3299 |
|
@@ -3872,6 +3939,27 @@ cpdef dict _inspect_function_pointers(): |
3872 | 3939 | global __nvmlDeviceSetRusdSettings_v1 |
3873 | 3940 | data["__nvmlDeviceSetRusdSettings_v1"] = <intptr_t>__nvmlDeviceSetRusdSettings_v1 |
3874 | 3941 |
|
| 3942 | + global __nvmlDeviceVgpuForceGspUnload |
| 3943 | + data["__nvmlDeviceVgpuForceGspUnload"] = <intptr_t>__nvmlDeviceVgpuForceGspUnload |
| 3944 | + |
| 3945 | + global __nvmlDeviceGetVgpuSchedulerState_v2 |
| 3946 | + data["__nvmlDeviceGetVgpuSchedulerState_v2"] = <intptr_t>__nvmlDeviceGetVgpuSchedulerState_v2 |
| 3947 | + |
| 3948 | + global __nvmlGpuInstanceGetVgpuSchedulerState_v2 |
| 3949 | + data["__nvmlGpuInstanceGetVgpuSchedulerState_v2"] = <intptr_t>__nvmlGpuInstanceGetVgpuSchedulerState_v2 |
| 3950 | + |
| 3951 | + global __nvmlDeviceGetVgpuSchedulerLog_v2 |
| 3952 | + data["__nvmlDeviceGetVgpuSchedulerLog_v2"] = <intptr_t>__nvmlDeviceGetVgpuSchedulerLog_v2 |
| 3953 | + |
| 3954 | + global __nvmlGpuInstanceGetVgpuSchedulerLog_v2 |
| 3955 | + data["__nvmlGpuInstanceGetVgpuSchedulerLog_v2"] = <intptr_t>__nvmlGpuInstanceGetVgpuSchedulerLog_v2 |
| 3956 | + |
| 3957 | + global __nvmlDeviceSetVgpuSchedulerState_v2 |
| 3958 | + data["__nvmlDeviceSetVgpuSchedulerState_v2"] = <intptr_t>__nvmlDeviceSetVgpuSchedulerState_v2 |
| 3959 | + |
| 3960 | + global __nvmlGpuInstanceSetVgpuSchedulerState_v2 |
| 3961 | + data["__nvmlGpuInstanceSetVgpuSchedulerState_v2"] = <intptr_t>__nvmlGpuInstanceSetVgpuSchedulerState_v2 |
| 3962 | + |
3875 | 3963 | func_ptrs = data |
3876 | 3964 | return data |
3877 | 3965 |
|
@@ -5167,6 +5255,16 @@ cdef nvmlReturn_t _nvmlDeviceGetComputeRunningProcesses_v3(nvmlDevice_t device, |
5167 | 5255 | device, infoCount, infos) |
5168 | 5256 |
|
5169 | 5257 |
|
| 5258 | +cdef nvmlReturn_t _nvmlDeviceGetGraphicsRunningProcesses_v3(nvmlDevice_t device, unsigned int* infoCount, nvmlProcessInfo_t* infos) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 5259 | + global __nvmlDeviceGetGraphicsRunningProcesses_v3 |
| 5260 | + _check_or_init_nvml() |
| 5261 | + if __nvmlDeviceGetGraphicsRunningProcesses_v3 == NULL: |
| 5262 | + with gil: |
| 5263 | + raise FunctionNotFoundError("function nvmlDeviceGetGraphicsRunningProcesses_v3 is not found") |
| 5264 | + return (<nvmlReturn_t (*)(nvmlDevice_t, unsigned int*, nvmlProcessInfo_t*) noexcept nogil>__nvmlDeviceGetGraphicsRunningProcesses_v3)( |
| 5265 | + device, infoCount, infos) |
| 5266 | + |
| 5267 | + |
5170 | 5268 | cdef nvmlReturn_t _nvmlDeviceGetMPSComputeRunningProcesses_v3(nvmlDevice_t device, unsigned int* infoCount, nvmlProcessInfo_t* infos) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
5171 | 5269 | global __nvmlDeviceGetMPSComputeRunningProcesses_v3 |
5172 | 5270 | _check_or_init_nvml() |
@@ -7315,3 +7413,73 @@ cdef nvmlReturn_t _nvmlDeviceSetRusdSettings_v1(nvmlDevice_t device, nvmlRusdSet |
7315 | 7413 | raise FunctionNotFoundError("function nvmlDeviceSetRusdSettings_v1 is not found") |
7316 | 7414 | return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlRusdSettings_v1_t*) noexcept nogil>__nvmlDeviceSetRusdSettings_v1)( |
7317 | 7415 | device, settings) |
| 7416 | + |
| 7417 | + |
| 7418 | +cdef nvmlReturn_t _nvmlDeviceVgpuForceGspUnload(nvmlDevice_t device) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 7419 | + global __nvmlDeviceVgpuForceGspUnload |
| 7420 | + _check_or_init_nvml() |
| 7421 | + if __nvmlDeviceVgpuForceGspUnload == NULL: |
| 7422 | + with gil: |
| 7423 | + raise FunctionNotFoundError("function nvmlDeviceVgpuForceGspUnload is not found") |
| 7424 | + return (<nvmlReturn_t (*)(nvmlDevice_t) noexcept nogil>__nvmlDeviceVgpuForceGspUnload)( |
| 7425 | + device) |
| 7426 | + |
| 7427 | + |
| 7428 | +cdef nvmlReturn_t _nvmlDeviceGetVgpuSchedulerState_v2(nvmlDevice_t device, nvmlVgpuSchedulerStateInfo_v2_t* pSchedulerStateInfo) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 7429 | + global __nvmlDeviceGetVgpuSchedulerState_v2 |
| 7430 | + _check_or_init_nvml() |
| 7431 | + if __nvmlDeviceGetVgpuSchedulerState_v2 == NULL: |
| 7432 | + with gil: |
| 7433 | + raise FunctionNotFoundError("function nvmlDeviceGetVgpuSchedulerState_v2 is not found") |
| 7434 | + return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlVgpuSchedulerStateInfo_v2_t*) noexcept nogil>__nvmlDeviceGetVgpuSchedulerState_v2)( |
| 7435 | + device, pSchedulerStateInfo) |
| 7436 | + |
| 7437 | + |
| 7438 | +cdef nvmlReturn_t _nvmlGpuInstanceGetVgpuSchedulerState_v2(nvmlGpuInstance_t gpuInstance, nvmlVgpuSchedulerStateInfo_v2_t* pSchedulerStateInfo) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 7439 | + global __nvmlGpuInstanceGetVgpuSchedulerState_v2 |
| 7440 | + _check_or_init_nvml() |
| 7441 | + if __nvmlGpuInstanceGetVgpuSchedulerState_v2 == NULL: |
| 7442 | + with gil: |
| 7443 | + raise FunctionNotFoundError("function nvmlGpuInstanceGetVgpuSchedulerState_v2 is not found") |
| 7444 | + return (<nvmlReturn_t (*)(nvmlGpuInstance_t, nvmlVgpuSchedulerStateInfo_v2_t*) noexcept nogil>__nvmlGpuInstanceGetVgpuSchedulerState_v2)( |
| 7445 | + gpuInstance, pSchedulerStateInfo) |
| 7446 | + |
| 7447 | + |
| 7448 | +cdef nvmlReturn_t _nvmlDeviceGetVgpuSchedulerLog_v2(nvmlDevice_t device, nvmlVgpuSchedulerLogInfo_v2_t* pSchedulerLogInfo) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 7449 | + global __nvmlDeviceGetVgpuSchedulerLog_v2 |
| 7450 | + _check_or_init_nvml() |
| 7451 | + if __nvmlDeviceGetVgpuSchedulerLog_v2 == NULL: |
| 7452 | + with gil: |
| 7453 | + raise FunctionNotFoundError("function nvmlDeviceGetVgpuSchedulerLog_v2 is not found") |
| 7454 | + return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlVgpuSchedulerLogInfo_v2_t*) noexcept nogil>__nvmlDeviceGetVgpuSchedulerLog_v2)( |
| 7455 | + device, pSchedulerLogInfo) |
| 7456 | + |
| 7457 | + |
| 7458 | +cdef nvmlReturn_t _nvmlGpuInstanceGetVgpuSchedulerLog_v2(nvmlGpuInstance_t gpuInstance, nvmlVgpuSchedulerLogInfo_v2_t* pSchedulerLogInfo) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 7459 | + global __nvmlGpuInstanceGetVgpuSchedulerLog_v2 |
| 7460 | + _check_or_init_nvml() |
| 7461 | + if __nvmlGpuInstanceGetVgpuSchedulerLog_v2 == NULL: |
| 7462 | + with gil: |
| 7463 | + raise FunctionNotFoundError("function nvmlGpuInstanceGetVgpuSchedulerLog_v2 is not found") |
| 7464 | + return (<nvmlReturn_t (*)(nvmlGpuInstance_t, nvmlVgpuSchedulerLogInfo_v2_t*) noexcept nogil>__nvmlGpuInstanceGetVgpuSchedulerLog_v2)( |
| 7465 | + gpuInstance, pSchedulerLogInfo) |
| 7466 | + |
| 7467 | + |
| 7468 | +cdef nvmlReturn_t _nvmlDeviceSetVgpuSchedulerState_v2(nvmlDevice_t device, nvmlVgpuSchedulerState_v2_t* pSchedulerState) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 7469 | + global __nvmlDeviceSetVgpuSchedulerState_v2 |
| 7470 | + _check_or_init_nvml() |
| 7471 | + if __nvmlDeviceSetVgpuSchedulerState_v2 == NULL: |
| 7472 | + with gil: |
| 7473 | + raise FunctionNotFoundError("function nvmlDeviceSetVgpuSchedulerState_v2 is not found") |
| 7474 | + return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlVgpuSchedulerState_v2_t*) noexcept nogil>__nvmlDeviceSetVgpuSchedulerState_v2)( |
| 7475 | + device, pSchedulerState) |
| 7476 | + |
| 7477 | + |
| 7478 | +cdef nvmlReturn_t _nvmlGpuInstanceSetVgpuSchedulerState_v2(nvmlGpuInstance_t gpuInstance, nvmlVgpuSchedulerState_v2_t* pSchedulerState) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil: |
| 7479 | + global __nvmlGpuInstanceSetVgpuSchedulerState_v2 |
| 7480 | + _check_or_init_nvml() |
| 7481 | + if __nvmlGpuInstanceSetVgpuSchedulerState_v2 == NULL: |
| 7482 | + with gil: |
| 7483 | + raise FunctionNotFoundError("function nvmlGpuInstanceSetVgpuSchedulerState_v2 is not found") |
| 7484 | + return (<nvmlReturn_t (*)(nvmlGpuInstance_t, nvmlVgpuSchedulerState_v2_t*) noexcept nogil>__nvmlGpuInstanceSetVgpuSchedulerState_v2)( |
| 7485 | + gpuInstance, pSchedulerState) |
0 commit comments