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
14 changes: 14 additions & 0 deletions cuda_bindings/cuda/bindings/cyruntime_types.pxi.in
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
3 changes: 2 additions & 1 deletion cuda_bindings/cuda/bindings/driver.pyx.in
Original file line number Diff line number Diff line change
Expand Up @@ -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}}

Expand Down
26 changes: 26 additions & 0 deletions cuda_bindings/cuda/bindings/runtime.pyx.in
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
2 changes: 1 addition & 1 deletion cuda_bindings/docs/source/module/driver.rst
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
Loading