Merge branch 'main' into faster-vector

Author: mdboom

cuda_bindings/build_hooks.py

@@ -2901,25 +2901,25 @@ 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):
     with nogil:
-        __status__ = cuFileReadAsync(<Handle>fh, <void*>buf_ptr_base, <size_t*>size_p, <off_t*>file_offset_p, <off_t*>buf_ptr_offset_p, <ssize_t*>bytes_read_p, <void*>stream)
+        __status__ = cuFileReadAsync(<Handle>fh, <void*>buf_ptr_base, <size_t*>size_p, <off_t*>file_offset_p, <off_t*>buf_ptr_offset_p, <ssize_t*>bytes_read_p, <CUstream>stream)
     check_status(__status__)
 
 
 cpdef write_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_written_p, intptr_t stream):
     with nogil:
-        __status__ = cuFileWriteAsync(<Handle>fh, <void*>buf_ptr_base, <size_t*>size_p, <off_t*>file_offset_p, <off_t*>buf_ptr_offset_p, <ssize_t*>bytes_written_p, <void*>stream)
+        __status__ = cuFileWriteAsync(<Handle>fh, <void*>buf_ptr_base, <size_t*>size_p, <off_t*>file_offset_p, <off_t*>buf_ptr_offset_p, <ssize_t*>bytes_written_p, <CUstream>stream)
     check_status(__status__)
 
 
 cpdef stream_register(intptr_t stream, unsigned flags):
     with nogil:
-        __status__ = cuFileStreamRegister(<void*>stream, flags)
+        __status__ = cuFileStreamRegister(<CUstream>stream, flags)
     check_status(__status__)
 
 
 cpdef stream_deregister(intptr_t stream):
     with nogil:
-        __status__ = cuFileStreamDeregister(<void*>stream)
+        __status__ = cuFileStreamDeregister(<CUstream>stream)
     check_status(__status__)
 
 

cuda_bindings/cuda/bindings/cufile.pyx

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 #
-# This code was automatically generated across versions from 12.9.1 to 13.1.1. Do not modify it directly.
+# This code was automatically generated across versions from 12.9.1 to 13.1.1, generator version 0.3.1.dev1283+gc7bc6fa75. Do not modify it directly.
 
 from libc.stdint cimport intptr_t
 
@@ -342,7 +342,6 @@ cpdef object device_get_grid_licensable_features_v4(intptr_t device)
 cpdef unsigned int get_vgpu_driver_capabilities(int capability) except? 0
 cpdef unsigned int device_get_vgpu_capabilities(intptr_t device, int capability) except? 0
 cpdef str vgpu_type_get_class(unsigned int vgpu_type_id)
-cpdef str vgpu_type_get_name(unsigned int vgpu_type_id)
 cpdef unsigned int vgpu_type_get_gpu_instance_profile_id(unsigned int vgpu_type_id) except? 0
 cpdef tuple vgpu_type_get_device_id(unsigned int vgpu_type_id)
 cpdef unsigned long long vgpu_type_get_framebuffer_size(unsigned int vgpu_type_id) except? 0

cuda_bindings/cuda/bindings/nvml.pxd

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 #
-# This code was automatically generated across versions from 12.9.1 to 13.1.1. Do not modify it directly.
+# This code was automatically generated across versions from 12.9.1 to 13.1.1, generator version 0.3.1.dev1283+gc7bc6fa75. Do not modify it directly.
 
 cimport cython  # NOQA
 
@@ -23189,22 +23189,6 @@ cpdef str vgpu_type_get_class(unsigned int vgpu_type_id):
     return cpython.PyUnicode_FromString(vgpu_type_class)
 
 
-cpdef str vgpu_type_get_name(unsigned int vgpu_type_id):
-    """Retrieve the vGPU type name.
-
-    Args:
-        vgpu_type_id (unsigned int): Handle to vGPU type.
-
-    .. seealso:: `nvmlVgpuTypeGetName`
-    """
-    cdef unsigned int size = 64
-    cdef char[64] vgpu_type_name
-    with nogil:
-        __status__ = nvmlVgpuTypeGetName(<nvmlVgpuTypeId_t>vgpu_type_id, vgpu_type_name, <unsigned int*>size)
-    check_status(__status__)
-    return cpython.PyUnicode_FromString(vgpu_type_name)
-
-
 cpdef unsigned int vgpu_type_get_gpu_instance_profile_id(unsigned int vgpu_type_id) except? 0:
     """Retrieve the GPU Instance Profile ID for the given vGPU type ID. The API will return a valid GPU Instance Profile ID for the MIG capable vGPU types, else INVALID_GPU_INSTANCE_PROFILE_ID is returned.
 
@@ -26228,3 +26212,19 @@ cpdef str device_get_current_clock_freqs(intptr_t device):
         __status__ = nvmlDeviceGetCurrentClockFreqs(<Device>device, current_clock_freqs)
     check_status(__status__)
     return cpython.PyUnicode_FromString(current_clock_freqs[0].str)
+
+
+cpdef str vgpu_type_get_name(unsigned int vgpu_type_id):
+    """Retrieve the vGPU type name.
+
+    Args:
+        vgpu_type_id (unsigned int): Handle to vGPU type.
+
+    .. seealso:: `nvmlVgpuTypeGetName`
+    """
+    cdef unsigned int[1] size = [64]
+    cdef char[64] vgpu_type_name
+    with nogil:
+        __status__ = nvmlVgpuTypeGetName(<nvmlVgpuTypeId_t>vgpu_type_id, vgpu_type_name, <unsigned int*>size)
+    check_status(__status__)
+    return cpython.PyUnicode_FromString(vgpu_type_name)

cuda_bindings/cuda/bindings/nvml.pyx

@@ -7,7 +7,8 @@ requires = [
     "cython>=3.2,<3.3",
     "pyclibrary>=0.1.7",
 ]
-build-backend = "setuptools.build_meta"
+build-backend = "build_hooks"
+backend-path = ["."]
 
 [project]
 name = "cuda-bindings"