Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 5 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/nvml.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
#
# This code was automatically generated across versions from 12.9.1 to 13.3.0. Do not modify it directly.

# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=ed4c433854399c2e4f1adc3ffcfc37901e1be29c5aa50728498c87225edf92b1
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=c2cc3cd086b5aeea5fad7ca17600d0102691a3cb354b916f5c086c383d77df19
from ..cynvml cimport *


Expand Down Expand Up @@ -363,3 +363,7 @@ cdef nvmlReturn_t _nvmlDeviceGetVgpuSchedulerLog_v2(nvmlDevice_t device, nvmlVgp
cdef nvmlReturn_t _nvmlGpuInstanceGetVgpuSchedulerLog_v2(nvmlGpuInstance_t gpuInstance, nvmlVgpuSchedulerLogInfo_v2_t* pSchedulerLogInfo) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceSetVgpuSchedulerState_v2(nvmlDevice_t device, nvmlVgpuSchedulerState_v2_t* pSchedulerState) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlGpuInstanceSetVgpuSchedulerState_v2(nvmlGpuInstance_t gpuInstance, nvmlVgpuSchedulerState_v2_t* pSchedulerState) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlSystemGetCPER_v1(nvmlGetCPER_v1_t* cper) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceGetBBXTimeData_v1(nvmlDevice_t device, nvmlBBXTimeData_v1_t* timeData) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceGetAccountingStats_v2(nvmlDevice_t device, nvmlAccountingStats_v2_t* stats) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
cdef nvmlReturn_t _nvmlDeviceGetRemappedRows_v2(nvmlDevice_t device, nvmlRemappedRowsInfo_v2_t* info) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil
86 changes: 85 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/nvml_linux.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# SPDX-License-Identifier: Apache-2.0
#
# This code was automatically generated across versions from 12.9.1 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=646d902675de6987cff08e5702d3b2bff889e6a827273e57413aa5de45a5897a
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=45cd03eeb8717b33a1e11d83ae99ac8d6b580e4c57fad99d185614bd0047faf3


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -415,6 +415,10 @@ cdef void* __nvmlDeviceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlGpuInstanceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlDeviceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlGpuInstanceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlSystemGetCPER_v1 = NULL
cdef void* __nvmlDeviceGetBBXTimeData_v1 = NULL
cdef void* __nvmlDeviceGetAccountingStats_v2 = NULL
cdef void* __nvmlDeviceGetRemappedRows_v2 = NULL

cdef int _init_nvml() except -1 nogil:
global _cyb___py_nvml_init
Expand Down Expand Up @@ -2879,6 +2883,34 @@ cdef int _init_nvml() except -1 nogil:
handle = load_library()
__nvmlGpuInstanceSetVgpuSchedulerState_v2 = _cyb_dlsym(handle, 'nvmlGpuInstanceSetVgpuSchedulerState_v2')

global __nvmlSystemGetCPER_v1
__nvmlSystemGetCPER_v1 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlSystemGetCPER_v1')
if __nvmlSystemGetCPER_v1 == NULL:
if handle == NULL:
handle = load_library()
__nvmlSystemGetCPER_v1 = _cyb_dlsym(handle, 'nvmlSystemGetCPER_v1')

global __nvmlDeviceGetBBXTimeData_v1
__nvmlDeviceGetBBXTimeData_v1 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlDeviceGetBBXTimeData_v1')
if __nvmlDeviceGetBBXTimeData_v1 == NULL:
if handle == NULL:
handle = load_library()
__nvmlDeviceGetBBXTimeData_v1 = _cyb_dlsym(handle, 'nvmlDeviceGetBBXTimeData_v1')

global __nvmlDeviceGetAccountingStats_v2
__nvmlDeviceGetAccountingStats_v2 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlDeviceGetAccountingStats_v2')
if __nvmlDeviceGetAccountingStats_v2 == NULL:
if handle == NULL:
handle = load_library()
__nvmlDeviceGetAccountingStats_v2 = _cyb_dlsym(handle, 'nvmlDeviceGetAccountingStats_v2')

global __nvmlDeviceGetRemappedRows_v2
__nvmlDeviceGetRemappedRows_v2 = _cyb_dlsym(_cyb_RTLD_DEFAULT, 'nvmlDeviceGetRemappedRows_v2')
if __nvmlDeviceGetRemappedRows_v2 == NULL:
if handle == NULL:
handle = load_library()
__nvmlDeviceGetRemappedRows_v2 = _cyb_dlsym(handle, 'nvmlDeviceGetRemappedRows_v2')

_cyb_atomic_int_store(<int *>&_cyb___py_nvml_init, 1)
return 0

Expand Down Expand Up @@ -3948,6 +3980,18 @@ cpdef dict _inspect_function_pointers():

global __nvmlGpuInstanceSetVgpuSchedulerState_v2
data["__nvmlGpuInstanceSetVgpuSchedulerState_v2"] = <_cyb_intptr_t>__nvmlGpuInstanceSetVgpuSchedulerState_v2

global __nvmlSystemGetCPER_v1
data["__nvmlSystemGetCPER_v1"] = <_cyb_intptr_t>__nvmlSystemGetCPER_v1

global __nvmlDeviceGetBBXTimeData_v1
data["__nvmlDeviceGetBBXTimeData_v1"] = <_cyb_intptr_t>__nvmlDeviceGetBBXTimeData_v1

global __nvmlDeviceGetAccountingStats_v2
data["__nvmlDeviceGetAccountingStats_v2"] = <_cyb_intptr_t>__nvmlDeviceGetAccountingStats_v2

global __nvmlDeviceGetRemappedRows_v2
data["__nvmlDeviceGetRemappedRows_v2"] = <_cyb_intptr_t>__nvmlDeviceGetRemappedRows_v2
_cyb_func_ptrs = data
return data

Expand Down Expand Up @@ -7478,3 +7522,43 @@ cdef nvmlReturn_t _nvmlGpuInstanceSetVgpuSchedulerState_v2(nvmlGpuInstance_t gpu
raise FunctionNotFoundError("function nvmlGpuInstanceSetVgpuSchedulerState_v2 is not found")
return (<nvmlReturn_t (*)(nvmlGpuInstance_t, nvmlVgpuSchedulerState_v2_t*) noexcept nogil>__nvmlGpuInstanceSetVgpuSchedulerState_v2)(
gpuInstance, pSchedulerState)


cdef nvmlReturn_t _nvmlSystemGetCPER_v1(nvmlGetCPER_v1_t* cper) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlSystemGetCPER_v1
_check_or_init_nvml()
if __nvmlSystemGetCPER_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlSystemGetCPER_v1 is not found")
return (<nvmlReturn_t (*)(nvmlGetCPER_v1_t*) noexcept nogil>__nvmlSystemGetCPER_v1)(
cper)


cdef nvmlReturn_t _nvmlDeviceGetBBXTimeData_v1(nvmlDevice_t device, nvmlBBXTimeData_v1_t* timeData) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetBBXTimeData_v1
_check_or_init_nvml()
if __nvmlDeviceGetBBXTimeData_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetBBXTimeData_v1 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlBBXTimeData_v1_t*) noexcept nogil>__nvmlDeviceGetBBXTimeData_v1)(
device, timeData)


cdef nvmlReturn_t _nvmlDeviceGetAccountingStats_v2(nvmlDevice_t device, nvmlAccountingStats_v2_t* stats) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetAccountingStats_v2
_check_or_init_nvml()
if __nvmlDeviceGetAccountingStats_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetAccountingStats_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlAccountingStats_v2_t*) noexcept nogil>__nvmlDeviceGetAccountingStats_v2)(
device, stats)


cdef nvmlReturn_t _nvmlDeviceGetRemappedRows_v2(nvmlDevice_t device, nvmlRemappedRowsInfo_v2_t* info) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetRemappedRows_v2
_check_or_init_nvml()
if __nvmlDeviceGetRemappedRows_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetRemappedRows_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlRemappedRowsInfo_v2_t*) noexcept nogil>__nvmlDeviceGetRemappedRows_v2)(
device, info)
70 changes: 69 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/nvml_windows.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
# SPDX-License-Identifier: Apache-2.0
#
# This code was automatically generated across versions from 12.9.1 to 13.3.0. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=831330186c4a7bb029b953be6dd3cda119a7f10c56fa9378ab621fbc4374f9d1
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=d7b5ba031ed135b60903431812f9883efdd554268990e04003d5ead5674466eb


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -412,6 +412,10 @@ cdef void* __nvmlDeviceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlGpuInstanceGetVgpuSchedulerLog_v2 = NULL
cdef void* __nvmlDeviceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlGpuInstanceSetVgpuSchedulerState_v2 = NULL
cdef void* __nvmlSystemGetCPER_v1 = NULL
cdef void* __nvmlDeviceGetBBXTimeData_v1 = NULL
cdef void* __nvmlDeviceGetAccountingStats_v2 = NULL
cdef void* __nvmlDeviceGetRemappedRows_v2 = NULL

cdef int _init_nvml() except -1 nogil:
global _cyb___py_nvml_init
Expand Down Expand Up @@ -1475,6 +1479,18 @@ cdef int _init_nvml() except -1 nogil:
global __nvmlGpuInstanceSetVgpuSchedulerState_v2
__nvmlGpuInstanceSetVgpuSchedulerState_v2 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlGpuInstanceSetVgpuSchedulerState_v2')

global __nvmlSystemGetCPER_v1
__nvmlSystemGetCPER_v1 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlSystemGetCPER_v1')

global __nvmlDeviceGetBBXTimeData_v1
__nvmlDeviceGetBBXTimeData_v1 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlDeviceGetBBXTimeData_v1')

global __nvmlDeviceGetAccountingStats_v2
__nvmlDeviceGetAccountingStats_v2 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlDeviceGetAccountingStats_v2')

global __nvmlDeviceGetRemappedRows_v2
__nvmlDeviceGetRemappedRows_v2 = _cyb_GetProcAddress(<HMODULE>handle, 'nvmlDeviceGetRemappedRows_v2')

_cyb_atomic_int_store(<int *>&_cyb___py_nvml_init, 1)
return 0

Expand Down Expand Up @@ -2544,6 +2560,18 @@ cpdef dict _inspect_function_pointers():

global __nvmlGpuInstanceSetVgpuSchedulerState_v2
data["__nvmlGpuInstanceSetVgpuSchedulerState_v2"] = <_cyb_intptr_t>__nvmlGpuInstanceSetVgpuSchedulerState_v2

global __nvmlSystemGetCPER_v1
data["__nvmlSystemGetCPER_v1"] = <_cyb_intptr_t>__nvmlSystemGetCPER_v1

global __nvmlDeviceGetBBXTimeData_v1
data["__nvmlDeviceGetBBXTimeData_v1"] = <_cyb_intptr_t>__nvmlDeviceGetBBXTimeData_v1

global __nvmlDeviceGetAccountingStats_v2
data["__nvmlDeviceGetAccountingStats_v2"] = <_cyb_intptr_t>__nvmlDeviceGetAccountingStats_v2

global __nvmlDeviceGetRemappedRows_v2
data["__nvmlDeviceGetRemappedRows_v2"] = <_cyb_intptr_t>__nvmlDeviceGetRemappedRows_v2
_cyb_func_ptrs = data
return data

Expand Down Expand Up @@ -6073,3 +6101,43 @@ cdef nvmlReturn_t _nvmlGpuInstanceSetVgpuSchedulerState_v2(nvmlGpuInstance_t gpu
raise FunctionNotFoundError("function nvmlGpuInstanceSetVgpuSchedulerState_v2 is not found")
return (<nvmlReturn_t (*)(nvmlGpuInstance_t, nvmlVgpuSchedulerState_v2_t*) noexcept nogil>__nvmlGpuInstanceSetVgpuSchedulerState_v2)(
gpuInstance, pSchedulerState)


cdef nvmlReturn_t _nvmlSystemGetCPER_v1(nvmlGetCPER_v1_t* cper) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlSystemGetCPER_v1
_check_or_init_nvml()
if __nvmlSystemGetCPER_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlSystemGetCPER_v1 is not found")
return (<nvmlReturn_t (*)(nvmlGetCPER_v1_t*) noexcept nogil>__nvmlSystemGetCPER_v1)(
cper)


cdef nvmlReturn_t _nvmlDeviceGetBBXTimeData_v1(nvmlDevice_t device, nvmlBBXTimeData_v1_t* timeData) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetBBXTimeData_v1
_check_or_init_nvml()
if __nvmlDeviceGetBBXTimeData_v1 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetBBXTimeData_v1 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlBBXTimeData_v1_t*) noexcept nogil>__nvmlDeviceGetBBXTimeData_v1)(
device, timeData)


cdef nvmlReturn_t _nvmlDeviceGetAccountingStats_v2(nvmlDevice_t device, nvmlAccountingStats_v2_t* stats) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetAccountingStats_v2
_check_or_init_nvml()
if __nvmlDeviceGetAccountingStats_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetAccountingStats_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlAccountingStats_v2_t*) noexcept nogil>__nvmlDeviceGetAccountingStats_v2)(
device, stats)


cdef nvmlReturn_t _nvmlDeviceGetRemappedRows_v2(nvmlDevice_t device, nvmlRemappedRowsInfo_v2_t* info) except?_NVMLRETURN_T_INTERNAL_LOADING_ERROR nogil:
global __nvmlDeviceGetRemappedRows_v2
_check_or_init_nvml()
if __nvmlDeviceGetRemappedRows_v2 == NULL:
with gil:
raise FunctionNotFoundError("function nvmlDeviceGetRemappedRows_v2 is not found")
return (<nvmlReturn_t (*)(nvmlDevice_t, nvmlRemappedRowsInfo_v2_t*) noexcept nogil>__nvmlDeviceGetRemappedRows_v2)(
device, info)
Loading