Skip to content
Open
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/cudla.pxd
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# SPDX-License-Identifier: Apache-2.0

# This code was automatically generated across versions from 1.5.0 to 13.4.1. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=f6b70193e4ca3c62bd5749b3d19d305f0e268225bf1a6dcf6b95091cb0511791
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=6eb337830ce6124c51007631ed0be6c3232fcd43f6d1e53f28b060ec0adf97f8


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -56,7 +56,7 @@ cpdef intptr_t create_device(uint64_t device, uint32_t flags) except *
cpdef intptr_t mem_register(intptr_t dev_handle, intptr_t ptr, size_t size, uint32_t flags) except *
cpdef intptr_t module_load_from_memory(intptr_t dev_handle, p_module, size_t module_size, uint32_t flags) except *
cpdef module_unload(intptr_t h_module, uint32_t flags)
cpdef submit_task(intptr_t dev_handle, intptr_t ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags)
cpdef submit_task(intptr_t dev_handle, ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags)
cpdef object device_get_attribute(intptr_t dev_handle, int attrib)
cpdef mem_unregister(intptr_t dev_handle, intptr_t dev_ptr)
cpdef int get_last_error(intptr_t dev_handle) except? 0
Expand Down
7 changes: 4 additions & 3 deletions cuda_bindings/cuda/bindings/cudla.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# SPDX-License-Identifier: Apache-2.0

# This code was automatically generated across versions from 1.5.0 to 13.4.1. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=dacac686fe1821a7d01ad45a8337f4d35d53fc8fafa3892cd3fdcf4b0ec83b1a
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=88aaad5ca6606a0c26d3f7c948ab76a2695b97c973cbd771bbb66b2f6c29fef2


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -1814,9 +1814,10 @@ cpdef module_unload(intptr_t h_module, uint32_t flags):
check_status(__status__)


cpdef submit_task(intptr_t dev_handle, intptr_t ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags):
cpdef submit_task(intptr_t dev_handle, ptr_to_tasks, uint32_t num_tasks, intptr_t stream, uint32_t flags):
cdef intptr_t _ptr_to_tasks_ptr_ = int(ptr_to_tasks)
with nogil:
__status__ = cudlaSubmitTask(<const DevHandle>dev_handle, <const cudlaTask* const>ptr_to_tasks, <const uint32_t>num_tasks, <void* const>stream, <const uint32_t>flags)
__status__ = cudlaSubmitTask(<const DevHandle>dev_handle, <const cudlaTask* const>_ptr_to_tasks_ptr_, <const uint32_t>num_tasks, <void* const>stream, <const uint32_t>flags)
check_status(__status__)


Expand Down
14 changes: 7 additions & 7 deletions cuda_bindings/cuda/bindings/cufile.pxd
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.4.1. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=616e51ad15aa071792eceba5db17a6fb2b845b219395e2fdcc6278d3621f52de
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=5b9d3a05eab53628d366c602ae3125fede0d25b2cc48137e2b1f7dcb29068650



Expand Down Expand Up @@ -53,7 +53,7 @@ ctypedef CUfileP2PFlags_t _P2PFlags
# Functions
###############################################################################

cpdef intptr_t handle_register(intptr_t descr) except? 0
cpdef intptr_t handle_register(descr) except? 0
cpdef void handle_deregister(intptr_t fh) except*
cpdef buf_register(intptr_t buf_ptr_base, size_t length, int flags)
cpdef buf_deregister(intptr_t buf_ptr_base)
Expand All @@ -65,8 +65,8 @@ cpdef driver_set_max_direct_io_size(size_t max_direct_io_size)
cpdef driver_set_max_cache_size(size_t max_cache_size)
cpdef driver_set_max_pinned_mem_size(size_t max_pinned_size)
cpdef intptr_t batch_io_set_up(unsigned nr) except? 0
cpdef batch_io_submit(intptr_t batch_idp, unsigned nr, intptr_t iocbp, unsigned int flags)
cpdef batch_io_get_status(intptr_t batch_idp, unsigned min_nr, intptr_t nr, intptr_t iocbp, intptr_t timeout)
cpdef batch_io_submit(intptr_t batch_idp, unsigned nr, iocbp, unsigned int flags)
cpdef batch_io_get_status(intptr_t batch_idp, unsigned min_nr, intptr_t nr, iocbp, intptr_t timeout)
cpdef batch_io_cancel(intptr_t batch_idp)
cpdef void batch_io_destroy(intptr_t batch_idp) except*
cpdef read_async(intptr_t fh, intptr_t buf_ptr_base, intptr_t size_p, intptr_t file_offset_p, intptr_t buf_ptr_offset_p, intptr_t bytes_read_p, intptr_t stream)
Expand All @@ -86,9 +86,9 @@ cpdef int get_stats_level() except? 0
cpdef stats_start()
cpdef stats_stop()
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 get_stats_l1(stats)
cpdef get_stats_l2(stats)
cpdef get_stats_l3(stats)
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)
67 changes: 39 additions & 28 deletions cuda_bindings/cuda/bindings/cufile.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.4.1. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=28643e63cf615e30409047a61ccc54e84bb0739fdec43d3e0ed10ef3759c013f
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=df140bcd5b3c6cfb01ff11bc6275a6048c37bee1db8d80a1da19b176df6af870


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -445,9 +445,10 @@ cdef class IOEvents:
return self._data.ctypes.data

def __int__(self):
if self._data.size > 1:
raise TypeError("int() argument must be a bytes-like object of size 1. "
"To get the pointer address of an array, use .ptr")
if self._data.size > 1 and not self._data.flags["C_CONTIGUOUS"]:
raise TypeError("int() argument must be a bytes-like object of size 1, or a "
"C-contiguous array. To get the pointer address of a "
"non-contiguous array, use .ptr")
return self._data.ctypes.data

def __len__(self):
Expand Down Expand Up @@ -784,9 +785,10 @@ cdef class PerGpuStats:
return self._data.ctypes.data

def __int__(self):
if self._data.size > 1:
raise TypeError("int() argument must be a bytes-like object of size 1. "
"To get the pointer address of an array, use .ptr")
if self._data.size > 1 and not self._data.flags["C_CONTIGUOUS"]:
raise TypeError("int() argument must be a bytes-like object of size 1, or a "
"C-contiguous array. To get the pointer address of a "
"non-contiguous array, use .ptr")
return self._data.ctypes.data

def __len__(self):
Expand Down Expand Up @@ -1247,9 +1249,10 @@ cdef class IOVec:
return self._data.ctypes.data

def __int__(self):
if self._data.size > 1:
raise TypeError("int() argument must be a bytes-like object of size 1. "
"To get the pointer address of an array, use .ptr")
if self._data.size > 1 and not self._data.flags["C_CONTIGUOUS"]:
raise TypeError("int() argument must be a bytes-like object of size 1, or a "
"C-contiguous array. To get the pointer address of a "
"non-contiguous array, use .ptr")
return self._data.ctypes.data

def __len__(self):
Expand Down Expand Up @@ -1405,9 +1408,10 @@ cdef class Descr:
return self._data.ctypes.data

def __int__(self):
if self._data.size > 1:
raise TypeError("int() argument must be a bytes-like object of size 1. "
"To get the pointer address of an array, use .ptr")
if self._data.size > 1 and not self._data.flags["C_CONTIGUOUS"]:
raise TypeError("int() argument must be a bytes-like object of size 1, or a "
"C-contiguous array. To get the pointer address of a "
"non-contiguous array, use .ptr")
return self._data.ctypes.data

def __len__(self):
Expand Down Expand Up @@ -2604,9 +2608,10 @@ cdef class IOParams:
return self._data.ctypes.data

def __int__(self):
if self._data.size > 1:
raise TypeError("int() argument must be a bytes-like object of size 1. "
"To get the pointer address of an array, use .ptr")
if self._data.size > 1 and not self._data.flags["C_CONTIGUOUS"]:
raise TypeError("int() argument must be a bytes-like object of size 1, or a "
"C-contiguous array. To get the pointer address of a "
"non-contiguous array, use .ptr")
return self._data.ctypes.data

def __len__(self):
Expand Down Expand Up @@ -3324,7 +3329,7 @@ cdef int check_status(ReturnT status) except 1 nogil:
# Wrapper functions
###############################################################################

cpdef intptr_t handle_register(intptr_t descr) except? 0:
cpdef intptr_t handle_register(descr) except? 0:
"""cuFileHandleRegister is required, and performs extra checking that is memoized to provide increased performance on later cuFile operations.

Args:
Expand All @@ -3335,9 +3340,10 @@ cpdef intptr_t handle_register(intptr_t descr) except? 0:

.. seealso:: `cuFileHandleRegister`
"""
cdef intptr_t _descr_ptr_ = int(descr)
cdef Handle fh
with nogil:
__status__ = cuFileHandleRegister(&fh, <CUfileDescr_t*>descr)
__status__ = cuFileHandleRegister(&fh, <CUfileDescr_t*>_descr_ptr_)
check_status(__status__)
return <intptr_t>fh

Expand Down Expand Up @@ -3480,15 +3486,17 @@ cpdef intptr_t batch_io_set_up(unsigned nr) except? 0:
return <intptr_t>batch_idp


cpdef batch_io_submit(intptr_t batch_idp, unsigned nr, intptr_t iocbp, unsigned int flags):
cpdef batch_io_submit(intptr_t batch_idp, unsigned nr, iocbp, unsigned int flags):
cdef intptr_t _iocbp_ptr_ = int(iocbp)
with nogil:
__status__ = cuFileBatchIOSubmit(<BatchHandle>batch_idp, nr, <CUfileIOParams_t*>iocbp, flags)
__status__ = cuFileBatchIOSubmit(<BatchHandle>batch_idp, nr, <CUfileIOParams_t*>_iocbp_ptr_, flags)
check_status(__status__)


cpdef batch_io_get_status(intptr_t batch_idp, unsigned min_nr, intptr_t nr, intptr_t iocbp, intptr_t timeout):
cpdef batch_io_get_status(intptr_t batch_idp, unsigned min_nr, intptr_t nr, iocbp, intptr_t timeout):
cdef intptr_t _iocbp_ptr_ = int(iocbp)
with nogil:
__status__ = cuFileBatchIOGetStatus(<BatchHandle>batch_idp, min_nr, <unsigned*>nr, <CUfileIOEvents_t*>iocbp, <timespec*>timeout)
__status__ = cuFileBatchIOGetStatus(<BatchHandle>batch_idp, min_nr, <unsigned*>nr, <CUfileIOEvents_t*>_iocbp_ptr_, <timespec*>timeout)
check_status(__status__)


Expand Down Expand Up @@ -3666,7 +3674,7 @@ cpdef stats_reset():
check_status(__status__)


cpdef get_stats_l1(intptr_t stats):
cpdef get_stats_l1(stats):
"""Get Level 1 cuFile statistics.

Args:
Expand All @@ -3675,12 +3683,13 @@ cpdef get_stats_l1(intptr_t stats):

.. seealso:: `cuFileGetStatsL1`
"""
cdef intptr_t _stats_ptr_ = int(stats)
with nogil:
__status__ = cuFileGetStatsL1(<CUfileStatsLevel1_t*>stats)
__status__ = cuFileGetStatsL1(<CUfileStatsLevel1_t*>_stats_ptr_)
check_status(__status__)


cpdef get_stats_l2(intptr_t stats):
cpdef get_stats_l2(stats):
"""Get Level 2 cuFile statistics.

Args:
Expand All @@ -3689,12 +3698,13 @@ cpdef get_stats_l2(intptr_t stats):

.. seealso:: `cuFileGetStatsL2`
"""
cdef intptr_t _stats_ptr_ = int(stats)
with nogil:
__status__ = cuFileGetStatsL2(<CUfileStatsLevel2_t*>stats)
__status__ = cuFileGetStatsL2(<CUfileStatsLevel2_t*>_stats_ptr_)
check_status(__status__)


cpdef get_stats_l3(intptr_t stats):
cpdef get_stats_l3(stats):
"""Get Level 3 cuFile statistics.

Args:
Expand All @@ -3703,8 +3713,9 @@ cpdef get_stats_l3(intptr_t stats):

.. seealso:: `cuFileGetStatsL3`
"""
cdef intptr_t _stats_ptr_ = int(stats)
with nogil:
__status__ = cuFileGetStatsL3(<CUfileStatsLevel3_t*>stats)
__status__ = cuFileGetStatsL3(<CUfileStatsLevel3_t*>_stats_ptr_)
check_status(__status__)


Expand Down
32 changes: 16 additions & 16 deletions cuda_bindings/cuda/bindings/nvml.pxd
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.4.1. Do not modify it directly.
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=635b329217b9c57fce06de4ffd743372a120b5039eb247f49205bd4d2baf663c
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=61e90d2304af3d2c6e4146a16977ae5514a560d909ea67c778cd955aafc29642



Expand Down Expand Up @@ -241,7 +241,7 @@ cpdef int device_get_mem_clk_vf_offset(intptr_t device) except? 0
cpdef tuple device_get_min_max_clock_of_p_state(intptr_t device, int type, int pstate)
cpdef tuple device_get_gpc_clk_min_max_vf_offset(intptr_t device)
cpdef tuple device_get_mem_clk_min_max_vf_offset(intptr_t device)
cpdef device_set_clock_offsets(intptr_t device, intptr_t info)
cpdef device_set_clock_offsets(intptr_t device, info)
cpdef unsigned int device_get_power_management_limit(intptr_t device) except? 0
cpdef tuple device_get_power_management_limit_constraints(intptr_t device)
cpdef unsigned int device_get_power_management_default_limit(intptr_t device) except? 0
Expand Down Expand Up @@ -338,16 +338,16 @@ cpdef system_set_nvlink_bw_mode(unsigned int nvlink_bw_mode)
cpdef unsigned int system_get_nvlink_bw_mode() except? 0
cpdef object device_get_nvlink_supported_bw_modes(intptr_t device)
cpdef object device_get_nvlink_bw_mode(intptr_t device)
cpdef device_set_nvlink_bw_mode(intptr_t device, intptr_t set_bw_mode)
cpdef device_set_nvlink_bw_mode(intptr_t device, set_bw_mode)
cpdef intptr_t event_set_create() except? 0
cpdef device_register_events(intptr_t device, unsigned long long event_types, intptr_t set)
cpdef unsigned long long device_get_supported_event_types(intptr_t device) except? 0
cpdef object event_set_wait_v2(intptr_t set, unsigned int timeoutms)
cpdef event_set_free(intptr_t set)
cpdef device_modify_drain_state(intptr_t pci_info, int new_state)
cpdef int device_query_drain_state(intptr_t pci_info) except? -1
cpdef device_remove_gpu_v2(intptr_t pci_info, int gpu_state, int link_state)
cpdef device_discover_gpus(intptr_t pci_info)
cpdef device_modify_drain_state(pci_info, int new_state)
cpdef int device_query_drain_state(pci_info) except? -1
cpdef device_remove_gpu_v2(pci_info, int gpu_state, int link_state)
cpdef device_discover_gpus(pci_info)
cpdef int device_get_virtualization_mode(intptr_t device) except? -1
cpdef int device_get_host_vgpu_mode(intptr_t device) except? -1
cpdef device_set_virtualization_mode(intptr_t device, int virtual_mode)
Expand Down Expand Up @@ -385,15 +385,15 @@ cpdef unsigned int vgpu_instance_get_gpu_instance_id(unsigned int vgpu_instance)
cpdef str vgpu_instance_get_gpu_pci_id(unsigned int vgpu_instance)
cpdef unsigned int vgpu_type_get_capabilities(unsigned int vgpu_type_id, int capability) except? 0
cpdef str vgpu_instance_get_mdev_uuid(unsigned int vgpu_instance)
cpdef gpu_instance_set_vgpu_scheduler_state(intptr_t gpu_instance, intptr_t p_scheduler)
cpdef gpu_instance_set_vgpu_scheduler_state(intptr_t gpu_instance, p_scheduler)
cpdef object gpu_instance_get_vgpu_scheduler_state(intptr_t gpu_instance)
cpdef object gpu_instance_get_vgpu_scheduler_log(intptr_t gpu_instance)
cpdef str device_get_pgpu_metadata_string(intptr_t device)
cpdef object device_get_vgpu_scheduler_log(intptr_t device)
cpdef object device_get_vgpu_scheduler_state(intptr_t device)
cpdef object device_get_vgpu_scheduler_capabilities(intptr_t device)
cpdef device_set_vgpu_scheduler_state(intptr_t device, intptr_t p_scheduler_state)
cpdef set_vgpu_version(intptr_t vgpu_version)
cpdef set_vgpu_version(vgpu_version)
cpdef tuple device_get_vgpu_process_utilization(intptr_t device, unsigned long long last_seen_time_stamp)
cpdef int vgpu_instance_get_accounting_mode(unsigned int vgpu_instance) except? -1
cpdef object vgpu_instance_get_accounting_pids(unsigned int vgpu_instance)
Expand All @@ -407,15 +407,15 @@ cpdef tuple device_get_mig_mode(intptr_t device)
cpdef object device_get_gpu_instance_possible_placements_v2(intptr_t device, unsigned int profile_id)
cpdef unsigned int device_get_gpu_instance_remaining_capacity(intptr_t device, unsigned int profile_id) except? 0
cpdef intptr_t device_create_gpu_instance(intptr_t device, unsigned int profile_id) except? 0
cpdef intptr_t device_create_gpu_instance_with_placement(intptr_t device, unsigned int profile_id, intptr_t placement) except? 0
cpdef intptr_t device_create_gpu_instance_with_placement(intptr_t device, unsigned int profile_id, placement) except? 0
cpdef gpu_instance_destroy(intptr_t gpu_instance)
cpdef intptr_t device_get_gpu_instance_by_id(intptr_t device, unsigned int id) except? 0
cpdef object gpu_instance_get_info(intptr_t gpu_instance)
cpdef object gpu_instance_get_compute_instance_profile_info_v(intptr_t gpu_instance, unsigned int profile, unsigned int eng_profile)
cpdef unsigned int gpu_instance_get_compute_instance_remaining_capacity(intptr_t gpu_instance, unsigned int profile_id) except? 0
cpdef object gpu_instance_get_compute_instance_possible_placements(intptr_t gpu_instance, unsigned int profile_id)
cpdef intptr_t gpu_instance_create_compute_instance(intptr_t gpu_instance, unsigned int profile_id) except? 0
cpdef intptr_t gpu_instance_create_compute_instance_with_placement(intptr_t gpu_instance, unsigned int profile_id, intptr_t placement) except? 0
cpdef intptr_t gpu_instance_create_compute_instance_with_placement(intptr_t gpu_instance, unsigned int profile_id, placement) except? 0
cpdef compute_instance_destroy(intptr_t compute_instance)
cpdef intptr_t gpu_instance_get_compute_instance_by_id(intptr_t gpu_instance, unsigned int id) except? 0
cpdef object compute_instance_get_info_v2(intptr_t compute_instance)
Expand All @@ -431,26 +431,26 @@ cpdef device_power_smoothing_set_state(intptr_t device, intptr_t state)
cpdef object device_get_addressing_mode(intptr_t device)
cpdef object device_get_repair_status(intptr_t device)
cpdef object device_get_power_mizer_mode_v1(intptr_t device)
cpdef device_set_power_mizer_mode_v1(intptr_t device, intptr_t power_mizer_mode)
cpdef device_set_power_mizer_mode_v1(intptr_t device, power_mizer_mode)
cpdef device_vgpu_force_gsp_unload(intptr_t device)
cpdef object device_get_vgpu_scheduler_state_v2(intptr_t device)
cpdef object gpu_instance_get_vgpu_scheduler_state_v2(intptr_t gpu_instance)
cpdef object device_get_vgpu_scheduler_log_v2(intptr_t device)
cpdef object gpu_instance_get_vgpu_scheduler_log_v2(intptr_t gpu_instance)
cpdef device_set_vgpu_scheduler_state_v2(intptr_t device, intptr_t p_scheduler_state)
cpdef gpu_instance_set_vgpu_scheduler_state_v2(intptr_t gpu_instance, intptr_t p_scheduler_state)
cpdef device_set_vgpu_scheduler_state_v2(intptr_t device, p_scheduler_state)
cpdef gpu_instance_set_vgpu_scheduler_state_v2(intptr_t gpu_instance, p_scheduler_state)
cpdef object system_get_cper_v1()
cpdef object device_get_bbx_time_data_v1(intptr_t device)
cpdef object device_get_accounting_stats_v2(intptr_t device)
cpdef object device_get_remapped_rows_v2(intptr_t device)
cpdef device_set_adaptive_tgp_mode_v1(intptr_t device, int mode)
cpdef object device_get_adaptive_tgp_mode_info_v1(intptr_t device)
cpdef device_set_memory_limits_v1(intptr_t device, intptr_t limits)
cpdef device_set_memory_limits_v1(intptr_t device, limits)
cpdef object device_get_memory_limits_v1(intptr_t device)
cpdef object device_get_gpu_fabric_info_v4(intptr_t device)
cpdef object device_perf_metrics_get_samples_v1(intptr_t device)
cpdef object device_set_nvlink_bw_mode_async_v1(intptr_t device)
cpdef object device_get_nv_link_telemetry_samples_v1(intptr_t device)
cpdef event_set_register_gpu_operational_events_v1(intptr_t event_set, intptr_t config)
cpdef event_set_register_gpu_operational_events_v1(intptr_t event_set, config)
cpdef object event_set_get_context_count_v1(intptr_t set)
cpdef object device_get_bank_remapper_status_v1(intptr_t device)
Loading
Loading