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
18 changes: 11 additions & 7 deletions cuda_bindings/cuda/bindings/_internal/_fast_enum.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
# SPDX-License-Identifier: Apache-2.0


# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=581469c1fadb5f72c43b478d73f2b905562136c8723a25e2f4240b5b681e2894
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=35d522fae601127f3ffc9d0de45d9adf2572934ee3308f5d99567aa8efd5ae77
"""
This is a replacement for the stdlib enum.IntEnum.

Expand Down Expand Up @@ -33,10 +33,14 @@ def __init__(cls, name, bases, namespace):
else:
continue

singleton = int.__new__(cls, value)
singleton.__doc__ = doc
singleton._name = name
cls.__singletons__[value] = singleton
# A name sharing a value with an already-processed member is an
# alias (e.g. a deprecated name kept for backward compatibility):
# it resolves to the same singleton, but isn't a distinct member.
if (singleton := cls.__singletons__.get(value)) is None:
singleton = int.__new__(cls, value)
singleton.__doc__ = doc
singleton._name = name
cls.__singletons__[value] = singleton
cls.__members__[name] = singleton

for name, member in cls.__members__.items():
Expand All @@ -46,10 +50,10 @@ def __repr__(cls) -> str:
return f"<enum '{cls.__name__}'>"

def __len__(cls) -> int:
return len(cls.__members__)
return len(cls.__singletons__)

def __iter__(cls) -> Iterator["FastEnum"]:
return iter(cls.__members__.values())
return iter(cls.__singletons__.values())

def __contains__(cls, item: Any) -> bool:
return item in cls.__singletons__
Expand Down
32 changes: 21 additions & 11 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=69ecb88f06055f5cb48ca1cfc02d37bbbd546020f594d06810a1b70e78059e89
# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=d993ca63a75f175f1e05248eb9e18c79cd0e61ef41e6f13ca05cc3ff1ff91ef6


# <<<< PREAMBLE CONTENT >>>>
Expand Down Expand Up @@ -1627,23 +1627,30 @@ class FenceType(_cyb_IntEnum):
"""
See `cudlaFenceType`.
"""
NVSCISYNC_FENCE = CUDLA_NVSCISYNC_FENCE
NVSCISYNC_FENCE_SOF = CUDLA_NVSCISYNC_FENCE_SOF
FENCE = CUDLA_NVSCISYNC_FENCE
NVSCISYNC_FENCE = FENCE # backward-compat alias
FENCE_SOF = CUDLA_NVSCISYNC_FENCE_SOF
NVSCISYNC_FENCE_SOF = FENCE_SOF # backward-compat alias

class ModuleLoadFlags(_cyb_IntEnum):
"""
See `cudlaModuleLoadFlags`.
"""
MODULE_DEFAULT = CUDLA_MODULE_DEFAULT
MODULE_ENABLE_FAULT_DIAGNOSTICS = CUDLA_MODULE_ENABLE_FAULT_DIAGNOSTICS
DEFAULT = CUDLA_MODULE_DEFAULT
MODULE_DEFAULT = DEFAULT # backward-compat alias
ENABLE_FAULT_DIAGNOSTICS = CUDLA_MODULE_ENABLE_FAULT_DIAGNOSTICS
MODULE_ENABLE_FAULT_DIAGNOSTICS = ENABLE_FAULT_DIAGNOSTICS # backward-compat alias

class SubmissionFlags(_cyb_IntEnum):
"""
See `cudlaSubmissionFlags`.
"""
SUBMIT_NOOP = CUDLA_SUBMIT_NOOP
SUBMIT_SKIP_LOCK_ACQUIRE = CUDLA_SUBMIT_SKIP_LOCK_ACQUIRE
SUBMIT_DIAGNOSTICS_TASK = CUDLA_SUBMIT_DIAGNOSTICS_TASK
NOOP = CUDLA_SUBMIT_NOOP
SUBMIT_NOOP = NOOP # backward-compat alias
SKIP_LOCK_ACQUIRE = CUDLA_SUBMIT_SKIP_LOCK_ACQUIRE
SUBMIT_SKIP_LOCK_ACQUIRE = SKIP_LOCK_ACQUIRE # backward-compat alias
DIAGNOSTICS_TASK = CUDLA_SUBMIT_DIAGNOSTICS_TASK
SUBMIT_DIAGNOSTICS_TASK = DIAGNOSTICS_TASK # backward-compat alias

class AccessPermissionFlags(_cyb_IntEnum):
"""
Expand All @@ -1664,9 +1671,12 @@ class ScratchMemoryConfig(_cyb_IntEnum):
"""
See `cudlaScratchMemoryConfig`.
"""
SCRATCH_MEMORY_DEFAULT = CUDLA_SCRATCH_MEMORY_DEFAULT
SCRATCH_MEMORY_SHARED_STATIC = CUDLA_SCRATCH_MEMORY_SHARED_STATIC
MAX = CUDLA_SCRATCH_MEMORY_CONFIG_MAX
DEFAULT = CUDLA_SCRATCH_MEMORY_DEFAULT
SCRATCH_MEMORY_DEFAULT = DEFAULT # backward-compat alias
SHARED_STATIC = CUDLA_SCRATCH_MEMORY_SHARED_STATIC
SCRATCH_MEMORY_SHARED_STATIC = SHARED_STATIC # backward-compat alias
CONFIG_MAX = CUDLA_SCRATCH_MEMORY_CONFIG_MAX
MAX = CONFIG_MAX # backward-compat alias


###############################################################################
Expand Down
Loading
Loading