Merge branch 'main' into prep-13.0.3-release

Author: mdboom

.pre-commit-config.yaml

@@ -78,5 +78,12 @@ repos:
     hooks:
       - id: actionlint
 
+  - repo: https://github.com/MarcoGorelli/cython-lint
+    rev: "d9ff7ce99ef4f2ae8fba93079ca9d76c4651d4ac"  # frozen: v0.18.0
+    hooks:
+      - id: cython-lint
+        args: [--no-pycodestyle]
+        exclude: ^cuda_bindings/
+
 default_language_version:
       python: python3

cuda_bindings/cuda/bindings/cyruntime_types.pxi.in

@@ -1591,6 +1591,20 @@ cdef extern from "library_types.h":
 
     ctypedef cudaEmulationStrategy_t cudaEmulationStrategy
 
+    cdef enum cudaEmulationMantissaControl_t:
+        CUDA_EMULATION_MANTISSA_CONTROL_DYNAMIC = 0
+        CUDA_EMULATION_MANTISSA_CONTROL_FIXED = 1
+
+    ctypedef cudaEmulationMantissaControl_t cudaEmulationMantissaControl
+
+    cdef enum cudaEmulationSpecialValuesSupport_t:
+        CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NONE = 0
+        CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_INFINITY = 1
+        CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NAN = 2
+        CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_DEFAULT = 65535
+
+    ctypedef cudaEmulationSpecialValuesSupport_t cudaEmulationSpecialValuesSupport
+
     cdef enum libraryPropertyType_t:
         MAJOR_VERSION = 0
         MINOR_VERSION = 1

cuda_bindings/cuda/bindings/driver.pyx.in

@@ -4822,7 +4822,8 @@ class CUmemPool_attribute(IntEnum):
 
     #: (value type = int) Allow cuMemAllocAsync to insert new stream
     #: dependencies in order to establish the stream ordering required to
-    #: reuse a piece of memory released by cuFreeAsync (default enabled).
+    #: reuse a piece of memory released by cuMemFreeAsync (default
+    #: enabled).
     CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES = cydriver.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES{{endif}}
     {{if 'CU_MEMPOOL_ATTR_RELEASE_THRESHOLD' in found_values}}
 

cuda_bindings/cuda/bindings/runtime.pyx.in

@@ -1578,6 +1578,32 @@ class cudaEmulationStrategy(IntEnum):
 
 _dict_cudaEmulationStrategy = dict(((int(v), v) for k, v in cudaEmulationStrategy.__members__.items()))
 {{endif}}
+{{if 'cudaEmulationMantissaControl_t' in found_types}}
+
+class cudaEmulationMantissaControl(IntEnum):
+    """"""
+    {{if 'CUDA_EMULATION_MANTISSA_CONTROL_DYNAMIC' in found_values}}
+    CUDA_EMULATION_MANTISSA_CONTROL_DYNAMIC = cyruntime.cudaEmulationMantissaControl_t.CUDA_EMULATION_MANTISSA_CONTROL_DYNAMIC{{endif}}
+    {{if 'CUDA_EMULATION_MANTISSA_CONTROL_FIXED' in found_values}}
+    CUDA_EMULATION_MANTISSA_CONTROL_FIXED = cyruntime.cudaEmulationMantissaControl_t.CUDA_EMULATION_MANTISSA_CONTROL_FIXED{{endif}}
+
+_dict_cudaEmulationMantissaControl = dict(((int(v), v) for k, v in cudaEmulationMantissaControl.__members__.items()))
+{{endif}}
+{{if 'cudaEmulationSpecialValuesSupport_t' in found_types}}
+
+class cudaEmulationSpecialValuesSupport(IntEnum):
+    """"""
+    {{if 'CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NONE' in found_values}}
+    CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NONE = cyruntime.cudaEmulationSpecialValuesSupport_t.CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NONE{{endif}}
+    {{if 'CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_INFINITY' in found_values}}
+    CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_INFINITY = cyruntime.cudaEmulationSpecialValuesSupport_t.CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_INFINITY{{endif}}
+    {{if 'CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NAN' in found_values}}
+    CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NAN = cyruntime.cudaEmulationSpecialValuesSupport_t.CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_NAN{{endif}}
+    {{if 'CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_DEFAULT' in found_values}}
+    CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_DEFAULT = cyruntime.cudaEmulationSpecialValuesSupport_t.CUDA_EMULATION_SPECIAL_VALUES_SUPPORT_DEFAULT{{endif}}
+
+_dict_cudaEmulationSpecialValuesSupport = dict(((int(v), v) for k, v in cudaEmulationSpecialValuesSupport.__members__.items()))
+{{endif}}
 {{if 'libraryPropertyType_t' in found_types}}
 
 class libraryPropertyType(IntEnum):

cuda_bindings/docs/source/module/driver.rst

@@ -4851,7 +4851,7 @@ Data types used by CUDA driver
     .. autoattribute:: cuda.bindings.driver.CUmemPool_attribute.CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES
 
 
-        (value type = int) Allow cuMemAllocAsync to insert new stream dependencies in order to establish the stream ordering required to reuse a piece of memory released by cuFreeAsync (default enabled).
+        (value type = int) Allow cuMemAllocAsync to insert new stream dependencies in order to establish the stream ordering required to reuse a piece of memory released by cuMemFreeAsync (default enabled).
 
 
     .. autoattribute:: cuda.bindings.driver.CUmemPool_attribute.CU_MEMPOOL_ATTR_RELEASE_THRESHOLD

cuda_core/cuda/core/experimental/_context.pyx

@@ -4,7 +4,6 @@
 
 from dataclasses import dataclass
 
-from cuda.core.experimental._utils.clear_error_support import assert_type
 from cuda.core.experimental._utils.cuda_utils import driver
 
 

cuda_core/cuda/core/experimental/_memory.pyx

@@ -20,8 +20,6 @@ from cuda.core.experimental._utils.cuda_utils cimport (
 )
 
 import abc
-import array
-import contextlib
 import cython
 from dataclasses import dataclass, field
 from typing import Iterable, Literal, Optional, TYPE_CHECKING, TypeVar, Union
@@ -34,9 +32,6 @@ from cuda.core.experimental._dlpack import DLDeviceType, make_py_capsule
 from cuda.core.experimental._stream import Stream
 from cuda.core.experimental._utils.cuda_utils import ( driver, Transaction, get_binding_version )
 
-if platform.system() == "Linux":
-    import socket
-
 if TYPE_CHECKING:
     from ._device import Device
     import uuid

cuda_core/cuda/core/experimental/_memoryview.pyx

@@ -2,12 +2,10 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-cimport cython
-
 from ._dlpack cimport *
 
 import functools
-from typing import Any, Optional
+from typing import Optional
 
 import numpy
 
@@ -226,7 +224,7 @@ cdef class _StridedMemoryViewProxy:
 
 
 cdef StridedMemoryView view_as_dlpack(obj, stream_ptr, view=None):
-    cdef int dldevice, device_id, i
+    cdef int dldevice, device_id
     cdef bint is_device_accessible, is_readonly
     is_device_accessible = False
     dldevice, device_id = obj.__dlpack_device__()

cuda_core/cuda/core/experimental/_stream.pyx

@@ -18,7 +18,6 @@ from cuda.core.experimental._utils.cuda_utils cimport (
 )
 
 import cython
-import os
 import warnings
 from dataclasses import dataclass
 from typing import TYPE_CHECKING, Optional, Protocol, Union
@@ -29,7 +28,6 @@ if TYPE_CHECKING:
 from cuda.core.experimental._context import Context
 from cuda.core.experimental._event import Event, EventOptions
 from cuda.core.experimental._graph import GraphBuilder
-from cuda.core.experimental._utils.clear_error_support import assert_type
 from cuda.core.experimental._utils.cuda_utils import (
     driver,
 )