Skip to content
Merged
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
4 changes: 2 additions & 2 deletions cuda_bindings/cuda/bindings/cufile.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# 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.
# This code was automatically generated across versions from 12.9.1 to 13.2.0, generator version 0.3.1.dev1568+g289771de9.d20260413. Do not modify it directly.

from libc.stdint cimport intptr_t

Expand Down Expand Up @@ -80,6 +80,6 @@ cpdef stats_reset()
cpdef get_stats_l1(intptr_t stats)
cpdef get_stats_l2(intptr_t stats)
cpdef get_stats_l3(intptr_t stats)
cpdef size_t get_bar_size_in_kb(int gpu_ind_ex) except? 0
cpdef size_t get_bar_size_in_kb(int gpu_index) except? 0
cpdef set_parameter_posix_pool_slab_array(intptr_t size_values, intptr_t count_values, int len)
cpdef get_parameter_posix_pool_slab_array(intptr_t size_values, intptr_t count_values, int len)
25 changes: 3 additions & 22 deletions cuda_bindings/cuda/bindings/cufile.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# 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.
# This code was automatically generated across versions from 12.9.1 to 13.2.0, generator version 0.3.1.dev1568+g289771de9.d20260413. Do not modify it directly.

cimport cython # NOQA
from libc cimport errno
Expand Down Expand Up @@ -75,7 +75,6 @@ _py_anon_pod1_dtype = _numpy.dtype((
}
))


cdef class _py_anon_pod1:
"""Empty-initialize an instance of `cuda_bindings_cufile__anon_pod1`.

Expand Down Expand Up @@ -389,21 +388,17 @@ io_events_dtype = _get_io_events_dtype_offsets()

cdef class IOEvents:
"""Empty-initialize an array of `CUfileIOEvents_t`.

The resulting object is of length `size` and of dtype `io_events_dtype`.
If default-constructed, the instance represents a single struct.

Args:
size (int): number of structs, default=1.


.. seealso:: `CUfileIOEvents_t`
"""
cdef:
readonly object _data



def __init__(self, size=1):
arr = _numpy.empty(size, dtype=io_events_dtype)
self._data = arr.view(_numpy.recarray)
Expand Down Expand Up @@ -728,21 +723,17 @@ per_gpu_stats_dtype = _get_per_gpu_stats_dtype_offsets()

cdef class PerGpuStats:
"""Empty-initialize an array of `CUfilePerGpuStats_t`.

The resulting object is of length `size` and of dtype `per_gpu_stats_dtype`.
If default-constructed, the instance represents a single struct.

Args:
size (int): number of structs, default=1.


.. seealso:: `CUfilePerGpuStats_t`
"""
cdef:
readonly object _data



def __init__(self, size=1):
arr = _numpy.empty(size, dtype=per_gpu_stats_dtype)
self._data = arr.view(_numpy.recarray)
Expand Down Expand Up @@ -1192,21 +1183,17 @@ descr_dtype = _get_descr_dtype_offsets()

cdef class Descr:
"""Empty-initialize an array of `CUfileDescr_t`.

The resulting object is of length `size` and of dtype `descr_dtype`.
If default-constructed, the instance represents a single struct.

Args:
size (int): number of structs, default=1.


.. seealso:: `CUfileDescr_t`
"""
cdef:
readonly object _data



def __init__(self, size=1):
arr = _numpy.empty(size, dtype=descr_dtype)
self._data = arr.view(_numpy.recarray)
Expand Down Expand Up @@ -1349,7 +1336,6 @@ _py_anon_pod2_dtype = _numpy.dtype((
}
))


cdef class _py_anon_pod2:
"""Empty-initialize an instance of `cuda_bindings_cufile__anon_pod2`.

Expand Down Expand Up @@ -2144,21 +2130,17 @@ io_params_dtype = _get_io_params_dtype_offsets()

cdef class IOParams:
"""Empty-initialize an array of `CUfileIOParams_t`.

The resulting object is of length `size` and of dtype `io_params_dtype`.
If default-constructed, the instance represents a single struct.

Args:
size (int): number of structs, default=1.


.. seealso:: `CUfileIOParams_t`
"""
cdef:
readonly object _data



def __init__(self, size=1):
arr = _numpy.empty(size, dtype=io_params_dtype)
self._data = arr.view(_numpy.recarray)
Expand Down Expand Up @@ -2643,7 +2625,6 @@ cdef class StatsLevel3:
return obj



###############################################################################
# Enum
###############################################################################
Expand Down Expand Up @@ -3252,10 +3233,10 @@ cpdef get_stats_l3(intptr_t stats):
check_status(__status__)


cpdef size_t get_bar_size_in_kb(int gpu_ind_ex) except? 0:
cpdef size_t get_bar_size_in_kb(int gpu_index) except? 0:
cdef size_t bar_size
with nogil:
__status__ = cuFileGetBARSizeInKB(gpu_ind_ex, &bar_size)
__status__ = cuFileGetBARSizeInKB(gpu_index, &bar_size)
check_status(__status__)
return bar_size

Expand Down
3 changes: 2 additions & 1 deletion cuda_bindings/cuda/bindings/cycufile.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -197,9 +197,11 @@ cdef extern from '<cufile.h>':
cdef extern from '<cufile.h>':
ctypedef void* CUfileHandle_t 'CUfileHandle_t'


cdef extern from '<cufile.h>':
ctypedef void* CUfileBatchHandle_t 'CUfileBatchHandle_t'


cdef extern from '<cufile.h>':
ctypedef struct CUfileError_t 'CUfileError_t':
CUfileOpError err
Expand Down Expand Up @@ -367,7 +369,6 @@ cdef extern from '<cufile.h>':
CUfilePerGpuStats_t per_gpu_stats[16]



cdef extern from *:
"""
// This is the missing piece we need to supply to help Cython & C++ compilers.
Expand Down
69 changes: 68 additions & 1 deletion cuda_bindings/cuda/bindings/cynvml.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# 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.
# This code was automatically generated across versions from 12.9.1 to 13.2.0, generator version 0.3.1.dev1568+g289771de9.d20260413. Do not modify it directly.

from libc.stdint cimport int64_t

Expand Down Expand Up @@ -817,12 +817,19 @@ ctypedef struct nvmlPlatformInfo_v2_t 'nvmlPlatformInfo_v2_t':
unsigned char moduleId

ctypedef unsigned int nvmlDeviceArchitecture_t 'nvmlDeviceArchitecture_t'

ctypedef unsigned int nvmlBusType_t 'nvmlBusType_t'

ctypedef unsigned int nvmlFanControlPolicy_t 'nvmlFanControlPolicy_t'

ctypedef unsigned int nvmlPowerSource_t 'nvmlPowerSource_t'

ctypedef unsigned char nvmlPowerScopeType_t 'nvmlPowerScopeType_t'

ctypedef unsigned int nvmlVgpuTypeId_t 'nvmlVgpuTypeId_t'

ctypedef unsigned int nvmlVgpuInstance_t 'nvmlVgpuInstance_t'

ctypedef struct nvmlVgpuHeterogeneousMode_v1_t 'nvmlVgpuHeterogeneousMode_v1_t':
unsigned int version
unsigned int mode
Expand Down Expand Up @@ -862,11 +869,13 @@ ctypedef struct nvmlConfComputeGetKeyRotationThresholdInfo_v1_t 'nvmlConfCompute
unsigned long long attackerAdvantage

ctypedef unsigned char nvmlGpuFabricState_t 'nvmlGpuFabricState_t'

ctypedef struct nvmlSystemDriverBranchInfo_v1_t 'nvmlSystemDriverBranchInfo_v1_t':
unsigned int version
char branch[80]

ctypedef unsigned int nvmlAffinityScope_t 'nvmlAffinityScope_t'

ctypedef struct nvmlTemperature_v1_t 'nvmlTemperature_v1_t':
unsigned int version
nvmlTemperatureSensors_t sensorType
Expand Down Expand Up @@ -915,12 +924,19 @@ ctypedef struct nvmlPdi_v1_t 'nvmlPdi_v1_t':
unsigned long long value

ctypedef void* nvmlDevice_t 'nvmlDevice_t'

ctypedef void* nvmlGpuInstance_t 'nvmlGpuInstance_t'

ctypedef void* nvmlUnit_t 'nvmlUnit_t'

ctypedef void* nvmlEventSet_t 'nvmlEventSet_t'

ctypedef void* nvmlSystemEventSet_t 'nvmlSystemEventSet_t'

ctypedef void* nvmlComputeInstance_t 'nvmlComputeInstance_t'

ctypedef void* nvmlGpmSample_t 'nvmlGpmSample_t'

ctypedef struct nvmlPciInfo_t 'nvmlPciInfo_t':
char busIdLegacy[16]
unsigned int domain
Expand Down Expand Up @@ -1383,15 +1399,25 @@ ctypedef struct nvmlVgpuSchedulerState_v2_t 'nvmlVgpuSchedulerState_v2_t':
unsigned int frequency

ctypedef nvmlPciInfoExt_v1_t nvmlPciInfoExt_t 'nvmlPciInfoExt_t'

ctypedef nvmlCoolerInfo_v1_t nvmlCoolerInfo_t 'nvmlCoolerInfo_t'

ctypedef nvmlDramEncryptionInfo_v1_t nvmlDramEncryptionInfo_t 'nvmlDramEncryptionInfo_t'

ctypedef nvmlMarginTemperature_v1_t nvmlMarginTemperature_t 'nvmlMarginTemperature_t'

ctypedef nvmlClockOffset_v1_t nvmlClockOffset_t 'nvmlClockOffset_t'

ctypedef nvmlFanSpeedInfo_v1_t nvmlFanSpeedInfo_t 'nvmlFanSpeedInfo_t'

ctypedef nvmlDevicePerfModes_v1_t nvmlDevicePerfModes_t 'nvmlDevicePerfModes_t'

ctypedef nvmlDeviceCurrentClockFreqs_v1_t nvmlDeviceCurrentClockFreqs_t 'nvmlDeviceCurrentClockFreqs_t'

ctypedef nvmlEccSramErrorStatus_v1_t nvmlEccSramErrorStatus_t 'nvmlEccSramErrorStatus_t'

ctypedef nvmlPlatformInfo_v2_t nvmlPlatformInfo_t 'nvmlPlatformInfo_t'

ctypedef struct nvmlPowerValue_v2_t 'nvmlPowerValue_v2_t':
unsigned int version
nvmlPowerScopeType_t powerScope
Expand Down Expand Up @@ -1466,13 +1492,21 @@ ctypedef struct nvmlFBCSessionInfo_t 'nvmlFBCSessionInfo_t':
unsigned int averageLatency

ctypedef nvmlVgpuHeterogeneousMode_v1_t nvmlVgpuHeterogeneousMode_t 'nvmlVgpuHeterogeneousMode_t'

ctypedef nvmlVgpuPlacementId_v1_t nvmlVgpuPlacementId_t 'nvmlVgpuPlacementId_t'

ctypedef nvmlVgpuPlacementList_v2_t nvmlVgpuPlacementList_t 'nvmlVgpuPlacementList_t'

ctypedef nvmlVgpuTypeBar1Info_v1_t nvmlVgpuTypeBar1Info_t 'nvmlVgpuTypeBar1Info_t'

ctypedef nvmlVgpuRuntimeState_v1_t nvmlVgpuRuntimeState_t 'nvmlVgpuRuntimeState_t'

ctypedef nvmlSystemConfComputeSettings_v1_t nvmlSystemConfComputeSettings_t 'nvmlSystemConfComputeSettings_t'

ctypedef nvmlConfComputeSetKeyRotationThresholdInfo_v1_t nvmlConfComputeSetKeyRotationThresholdInfo_t 'nvmlConfComputeSetKeyRotationThresholdInfo_t'

ctypedef nvmlConfComputeGetKeyRotationThresholdInfo_v1_t nvmlConfComputeGetKeyRotationThresholdInfo_t 'nvmlConfComputeGetKeyRotationThresholdInfo_t'

ctypedef struct nvmlGpuFabricInfo_t 'nvmlGpuFabricInfo_t':
unsigned char clusterUuid[16]
nvmlReturn_t status
Expand All @@ -1497,16 +1531,27 @@ ctypedef struct nvmlGpuFabricInfo_v3_t 'nvmlGpuFabricInfo_v3_t':
unsigned char healthSummary

ctypedef nvmlSystemDriverBranchInfo_v1_t nvmlSystemDriverBranchInfo_t 'nvmlSystemDriverBranchInfo_t'

ctypedef nvmlTemperature_v1_t nvmlTemperature_t 'nvmlTemperature_t'

ctypedef nvmlNvlinkSupportedBwModes_v1_t nvmlNvlinkSupportedBwModes_t 'nvmlNvlinkSupportedBwModes_t'

ctypedef nvmlNvlinkGetBwMode_v1_t nvmlNvlinkGetBwMode_t 'nvmlNvlinkGetBwMode_t'

ctypedef nvmlNvlinkSetBwMode_v1_t nvmlNvlinkSetBwMode_t 'nvmlNvlinkSetBwMode_t'

ctypedef nvmlDeviceCapabilities_v1_t nvmlDeviceCapabilities_t 'nvmlDeviceCapabilities_t'

ctypedef nvmlPowerSmoothingProfile_v1_t nvmlPowerSmoothingProfile_t 'nvmlPowerSmoothingProfile_t'

ctypedef nvmlPowerSmoothingState_v1_t nvmlPowerSmoothingState_t 'nvmlPowerSmoothingState_t'

ctypedef nvmlDeviceAddressingMode_v1_t nvmlDeviceAddressingMode_t 'nvmlDeviceAddressingMode_t'

ctypedef nvmlRepairStatus_v1_t nvmlRepairStatus_t 'nvmlRepairStatus_t'

ctypedef nvmlPdi_v1_t nvmlPdi_t 'nvmlPdi_t'

ctypedef struct nvmlEventData_t 'nvmlEventData_t':
nvmlDevice_t device
unsigned long long eventType
Expand Down Expand Up @@ -1706,20 +1751,29 @@ ctypedef struct nvmlVgpuSchedulerLogInfo_v2_t 'nvmlVgpuSchedulerLogInfo_v2_t':
nvmlVgpuSchedulerLogEntry_v2_t logEntries[200]

ctypedef nvmlVgpuTypeIdInfo_v1_t nvmlVgpuTypeIdInfo_t 'nvmlVgpuTypeIdInfo_t'

ctypedef nvmlVgpuTypeMaxInstance_v1_t nvmlVgpuTypeMaxInstance_t 'nvmlVgpuTypeMaxInstance_t'

ctypedef nvmlVgpuCreatablePlacementInfo_v1_t nvmlVgpuCreatablePlacementInfo_t 'nvmlVgpuCreatablePlacementInfo_t'

ctypedef struct nvmlVgpuProcessesUtilizationInfo_v1_t 'nvmlVgpuProcessesUtilizationInfo_v1_t':
unsigned int version
unsigned int vgpuProcessCount
unsigned long long lastSeenTimeStamp
nvmlVgpuProcessUtilizationInfo_v1_t* vgpuProcUtilArray

ctypedef nvmlActiveVgpuInstanceInfo_v1_t nvmlActiveVgpuInstanceInfo_t 'nvmlActiveVgpuInstanceInfo_t'

ctypedef nvmlGpuFabricInfo_v3_t nvmlGpuFabricInfoV_t 'nvmlGpuFabricInfoV_t'

ctypedef nvmlSystemEventSetCreateRequest_v1_t nvmlSystemEventSetCreateRequest_t 'nvmlSystemEventSetCreateRequest_t'

ctypedef nvmlSystemEventSetFreeRequest_v1_t nvmlSystemEventSetFreeRequest_t 'nvmlSystemEventSetFreeRequest_t'

ctypedef nvmlSystemRegisterEventRequest_v1_t nvmlSystemRegisterEventRequest_t 'nvmlSystemRegisterEventRequest_t'

ctypedef nvmlProcessDetailList_v1_t nvmlProcessDetailList_t 'nvmlProcessDetailList_t'

ctypedef struct nvmlVgpuInstancesUtilizationInfo_v1_t 'nvmlVgpuInstancesUtilizationInfo_v1_t':
unsigned int version
nvmlValueType_t sampleValType
Expand All @@ -1733,7 +1787,9 @@ ctypedef struct nvmlPRMCounter_v1_t 'nvmlPRMCounter_v1_t':
nvmlPRMCounterValue_v1_t counterValue

ctypedef nvmlUUID_v1_t nvmlUUID_t 'nvmlUUID_t'

ctypedef nvmlProcessesUtilizationInfo_v1_t nvmlProcessesUtilizationInfo_t 'nvmlProcessesUtilizationInfo_t'

ctypedef struct nvmlVgpuSchedulerLog_t 'nvmlVgpuSchedulerLog_t':
unsigned int engineId
unsigned int schedulerPolicy
Expand Down Expand Up @@ -1781,6 +1837,7 @@ ctypedef struct nvmlGridLicensableFeatures_t 'nvmlGridLicensableFeatures_t':
nvmlGridLicensableFeature_t gridLicensableFeatures[3]

ctypedef nvmlSystemEventSetWaitRequest_v1_t nvmlSystemEventSetWaitRequest_t 'nvmlSystemEventSetWaitRequest_t'

ctypedef struct nvmlGpmMetricsGet_t 'nvmlGpmMetricsGet_t':
unsigned int version
unsigned int numMetrics
Expand All @@ -1789,29 +1846,39 @@ ctypedef struct nvmlGpmMetricsGet_t 'nvmlGpmMetricsGet_t':
nvmlGpmMetric_t metrics[333]

ctypedef nvmlWorkloadPowerProfileInfo_v1_t nvmlWorkloadPowerProfileInfo_t 'nvmlWorkloadPowerProfileInfo_t'

ctypedef nvmlWorkloadPowerProfileCurrentProfiles_v1_t nvmlWorkloadPowerProfileCurrentProfiles_t 'nvmlWorkloadPowerProfileCurrentProfiles_t'

ctypedef nvmlWorkloadPowerProfileRequestedProfiles_v1_t nvmlWorkloadPowerProfileRequestedProfiles_t 'nvmlWorkloadPowerProfileRequestedProfiles_t'

ctypedef nvmlEccSramUniqueUncorrectedErrorCounts_v1_t nvmlEccSramUniqueUncorrectedErrorCounts_t 'nvmlEccSramUniqueUncorrectedErrorCounts_t'

ctypedef struct nvmlNvLinkInfo_v2_t 'nvmlNvLinkInfo_v2_t':
unsigned int version
unsigned int isNvleEnabled
nvmlNvlinkFirmwareInfo_t firmwareInfo

ctypedef nvmlVgpuProcessesUtilizationInfo_v1_t nvmlVgpuProcessesUtilizationInfo_t 'nvmlVgpuProcessesUtilizationInfo_t'

ctypedef nvmlVgpuInstancesUtilizationInfo_v1_t nvmlVgpuInstancesUtilizationInfo_t 'nvmlVgpuInstancesUtilizationInfo_t'

ctypedef struct nvmlPRMCounterList_v1_t 'nvmlPRMCounterList_v1_t':
unsigned int numCounters
nvmlPRMCounter_v1_t* counters

ctypedef nvmlVgpuSchedulerStateInfo_v1_t nvmlVgpuSchedulerStateInfo_t 'nvmlVgpuSchedulerStateInfo_t'

ctypedef nvmlVgpuSchedulerLogInfo_v1_t nvmlVgpuSchedulerLogInfo_t 'nvmlVgpuSchedulerLogInfo_t'

ctypedef nvmlVgpuSchedulerState_v1_t nvmlVgpuSchedulerState_t 'nvmlVgpuSchedulerState_t'

ctypedef struct nvmlWorkloadPowerProfileProfilesInfo_v1_t 'nvmlWorkloadPowerProfileProfilesInfo_v1_t':
unsigned int version
nvmlMask255_t perfProfilesMask
nvmlWorkloadPowerProfileInfo_t perfProfile[255]

ctypedef nvmlNvLinkInfo_v2_t nvmlNvLinkInfo_t 'nvmlNvLinkInfo_t'

ctypedef nvmlWorkloadPowerProfileProfilesInfo_v1_t nvmlWorkloadPowerProfileProfilesInfo_t 'nvmlWorkloadPowerProfileProfilesInfo_t'


Expand Down
Loading
Loading