# Copyright 2021-2024 NVIDIA Corporation. All rights reserved. # # Please refer to the NVIDIA end user license agreement (EULA) associated # with this source code for terms and conditions that govern your use of # this software. Any use, reproduction, disclosure, or distribution of # this software and related documentation outside the terms of the EULA # is strictly prohibited. from typing import List, Tuple, Any, Optional from enum import IntEnum import cython import ctypes from libc.stdlib cimport calloc, free from libc cimport string from libc.stdint cimport int32_t, uint32_t, int64_t, uint64_t from libc.stddef cimport wchar_t from libc.limits cimport CHAR_MIN from libcpp.vector cimport vector from cpython.buffer cimport PyObject_CheckBuffer, PyObject_GetBuffer, PyBuffer_Release, PyBUF_SIMPLE, PyBUF_ANY_CONTIGUOUS from cpython.bytes cimport PyBytes_FromStringAndSize ctypedef unsigned long long signed_char_ptr ctypedef unsigned long long unsigned_char_ptr ctypedef unsigned long long char_ptr ctypedef unsigned long long short_ptr ctypedef unsigned long long unsigned_short_ptr ctypedef unsigned long long int_ptr ctypedef unsigned long long long_int_ptr ctypedef unsigned long long long_long_int_ptr ctypedef unsigned long long unsigned_int_ptr ctypedef unsigned long long unsigned_long_int_ptr ctypedef unsigned long long unsigned_long_long_int_ptr ctypedef unsigned long long uint32_t_ptr ctypedef unsigned long long uint64_t_ptr ctypedef unsigned long long int32_t_ptr ctypedef unsigned long long int64_t_ptr ctypedef unsigned long long unsigned_ptr ctypedef unsigned long long unsigned_long_long_ptr ctypedef unsigned long long long_long_ptr ctypedef unsigned long long size_t_ptr ctypedef unsigned long long float_ptr ctypedef unsigned long long double_ptr ctypedef unsigned long long void_ptr #: CUDA API version number CUDA_VERSION = ccuda.CUDA_VERSION #: CUDA IPC handle size CU_IPC_HANDLE_SIZE = ccuda.CU_IPC_HANDLE_SIZE #: Legacy stream handle #: #: Stream handle that can be passed as a CUstream to use an implicit stream #: with legacy synchronization behavior. #: #: See details of the \link_sync_behavior CU_STREAM_LEGACY = ccuda.CU_STREAM_LEGACY #: Per-thread stream handle #: #: Stream handle that can be passed as a CUstream to use an implicit stream #: with per-thread synchronization behavior. #: #: See details of the \link_sync_behavior CU_STREAM_PER_THREAD = ccuda.CU_STREAM_PER_THREAD CU_COMPUTE_ACCELERATED_TARGET_BASE = ccuda.CU_COMPUTE_ACCELERATED_TARGET_BASE #: Conditional node handle flags Default value is applied when graph is #: launched. CU_GRAPH_COND_ASSIGN_DEFAULT = ccuda.CU_GRAPH_COND_ASSIGN_DEFAULT #: This port activates when the kernel has finished executing. CU_GRAPH_KERNEL_NODE_PORT_DEFAULT = ccuda.CU_GRAPH_KERNEL_NODE_PORT_DEFAULT #: This port activates when all blocks of the kernel have performed #: cudaTriggerProgrammaticLaunchCompletion() or have terminated. It must be #: used with edge type :py:obj:`~.CU_GRAPH_DEPENDENCY_TYPE_PROGRAMMATIC`. #: See also :py:obj:`~.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT`. CU_GRAPH_KERNEL_NODE_PORT_PROGRAMMATIC = ccuda.CU_GRAPH_KERNEL_NODE_PORT_PROGRAMMATIC #: This port activates when all blocks of the kernel have begun execution. #: See also :py:obj:`~.CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT`. CU_GRAPH_KERNEL_NODE_PORT_LAUNCH_ORDER = ccuda.CU_GRAPH_KERNEL_NODE_PORT_LAUNCH_ORDER CU_KERNEL_NODE_ATTRIBUTE_ACCESS_POLICY_WINDOW = ccuda.CU_KERNEL_NODE_ATTRIBUTE_ACCESS_POLICY_WINDOW CU_KERNEL_NODE_ATTRIBUTE_COOPERATIVE = ccuda.CU_KERNEL_NODE_ATTRIBUTE_COOPERATIVE CU_KERNEL_NODE_ATTRIBUTE_CLUSTER_DIMENSION = ccuda.CU_KERNEL_NODE_ATTRIBUTE_CLUSTER_DIMENSION CU_KERNEL_NODE_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE = ccuda.CU_KERNEL_NODE_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE CU_KERNEL_NODE_ATTRIBUTE_PRIORITY = ccuda.CU_KERNEL_NODE_ATTRIBUTE_PRIORITY CU_KERNEL_NODE_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP = ccuda.CU_KERNEL_NODE_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP CU_KERNEL_NODE_ATTRIBUTE_MEM_SYNC_DOMAIN = ccuda.CU_KERNEL_NODE_ATTRIBUTE_MEM_SYNC_DOMAIN CU_KERNEL_NODE_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE = ccuda.CU_KERNEL_NODE_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE CU_KERNEL_NODE_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT = ccuda.CU_KERNEL_NODE_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT CU_STREAM_ATTRIBUTE_ACCESS_POLICY_WINDOW = ccuda.CU_STREAM_ATTRIBUTE_ACCESS_POLICY_WINDOW CU_STREAM_ATTRIBUTE_SYNCHRONIZATION_POLICY = ccuda.CU_STREAM_ATTRIBUTE_SYNCHRONIZATION_POLICY CU_STREAM_ATTRIBUTE_PRIORITY = ccuda.CU_STREAM_ATTRIBUTE_PRIORITY CU_STREAM_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP = ccuda.CU_STREAM_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP CU_STREAM_ATTRIBUTE_MEM_SYNC_DOMAIN = ccuda.CU_STREAM_ATTRIBUTE_MEM_SYNC_DOMAIN #: If set, host memory is portable between CUDA contexts. Flag for #: :py:obj:`~.cuMemHostAlloc()` CU_MEMHOSTALLOC_PORTABLE = ccuda.CU_MEMHOSTALLOC_PORTABLE #: If set, host memory is mapped into CUDA address space and #: :py:obj:`~.cuMemHostGetDevicePointer()` may be called on the host #: pointer. Flag for :py:obj:`~.cuMemHostAlloc()` CU_MEMHOSTALLOC_DEVICEMAP = ccuda.CU_MEMHOSTALLOC_DEVICEMAP #: If set, host memory is allocated as write-combined - fast to write, #: faster to DMA, slow to read except via SSE4 streaming load instruction #: (MOVNTDQA). Flag for :py:obj:`~.cuMemHostAlloc()` CU_MEMHOSTALLOC_WRITECOMBINED = ccuda.CU_MEMHOSTALLOC_WRITECOMBINED #: If set, host memory is portable between CUDA contexts. Flag for #: :py:obj:`~.cuMemHostRegister()` CU_MEMHOSTREGISTER_PORTABLE = ccuda.CU_MEMHOSTREGISTER_PORTABLE #: If set, host memory is mapped into CUDA address space and #: :py:obj:`~.cuMemHostGetDevicePointer()` may be called on the host #: pointer. Flag for :py:obj:`~.cuMemHostRegister()` CU_MEMHOSTREGISTER_DEVICEMAP = ccuda.CU_MEMHOSTREGISTER_DEVICEMAP #: If set, the passed memory pointer is treated as pointing to some memory- #: mapped I/O space, e.g. belonging to a third-party PCIe device. On #: Windows the flag is a no-op. On Linux that memory is marked as non #: cache-coherent for the GPU and is expected to be physically contiguous. #: It may return :py:obj:`~.CUDA_ERROR_NOT_PERMITTED` if run as an #: unprivileged user, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` on older Linux #: kernel versions. On all other platforms, it is not supported and #: :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` is returned. Flag for #: :py:obj:`~.cuMemHostRegister()` CU_MEMHOSTREGISTER_IOMEMORY = ccuda.CU_MEMHOSTREGISTER_IOMEMORY #: If set, the passed memory pointer is treated as pointing to memory that #: is considered read-only by the device. On platforms without #: :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, #: this flag is required in order to register memory mapped to the CPU as #: read-only. Support for the use of this flag can be queried from the #: device attribute #: :py:obj:`~.CU_DEVICE_ATTRIBUTE_READ_ONLY_HOST_REGISTER_SUPPORTED`. Using #: this flag with a current context associated with a device that does not #: have this attribute set will cause :py:obj:`~.cuMemHostRegister` to #: error with :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`. CU_MEMHOSTREGISTER_READ_ONLY = ccuda.CU_MEMHOSTREGISTER_READ_ONLY #: Indicates that the layered sparse CUDA array or CUDA mipmapped array has #: a single mip tail region for all layers CU_ARRAY_SPARSE_PROPERTIES_SINGLE_MIPTAIL = ccuda.CU_ARRAY_SPARSE_PROPERTIES_SINGLE_MIPTAIL #: Size of tensor map descriptor CU_TENSOR_MAP_NUM_QWORDS = ccuda.CU_TENSOR_MAP_NUM_QWORDS #: Indicates that the external memory object is a dedicated resource CUDA_EXTERNAL_MEMORY_DEDICATED = ccuda.CUDA_EXTERNAL_MEMORY_DEDICATED #: When the `flags` parameter of #: :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS` contains this flag, it #: indicates that signaling an external semaphore object should skip #: performing appropriate memory synchronization operations over all the #: external memory objects that are imported as #: :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF`, which otherwise are #: performed by default to ensure data coherency with other importers of #: the same NvSciBuf memory objects. CUDA_EXTERNAL_SEMAPHORE_SIGNAL_SKIP_NVSCIBUF_MEMSYNC = ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_SKIP_NVSCIBUF_MEMSYNC #: When the `flags` parameter of #: :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS` contains this flag, it #: indicates that waiting on an external semaphore object should skip #: performing appropriate memory synchronization operations over all the #: external memory objects that are imported as #: :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF`, which otherwise are #: performed by default to ensure data coherency with other importers of #: the same NvSciBuf memory objects. CUDA_EXTERNAL_SEMAPHORE_WAIT_SKIP_NVSCIBUF_MEMSYNC = ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_SKIP_NVSCIBUF_MEMSYNC #: When `flags` of :py:obj:`~.cuDeviceGetNvSciSyncAttributes` is set to #: this, it indicates that application needs signaler specific #: NvSciSyncAttr to be filled by #: :py:obj:`~.cuDeviceGetNvSciSyncAttributes`. CUDA_NVSCISYNC_ATTR_SIGNAL = ccuda.CUDA_NVSCISYNC_ATTR_SIGNAL #: When `flags` of :py:obj:`~.cuDeviceGetNvSciSyncAttributes` is set to #: this, it indicates that application needs waiter specific NvSciSyncAttr #: to be filled by :py:obj:`~.cuDeviceGetNvSciSyncAttributes`. CUDA_NVSCISYNC_ATTR_WAIT = ccuda.CUDA_NVSCISYNC_ATTR_WAIT #: This flag if set indicates that the memory will be used as a tile pool. CU_MEM_CREATE_USAGE_TILE_POOL = ccuda.CU_MEM_CREATE_USAGE_TILE_POOL #: If set, each kernel launched as part of #: :py:obj:`~.cuLaunchCooperativeKernelMultiDevice` only waits for prior #: work in the stream corresponding to that GPU to complete before the #: kernel begins execution. CUDA_COOPERATIVE_LAUNCH_MULTI_DEVICE_NO_PRE_LAUNCH_SYNC = ccuda.CUDA_COOPERATIVE_LAUNCH_MULTI_DEVICE_NO_PRE_LAUNCH_SYNC #: If set, any subsequent work pushed in a stream that participated in a #: call to :py:obj:`~.cuLaunchCooperativeKernelMultiDevice` will only wait #: for the kernel launched on the GPU corresponding to that stream to #: complete before it begins execution. CUDA_COOPERATIVE_LAUNCH_MULTI_DEVICE_NO_POST_LAUNCH_SYNC = ccuda.CUDA_COOPERATIVE_LAUNCH_MULTI_DEVICE_NO_POST_LAUNCH_SYNC #: If set, the CUDA array is a collection of layers, where each layer is #: either a 1D or a 2D array and the Depth member of #: CUDA_ARRAY3D_DESCRIPTOR specifies the number of layers, not the depth of #: a 3D array. CUDA_ARRAY3D_LAYERED = ccuda.CUDA_ARRAY3D_LAYERED #: Deprecated, use CUDA_ARRAY3D_LAYERED CUDA_ARRAY3D_2DARRAY = ccuda.CUDA_ARRAY3D_2DARRAY #: This flag must be set in order to bind a surface reference to the CUDA #: array CUDA_ARRAY3D_SURFACE_LDST = ccuda.CUDA_ARRAY3D_SURFACE_LDST #: If set, the CUDA array is a collection of six 2D arrays, representing #: faces of a cube. The width of such a CUDA array must be equal to its #: height, and Depth must be six. If :py:obj:`~.CUDA_ARRAY3D_LAYERED` flag #: is also set, then the CUDA array is a collection of cubemaps and Depth #: must be a multiple of six. CUDA_ARRAY3D_CUBEMAP = ccuda.CUDA_ARRAY3D_CUBEMAP #: This flag must be set in order to perform texture gather operations on a #: CUDA array. CUDA_ARRAY3D_TEXTURE_GATHER = ccuda.CUDA_ARRAY3D_TEXTURE_GATHER #: This flag if set indicates that the CUDA array is a DEPTH_TEXTURE. CUDA_ARRAY3D_DEPTH_TEXTURE = ccuda.CUDA_ARRAY3D_DEPTH_TEXTURE #: This flag indicates that the CUDA array may be bound as a color target #: in an external graphics API CUDA_ARRAY3D_COLOR_ATTACHMENT = ccuda.CUDA_ARRAY3D_COLOR_ATTACHMENT #: This flag if set indicates that the CUDA array or CUDA mipmapped array #: is a sparse CUDA array or CUDA mipmapped array respectively CUDA_ARRAY3D_SPARSE = ccuda.CUDA_ARRAY3D_SPARSE #: This flag if set indicates that the CUDA array or CUDA mipmapped array #: will allow deferred memory mapping CUDA_ARRAY3D_DEFERRED_MAPPING = ccuda.CUDA_ARRAY3D_DEFERRED_MAPPING #: This flag indicates that the CUDA array will be used for hardware #: accelerated video encode/decode operations. CUDA_ARRAY3D_VIDEO_ENCODE_DECODE = ccuda.CUDA_ARRAY3D_VIDEO_ENCODE_DECODE #: Override the texref format with a format inferred from the array. Flag #: for :py:obj:`~.cuTexRefSetArray()` CU_TRSA_OVERRIDE_FORMAT = ccuda.CU_TRSA_OVERRIDE_FORMAT #: Read the texture as integers rather than promoting the values to floats #: in the range [0,1]. Flag for :py:obj:`~.cuTexRefSetFlags()` and #: :py:obj:`~.cuTexObjectCreate()` CU_TRSF_READ_AS_INTEGER = ccuda.CU_TRSF_READ_AS_INTEGER #: Use normalized texture coordinates in the range [0,1) instead of #: [0,dim). Flag for :py:obj:`~.cuTexRefSetFlags()` and #: :py:obj:`~.cuTexObjectCreate()` CU_TRSF_NORMALIZED_COORDINATES = ccuda.CU_TRSF_NORMALIZED_COORDINATES #: Perform sRGB->linear conversion during texture read. Flag for #: :py:obj:`~.cuTexRefSetFlags()` and :py:obj:`~.cuTexObjectCreate()` CU_TRSF_SRGB = ccuda.CU_TRSF_SRGB #: Disable any trilinear filtering optimizations. Flag for #: :py:obj:`~.cuTexRefSetFlags()` and :py:obj:`~.cuTexObjectCreate()` CU_TRSF_DISABLE_TRILINEAR_OPTIMIZATION = ccuda.CU_TRSF_DISABLE_TRILINEAR_OPTIMIZATION #: Enable seamless cube map filtering. Flag for #: :py:obj:`~.cuTexObjectCreate()` CU_TRSF_SEAMLESS_CUBEMAP = ccuda.CU_TRSF_SEAMLESS_CUBEMAP #: C++ compile time constant for CU_LAUNCH_PARAM_END CU_LAUNCH_PARAM_END_AS_INT = ccuda.CU_LAUNCH_PARAM_END_AS_INT #: End of array terminator for the `extra` parameter to #: :py:obj:`~.cuLaunchKernel` CU_LAUNCH_PARAM_END = ccuda.CU_LAUNCH_PARAM_END #: C++ compile time constant for CU_LAUNCH_PARAM_BUFFER_POINTER CU_LAUNCH_PARAM_BUFFER_POINTER_AS_INT = ccuda.CU_LAUNCH_PARAM_BUFFER_POINTER_AS_INT #: Indicator that the next value in the `extra` parameter to #: :py:obj:`~.cuLaunchKernel` will be a pointer to a buffer containing all #: kernel parameters used for launching kernel `f`. This buffer needs to #: honor all alignment/padding requirements of the individual parameters. #: If :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_SIZE` is not also specified in the #: `extra` array, then :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_POINTER` will have #: no effect. CU_LAUNCH_PARAM_BUFFER_POINTER = ccuda.CU_LAUNCH_PARAM_BUFFER_POINTER #: C++ compile time constant for CU_LAUNCH_PARAM_BUFFER_SIZE CU_LAUNCH_PARAM_BUFFER_SIZE_AS_INT = ccuda.CU_LAUNCH_PARAM_BUFFER_SIZE_AS_INT #: Indicator that the next value in the `extra` parameter to #: :py:obj:`~.cuLaunchKernel` will be a pointer to a size_t which contains #: the size of the buffer specified with #: :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_POINTER`. It is required that #: :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_POINTER` also be specified in the #: `extra` array if the value associated with #: :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_SIZE` is not zero. CU_LAUNCH_PARAM_BUFFER_SIZE = ccuda.CU_LAUNCH_PARAM_BUFFER_SIZE #: For texture references loaded into the module, use default texunit from #: texture reference. CU_PARAM_TR_DEFAULT = ccuda.CU_PARAM_TR_DEFAULT #: Device that represents the CPU CU_DEVICE_CPU = ccuda.CU_DEVICE_CPU #: Device that represents an invalid device CU_DEVICE_INVALID = ccuda.CU_DEVICE_INVALID RESOURCE_ABI_VERSION = ccuda.RESOURCE_ABI_VERSION RESOURCE_ABI_EXTERNAL_BYTES = ccuda.RESOURCE_ABI_EXTERNAL_BYTES #: Maximum number of planes per frame MAX_PLANES = ccuda.MAX_PLANES #: Indicates that timeout for :py:obj:`~.cuEGLStreamConsumerAcquireFrame` #: is infinite. CUDA_EGL_INFINITE_TIMEOUT = ccuda.CUDA_EGL_INFINITE_TIMEOUT {{if 'CUipcMem_flags_enum' in found_types}} class CUipcMem_flags(IntEnum): """ CUDA Ipc Mem Flags """ {{if 'CU_IPC_MEM_LAZY_ENABLE_PEER_ACCESS' in found_values}} #: Automatically enable peer access between remote devices as needed CU_IPC_MEM_LAZY_ENABLE_PEER_ACCESS = ccuda.CUipcMem_flags_enum.CU_IPC_MEM_LAZY_ENABLE_PEER_ACCESS{{endif}} {{endif}} {{if 'CUmemAttach_flags_enum' in found_types}} class CUmemAttach_flags(IntEnum): """ CUDA Mem Attach Flags """ {{if 'CU_MEM_ATTACH_GLOBAL' in found_values}} #: Memory can be accessed by any stream on any device CU_MEM_ATTACH_GLOBAL = ccuda.CUmemAttach_flags_enum.CU_MEM_ATTACH_GLOBAL{{endif}} {{if 'CU_MEM_ATTACH_HOST' in found_values}} #: Memory cannot be accessed by any stream on any device CU_MEM_ATTACH_HOST = ccuda.CUmemAttach_flags_enum.CU_MEM_ATTACH_HOST{{endif}} {{if 'CU_MEM_ATTACH_SINGLE' in found_values}} #: Memory can only be accessed by a single stream on the associated #: device CU_MEM_ATTACH_SINGLE = ccuda.CUmemAttach_flags_enum.CU_MEM_ATTACH_SINGLE{{endif}} {{endif}} {{if 'CUctx_flags_enum' in found_types}} class CUctx_flags(IntEnum): """ Context creation flags """ {{if 'CU_CTX_SCHED_AUTO' in found_values}} #: Automatic scheduling CU_CTX_SCHED_AUTO = ccuda.CUctx_flags_enum.CU_CTX_SCHED_AUTO{{endif}} {{if 'CU_CTX_SCHED_SPIN' in found_values}} #: Set spin as default scheduling CU_CTX_SCHED_SPIN = ccuda.CUctx_flags_enum.CU_CTX_SCHED_SPIN{{endif}} {{if 'CU_CTX_SCHED_YIELD' in found_values}} #: Set yield as default scheduling CU_CTX_SCHED_YIELD = ccuda.CUctx_flags_enum.CU_CTX_SCHED_YIELD{{endif}} {{if 'CU_CTX_SCHED_BLOCKING_SYNC' in found_values}} #: Set blocking synchronization as default scheduling CU_CTX_SCHED_BLOCKING_SYNC = ccuda.CUctx_flags_enum.CU_CTX_SCHED_BLOCKING_SYNC{{endif}} {{if 'CU_CTX_BLOCKING_SYNC' in found_values}} #: Set blocking synchronization as default scheduling [Deprecated] CU_CTX_BLOCKING_SYNC = ccuda.CUctx_flags_enum.CU_CTX_BLOCKING_SYNC{{endif}} {{if 'CU_CTX_SCHED_MASK' in found_values}} CU_CTX_SCHED_MASK = ccuda.CUctx_flags_enum.CU_CTX_SCHED_MASK{{endif}} {{if 'CU_CTX_MAP_HOST' in found_values}} #: [Deprecated] CU_CTX_MAP_HOST = ccuda.CUctx_flags_enum.CU_CTX_MAP_HOST{{endif}} {{if 'CU_CTX_LMEM_RESIZE_TO_MAX' in found_values}} #: Keep local memory allocation after launch CU_CTX_LMEM_RESIZE_TO_MAX = ccuda.CUctx_flags_enum.CU_CTX_LMEM_RESIZE_TO_MAX{{endif}} {{if 'CU_CTX_COREDUMP_ENABLE' in found_values}} #: Trigger coredumps from exceptions in this context CU_CTX_COREDUMP_ENABLE = ccuda.CUctx_flags_enum.CU_CTX_COREDUMP_ENABLE{{endif}} {{if 'CU_CTX_USER_COREDUMP_ENABLE' in found_values}} #: Enable user pipe to trigger coredumps in this context CU_CTX_USER_COREDUMP_ENABLE = ccuda.CUctx_flags_enum.CU_CTX_USER_COREDUMP_ENABLE{{endif}} {{if 'CU_CTX_SYNC_MEMOPS' in found_values}} #: Ensure synchronous memory operations on this context will #: synchronize CU_CTX_SYNC_MEMOPS = ccuda.CUctx_flags_enum.CU_CTX_SYNC_MEMOPS{{endif}} {{if 'CU_CTX_FLAGS_MASK' in found_values}} CU_CTX_FLAGS_MASK = ccuda.CUctx_flags_enum.CU_CTX_FLAGS_MASK{{endif}} {{endif}} {{if 'CUevent_sched_flags_enum' in found_types}} class CUevent_sched_flags(IntEnum): """ Event sched flags """ {{if 'CU_EVENT_SCHED_AUTO' in found_values}} #: Automatic scheduling CU_EVENT_SCHED_AUTO = ccuda.CUevent_sched_flags_enum.CU_EVENT_SCHED_AUTO{{endif}} {{if 'CU_EVENT_SCHED_SPIN' in found_values}} #: Set spin as default scheduling CU_EVENT_SCHED_SPIN = ccuda.CUevent_sched_flags_enum.CU_EVENT_SCHED_SPIN{{endif}} {{if 'CU_EVENT_SCHED_YIELD' in found_values}} #: Set yield as default scheduling CU_EVENT_SCHED_YIELD = ccuda.CUevent_sched_flags_enum.CU_EVENT_SCHED_YIELD{{endif}} {{if 'CU_EVENT_SCHED_BLOCKING_SYNC' in found_values}} #: Set blocking synchronization as default scheduling CU_EVENT_SCHED_BLOCKING_SYNC = ccuda.CUevent_sched_flags_enum.CU_EVENT_SCHED_BLOCKING_SYNC{{endif}} {{endif}} {{if 'cl_event_flags_enum' in found_types}} class cl_event_flags(IntEnum): """ NVCL event scheduling flags """ {{if 'NVCL_EVENT_SCHED_AUTO' in found_values}} #: Automatic scheduling NVCL_EVENT_SCHED_AUTO = ccuda.cl_event_flags_enum.NVCL_EVENT_SCHED_AUTO{{endif}} {{if 'NVCL_EVENT_SCHED_SPIN' in found_values}} #: Set spin as default scheduling NVCL_EVENT_SCHED_SPIN = ccuda.cl_event_flags_enum.NVCL_EVENT_SCHED_SPIN{{endif}} {{if 'NVCL_EVENT_SCHED_YIELD' in found_values}} #: Set yield as default scheduling NVCL_EVENT_SCHED_YIELD = ccuda.cl_event_flags_enum.NVCL_EVENT_SCHED_YIELD{{endif}} {{if 'NVCL_EVENT_SCHED_BLOCKING_SYNC' in found_values}} #: Set blocking synchronization as default scheduling NVCL_EVENT_SCHED_BLOCKING_SYNC = ccuda.cl_event_flags_enum.NVCL_EVENT_SCHED_BLOCKING_SYNC{{endif}} {{endif}} {{if 'cl_context_flags_enum' in found_types}} class cl_context_flags(IntEnum): """ NVCL context scheduling flags """ {{if 'NVCL_CTX_SCHED_AUTO' in found_values}} #: Automatic scheduling NVCL_CTX_SCHED_AUTO = ccuda.cl_context_flags_enum.NVCL_CTX_SCHED_AUTO{{endif}} {{if 'NVCL_CTX_SCHED_SPIN' in found_values}} #: Set spin as default scheduling NVCL_CTX_SCHED_SPIN = ccuda.cl_context_flags_enum.NVCL_CTX_SCHED_SPIN{{endif}} {{if 'NVCL_CTX_SCHED_YIELD' in found_values}} #: Set yield as default scheduling NVCL_CTX_SCHED_YIELD = ccuda.cl_context_flags_enum.NVCL_CTX_SCHED_YIELD{{endif}} {{if 'NVCL_CTX_SCHED_BLOCKING_SYNC' in found_values}} #: Set blocking synchronization as default scheduling NVCL_CTX_SCHED_BLOCKING_SYNC = ccuda.cl_context_flags_enum.NVCL_CTX_SCHED_BLOCKING_SYNC{{endif}} {{endif}} {{if 'CUstream_flags_enum' in found_types}} class CUstream_flags(IntEnum): """ Stream creation flags """ {{if 'CU_STREAM_DEFAULT' in found_values}} #: Default stream flag CU_STREAM_DEFAULT = ccuda.CUstream_flags_enum.CU_STREAM_DEFAULT{{endif}} {{if 'CU_STREAM_NON_BLOCKING' in found_values}} #: Stream does not synchronize with stream 0 (the NULL stream) CU_STREAM_NON_BLOCKING = ccuda.CUstream_flags_enum.CU_STREAM_NON_BLOCKING{{endif}} {{endif}} {{if 'CUevent_flags_enum' in found_types}} class CUevent_flags(IntEnum): """ Event creation flags """ {{if 'CU_EVENT_DEFAULT' in found_values}} #: Default event flag CU_EVENT_DEFAULT = ccuda.CUevent_flags_enum.CU_EVENT_DEFAULT{{endif}} {{if 'CU_EVENT_BLOCKING_SYNC' in found_values}} #: Event uses blocking synchronization CU_EVENT_BLOCKING_SYNC = ccuda.CUevent_flags_enum.CU_EVENT_BLOCKING_SYNC{{endif}} {{if 'CU_EVENT_DISABLE_TIMING' in found_values}} #: Event will not record timing data CU_EVENT_DISABLE_TIMING = ccuda.CUevent_flags_enum.CU_EVENT_DISABLE_TIMING{{endif}} {{if 'CU_EVENT_INTERPROCESS' in found_values}} #: Event is suitable for interprocess use. CU_EVENT_DISABLE_TIMING must #: be set CU_EVENT_INTERPROCESS = ccuda.CUevent_flags_enum.CU_EVENT_INTERPROCESS{{endif}} {{endif}} {{if 'CUevent_record_flags_enum' in found_types}} class CUevent_record_flags(IntEnum): """ Event record flags """ {{if 'CU_EVENT_RECORD_DEFAULT' in found_values}} #: Default event record flag CU_EVENT_RECORD_DEFAULT = ccuda.CUevent_record_flags_enum.CU_EVENT_RECORD_DEFAULT{{endif}} {{if 'CU_EVENT_RECORD_EXTERNAL' in found_values}} #: When using stream capture, create an event record node instead of #: the default behavior. This flag is invalid when used outside of #: capture. CU_EVENT_RECORD_EXTERNAL = ccuda.CUevent_record_flags_enum.CU_EVENT_RECORD_EXTERNAL{{endif}} {{endif}} {{if 'CUevent_wait_flags_enum' in found_types}} class CUevent_wait_flags(IntEnum): """ Event wait flags """ {{if 'CU_EVENT_WAIT_DEFAULT' in found_values}} #: Default event wait flag CU_EVENT_WAIT_DEFAULT = ccuda.CUevent_wait_flags_enum.CU_EVENT_WAIT_DEFAULT{{endif}} {{if 'CU_EVENT_WAIT_EXTERNAL' in found_values}} #: When using stream capture, create an event wait node instead of the #: default behavior. This flag is invalid when used outside of capture. CU_EVENT_WAIT_EXTERNAL = ccuda.CUevent_wait_flags_enum.CU_EVENT_WAIT_EXTERNAL{{endif}} {{endif}} {{if 'CUstreamWaitValue_flags_enum' in found_types}} class CUstreamWaitValue_flags(IntEnum): """ Flags for :py:obj:`~.cuStreamWaitValue32` and :py:obj:`~.cuStreamWaitValue64` """ {{if 'CU_STREAM_WAIT_VALUE_GEQ' in found_values}} #: Wait until (int32_t)(*addr - value) >= 0 (or int64_t for 64 bit #: values). Note this is a cyclic comparison which ignores wraparound. #: (Default behavior.) CU_STREAM_WAIT_VALUE_GEQ = ccuda.CUstreamWaitValue_flags_enum.CU_STREAM_WAIT_VALUE_GEQ{{endif}} {{if 'CU_STREAM_WAIT_VALUE_EQ' in found_values}} #: Wait until *addr == value. CU_STREAM_WAIT_VALUE_EQ = ccuda.CUstreamWaitValue_flags_enum.CU_STREAM_WAIT_VALUE_EQ{{endif}} {{if 'CU_STREAM_WAIT_VALUE_AND' in found_values}} #: Wait until (*addr & value) != 0. CU_STREAM_WAIT_VALUE_AND = ccuda.CUstreamWaitValue_flags_enum.CU_STREAM_WAIT_VALUE_AND{{endif}} {{if 'CU_STREAM_WAIT_VALUE_NOR' in found_values}} #: Wait until ~(*addr | value) != 0. Support for this operation can be #: queried with :py:obj:`~.cuDeviceGetAttribute()` and #: :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR`. CU_STREAM_WAIT_VALUE_NOR = ccuda.CUstreamWaitValue_flags_enum.CU_STREAM_WAIT_VALUE_NOR{{endif}} {{if 'CU_STREAM_WAIT_VALUE_FLUSH' in found_values}} #: Follow the wait operation with a flush of outstanding remote writes. #: This means that, if a remote write operation is guaranteed to have #: reached the device before the wait can be satisfied, that write is #: guaranteed to be visible to downstream device work. The device is #: permitted to reorder remote writes internally. For example, this #: flag would be required if two remote writes arrive in a defined #: order, the wait is satisfied by the second write, and downstream #: work needs to observe the first write. Support for this operation is #: restricted to selected platforms and can be queried with #: :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_FLUSH_REMOTE_WRITES`. CU_STREAM_WAIT_VALUE_FLUSH = ccuda.CUstreamWaitValue_flags_enum.CU_STREAM_WAIT_VALUE_FLUSH{{endif}} {{endif}} {{if 'CUstreamWriteValue_flags_enum' in found_types}} class CUstreamWriteValue_flags(IntEnum): """ Flags for :py:obj:`~.cuStreamWriteValue32` """ {{if 'CU_STREAM_WRITE_VALUE_DEFAULT' in found_values}} #: Default behavior CU_STREAM_WRITE_VALUE_DEFAULT = ccuda.CUstreamWriteValue_flags_enum.CU_STREAM_WRITE_VALUE_DEFAULT{{endif}} {{if 'CU_STREAM_WRITE_VALUE_NO_MEMORY_BARRIER' in found_values}} #: Permits the write to be reordered with writes which were issued #: before it, as a performance optimization. Normally, #: :py:obj:`~.cuStreamWriteValue32` will provide a memory fence before #: the write, which has similar semantics to __threadfence_system() but #: is scoped to the stream rather than a CUDA thread. This flag is not #: supported in the v2 API. CU_STREAM_WRITE_VALUE_NO_MEMORY_BARRIER = ccuda.CUstreamWriteValue_flags_enum.CU_STREAM_WRITE_VALUE_NO_MEMORY_BARRIER{{endif}} {{endif}} {{if 'CUstreamBatchMemOpType_enum' in found_types}} class CUstreamBatchMemOpType(IntEnum): """ Operations for :py:obj:`~.cuStreamBatchMemOp` """ {{if 'CU_STREAM_MEM_OP_WAIT_VALUE_32' in found_values}} #: Represents a :py:obj:`~.cuStreamWaitValue32` operation CU_STREAM_MEM_OP_WAIT_VALUE_32 = ccuda.CUstreamBatchMemOpType_enum.CU_STREAM_MEM_OP_WAIT_VALUE_32{{endif}} {{if 'CU_STREAM_MEM_OP_WRITE_VALUE_32' in found_values}} #: Represents a :py:obj:`~.cuStreamWriteValue32` operation CU_STREAM_MEM_OP_WRITE_VALUE_32 = ccuda.CUstreamBatchMemOpType_enum.CU_STREAM_MEM_OP_WRITE_VALUE_32{{endif}} {{if 'CU_STREAM_MEM_OP_FLUSH_REMOTE_WRITES' in found_values}} #: This has the same effect as :py:obj:`~.CU_STREAM_WAIT_VALUE_FLUSH`, #: but as a standalone operation. CU_STREAM_MEM_OP_FLUSH_REMOTE_WRITES = ccuda.CUstreamBatchMemOpType_enum.CU_STREAM_MEM_OP_FLUSH_REMOTE_WRITES{{endif}} {{if 'CU_STREAM_MEM_OP_WAIT_VALUE_64' in found_values}} #: Represents a :py:obj:`~.cuStreamWaitValue64` operation CU_STREAM_MEM_OP_WAIT_VALUE_64 = ccuda.CUstreamBatchMemOpType_enum.CU_STREAM_MEM_OP_WAIT_VALUE_64{{endif}} {{if 'CU_STREAM_MEM_OP_WRITE_VALUE_64' in found_values}} #: Represents a :py:obj:`~.cuStreamWriteValue64` operation CU_STREAM_MEM_OP_WRITE_VALUE_64 = ccuda.CUstreamBatchMemOpType_enum.CU_STREAM_MEM_OP_WRITE_VALUE_64{{endif}} {{if 'CU_STREAM_MEM_OP_BARRIER' in found_values}} #: Insert a memory barrier of the specified type CU_STREAM_MEM_OP_BARRIER = ccuda.CUstreamBatchMemOpType_enum.CU_STREAM_MEM_OP_BARRIER{{endif}} {{endif}} {{if 'CUstreamMemoryBarrier_flags_enum' in found_types}} class CUstreamMemoryBarrier_flags(IntEnum): """ Flags for :py:obj:`~.cuStreamMemoryBarrier` """ {{if 'CU_STREAM_MEMORY_BARRIER_TYPE_SYS' in found_values}} #: System-wide memory barrier. CU_STREAM_MEMORY_BARRIER_TYPE_SYS = ccuda.CUstreamMemoryBarrier_flags_enum.CU_STREAM_MEMORY_BARRIER_TYPE_SYS{{endif}} {{if 'CU_STREAM_MEMORY_BARRIER_TYPE_GPU' in found_values}} #: Limit memory barrier scope to the GPU. CU_STREAM_MEMORY_BARRIER_TYPE_GPU = ccuda.CUstreamMemoryBarrier_flags_enum.CU_STREAM_MEMORY_BARRIER_TYPE_GPU{{endif}} {{endif}} {{if 'CUoccupancy_flags_enum' in found_types}} class CUoccupancy_flags(IntEnum): """ Occupancy calculator flag """ {{if 'CU_OCCUPANCY_DEFAULT' in found_values}} #: Default behavior CU_OCCUPANCY_DEFAULT = ccuda.CUoccupancy_flags_enum.CU_OCCUPANCY_DEFAULT{{endif}} {{if 'CU_OCCUPANCY_DISABLE_CACHING_OVERRIDE' in found_values}} #: Assume global caching is enabled and cannot be automatically turned #: off CU_OCCUPANCY_DISABLE_CACHING_OVERRIDE = ccuda.CUoccupancy_flags_enum.CU_OCCUPANCY_DISABLE_CACHING_OVERRIDE{{endif}} {{endif}} {{if 'CUstreamUpdateCaptureDependencies_flags_enum' in found_types}} class CUstreamUpdateCaptureDependencies_flags(IntEnum): """ Flags for :py:obj:`~.cuStreamUpdateCaptureDependencies` """ {{if 'CU_STREAM_ADD_CAPTURE_DEPENDENCIES' in found_values}} #: Add new nodes to the dependency set CU_STREAM_ADD_CAPTURE_DEPENDENCIES = ccuda.CUstreamUpdateCaptureDependencies_flags_enum.CU_STREAM_ADD_CAPTURE_DEPENDENCIES{{endif}} {{if 'CU_STREAM_SET_CAPTURE_DEPENDENCIES' in found_values}} #: Replace the dependency set with the new nodes CU_STREAM_SET_CAPTURE_DEPENDENCIES = ccuda.CUstreamUpdateCaptureDependencies_flags_enum.CU_STREAM_SET_CAPTURE_DEPENDENCIES{{endif}} {{endif}} {{if 'CUasyncNotificationType_enum' in found_types}} class CUasyncNotificationType(IntEnum): """ Types of async notification that can be sent """ {{if 'CU_ASYNC_NOTIFICATION_TYPE_OVER_BUDGET' in found_values}} CU_ASYNC_NOTIFICATION_TYPE_OVER_BUDGET = ccuda.CUasyncNotificationType_enum.CU_ASYNC_NOTIFICATION_TYPE_OVER_BUDGET{{endif}} {{endif}} {{if 'CUarray_format_enum' in found_types}} class CUarray_format(IntEnum): """ Array formats """ {{if 'CU_AD_FORMAT_UNSIGNED_INT8' in found_values}} #: Unsigned 8-bit integers CU_AD_FORMAT_UNSIGNED_INT8 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNSIGNED_INT8{{endif}} {{if 'CU_AD_FORMAT_UNSIGNED_INT16' in found_values}} #: Unsigned 16-bit integers CU_AD_FORMAT_UNSIGNED_INT16 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNSIGNED_INT16{{endif}} {{if 'CU_AD_FORMAT_UNSIGNED_INT32' in found_values}} #: Unsigned 32-bit integers CU_AD_FORMAT_UNSIGNED_INT32 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNSIGNED_INT32{{endif}} {{if 'CU_AD_FORMAT_SIGNED_INT8' in found_values}} #: Signed 8-bit integers CU_AD_FORMAT_SIGNED_INT8 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SIGNED_INT8{{endif}} {{if 'CU_AD_FORMAT_SIGNED_INT16' in found_values}} #: Signed 16-bit integers CU_AD_FORMAT_SIGNED_INT16 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SIGNED_INT16{{endif}} {{if 'CU_AD_FORMAT_SIGNED_INT32' in found_values}} #: Signed 32-bit integers CU_AD_FORMAT_SIGNED_INT32 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SIGNED_INT32{{endif}} {{if 'CU_AD_FORMAT_HALF' in found_values}} #: 16-bit floating point CU_AD_FORMAT_HALF = ccuda.CUarray_format_enum.CU_AD_FORMAT_HALF{{endif}} {{if 'CU_AD_FORMAT_FLOAT' in found_values}} #: 32-bit floating point CU_AD_FORMAT_FLOAT = ccuda.CUarray_format_enum.CU_AD_FORMAT_FLOAT{{endif}} {{if 'CU_AD_FORMAT_BC1_UNORM' in found_values}} #: 4 channel unsigned normalized block-compressed (BC1 compression) #: format CU_AD_FORMAT_BC1_UNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC1_UNORM{{endif}} {{if 'CU_AD_FORMAT_BC1_UNORM_SRGB' in found_values}} #: 4 channel unsigned normalized block-compressed (BC1 compression) #: format with sRGB encoding CU_AD_FORMAT_BC1_UNORM_SRGB = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC1_UNORM_SRGB{{endif}} {{if 'CU_AD_FORMAT_BC2_UNORM' in found_values}} #: 4 channel unsigned normalized block-compressed (BC2 compression) #: format CU_AD_FORMAT_BC2_UNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC2_UNORM{{endif}} {{if 'CU_AD_FORMAT_BC2_UNORM_SRGB' in found_values}} #: 4 channel unsigned normalized block-compressed (BC2 compression) #: format with sRGB encoding CU_AD_FORMAT_BC2_UNORM_SRGB = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC2_UNORM_SRGB{{endif}} {{if 'CU_AD_FORMAT_BC3_UNORM' in found_values}} #: 4 channel unsigned normalized block-compressed (BC3 compression) #: format CU_AD_FORMAT_BC3_UNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC3_UNORM{{endif}} {{if 'CU_AD_FORMAT_BC3_UNORM_SRGB' in found_values}} #: 4 channel unsigned normalized block-compressed (BC3 compression) #: format with sRGB encoding CU_AD_FORMAT_BC3_UNORM_SRGB = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC3_UNORM_SRGB{{endif}} {{if 'CU_AD_FORMAT_BC4_UNORM' in found_values}} #: 1 channel unsigned normalized block-compressed (BC4 compression) #: format CU_AD_FORMAT_BC4_UNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC4_UNORM{{endif}} {{if 'CU_AD_FORMAT_BC4_SNORM' in found_values}} #: 1 channel signed normalized block-compressed (BC4 compression) #: format CU_AD_FORMAT_BC4_SNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC4_SNORM{{endif}} {{if 'CU_AD_FORMAT_BC5_UNORM' in found_values}} #: 2 channel unsigned normalized block-compressed (BC5 compression) #: format CU_AD_FORMAT_BC5_UNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC5_UNORM{{endif}} {{if 'CU_AD_FORMAT_BC5_SNORM' in found_values}} #: 2 channel signed normalized block-compressed (BC5 compression) #: format CU_AD_FORMAT_BC5_SNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC5_SNORM{{endif}} {{if 'CU_AD_FORMAT_BC6H_UF16' in found_values}} #: 3 channel unsigned half-float block-compressed (BC6H compression) #: format CU_AD_FORMAT_BC6H_UF16 = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC6H_UF16{{endif}} {{if 'CU_AD_FORMAT_BC6H_SF16' in found_values}} #: 3 channel signed half-float block-compressed (BC6H compression) #: format CU_AD_FORMAT_BC6H_SF16 = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC6H_SF16{{endif}} {{if 'CU_AD_FORMAT_BC7_UNORM' in found_values}} #: 4 channel unsigned normalized block-compressed (BC7 compression) #: format CU_AD_FORMAT_BC7_UNORM = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC7_UNORM{{endif}} {{if 'CU_AD_FORMAT_BC7_UNORM_SRGB' in found_values}} #: 4 channel unsigned normalized block-compressed (BC7 compression) #: format with sRGB encoding CU_AD_FORMAT_BC7_UNORM_SRGB = ccuda.CUarray_format_enum.CU_AD_FORMAT_BC7_UNORM_SRGB{{endif}} {{if 'CU_AD_FORMAT_P010' in found_values}} #: 10-bit YUV planar format, with 4:2:0 sampling CU_AD_FORMAT_P010 = ccuda.CUarray_format_enum.CU_AD_FORMAT_P010{{endif}} {{if 'CU_AD_FORMAT_P016' in found_values}} #: 16-bit YUV planar format, with 4:2:0 sampling CU_AD_FORMAT_P016 = ccuda.CUarray_format_enum.CU_AD_FORMAT_P016{{endif}} {{if 'CU_AD_FORMAT_NV16' in found_values}} #: 8-bit YUV planar format, with 4:2:2 sampling CU_AD_FORMAT_NV16 = ccuda.CUarray_format_enum.CU_AD_FORMAT_NV16{{endif}} {{if 'CU_AD_FORMAT_P210' in found_values}} #: 10-bit YUV planar format, with 4:2:2 sampling CU_AD_FORMAT_P210 = ccuda.CUarray_format_enum.CU_AD_FORMAT_P210{{endif}} {{if 'CU_AD_FORMAT_P216' in found_values}} #: 16-bit YUV planar format, with 4:2:2 sampling CU_AD_FORMAT_P216 = ccuda.CUarray_format_enum.CU_AD_FORMAT_P216{{endif}} {{if 'CU_AD_FORMAT_YUY2' in found_values}} #: 2 channel, 8-bit YUV packed planar format, with 4:2:2 sampling CU_AD_FORMAT_YUY2 = ccuda.CUarray_format_enum.CU_AD_FORMAT_YUY2{{endif}} {{if 'CU_AD_FORMAT_Y210' in found_values}} #: 2 channel, 10-bit YUV packed planar format, with 4:2:2 sampling CU_AD_FORMAT_Y210 = ccuda.CUarray_format_enum.CU_AD_FORMAT_Y210{{endif}} {{if 'CU_AD_FORMAT_Y216' in found_values}} #: 2 channel, 16-bit YUV packed planar format, with 4:2:2 sampling CU_AD_FORMAT_Y216 = ccuda.CUarray_format_enum.CU_AD_FORMAT_Y216{{endif}} {{if 'CU_AD_FORMAT_AYUV' in found_values}} #: 4 channel, 8-bit YUV packed planar format, with 4:4:4 sampling CU_AD_FORMAT_AYUV = ccuda.CUarray_format_enum.CU_AD_FORMAT_AYUV{{endif}} {{if 'CU_AD_FORMAT_Y410' in found_values}} #: 10-bit YUV packed planar format, with 4:4:4 sampling CU_AD_FORMAT_Y410 = ccuda.CUarray_format_enum.CU_AD_FORMAT_Y410{{endif}} {{if 'CU_AD_FORMAT_NV12' in found_values}} #: 8-bit YUV planar format, with 4:2:0 sampling CU_AD_FORMAT_NV12 = ccuda.CUarray_format_enum.CU_AD_FORMAT_NV12{{endif}} {{if 'CU_AD_FORMAT_Y416' in found_values}} #: 4 channel, 12-bit YUV packed planar format, with 4:4:4 sampling CU_AD_FORMAT_Y416 = ccuda.CUarray_format_enum.CU_AD_FORMAT_Y416{{endif}} {{if 'CU_AD_FORMAT_Y444_PLANAR8' in found_values}} #: 3 channel 8-bit YUV planar format, with 4:4:4 sampling CU_AD_FORMAT_Y444_PLANAR8 = ccuda.CUarray_format_enum.CU_AD_FORMAT_Y444_PLANAR8{{endif}} {{if 'CU_AD_FORMAT_Y444_PLANAR10' in found_values}} #: 3 channel 10-bit YUV planar format, with 4:4:4 sampling CU_AD_FORMAT_Y444_PLANAR10 = ccuda.CUarray_format_enum.CU_AD_FORMAT_Y444_PLANAR10{{endif}} {{if 'CU_AD_FORMAT_UNORM_INT8X1' in found_values}} #: 1 channel unsigned 8-bit normalized integer CU_AD_FORMAT_UNORM_INT8X1 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNORM_INT8X1{{endif}} {{if 'CU_AD_FORMAT_UNORM_INT8X2' in found_values}} #: 2 channel unsigned 8-bit normalized integer CU_AD_FORMAT_UNORM_INT8X2 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNORM_INT8X2{{endif}} {{if 'CU_AD_FORMAT_UNORM_INT8X4' in found_values}} #: 4 channel unsigned 8-bit normalized integer CU_AD_FORMAT_UNORM_INT8X4 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNORM_INT8X4{{endif}} {{if 'CU_AD_FORMAT_UNORM_INT16X1' in found_values}} #: 1 channel unsigned 16-bit normalized integer CU_AD_FORMAT_UNORM_INT16X1 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNORM_INT16X1{{endif}} {{if 'CU_AD_FORMAT_UNORM_INT16X2' in found_values}} #: 2 channel unsigned 16-bit normalized integer CU_AD_FORMAT_UNORM_INT16X2 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNORM_INT16X2{{endif}} {{if 'CU_AD_FORMAT_UNORM_INT16X4' in found_values}} #: 4 channel unsigned 16-bit normalized integer CU_AD_FORMAT_UNORM_INT16X4 = ccuda.CUarray_format_enum.CU_AD_FORMAT_UNORM_INT16X4{{endif}} {{if 'CU_AD_FORMAT_SNORM_INT8X1' in found_values}} #: 1 channel signed 8-bit normalized integer CU_AD_FORMAT_SNORM_INT8X1 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SNORM_INT8X1{{endif}} {{if 'CU_AD_FORMAT_SNORM_INT8X2' in found_values}} #: 2 channel signed 8-bit normalized integer CU_AD_FORMAT_SNORM_INT8X2 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SNORM_INT8X2{{endif}} {{if 'CU_AD_FORMAT_SNORM_INT8X4' in found_values}} #: 4 channel signed 8-bit normalized integer CU_AD_FORMAT_SNORM_INT8X4 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SNORM_INT8X4{{endif}} {{if 'CU_AD_FORMAT_SNORM_INT16X1' in found_values}} #: 1 channel signed 16-bit normalized integer CU_AD_FORMAT_SNORM_INT16X1 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SNORM_INT16X1{{endif}} {{if 'CU_AD_FORMAT_SNORM_INT16X2' in found_values}} #: 2 channel signed 16-bit normalized integer CU_AD_FORMAT_SNORM_INT16X2 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SNORM_INT16X2{{endif}} {{if 'CU_AD_FORMAT_SNORM_INT16X4' in found_values}} #: 4 channel signed 16-bit normalized integer CU_AD_FORMAT_SNORM_INT16X4 = ccuda.CUarray_format_enum.CU_AD_FORMAT_SNORM_INT16X4{{endif}} {{if 'CU_AD_FORMAT_MAX' in found_values}} CU_AD_FORMAT_MAX = ccuda.CUarray_format_enum.CU_AD_FORMAT_MAX{{endif}} {{endif}} {{if 'CUaddress_mode_enum' in found_types}} class CUaddress_mode(IntEnum): """ Texture reference addressing modes """ {{if 'CU_TR_ADDRESS_MODE_WRAP' in found_values}} #: Wrapping address mode CU_TR_ADDRESS_MODE_WRAP = ccuda.CUaddress_mode_enum.CU_TR_ADDRESS_MODE_WRAP{{endif}} {{if 'CU_TR_ADDRESS_MODE_CLAMP' in found_values}} #: Clamp to edge address mode CU_TR_ADDRESS_MODE_CLAMP = ccuda.CUaddress_mode_enum.CU_TR_ADDRESS_MODE_CLAMP{{endif}} {{if 'CU_TR_ADDRESS_MODE_MIRROR' in found_values}} #: Mirror address mode CU_TR_ADDRESS_MODE_MIRROR = ccuda.CUaddress_mode_enum.CU_TR_ADDRESS_MODE_MIRROR{{endif}} {{if 'CU_TR_ADDRESS_MODE_BORDER' in found_values}} #: Border address mode CU_TR_ADDRESS_MODE_BORDER = ccuda.CUaddress_mode_enum.CU_TR_ADDRESS_MODE_BORDER{{endif}} {{endif}} {{if 'CUfilter_mode_enum' in found_types}} class CUfilter_mode(IntEnum): """ Texture reference filtering modes """ {{if 'CU_TR_FILTER_MODE_POINT' in found_values}} #: Point filter mode CU_TR_FILTER_MODE_POINT = ccuda.CUfilter_mode_enum.CU_TR_FILTER_MODE_POINT{{endif}} {{if 'CU_TR_FILTER_MODE_LINEAR' in found_values}} #: Linear filter mode CU_TR_FILTER_MODE_LINEAR = ccuda.CUfilter_mode_enum.CU_TR_FILTER_MODE_LINEAR{{endif}} {{endif}} {{if 'CUdevice_attribute_enum' in found_types}} class CUdevice_attribute(IntEnum): """ Device properties """ {{if 'CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK' in found_values}} #: Maximum number of threads per block CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X' in found_values}} #: Maximum block dimension X CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Y' in found_values}} #: Maximum block dimension Y CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Y = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Y{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Z' in found_values}} #: Maximum block dimension Z CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Z = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Z{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X' in found_values}} #: Maximum grid dimension X CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Y' in found_values}} #: Maximum grid dimension Y CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Y = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Y{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Z' in found_values}} #: Maximum grid dimension Z CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Z = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Z{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK' in found_values}} #: Maximum shared memory available per block in bytes CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_SHARED_MEMORY_PER_BLOCK' in found_values}} #: Deprecated, use CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK CU_DEVICE_ATTRIBUTE_SHARED_MEMORY_PER_BLOCK = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_SHARED_MEMORY_PER_BLOCK{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_TOTAL_CONSTANT_MEMORY' in found_values}} #: Memory available on device for constant variables in a CUDA C kernel #: in bytes CU_DEVICE_ATTRIBUTE_TOTAL_CONSTANT_MEMORY = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_TOTAL_CONSTANT_MEMORY{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_WARP_SIZE' in found_values}} #: Warp size in threads CU_DEVICE_ATTRIBUTE_WARP_SIZE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_WARP_SIZE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_PITCH' in found_values}} #: Maximum pitch in bytes allowed by memory copies CU_DEVICE_ATTRIBUTE_MAX_PITCH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_PITCH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK' in found_values}} #: Maximum number of 32-bit registers available per block CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_REGISTERS_PER_BLOCK' in found_values}} #: Deprecated, use CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK CU_DEVICE_ATTRIBUTE_REGISTERS_PER_BLOCK = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_REGISTERS_PER_BLOCK{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CLOCK_RATE' in found_values}} #: Typical clock frequency in kilohertz CU_DEVICE_ATTRIBUTE_CLOCK_RATE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CLOCK_RATE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT' in found_values}} #: Alignment requirement for textures CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GPU_OVERLAP' in found_values}} #: Device can possibly copy memory and execute a kernel concurrently. #: Deprecated. Use instead CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT. CU_DEVICE_ATTRIBUTE_GPU_OVERLAP = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GPU_OVERLAP{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT' in found_values}} #: Number of multiprocessors on device CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT' in found_values}} #: Specifies whether there is a run time limit on kernels CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_INTEGRATED' in found_values}} #: Device is integrated with host memory CU_DEVICE_ATTRIBUTE_INTEGRATED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_INTEGRATED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY' in found_values}} #: Device can map host memory into CUDA address space CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_COMPUTE_MODE' in found_values}} #: Compute mode (See :py:obj:`~.CUcomputemode` for details) CU_DEVICE_ATTRIBUTE_COMPUTE_MODE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_COMPUTE_MODE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_WIDTH' in found_values}} #: Maximum 1D texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_WIDTH' in found_values}} #: Maximum 2D texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_HEIGHT' in found_values}} #: Maximum 2D texture height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH' in found_values}} #: Maximum 3D texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT' in found_values}} #: Maximum 3D texture height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH' in found_values}} #: Maximum 3D texture depth CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_WIDTH' in found_values}} #: Maximum 2D layered texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_WIDTH' in found_values}} #: Deprecated, use CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_WIDTH CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_HEIGHT' in found_values}} #: Maximum 2D layered texture height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_HEIGHT' in found_values}} #: Deprecated, use CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_HEIGHT CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_LAYERS' in found_values}} #: Maximum layers in a 2D layered texture CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_LAYERS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_LAYERS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_NUMSLICES' in found_values}} #: Deprecated, use CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_LAYERS CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_NUMSLICES = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_ARRAY_NUMSLICES{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_SURFACE_ALIGNMENT' in found_values}} #: Alignment requirement for surfaces CU_DEVICE_ATTRIBUTE_SURFACE_ALIGNMENT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_SURFACE_ALIGNMENT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS' in found_values}} #: Device can possibly execute multiple kernels concurrently CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_ECC_ENABLED' in found_values}} #: Device has ECC support enabled CU_DEVICE_ATTRIBUTE_ECC_ENABLED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_ECC_ENABLED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_PCI_BUS_ID' in found_values}} #: PCI bus ID of the device CU_DEVICE_ATTRIBUTE_PCI_BUS_ID = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_PCI_BUS_ID{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_PCI_DEVICE_ID' in found_values}} #: PCI device ID of the device CU_DEVICE_ATTRIBUTE_PCI_DEVICE_ID = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_PCI_DEVICE_ID{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_TCC_DRIVER' in found_values}} #: Device is using TCC driver model CU_DEVICE_ATTRIBUTE_TCC_DRIVER = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_TCC_DRIVER{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE' in found_values}} #: Peak memory clock frequency in kilohertz CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH' in found_values}} #: Global memory bus width in bits CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_L2_CACHE_SIZE' in found_values}} #: Size of L2 cache in bytes CU_DEVICE_ATTRIBUTE_L2_CACHE_SIZE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_L2_CACHE_SIZE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR' in found_values}} #: Maximum resident threads per multiprocessor CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT' in found_values}} #: Number of asynchronous engines CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_ASYNC_ENGINE_COUNT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING' in found_values}} #: Device shares a unified address space with the host CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_WIDTH' in found_values}} #: Maximum 1D layered texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_LAYERS' in found_values}} #: Maximum layers in a 1D layered texture CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_LAYERS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_LAYERS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_TEX2D_GATHER' in found_values}} #: Deprecated, do not use. CU_DEVICE_ATTRIBUTE_CAN_TEX2D_GATHER = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_TEX2D_GATHER{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_WIDTH' in found_values}} #: Maximum 2D texture width if CUDA_ARRAY3D_TEXTURE_GATHER is set CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_HEIGHT' in found_values}} #: Maximum 2D texture height if CUDA_ARRAY3D_TEXTURE_GATHER is set CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH_ALTERNATE' in found_values}} #: Alternate maximum 3D texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH_ALTERNATE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH_ALTERNATE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT_ALTERNATE' in found_values}} #: Alternate maximum 3D texture height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT_ALTERNATE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT_ALTERNATE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH_ALTERNATE' in found_values}} #: Alternate maximum 3D texture depth CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH_ALTERNATE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH_ALTERNATE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_PCI_DOMAIN_ID' in found_values}} #: PCI domain ID of the device CU_DEVICE_ATTRIBUTE_PCI_DOMAIN_ID = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_PCI_DOMAIN_ID{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_TEXTURE_PITCH_ALIGNMENT' in found_values}} #: Pitch alignment requirement for textures CU_DEVICE_ATTRIBUTE_TEXTURE_PITCH_ALIGNMENT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_TEXTURE_PITCH_ALIGNMENT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_WIDTH' in found_values}} #: Maximum cubemap texture width/height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_WIDTH' in found_values}} #: Maximum cubemap layered texture width/height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_LAYERS' in found_values}} #: Maximum layers in a cubemap layered texture CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_LAYERS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_LAYERS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_WIDTH' in found_values}} #: Maximum 1D surface width CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_WIDTH' in found_values}} #: Maximum 2D surface width CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_HEIGHT' in found_values}} #: Maximum 2D surface height CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_WIDTH' in found_values}} #: Maximum 3D surface width CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_HEIGHT' in found_values}} #: Maximum 3D surface height CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_DEPTH' in found_values}} #: Maximum 3D surface depth CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_DEPTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_DEPTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_WIDTH' in found_values}} #: Maximum 1D layered surface width CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_LAYERS' in found_values}} #: Maximum layers in a 1D layered surface CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_LAYERS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_LAYERS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_WIDTH' in found_values}} #: Maximum 2D layered surface width CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_HEIGHT' in found_values}} #: Maximum 2D layered surface height CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_LAYERS' in found_values}} #: Maximum layers in a 2D layered surface CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_LAYERS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_LAYERS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_WIDTH' in found_values}} #: Maximum cubemap surface width CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_WIDTH' in found_values}} #: Maximum cubemap layered surface width CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_LAYERS' in found_values}} #: Maximum layers in a cubemap layered surface CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_LAYERS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_LAYERS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LINEAR_WIDTH' in found_values}} #: Deprecated, do not use. Use cudaDeviceGetTexture1DLinearMaxWidth() #: or :py:obj:`~.cuDeviceGetTexture1DLinearMaxWidth()` instead. CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LINEAR_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LINEAR_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_WIDTH' in found_values}} #: Maximum 2D linear texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_HEIGHT' in found_values}} #: Maximum 2D linear texture height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_PITCH' in found_values}} #: Maximum 2D linear texture pitch in bytes CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_PITCH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_PITCH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_WIDTH' in found_values}} #: Maximum mipmapped 2D texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_HEIGHT' in found_values}} #: Maximum mipmapped 2D texture height CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_HEIGHT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_HEIGHT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR' in found_values}} #: Major compute capability version number CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR' in found_values}} #: Minor compute capability version number CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_MIPMAPPED_WIDTH' in found_values}} #: Maximum mipmapped 1D texture width CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_MIPMAPPED_WIDTH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_MIPMAPPED_WIDTH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_STREAM_PRIORITIES_SUPPORTED' in found_values}} #: Device supports stream priorities CU_DEVICE_ATTRIBUTE_STREAM_PRIORITIES_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_STREAM_PRIORITIES_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GLOBAL_L1_CACHE_SUPPORTED' in found_values}} #: Device supports caching globals in L1 CU_DEVICE_ATTRIBUTE_GLOBAL_L1_CACHE_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GLOBAL_L1_CACHE_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_LOCAL_L1_CACHE_SUPPORTED' in found_values}} #: Device supports caching locals in L1 CU_DEVICE_ATTRIBUTE_LOCAL_L1_CACHE_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_LOCAL_L1_CACHE_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_MULTIPROCESSOR' in found_values}} #: Maximum shared memory available per multiprocessor in bytes CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_MULTIPROCESSOR = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_MULTIPROCESSOR{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_MULTIPROCESSOR' in found_values}} #: Maximum number of 32-bit registers available per multiprocessor CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_MULTIPROCESSOR = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_MULTIPROCESSOR{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY' in found_values}} #: Device can allocate managed memory on this system CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD' in found_values}} #: Device is on a multi-GPU board CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD_GROUP_ID' in found_values}} #: Unique id for a group of devices on the same multi-GPU board CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD_GROUP_ID = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD_GROUP_ID{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_HOST_NATIVE_ATOMIC_SUPPORTED' in found_values}} #: Link between the device and the host supports native atomic #: operations (this is a placeholder attribute, and is not supported on #: any current hardware) CU_DEVICE_ATTRIBUTE_HOST_NATIVE_ATOMIC_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_HOST_NATIVE_ATOMIC_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_SINGLE_TO_DOUBLE_PRECISION_PERF_RATIO' in found_values}} #: Ratio of single precision performance (in floating-point operations #: per second) to double precision performance CU_DEVICE_ATTRIBUTE_SINGLE_TO_DOUBLE_PRECISION_PERF_RATIO = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_SINGLE_TO_DOUBLE_PRECISION_PERF_RATIO{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS' in found_values}} #: Device supports coherently accessing pageable memory without calling #: cudaHostRegister on it CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS' in found_values}} #: Device can coherently access managed memory concurrently with the #: CPU CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_COMPUTE_PREEMPTION_SUPPORTED' in found_values}} #: Device supports compute preemption. CU_DEVICE_ATTRIBUTE_COMPUTE_PREEMPTION_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_COMPUTE_PREEMPTION_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_USE_HOST_POINTER_FOR_REGISTERED_MEM' in found_values}} #: Device can access host registered memory at the same virtual address #: as the CPU CU_DEVICE_ATTRIBUTE_CAN_USE_HOST_POINTER_FOR_REGISTERED_MEM = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_USE_HOST_POINTER_FOR_REGISTERED_MEM{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_MEM_OPS_V1' in found_values}} #: Deprecated, along with v1 MemOps API, :py:obj:`~.cuStreamBatchMemOp` #: and related APIs are supported. CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_MEM_OPS_V1 = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_MEM_OPS_V1{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS_V1' in found_values}} #: Deprecated, along with v1 MemOps API, 64-bit operations are #: supported in :py:obj:`~.cuStreamBatchMemOp` and related APIs. CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS_V1 = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS_V1{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR_V1' in found_values}} #: Deprecated, along with v1 MemOps API, #: :py:obj:`~.CU_STREAM_WAIT_VALUE_NOR` is supported. CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR_V1 = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR_V1{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH' in found_values}} #: Device supports launching cooperative kernels via #: :py:obj:`~.cuLaunchCooperativeKernel` CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_COOPERATIVE_MULTI_DEVICE_LAUNCH' in found_values}} #: Deprecated, :py:obj:`~.cuLaunchCooperativeKernelMultiDevice` is #: deprecated. CU_DEVICE_ATTRIBUTE_COOPERATIVE_MULTI_DEVICE_LAUNCH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_COOPERATIVE_MULTI_DEVICE_LAUNCH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK_OPTIN' in found_values}} #: Maximum optin shared memory per block CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK_OPTIN = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK_OPTIN{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_FLUSH_REMOTE_WRITES' in found_values}} #: The :py:obj:`~.CU_STREAM_WAIT_VALUE_FLUSH` flag and the #: :py:obj:`~.CU_STREAM_MEM_OP_FLUSH_REMOTE_WRITES` MemOp are supported #: on the device. See :py:obj:`~.Stream Memory Operations` for #: additional details. CU_DEVICE_ATTRIBUTE_CAN_FLUSH_REMOTE_WRITES = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_FLUSH_REMOTE_WRITES{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_HOST_REGISTER_SUPPORTED' in found_values}} #: Device supports host memory registration via #: :py:obj:`~.cudaHostRegister`. CU_DEVICE_ATTRIBUTE_HOST_REGISTER_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_HOST_REGISTER_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES' in found_values}} #: Device accesses pageable memory via the host's page tables. CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_DIRECT_MANAGED_MEM_ACCESS_FROM_HOST' in found_values}} #: The host can directly access managed memory on the device without #: migration. CU_DEVICE_ATTRIBUTE_DIRECT_MANAGED_MEM_ACCESS_FROM_HOST = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_DIRECT_MANAGED_MEM_ACCESS_FROM_HOST{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_VIRTUAL_ADDRESS_MANAGEMENT_SUPPORTED' in found_values}} #: Deprecated, Use #: CU_DEVICE_ATTRIBUTE_VIRTUAL_MEMORY_MANAGEMENT_SUPPORTED CU_DEVICE_ATTRIBUTE_VIRTUAL_ADDRESS_MANAGEMENT_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_VIRTUAL_ADDRESS_MANAGEMENT_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_VIRTUAL_MEMORY_MANAGEMENT_SUPPORTED' in found_values}} #: Device supports virtual memory management APIs like #: :py:obj:`~.cuMemAddressReserve`, :py:obj:`~.cuMemCreate`, #: :py:obj:`~.cuMemMap` and related APIs CU_DEVICE_ATTRIBUTE_VIRTUAL_MEMORY_MANAGEMENT_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_VIRTUAL_MEMORY_MANAGEMENT_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_POSIX_FILE_DESCRIPTOR_SUPPORTED' in found_values}} #: Device supports exporting memory to a posix file descriptor with #: :py:obj:`~.cuMemExportToShareableHandle`, if requested via #: :py:obj:`~.cuMemCreate` CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_POSIX_FILE_DESCRIPTOR_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_POSIX_FILE_DESCRIPTOR_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_HANDLE_SUPPORTED' in found_values}} #: Device supports exporting memory to a Win32 NT handle with #: :py:obj:`~.cuMemExportToShareableHandle`, if requested via #: :py:obj:`~.cuMemCreate` CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_HANDLE_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_HANDLE_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_KMT_HANDLE_SUPPORTED' in found_values}} #: Device supports exporting memory to a Win32 KMT handle with #: :py:obj:`~.cuMemExportToShareableHandle`, if requested via #: :py:obj:`~.cuMemCreate` CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_KMT_HANDLE_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_KMT_HANDLE_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_BLOCKS_PER_MULTIPROCESSOR' in found_values}} #: Maximum number of blocks per multiprocessor CU_DEVICE_ATTRIBUTE_MAX_BLOCKS_PER_MULTIPROCESSOR = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_BLOCKS_PER_MULTIPROCESSOR{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GENERIC_COMPRESSION_SUPPORTED' in found_values}} #: Device supports compression of memory CU_DEVICE_ATTRIBUTE_GENERIC_COMPRESSION_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GENERIC_COMPRESSION_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_PERSISTING_L2_CACHE_SIZE' in found_values}} #: Maximum L2 persisting lines capacity setting in bytes. CU_DEVICE_ATTRIBUTE_MAX_PERSISTING_L2_CACHE_SIZE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_PERSISTING_L2_CACHE_SIZE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX_ACCESS_POLICY_WINDOW_SIZE' in found_values}} #: Maximum value of :py:obj:`~.CUaccessPolicyWindow.num_bytes`. CU_DEVICE_ATTRIBUTE_MAX_ACCESS_POLICY_WINDOW_SIZE = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX_ACCESS_POLICY_WINDOW_SIZE{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WITH_CUDA_VMM_SUPPORTED' in found_values}} #: Device supports specifying the GPUDirect RDMA flag with #: :py:obj:`~.cuMemCreate` CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WITH_CUDA_VMM_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WITH_CUDA_VMM_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_RESERVED_SHARED_MEMORY_PER_BLOCK' in found_values}} #: Shared memory reserved by CUDA driver per block in bytes CU_DEVICE_ATTRIBUTE_RESERVED_SHARED_MEMORY_PER_BLOCK = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_RESERVED_SHARED_MEMORY_PER_BLOCK{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_SPARSE_CUDA_ARRAY_SUPPORTED' in found_values}} #: Device supports sparse CUDA arrays and sparse CUDA mipmapped arrays CU_DEVICE_ATTRIBUTE_SPARSE_CUDA_ARRAY_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_SPARSE_CUDA_ARRAY_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_READ_ONLY_HOST_REGISTER_SUPPORTED' in found_values}} #: Device supports using the :py:obj:`~.cuMemHostRegister` flag #: :py:obj:`~.CU_MEMHOSTERGISTER_READ_ONLY` to register memory that #: must be mapped as read-only to the GPU CU_DEVICE_ATTRIBUTE_READ_ONLY_HOST_REGISTER_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_READ_ONLY_HOST_REGISTER_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_TIMELINE_SEMAPHORE_INTEROP_SUPPORTED' in found_values}} #: External timeline semaphore interop is supported on the device CU_DEVICE_ATTRIBUTE_TIMELINE_SEMAPHORE_INTEROP_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_TIMELINE_SEMAPHORE_INTEROP_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MEMORY_POOLS_SUPPORTED' in found_values}} #: Device supports using the :py:obj:`~.cuMemAllocAsync` and #: :py:obj:`~.cuMemPool` family of APIs CU_DEVICE_ATTRIBUTE_MEMORY_POOLS_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MEMORY_POOLS_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_SUPPORTED' in found_values}} #: Device supports GPUDirect RDMA APIs, like nvidia_p2p_get_pages (see #: https://docs.nvidia.com/cuda/gpudirect-rdma for more information) CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_FLUSH_WRITES_OPTIONS' in found_values}} #: The returned attribute shall be interpreted as a bitmask, where the #: individual bits are described by the #: :py:obj:`~.CUflushGPUDirectRDMAWritesOptions` enum CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_FLUSH_WRITES_OPTIONS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_FLUSH_WRITES_OPTIONS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WRITES_ORDERING' in found_values}} #: GPUDirect RDMA writes to the device do not need to be flushed for #: consumers within the scope indicated by the returned attribute. See #: :py:obj:`~.CUGPUDirectRDMAWritesOrdering` for the numerical values #: returned here. CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WRITES_ORDERING = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WRITES_ORDERING{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MEMPOOL_SUPPORTED_HANDLE_TYPES' in found_values}} #: Handle types supported with mempool based IPC CU_DEVICE_ATTRIBUTE_MEMPOOL_SUPPORTED_HANDLE_TYPES = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MEMPOOL_SUPPORTED_HANDLE_TYPES{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CLUSTER_LAUNCH' in found_values}} #: Indicates device supports cluster launch CU_DEVICE_ATTRIBUTE_CLUSTER_LAUNCH = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CLUSTER_LAUNCH{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_DEFERRED_MAPPING_CUDA_ARRAY_SUPPORTED' in found_values}} #: Device supports deferred mapping CUDA arrays and CUDA mipmapped #: arrays CU_DEVICE_ATTRIBUTE_DEFERRED_MAPPING_CUDA_ARRAY_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_DEFERRED_MAPPING_CUDA_ARRAY_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS' in found_values}} #: 64-bit operations are supported in :py:obj:`~.cuStreamBatchMemOp` #: and related MemOp APIs. CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR' in found_values}} #: :py:obj:`~.CU_STREAM_WAIT_VALUE_NOR` is supported by MemOp APIs. CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_DMA_BUF_SUPPORTED' in found_values}} #: Device supports buffer sharing with dma_buf mechanism. CU_DEVICE_ATTRIBUTE_DMA_BUF_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_DMA_BUF_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED' in found_values}} #: Device supports IPC Events. CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MEM_SYNC_DOMAIN_COUNT' in found_values}} #: Number of memory domains the device supports. CU_DEVICE_ATTRIBUTE_MEM_SYNC_DOMAIN_COUNT = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MEM_SYNC_DOMAIN_COUNT{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_TENSOR_MAP_ACCESS_SUPPORTED' in found_values}} #: Device supports accessing memory using Tensor Map. CU_DEVICE_ATTRIBUTE_TENSOR_MAP_ACCESS_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_TENSOR_MAP_ACCESS_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_FABRIC_SUPPORTED' in found_values}} #: Device supports exporting memory to a fabric handle with #: :py:obj:`~.cuMemExportToShareableHandle()` or requested with #: :py:obj:`~.cuMemCreate()` CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_FABRIC_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_FABRIC_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_UNIFIED_FUNCTION_POINTERS' in found_values}} #: Device supports unified function pointers. CU_DEVICE_ATTRIBUTE_UNIFIED_FUNCTION_POINTERS = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_UNIFIED_FUNCTION_POINTERS{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_NUMA_CONFIG' in found_values}} #: NUMA configuration of a device: value is of type #: :py:obj:`~.CUdeviceNumaConfig` enum CU_DEVICE_ATTRIBUTE_NUMA_CONFIG = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_NUMA_CONFIG{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_NUMA_ID' in found_values}} #: NUMA node ID of the GPU memory CU_DEVICE_ATTRIBUTE_NUMA_ID = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_NUMA_ID{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED' in found_values}} #: Device supports switch multicast and reduction operations. CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MPS_ENABLED' in found_values}} #: Indicates if contexts created on this device will be shared via MPS CU_DEVICE_ATTRIBUTE_MPS_ENABLED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MPS_ENABLED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_HOST_NUMA_ID' in found_values}} #: NUMA ID of the host node closest to the device. Returns -1 when #: system does not support NUMA. CU_DEVICE_ATTRIBUTE_HOST_NUMA_ID = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_HOST_NUMA_ID{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_D3D12_CIG_SUPPORTED' in found_values}} #: Device supports CIG with D3D12. CU_DEVICE_ATTRIBUTE_D3D12_CIG_SUPPORTED = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_D3D12_CIG_SUPPORTED{{endif}} {{if 'CU_DEVICE_ATTRIBUTE_MAX' in found_values}} CU_DEVICE_ATTRIBUTE_MAX = ccuda.CUdevice_attribute_enum.CU_DEVICE_ATTRIBUTE_MAX{{endif}} {{endif}} {{if 'CUpointer_attribute_enum' in found_types}} class CUpointer_attribute(IntEnum): """ Pointer information """ {{if 'CU_POINTER_ATTRIBUTE_CONTEXT' in found_values}} #: The :py:obj:`~.CUcontext` on which a pointer was allocated or #: registered CU_POINTER_ATTRIBUTE_CONTEXT = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_CONTEXT{{endif}} {{if 'CU_POINTER_ATTRIBUTE_MEMORY_TYPE' in found_values}} #: The :py:obj:`~.CUmemorytype` describing the physical location of a #: pointer CU_POINTER_ATTRIBUTE_MEMORY_TYPE = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_MEMORY_TYPE{{endif}} {{if 'CU_POINTER_ATTRIBUTE_DEVICE_POINTER' in found_values}} #: The address at which a pointer's memory may be accessed on the #: device CU_POINTER_ATTRIBUTE_DEVICE_POINTER = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_DEVICE_POINTER{{endif}} {{if 'CU_POINTER_ATTRIBUTE_HOST_POINTER' in found_values}} #: The address at which a pointer's memory may be accessed on the host CU_POINTER_ATTRIBUTE_HOST_POINTER = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_HOST_POINTER{{endif}} {{if 'CU_POINTER_ATTRIBUTE_P2P_TOKENS' in found_values}} #: A pair of tokens for use with the nv-p2p.h Linux kernel interface CU_POINTER_ATTRIBUTE_P2P_TOKENS = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_P2P_TOKENS{{endif}} {{if 'CU_POINTER_ATTRIBUTE_SYNC_MEMOPS' in found_values}} #: Synchronize every synchronous memory operation initiated on this #: region CU_POINTER_ATTRIBUTE_SYNC_MEMOPS = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_SYNC_MEMOPS{{endif}} {{if 'CU_POINTER_ATTRIBUTE_BUFFER_ID' in found_values}} #: A process-wide unique ID for an allocated memory region CU_POINTER_ATTRIBUTE_BUFFER_ID = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_BUFFER_ID{{endif}} {{if 'CU_POINTER_ATTRIBUTE_IS_MANAGED' in found_values}} #: Indicates if the pointer points to managed memory CU_POINTER_ATTRIBUTE_IS_MANAGED = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_IS_MANAGED{{endif}} {{if 'CU_POINTER_ATTRIBUTE_DEVICE_ORDINAL' in found_values}} #: A device ordinal of a device on which a pointer was allocated or #: registered CU_POINTER_ATTRIBUTE_DEVICE_ORDINAL = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_DEVICE_ORDINAL{{endif}} {{if 'CU_POINTER_ATTRIBUTE_IS_LEGACY_CUDA_IPC_CAPABLE' in found_values}} #: 1 if this pointer maps to an allocation that is suitable for #: :py:obj:`~.cudaIpcGetMemHandle`, 0 otherwise CU_POINTER_ATTRIBUTE_IS_LEGACY_CUDA_IPC_CAPABLE = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_IS_LEGACY_CUDA_IPC_CAPABLE{{endif}} {{if 'CU_POINTER_ATTRIBUTE_RANGE_START_ADDR' in found_values}} #: Starting address for this requested pointer CU_POINTER_ATTRIBUTE_RANGE_START_ADDR = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_RANGE_START_ADDR{{endif}} {{if 'CU_POINTER_ATTRIBUTE_RANGE_SIZE' in found_values}} #: Size of the address range for this requested pointer CU_POINTER_ATTRIBUTE_RANGE_SIZE = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_RANGE_SIZE{{endif}} {{if 'CU_POINTER_ATTRIBUTE_MAPPED' in found_values}} #: 1 if this pointer is in a valid address range that is mapped to a #: backing allocation, 0 otherwise CU_POINTER_ATTRIBUTE_MAPPED = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_MAPPED{{endif}} {{if 'CU_POINTER_ATTRIBUTE_ALLOWED_HANDLE_TYPES' in found_values}} #: Bitmask of allowed :py:obj:`~.CUmemAllocationHandleType` for this #: allocation CU_POINTER_ATTRIBUTE_ALLOWED_HANDLE_TYPES = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_ALLOWED_HANDLE_TYPES{{endif}} {{if 'CU_POINTER_ATTRIBUTE_IS_GPU_DIRECT_RDMA_CAPABLE' in found_values}} #: 1 if the memory this pointer is referencing can be used with the #: GPUDirect RDMA API CU_POINTER_ATTRIBUTE_IS_GPU_DIRECT_RDMA_CAPABLE = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_IS_GPU_DIRECT_RDMA_CAPABLE{{endif}} {{if 'CU_POINTER_ATTRIBUTE_ACCESS_FLAGS' in found_values}} #: Returns the access flags the device associated with the current #: context has on the corresponding memory referenced by the pointer #: given CU_POINTER_ATTRIBUTE_ACCESS_FLAGS = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_ACCESS_FLAGS{{endif}} {{if 'CU_POINTER_ATTRIBUTE_MEMPOOL_HANDLE' in found_values}} #: Returns the mempool handle for the allocation if it was allocated #: from a mempool. Otherwise returns NULL. CU_POINTER_ATTRIBUTE_MEMPOOL_HANDLE = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_MEMPOOL_HANDLE{{endif}} {{if 'CU_POINTER_ATTRIBUTE_MAPPING_SIZE' in found_values}} #: Size of the actual underlying mapping that the pointer belongs to CU_POINTER_ATTRIBUTE_MAPPING_SIZE = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_MAPPING_SIZE{{endif}} {{if 'CU_POINTER_ATTRIBUTE_MAPPING_BASE_ADDR' in found_values}} #: The start address of the mapping that the pointer belongs to CU_POINTER_ATTRIBUTE_MAPPING_BASE_ADDR = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_MAPPING_BASE_ADDR{{endif}} {{if 'CU_POINTER_ATTRIBUTE_MEMORY_BLOCK_ID' in found_values}} #: A process-wide unique id corresponding to the physical allocation #: the pointer belongs to CU_POINTER_ATTRIBUTE_MEMORY_BLOCK_ID = ccuda.CUpointer_attribute_enum.CU_POINTER_ATTRIBUTE_MEMORY_BLOCK_ID{{endif}} {{endif}} {{if 'CUfunction_attribute_enum' in found_types}} class CUfunction_attribute(IntEnum): """ Function properties """ {{if 'CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK' in found_values}} #: The maximum number of threads per block, beyond which a launch of #: the function would fail. This number depends on both the function #: and the device on which the function is currently loaded. CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK{{endif}} {{if 'CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES' in found_values}} #: The size in bytes of statically-allocated shared memory required by #: this function. This does not include dynamically-allocated shared #: memory requested by the user at runtime. CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES{{endif}} {{if 'CU_FUNC_ATTRIBUTE_CONST_SIZE_BYTES' in found_values}} #: The size in bytes of user-allocated constant memory required by this #: function. CU_FUNC_ATTRIBUTE_CONST_SIZE_BYTES = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_CONST_SIZE_BYTES{{endif}} {{if 'CU_FUNC_ATTRIBUTE_LOCAL_SIZE_BYTES' in found_values}} #: The size in bytes of local memory used by each thread of this #: function. CU_FUNC_ATTRIBUTE_LOCAL_SIZE_BYTES = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_LOCAL_SIZE_BYTES{{endif}} {{if 'CU_FUNC_ATTRIBUTE_NUM_REGS' in found_values}} #: The number of registers used by each thread of this function. CU_FUNC_ATTRIBUTE_NUM_REGS = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_NUM_REGS{{endif}} {{if 'CU_FUNC_ATTRIBUTE_PTX_VERSION' in found_values}} #: The PTX virtual architecture version for which the function was #: compiled. This value is the major PTX version * 10 + the minor PTX #: version, so a PTX version 1.3 function would return the value 13. #: Note that this may return the undefined value of 0 for cubins #: compiled prior to CUDA 3.0. CU_FUNC_ATTRIBUTE_PTX_VERSION = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_PTX_VERSION{{endif}} {{if 'CU_FUNC_ATTRIBUTE_BINARY_VERSION' in found_values}} #: The binary architecture version for which the function was compiled. #: This value is the major binary version * 10 + the minor binary #: version, so a binary version 1.3 function would return the value 13. #: Note that this will return a value of 10 for legacy cubins that do #: not have a properly-encoded binary architecture version. CU_FUNC_ATTRIBUTE_BINARY_VERSION = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_BINARY_VERSION{{endif}} {{if 'CU_FUNC_ATTRIBUTE_CACHE_MODE_CA' in found_values}} #: The attribute to indicate whether the function has been compiled #: with user specified option "-Xptxas --dlcm=ca" set . CU_FUNC_ATTRIBUTE_CACHE_MODE_CA = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_CACHE_MODE_CA{{endif}} {{if 'CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES' in found_values}} #: The maximum size in bytes of dynamically-allocated shared memory #: that can be used by this function. If the user-specified dynamic #: shared memory size is larger than this value, the launch will fail. #: See :py:obj:`~.cuFuncSetAttribute`, :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES{{endif}} {{if 'CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT' in found_values}} #: On devices where the L1 cache and shared memory use the same #: hardware resources, this sets the shared memory carveout preference, #: in percent of the total shared memory. Refer to #: :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_MULTIPROCESSOR`. #: This is only a hint, and the driver can choose a different ratio if #: required to execute the function. See #: :py:obj:`~.cuFuncSetAttribute`, :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT{{endif}} {{if 'CU_FUNC_ATTRIBUTE_CLUSTER_SIZE_MUST_BE_SET' in found_values}} #: If this attribute is set, the kernel must launch with a valid #: cluster size specified. See :py:obj:`~.cuFuncSetAttribute`, #: :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_CLUSTER_SIZE_MUST_BE_SET = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_CLUSTER_SIZE_MUST_BE_SET{{endif}} {{if 'CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_WIDTH' in found_values}} #: The required cluster width in blocks. The values must either all be #: 0 or all be positive. The validity of the cluster dimensions is #: otherwise checked at launch time. #: #: If the value is set during compile time, it cannot be set at #: runtime. Setting it at runtime will return CUDA_ERROR_NOT_PERMITTED. #: See :py:obj:`~.cuFuncSetAttribute`, :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_WIDTH = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_WIDTH{{endif}} {{if 'CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_HEIGHT' in found_values}} #: The required cluster height in blocks. The values must either all be #: 0 or all be positive. The validity of the cluster dimensions is #: otherwise checked at launch time. #: #: If the value is set during compile time, it cannot be set at #: runtime. Setting it at runtime should return #: CUDA_ERROR_NOT_PERMITTED. See :py:obj:`~.cuFuncSetAttribute`, #: :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_HEIGHT = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_HEIGHT{{endif}} {{if 'CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_DEPTH' in found_values}} #: The required cluster depth in blocks. The values must either all be #: 0 or all be positive. The validity of the cluster dimensions is #: otherwise checked at launch time. #: #: If the value is set during compile time, it cannot be set at #: runtime. Setting it at runtime should return #: CUDA_ERROR_NOT_PERMITTED. See :py:obj:`~.cuFuncSetAttribute`, #: :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_DEPTH = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_DEPTH{{endif}} {{if 'CU_FUNC_ATTRIBUTE_NON_PORTABLE_CLUSTER_SIZE_ALLOWED' in found_values}} #: Whether the function can be launched with non-portable cluster size. #: 1 is allowed, 0 is disallowed. A non-portable cluster size may only #: function on the specific SKUs the program is tested on. The launch #: might fail if the program is run on a different hardware platform. #: #: CUDA API provides cudaOccupancyMaxActiveClusters to assist with #: checking whether the desired size can be launched on the current #: device. #: #: Portable Cluster Size #: #: A portable cluster size is guaranteed to be functional on all #: compute capabilities higher than the target compute capability. The #: portable cluster size for sm_90 is 8 blocks per cluster. This value #: may increase for future compute capabilities. #: #: The specific hardware unit may support higher cluster sizes that’s #: not guaranteed to be portable. See :py:obj:`~.cuFuncSetAttribute`, #: :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_NON_PORTABLE_CLUSTER_SIZE_ALLOWED = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_NON_PORTABLE_CLUSTER_SIZE_ALLOWED{{endif}} {{if 'CU_FUNC_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE' in found_values}} #: The block scheduling policy of a function. The value type is #: CUclusterSchedulingPolicy / cudaClusterSchedulingPolicy. See #: :py:obj:`~.cuFuncSetAttribute`, :py:obj:`~.cuKernelSetAttribute` CU_FUNC_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE{{endif}} {{if 'CU_FUNC_ATTRIBUTE_MAX' in found_values}} CU_FUNC_ATTRIBUTE_MAX = ccuda.CUfunction_attribute_enum.CU_FUNC_ATTRIBUTE_MAX{{endif}} {{endif}} {{if 'CUfunc_cache_enum' in found_types}} class CUfunc_cache(IntEnum): """ Function cache configurations """ {{if 'CU_FUNC_CACHE_PREFER_NONE' in found_values}} #: no preference for shared memory or L1 (default) CU_FUNC_CACHE_PREFER_NONE = ccuda.CUfunc_cache_enum.CU_FUNC_CACHE_PREFER_NONE{{endif}} {{if 'CU_FUNC_CACHE_PREFER_SHARED' in found_values}} #: prefer larger shared memory and smaller L1 cache CU_FUNC_CACHE_PREFER_SHARED = ccuda.CUfunc_cache_enum.CU_FUNC_CACHE_PREFER_SHARED{{endif}} {{if 'CU_FUNC_CACHE_PREFER_L1' in found_values}} #: prefer larger L1 cache and smaller shared memory CU_FUNC_CACHE_PREFER_L1 = ccuda.CUfunc_cache_enum.CU_FUNC_CACHE_PREFER_L1{{endif}} {{if 'CU_FUNC_CACHE_PREFER_EQUAL' in found_values}} #: prefer equal sized L1 cache and shared memory CU_FUNC_CACHE_PREFER_EQUAL = ccuda.CUfunc_cache_enum.CU_FUNC_CACHE_PREFER_EQUAL{{endif}} {{endif}} {{if 'CUsharedconfig_enum' in found_types}} class CUsharedconfig(IntEnum): """ [Deprecated] Shared memory configurations """ {{if 'CU_SHARED_MEM_CONFIG_DEFAULT_BANK_SIZE' in found_values}} #: set default shared memory bank size CU_SHARED_MEM_CONFIG_DEFAULT_BANK_SIZE = ccuda.CUsharedconfig_enum.CU_SHARED_MEM_CONFIG_DEFAULT_BANK_SIZE{{endif}} {{if 'CU_SHARED_MEM_CONFIG_FOUR_BYTE_BANK_SIZE' in found_values}} #: set shared memory bank width to four bytes CU_SHARED_MEM_CONFIG_FOUR_BYTE_BANK_SIZE = ccuda.CUsharedconfig_enum.CU_SHARED_MEM_CONFIG_FOUR_BYTE_BANK_SIZE{{endif}} {{if 'CU_SHARED_MEM_CONFIG_EIGHT_BYTE_BANK_SIZE' in found_values}} #: set shared memory bank width to eight bytes CU_SHARED_MEM_CONFIG_EIGHT_BYTE_BANK_SIZE = ccuda.CUsharedconfig_enum.CU_SHARED_MEM_CONFIG_EIGHT_BYTE_BANK_SIZE{{endif}} {{endif}} {{if 'CUshared_carveout_enum' in found_types}} class CUshared_carveout(IntEnum): """ Shared memory carveout configurations. These may be passed to :py:obj:`~.cuFuncSetAttribute` or :py:obj:`~.cuKernelSetAttribute` """ {{if 'CU_SHAREDMEM_CARVEOUT_DEFAULT' in found_values}} #: No preference for shared memory or L1 (default) CU_SHAREDMEM_CARVEOUT_DEFAULT = ccuda.CUshared_carveout_enum.CU_SHAREDMEM_CARVEOUT_DEFAULT{{endif}} {{if 'CU_SHAREDMEM_CARVEOUT_MAX_L1' in found_values}} #: Prefer maximum available L1 cache, minimum shared memory CU_SHAREDMEM_CARVEOUT_MAX_L1 = ccuda.CUshared_carveout_enum.CU_SHAREDMEM_CARVEOUT_MAX_L1{{endif}} {{if 'CU_SHAREDMEM_CARVEOUT_MAX_SHARED' in found_values}} #: Prefer maximum available shared memory, minimum L1 cache CU_SHAREDMEM_CARVEOUT_MAX_SHARED = ccuda.CUshared_carveout_enum.CU_SHAREDMEM_CARVEOUT_MAX_SHARED{{endif}} {{endif}} {{if 'CUmemorytype_enum' in found_types}} class CUmemorytype(IntEnum): """ Memory types """ {{if 'CU_MEMORYTYPE_HOST' in found_values}} #: Host memory CU_MEMORYTYPE_HOST = ccuda.CUmemorytype_enum.CU_MEMORYTYPE_HOST{{endif}} {{if 'CU_MEMORYTYPE_DEVICE' in found_values}} #: Device memory CU_MEMORYTYPE_DEVICE = ccuda.CUmemorytype_enum.CU_MEMORYTYPE_DEVICE{{endif}} {{if 'CU_MEMORYTYPE_ARRAY' in found_values}} #: Array memory CU_MEMORYTYPE_ARRAY = ccuda.CUmemorytype_enum.CU_MEMORYTYPE_ARRAY{{endif}} {{if 'CU_MEMORYTYPE_UNIFIED' in found_values}} #: Unified device or host memory CU_MEMORYTYPE_UNIFIED = ccuda.CUmemorytype_enum.CU_MEMORYTYPE_UNIFIED{{endif}} {{endif}} {{if 'CUcomputemode_enum' in found_types}} class CUcomputemode(IntEnum): """ Compute Modes """ {{if 'CU_COMPUTEMODE_DEFAULT' in found_values}} #: Default compute mode (Multiple contexts allowed per device) CU_COMPUTEMODE_DEFAULT = ccuda.CUcomputemode_enum.CU_COMPUTEMODE_DEFAULT{{endif}} {{if 'CU_COMPUTEMODE_PROHIBITED' in found_values}} #: Compute-prohibited mode (No contexts can be created on this device #: at this time) CU_COMPUTEMODE_PROHIBITED = ccuda.CUcomputemode_enum.CU_COMPUTEMODE_PROHIBITED{{endif}} {{if 'CU_COMPUTEMODE_EXCLUSIVE_PROCESS' in found_values}} #: Compute-exclusive-process mode (Only one context used by a single #: process can be present on this device at a time) CU_COMPUTEMODE_EXCLUSIVE_PROCESS = ccuda.CUcomputemode_enum.CU_COMPUTEMODE_EXCLUSIVE_PROCESS{{endif}} {{endif}} {{if 'CUmem_advise_enum' in found_types}} class CUmem_advise(IntEnum): """ Memory advise values """ {{if 'CU_MEM_ADVISE_SET_READ_MOSTLY' in found_values}} #: Data will mostly be read and only occasionally be written to CU_MEM_ADVISE_SET_READ_MOSTLY = ccuda.CUmem_advise_enum.CU_MEM_ADVISE_SET_READ_MOSTLY{{endif}} {{if 'CU_MEM_ADVISE_UNSET_READ_MOSTLY' in found_values}} #: Undo the effect of :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` CU_MEM_ADVISE_UNSET_READ_MOSTLY = ccuda.CUmem_advise_enum.CU_MEM_ADVISE_UNSET_READ_MOSTLY{{endif}} {{if 'CU_MEM_ADVISE_SET_PREFERRED_LOCATION' in found_values}} #: Set the preferred location for the data as the specified device CU_MEM_ADVISE_SET_PREFERRED_LOCATION = ccuda.CUmem_advise_enum.CU_MEM_ADVISE_SET_PREFERRED_LOCATION{{endif}} {{if 'CU_MEM_ADVISE_UNSET_PREFERRED_LOCATION' in found_values}} #: Clear the preferred location for the data CU_MEM_ADVISE_UNSET_PREFERRED_LOCATION = ccuda.CUmem_advise_enum.CU_MEM_ADVISE_UNSET_PREFERRED_LOCATION{{endif}} {{if 'CU_MEM_ADVISE_SET_ACCESSED_BY' in found_values}} #: Data will be accessed by the specified device, so prevent page #: faults as much as possible CU_MEM_ADVISE_SET_ACCESSED_BY = ccuda.CUmem_advise_enum.CU_MEM_ADVISE_SET_ACCESSED_BY{{endif}} {{if 'CU_MEM_ADVISE_UNSET_ACCESSED_BY' in found_values}} #: Let the Unified Memory subsystem decide on the page faulting policy #: for the specified device CU_MEM_ADVISE_UNSET_ACCESSED_BY = ccuda.CUmem_advise_enum.CU_MEM_ADVISE_UNSET_ACCESSED_BY{{endif}} {{endif}} {{if 'CUmem_range_attribute_enum' in found_types}} class CUmem_range_attribute(IntEnum): """ """ {{if 'CU_MEM_RANGE_ATTRIBUTE_READ_MOSTLY' in found_values}} #: Whether the range will mostly be read and only occasionally be #: written to CU_MEM_RANGE_ATTRIBUTE_READ_MOSTLY = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_READ_MOSTLY{{endif}} {{if 'CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION' in found_values}} #: The preferred location of the range CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION{{endif}} {{if 'CU_MEM_RANGE_ATTRIBUTE_ACCESSED_BY' in found_values}} #: Memory range has :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY` set for #: specified device CU_MEM_RANGE_ATTRIBUTE_ACCESSED_BY = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_ACCESSED_BY{{endif}} {{if 'CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION' in found_values}} #: The last location to which the range was prefetched CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION{{endif}} {{if 'CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_TYPE' in found_values}} #: The preferred location type of the range CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_TYPE = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_TYPE{{endif}} {{if 'CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_ID' in found_values}} #: The preferred location id of the range CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_ID = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_ID{{endif}} {{if 'CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_TYPE' in found_values}} #: The last location type to which the range was prefetched CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_TYPE = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_TYPE{{endif}} {{if 'CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_ID' in found_values}} #: The last location id to which the range was prefetched CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_ID = ccuda.CUmem_range_attribute_enum.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_ID{{endif}} {{endif}} {{if 'CUjit_option_enum' in found_types}} class CUjit_option(IntEnum): """ Online compiler and linker options """ {{if 'CU_JIT_MAX_REGISTERS' in found_values}} #: Max number of registers that a thread may use. #: Option type: unsigned int #: Applies to: compiler only CU_JIT_MAX_REGISTERS = ccuda.CUjit_option_enum.CU_JIT_MAX_REGISTERS{{endif}} {{if 'CU_JIT_THREADS_PER_BLOCK' in found_values}} #: IN: Specifies minimum number of threads per block to target #: compilation for #: OUT: Returns the number of threads the compiler actually targeted. #: This restricts the resource utilization of the compiler (e.g. max #: registers) such that a block with the given number of threads should #: be able to launch based on register limitations. Note, this option #: does not currently take into account any other resource limitations, #: such as shared memory utilization. #: Cannot be combined with :py:obj:`~.CU_JIT_TARGET`. #: Option type: unsigned int #: Applies to: compiler only CU_JIT_THREADS_PER_BLOCK = ccuda.CUjit_option_enum.CU_JIT_THREADS_PER_BLOCK{{endif}} {{if 'CU_JIT_WALL_TIME' in found_values}} #: Overwrites the option value with the total wall clock time, in #: milliseconds, spent in the compiler and linker #: Option type: float #: Applies to: compiler and linker CU_JIT_WALL_TIME = ccuda.CUjit_option_enum.CU_JIT_WALL_TIME{{endif}} {{if 'CU_JIT_INFO_LOG_BUFFER' in found_values}} #: Pointer to a buffer in which to print any log messages that are #: informational in nature (the buffer size is specified via option #: :py:obj:`~.CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES`) #: Option type: char * #: Applies to: compiler and linker CU_JIT_INFO_LOG_BUFFER = ccuda.CUjit_option_enum.CU_JIT_INFO_LOG_BUFFER{{endif}} {{if 'CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES' in found_values}} #: IN: Log buffer size in bytes. Log messages will be capped at this #: size (including null terminator) #: OUT: Amount of log buffer filled with messages #: Option type: unsigned int #: Applies to: compiler and linker CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES = ccuda.CUjit_option_enum.CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES{{endif}} {{if 'CU_JIT_ERROR_LOG_BUFFER' in found_values}} #: Pointer to a buffer in which to print any log messages that reflect #: errors (the buffer size is specified via option #: :py:obj:`~.CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES`) #: Option type: char * #: Applies to: compiler and linker CU_JIT_ERROR_LOG_BUFFER = ccuda.CUjit_option_enum.CU_JIT_ERROR_LOG_BUFFER{{endif}} {{if 'CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES' in found_values}} #: IN: Log buffer size in bytes. Log messages will be capped at this #: size (including null terminator) #: OUT: Amount of log buffer filled with messages #: Option type: unsigned int #: Applies to: compiler and linker CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES = ccuda.CUjit_option_enum.CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES{{endif}} {{if 'CU_JIT_OPTIMIZATION_LEVEL' in found_values}} #: Level of optimizations to apply to generated code (0 - 4), with 4 #: being the default and highest level of optimizations. #: Option type: unsigned int #: Applies to: compiler only CU_JIT_OPTIMIZATION_LEVEL = ccuda.CUjit_option_enum.CU_JIT_OPTIMIZATION_LEVEL{{endif}} {{if 'CU_JIT_TARGET_FROM_CUCONTEXT' in found_values}} #: No option value required. Determines the target based on the current #: attached context (default) #: Option type: No option value needed #: Applies to: compiler and linker CU_JIT_TARGET_FROM_CUCONTEXT = ccuda.CUjit_option_enum.CU_JIT_TARGET_FROM_CUCONTEXT{{endif}} {{if 'CU_JIT_TARGET' in found_values}} #: Target is chosen based on supplied :py:obj:`~.CUjit_target`. Cannot #: be combined with :py:obj:`~.CU_JIT_THREADS_PER_BLOCK`. #: Option type: unsigned int for enumerated type #: :py:obj:`~.CUjit_target` #: Applies to: compiler and linker CU_JIT_TARGET = ccuda.CUjit_option_enum.CU_JIT_TARGET{{endif}} {{if 'CU_JIT_FALLBACK_STRATEGY' in found_values}} #: Specifies choice of fallback strategy if matching cubin is not #: found. Choice is based on supplied :py:obj:`~.CUjit_fallback`. This #: option cannot be used with cuLink* APIs as the linker requires exact #: matches. #: Option type: unsigned int for enumerated type #: :py:obj:`~.CUjit_fallback` #: Applies to: compiler only CU_JIT_FALLBACK_STRATEGY = ccuda.CUjit_option_enum.CU_JIT_FALLBACK_STRATEGY{{endif}} {{if 'CU_JIT_GENERATE_DEBUG_INFO' in found_values}} #: Specifies whether to create debug information in output (-g) (0: #: false, default) #: Option type: int #: Applies to: compiler and linker CU_JIT_GENERATE_DEBUG_INFO = ccuda.CUjit_option_enum.CU_JIT_GENERATE_DEBUG_INFO{{endif}} {{if 'CU_JIT_LOG_VERBOSE' in found_values}} #: Generate verbose log messages (0: false, default) #: Option type: int #: Applies to: compiler and linker CU_JIT_LOG_VERBOSE = ccuda.CUjit_option_enum.CU_JIT_LOG_VERBOSE{{endif}} {{if 'CU_JIT_GENERATE_LINE_INFO' in found_values}} #: Generate line number information (-lineinfo) (0: false, default) #: Option type: int #: Applies to: compiler only CU_JIT_GENERATE_LINE_INFO = ccuda.CUjit_option_enum.CU_JIT_GENERATE_LINE_INFO{{endif}} {{if 'CU_JIT_CACHE_MODE' in found_values}} #: Specifies whether to enable caching explicitly (-dlcm) #: Choice is based on supplied :py:obj:`~.CUjit_cacheMode_enum`. #: Option type: unsigned int for enumerated type #: :py:obj:`~.CUjit_cacheMode_enum` #: Applies to: compiler only CU_JIT_CACHE_MODE = ccuda.CUjit_option_enum.CU_JIT_CACHE_MODE{{endif}} {{if 'CU_JIT_NEW_SM3X_OPT' in found_values}} #: [Deprecated] CU_JIT_NEW_SM3X_OPT = ccuda.CUjit_option_enum.CU_JIT_NEW_SM3X_OPT{{endif}} {{if 'CU_JIT_FAST_COMPILE' in found_values}} #: This jit option is used for internal purpose only. CU_JIT_FAST_COMPILE = ccuda.CUjit_option_enum.CU_JIT_FAST_COMPILE{{endif}} {{if 'CU_JIT_GLOBAL_SYMBOL_NAMES' in found_values}} #: Array of device symbol names that will be relocated to the #: corresponding host addresses stored in #: :py:obj:`~.CU_JIT_GLOBAL_SYMBOL_ADDRESSES`. #: Must contain :py:obj:`~.CU_JIT_GLOBAL_SYMBOL_COUNT` entries. #: When loading a device module, driver will relocate all encountered #: unresolved symbols to the host addresses. #: It is only allowed to register symbols that correspond to unresolved #: global variables. #: It is illegal to register the same device symbol at multiple #: addresses. #: Option type: const char ** #: Applies to: dynamic linker only CU_JIT_GLOBAL_SYMBOL_NAMES = ccuda.CUjit_option_enum.CU_JIT_GLOBAL_SYMBOL_NAMES{{endif}} {{if 'CU_JIT_GLOBAL_SYMBOL_ADDRESSES' in found_values}} #: Array of host addresses that will be used to relocate corresponding #: device symbols stored in :py:obj:`~.CU_JIT_GLOBAL_SYMBOL_NAMES`. #: Must contain :py:obj:`~.CU_JIT_GLOBAL_SYMBOL_COUNT` entries. #: Option type: void ** #: Applies to: dynamic linker only CU_JIT_GLOBAL_SYMBOL_ADDRESSES = ccuda.CUjit_option_enum.CU_JIT_GLOBAL_SYMBOL_ADDRESSES{{endif}} {{if 'CU_JIT_GLOBAL_SYMBOL_COUNT' in found_values}} #: Number of entries in :py:obj:`~.CU_JIT_GLOBAL_SYMBOL_NAMES` and #: :py:obj:`~.CU_JIT_GLOBAL_SYMBOL_ADDRESSES` arrays. #: Option type: unsigned int #: Applies to: dynamic linker only CU_JIT_GLOBAL_SYMBOL_COUNT = ccuda.CUjit_option_enum.CU_JIT_GLOBAL_SYMBOL_COUNT{{endif}} {{if 'CU_JIT_LTO' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_LTO = ccuda.CUjit_option_enum.CU_JIT_LTO{{endif}} {{if 'CU_JIT_FTZ' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_FTZ = ccuda.CUjit_option_enum.CU_JIT_FTZ{{endif}} {{if 'CU_JIT_PREC_DIV' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_PREC_DIV = ccuda.CUjit_option_enum.CU_JIT_PREC_DIV{{endif}} {{if 'CU_JIT_PREC_SQRT' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_PREC_SQRT = ccuda.CUjit_option_enum.CU_JIT_PREC_SQRT{{endif}} {{if 'CU_JIT_FMA' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_FMA = ccuda.CUjit_option_enum.CU_JIT_FMA{{endif}} {{if 'CU_JIT_REFERENCED_KERNEL_NAMES' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_REFERENCED_KERNEL_NAMES = ccuda.CUjit_option_enum.CU_JIT_REFERENCED_KERNEL_NAMES{{endif}} {{if 'CU_JIT_REFERENCED_KERNEL_COUNT' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_REFERENCED_KERNEL_COUNT = ccuda.CUjit_option_enum.CU_JIT_REFERENCED_KERNEL_COUNT{{endif}} {{if 'CU_JIT_REFERENCED_VARIABLE_NAMES' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_REFERENCED_VARIABLE_NAMES = ccuda.CUjit_option_enum.CU_JIT_REFERENCED_VARIABLE_NAMES{{endif}} {{if 'CU_JIT_REFERENCED_VARIABLE_COUNT' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_REFERENCED_VARIABLE_COUNT = ccuda.CUjit_option_enum.CU_JIT_REFERENCED_VARIABLE_COUNT{{endif}} {{if 'CU_JIT_OPTIMIZE_UNUSED_DEVICE_VARIABLES' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_OPTIMIZE_UNUSED_DEVICE_VARIABLES = ccuda.CUjit_option_enum.CU_JIT_OPTIMIZE_UNUSED_DEVICE_VARIABLES{{endif}} {{if 'CU_JIT_POSITION_INDEPENDENT_CODE' in found_values}} #: Generate position independent code (0: false) #: Option type: int #: Applies to: compiler only CU_JIT_POSITION_INDEPENDENT_CODE = ccuda.CUjit_option_enum.CU_JIT_POSITION_INDEPENDENT_CODE{{endif}} {{if 'CU_JIT_MIN_CTA_PER_SM' in found_values}} #: This option hints to the JIT compiler the minimum number of CTAs #: from the kernel’s grid to be mapped to a SM. This option is ignored #: when used together with :py:obj:`~.CU_JIT_MAX_REGISTERS` or #: :py:obj:`~.CU_JIT_THREADS_PER_BLOCK`. Optimizations based on this #: option need :py:obj:`~.CU_JIT_MAX_THREADS_PER_BLOCK` to be specified #: as well. For kernels already using PTX directive .minnctapersm, this #: option will be ignored by default. Use #: :py:obj:`~.CU_JIT_OVERRIDE_DIRECTIVE_VALUES` to let this option take #: precedence over the PTX directive. Option type: unsigned int #: Applies to: compiler only CU_JIT_MIN_CTA_PER_SM = ccuda.CUjit_option_enum.CU_JIT_MIN_CTA_PER_SM{{endif}} {{if 'CU_JIT_MAX_THREADS_PER_BLOCK' in found_values}} #: Maximum number threads in a thread block, computed as the product of #: the maximum extent specifed for each dimension of the block. This #: limit is guaranteed not to be exeeded in any invocation of the #: kernel. Exceeding the the maximum number of threads results in #: runtime error or kernel launch failure. For kernels already using #: PTX directive .maxntid, this option will be ignored by default. Use #: :py:obj:`~.CU_JIT_OVERRIDE_DIRECTIVE_VALUES` to let this option take #: precedence over the PTX directive. Option type: int #: Applies to: compiler only CU_JIT_MAX_THREADS_PER_BLOCK = ccuda.CUjit_option_enum.CU_JIT_MAX_THREADS_PER_BLOCK{{endif}} {{if 'CU_JIT_OVERRIDE_DIRECTIVE_VALUES' in found_values}} #: This option lets the values specified using #: :py:obj:`~.CU_JIT_MAX_REGISTERS`, #: :py:obj:`~.CU_JIT_THREADS_PER_BLOCK`, #: :py:obj:`~.CU_JIT_MAX_THREADS_PER_BLOCK` and #: :py:obj:`~.CU_JIT_MIN_CTA_PER_SM` take precedence over any PTX #: directives. (0: Disable, default; 1: Enable) Option type: int #: Applies to: compiler only CU_JIT_OVERRIDE_DIRECTIVE_VALUES = ccuda.CUjit_option_enum.CU_JIT_OVERRIDE_DIRECTIVE_VALUES{{endif}} {{if 'CU_JIT_NUM_OPTIONS' in found_values}} CU_JIT_NUM_OPTIONS = ccuda.CUjit_option_enum.CU_JIT_NUM_OPTIONS{{endif}} {{endif}} {{if 'CUjit_target_enum' in found_types}} class CUjit_target(IntEnum): """ Online compilation targets """ {{if 'CU_TARGET_COMPUTE_30' in found_values}} #: Compute device class 3.0 CU_TARGET_COMPUTE_30 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_30{{endif}} {{if 'CU_TARGET_COMPUTE_32' in found_values}} #: Compute device class 3.2 CU_TARGET_COMPUTE_32 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_32{{endif}} {{if 'CU_TARGET_COMPUTE_35' in found_values}} #: Compute device class 3.5 CU_TARGET_COMPUTE_35 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_35{{endif}} {{if 'CU_TARGET_COMPUTE_37' in found_values}} #: Compute device class 3.7 CU_TARGET_COMPUTE_37 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_37{{endif}} {{if 'CU_TARGET_COMPUTE_50' in found_values}} #: Compute device class 5.0 CU_TARGET_COMPUTE_50 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_50{{endif}} {{if 'CU_TARGET_COMPUTE_52' in found_values}} #: Compute device class 5.2 CU_TARGET_COMPUTE_52 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_52{{endif}} {{if 'CU_TARGET_COMPUTE_53' in found_values}} #: Compute device class 5.3 CU_TARGET_COMPUTE_53 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_53{{endif}} {{if 'CU_TARGET_COMPUTE_60' in found_values}} #: Compute device class 6.0. CU_TARGET_COMPUTE_60 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_60{{endif}} {{if 'CU_TARGET_COMPUTE_61' in found_values}} #: Compute device class 6.1. CU_TARGET_COMPUTE_61 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_61{{endif}} {{if 'CU_TARGET_COMPUTE_62' in found_values}} #: Compute device class 6.2. CU_TARGET_COMPUTE_62 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_62{{endif}} {{if 'CU_TARGET_COMPUTE_70' in found_values}} #: Compute device class 7.0. CU_TARGET_COMPUTE_70 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_70{{endif}} {{if 'CU_TARGET_COMPUTE_72' in found_values}} #: Compute device class 7.2. CU_TARGET_COMPUTE_72 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_72{{endif}} {{if 'CU_TARGET_COMPUTE_75' in found_values}} #: Compute device class 7.5. CU_TARGET_COMPUTE_75 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_75{{endif}} {{if 'CU_TARGET_COMPUTE_80' in found_values}} #: Compute device class 8.0. CU_TARGET_COMPUTE_80 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_80{{endif}} {{if 'CU_TARGET_COMPUTE_86' in found_values}} #: Compute device class 8.6. CU_TARGET_COMPUTE_86 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_86{{endif}} {{if 'CU_TARGET_COMPUTE_87' in found_values}} #: Compute device class 8.7. CU_TARGET_COMPUTE_87 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_87{{endif}} {{if 'CU_TARGET_COMPUTE_89' in found_values}} #: Compute device class 8.9. CU_TARGET_COMPUTE_89 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_89{{endif}} {{if 'CU_TARGET_COMPUTE_90' in found_values}} #: Compute device class 9.0. Compute device class 9.0. with accelerated #: features. CU_TARGET_COMPUTE_90 = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_90{{endif}} {{if 'CU_TARGET_COMPUTE_90A' in found_values}} CU_TARGET_COMPUTE_90A = ccuda.CUjit_target_enum.CU_TARGET_COMPUTE_90A{{endif}} {{endif}} {{if 'CUjit_fallback_enum' in found_types}} class CUjit_fallback(IntEnum): """ Cubin matching fallback strategies """ {{if 'CU_PREFER_PTX' in found_values}} #: Prefer to compile ptx if exact binary match not found CU_PREFER_PTX = ccuda.CUjit_fallback_enum.CU_PREFER_PTX{{endif}} {{if 'CU_PREFER_BINARY' in found_values}} #: Prefer to fall back to compatible binary code if exact match not #: found CU_PREFER_BINARY = ccuda.CUjit_fallback_enum.CU_PREFER_BINARY{{endif}} {{endif}} {{if 'CUjit_cacheMode_enum' in found_types}} class CUjit_cacheMode(IntEnum): """ Caching modes for dlcm """ {{if 'CU_JIT_CACHE_OPTION_NONE' in found_values}} #: Compile with no -dlcm flag specified CU_JIT_CACHE_OPTION_NONE = ccuda.CUjit_cacheMode_enum.CU_JIT_CACHE_OPTION_NONE{{endif}} {{if 'CU_JIT_CACHE_OPTION_CG' in found_values}} #: Compile with L1 cache disabled CU_JIT_CACHE_OPTION_CG = ccuda.CUjit_cacheMode_enum.CU_JIT_CACHE_OPTION_CG{{endif}} {{if 'CU_JIT_CACHE_OPTION_CA' in found_values}} #: Compile with L1 cache enabled CU_JIT_CACHE_OPTION_CA = ccuda.CUjit_cacheMode_enum.CU_JIT_CACHE_OPTION_CA{{endif}} {{endif}} {{if 'CUjitInputType_enum' in found_types}} class CUjitInputType(IntEnum): """ Device code formats """ {{if 'CU_JIT_INPUT_CUBIN' in found_values}} #: Compiled device-class-specific device code #: Applicable options: none CU_JIT_INPUT_CUBIN = ccuda.CUjitInputType_enum.CU_JIT_INPUT_CUBIN{{endif}} {{if 'CU_JIT_INPUT_PTX' in found_values}} #: PTX source code #: Applicable options: PTX compiler options CU_JIT_INPUT_PTX = ccuda.CUjitInputType_enum.CU_JIT_INPUT_PTX{{endif}} {{if 'CU_JIT_INPUT_FATBINARY' in found_values}} #: Bundle of multiple cubins and/or PTX of some device code #: Applicable options: PTX compiler options, #: :py:obj:`~.CU_JIT_FALLBACK_STRATEGY` CU_JIT_INPUT_FATBINARY = ccuda.CUjitInputType_enum.CU_JIT_INPUT_FATBINARY{{endif}} {{if 'CU_JIT_INPUT_OBJECT' in found_values}} #: Host object with embedded device code #: Applicable options: PTX compiler options, #: :py:obj:`~.CU_JIT_FALLBACK_STRATEGY` CU_JIT_INPUT_OBJECT = ccuda.CUjitInputType_enum.CU_JIT_INPUT_OBJECT{{endif}} {{if 'CU_JIT_INPUT_LIBRARY' in found_values}} #: Archive of host objects with embedded device code #: Applicable options: PTX compiler options, #: :py:obj:`~.CU_JIT_FALLBACK_STRATEGY` CU_JIT_INPUT_LIBRARY = ccuda.CUjitInputType_enum.CU_JIT_INPUT_LIBRARY{{endif}} {{if 'CU_JIT_INPUT_NVVM' in found_values}} #: [Deprecated] #: #: Only valid with LTO-IR compiled with toolkits prior to CUDA 12.0 CU_JIT_INPUT_NVVM = ccuda.CUjitInputType_enum.CU_JIT_INPUT_NVVM{{endif}} {{if 'CU_JIT_NUM_INPUT_TYPES' in found_values}} CU_JIT_NUM_INPUT_TYPES = ccuda.CUjitInputType_enum.CU_JIT_NUM_INPUT_TYPES{{endif}} {{endif}} {{if 'CUgraphicsRegisterFlags_enum' in found_types}} class CUgraphicsRegisterFlags(IntEnum): """ Flags to register a graphics resource """ {{if 'CU_GRAPHICS_REGISTER_FLAGS_NONE' in found_values}} CU_GRAPHICS_REGISTER_FLAGS_NONE = ccuda.CUgraphicsRegisterFlags_enum.CU_GRAPHICS_REGISTER_FLAGS_NONE{{endif}} {{if 'CU_GRAPHICS_REGISTER_FLAGS_READ_ONLY' in found_values}} CU_GRAPHICS_REGISTER_FLAGS_READ_ONLY = ccuda.CUgraphicsRegisterFlags_enum.CU_GRAPHICS_REGISTER_FLAGS_READ_ONLY{{endif}} {{if 'CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD' in found_values}} CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD = ccuda.CUgraphicsRegisterFlags_enum.CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD{{endif}} {{if 'CU_GRAPHICS_REGISTER_FLAGS_SURFACE_LDST' in found_values}} CU_GRAPHICS_REGISTER_FLAGS_SURFACE_LDST = ccuda.CUgraphicsRegisterFlags_enum.CU_GRAPHICS_REGISTER_FLAGS_SURFACE_LDST{{endif}} {{if 'CU_GRAPHICS_REGISTER_FLAGS_TEXTURE_GATHER' in found_values}} CU_GRAPHICS_REGISTER_FLAGS_TEXTURE_GATHER = ccuda.CUgraphicsRegisterFlags_enum.CU_GRAPHICS_REGISTER_FLAGS_TEXTURE_GATHER{{endif}} {{endif}} {{if 'CUgraphicsMapResourceFlags_enum' in found_types}} class CUgraphicsMapResourceFlags(IntEnum): """ Flags for mapping and unmapping interop resources """ {{if 'CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE' in found_values}} CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE = ccuda.CUgraphicsMapResourceFlags_enum.CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE{{endif}} {{if 'CU_GRAPHICS_MAP_RESOURCE_FLAGS_READ_ONLY' in found_values}} CU_GRAPHICS_MAP_RESOURCE_FLAGS_READ_ONLY = ccuda.CUgraphicsMapResourceFlags_enum.CU_GRAPHICS_MAP_RESOURCE_FLAGS_READ_ONLY{{endif}} {{if 'CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITE_DISCARD' in found_values}} CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITE_DISCARD = ccuda.CUgraphicsMapResourceFlags_enum.CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITE_DISCARD{{endif}} {{endif}} {{if 'CUarray_cubemap_face_enum' in found_types}} class CUarray_cubemap_face(IntEnum): """ Array indices for cube faces """ {{if 'CU_CUBEMAP_FACE_POSITIVE_X' in found_values}} #: Positive X face of cubemap CU_CUBEMAP_FACE_POSITIVE_X = ccuda.CUarray_cubemap_face_enum.CU_CUBEMAP_FACE_POSITIVE_X{{endif}} {{if 'CU_CUBEMAP_FACE_NEGATIVE_X' in found_values}} #: Negative X face of cubemap CU_CUBEMAP_FACE_NEGATIVE_X = ccuda.CUarray_cubemap_face_enum.CU_CUBEMAP_FACE_NEGATIVE_X{{endif}} {{if 'CU_CUBEMAP_FACE_POSITIVE_Y' in found_values}} #: Positive Y face of cubemap CU_CUBEMAP_FACE_POSITIVE_Y = ccuda.CUarray_cubemap_face_enum.CU_CUBEMAP_FACE_POSITIVE_Y{{endif}} {{if 'CU_CUBEMAP_FACE_NEGATIVE_Y' in found_values}} #: Negative Y face of cubemap CU_CUBEMAP_FACE_NEGATIVE_Y = ccuda.CUarray_cubemap_face_enum.CU_CUBEMAP_FACE_NEGATIVE_Y{{endif}} {{if 'CU_CUBEMAP_FACE_POSITIVE_Z' in found_values}} #: Positive Z face of cubemap CU_CUBEMAP_FACE_POSITIVE_Z = ccuda.CUarray_cubemap_face_enum.CU_CUBEMAP_FACE_POSITIVE_Z{{endif}} {{if 'CU_CUBEMAP_FACE_NEGATIVE_Z' in found_values}} #: Negative Z face of cubemap CU_CUBEMAP_FACE_NEGATIVE_Z = ccuda.CUarray_cubemap_face_enum.CU_CUBEMAP_FACE_NEGATIVE_Z{{endif}} {{endif}} {{if 'CUlimit_enum' in found_types}} class CUlimit(IntEnum): """ Limits """ {{if 'CU_LIMIT_STACK_SIZE' in found_values}} #: GPU thread stack size CU_LIMIT_STACK_SIZE = ccuda.CUlimit_enum.CU_LIMIT_STACK_SIZE{{endif}} {{if 'CU_LIMIT_PRINTF_FIFO_SIZE' in found_values}} #: GPU printf FIFO size CU_LIMIT_PRINTF_FIFO_SIZE = ccuda.CUlimit_enum.CU_LIMIT_PRINTF_FIFO_SIZE{{endif}} {{if 'CU_LIMIT_MALLOC_HEAP_SIZE' in found_values}} #: GPU malloc heap size CU_LIMIT_MALLOC_HEAP_SIZE = ccuda.CUlimit_enum.CU_LIMIT_MALLOC_HEAP_SIZE{{endif}} {{if 'CU_LIMIT_DEV_RUNTIME_SYNC_DEPTH' in found_values}} #: GPU device runtime launch synchronize depth CU_LIMIT_DEV_RUNTIME_SYNC_DEPTH = ccuda.CUlimit_enum.CU_LIMIT_DEV_RUNTIME_SYNC_DEPTH{{endif}} {{if 'CU_LIMIT_DEV_RUNTIME_PENDING_LAUNCH_COUNT' in found_values}} #: GPU device runtime pending launch count CU_LIMIT_DEV_RUNTIME_PENDING_LAUNCH_COUNT = ccuda.CUlimit_enum.CU_LIMIT_DEV_RUNTIME_PENDING_LAUNCH_COUNT{{endif}} {{if 'CU_LIMIT_MAX_L2_FETCH_GRANULARITY' in found_values}} #: A value between 0 and 128 that indicates the maximum fetch #: granularity of L2 (in Bytes). This is a hint CU_LIMIT_MAX_L2_FETCH_GRANULARITY = ccuda.CUlimit_enum.CU_LIMIT_MAX_L2_FETCH_GRANULARITY{{endif}} {{if 'CU_LIMIT_PERSISTING_L2_CACHE_SIZE' in found_values}} #: A size in bytes for L2 persisting lines cache size CU_LIMIT_PERSISTING_L2_CACHE_SIZE = ccuda.CUlimit_enum.CU_LIMIT_PERSISTING_L2_CACHE_SIZE{{endif}} {{if 'CU_LIMIT_SHMEM_SIZE' in found_values}} #: A maximum size in bytes of shared memory available to CUDA kernels #: on a CIG context. Can only be queried, cannot be set CU_LIMIT_SHMEM_SIZE = ccuda.CUlimit_enum.CU_LIMIT_SHMEM_SIZE{{endif}} {{if 'CU_LIMIT_CIG_ENABLED' in found_values}} #: A non-zero value indicates this CUDA context is a CIG-enabled #: context. Can only be queried, cannot be set CU_LIMIT_CIG_ENABLED = ccuda.CUlimit_enum.CU_LIMIT_CIG_ENABLED{{endif}} {{if 'CU_LIMIT_CIG_SHMEM_FALLBACK_ENABLED' in found_values}} #: When set to a non-zero value, CUDA will fail to launch a kernel on a #: CIG context, instead of using the fallback path, if the kernel uses #: more shared memory than available CU_LIMIT_CIG_SHMEM_FALLBACK_ENABLED = ccuda.CUlimit_enum.CU_LIMIT_CIG_SHMEM_FALLBACK_ENABLED{{endif}} {{if 'CU_LIMIT_MAX' in found_values}} CU_LIMIT_MAX = ccuda.CUlimit_enum.CU_LIMIT_MAX{{endif}} {{endif}} {{if 'CUresourcetype_enum' in found_types}} class CUresourcetype(IntEnum): """ Resource types """ {{if 'CU_RESOURCE_TYPE_ARRAY' in found_values}} #: Array resource CU_RESOURCE_TYPE_ARRAY = ccuda.CUresourcetype_enum.CU_RESOURCE_TYPE_ARRAY{{endif}} {{if 'CU_RESOURCE_TYPE_MIPMAPPED_ARRAY' in found_values}} #: Mipmapped array resource CU_RESOURCE_TYPE_MIPMAPPED_ARRAY = ccuda.CUresourcetype_enum.CU_RESOURCE_TYPE_MIPMAPPED_ARRAY{{endif}} {{if 'CU_RESOURCE_TYPE_LINEAR' in found_values}} #: Linear resource CU_RESOURCE_TYPE_LINEAR = ccuda.CUresourcetype_enum.CU_RESOURCE_TYPE_LINEAR{{endif}} {{if 'CU_RESOURCE_TYPE_PITCH2D' in found_values}} #: Pitch 2D resource CU_RESOURCE_TYPE_PITCH2D = ccuda.CUresourcetype_enum.CU_RESOURCE_TYPE_PITCH2D{{endif}} {{endif}} {{if 'CUaccessProperty_enum' in found_types}} class CUaccessProperty(IntEnum): """ Specifies performance hint with :py:obj:`~.CUaccessPolicyWindow` for hitProp and missProp members. """ {{if 'CU_ACCESS_PROPERTY_NORMAL' in found_values}} #: Normal cache persistence. CU_ACCESS_PROPERTY_NORMAL = ccuda.CUaccessProperty_enum.CU_ACCESS_PROPERTY_NORMAL{{endif}} {{if 'CU_ACCESS_PROPERTY_STREAMING' in found_values}} #: Streaming access is less likely to persit from cache. CU_ACCESS_PROPERTY_STREAMING = ccuda.CUaccessProperty_enum.CU_ACCESS_PROPERTY_STREAMING{{endif}} {{if 'CU_ACCESS_PROPERTY_PERSISTING' in found_values}} #: Persisting access is more likely to persist in cache. CU_ACCESS_PROPERTY_PERSISTING = ccuda.CUaccessProperty_enum.CU_ACCESS_PROPERTY_PERSISTING{{endif}} {{endif}} {{if 'CUgraphConditionalNodeType_enum' in found_types}} class CUgraphConditionalNodeType(IntEnum): """ Conditional node types """ {{if 'CU_GRAPH_COND_TYPE_IF' in found_values}} #: Conditional 'if' Node. Body executed once if condition value is non- #: zero. CU_GRAPH_COND_TYPE_IF = ccuda.CUgraphConditionalNodeType_enum.CU_GRAPH_COND_TYPE_IF{{endif}} {{if 'CU_GRAPH_COND_TYPE_WHILE' in found_values}} #: Conditional 'while' Node. Body executed repeatedly while condition #: value is non-zero. CU_GRAPH_COND_TYPE_WHILE = ccuda.CUgraphConditionalNodeType_enum.CU_GRAPH_COND_TYPE_WHILE{{endif}} {{endif}} {{if 'CUgraphNodeType_enum' in found_types}} class CUgraphNodeType(IntEnum): """ Graph node types """ {{if 'CU_GRAPH_NODE_TYPE_KERNEL' in found_values}} #: GPU kernel node CU_GRAPH_NODE_TYPE_KERNEL = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_KERNEL{{endif}} {{if 'CU_GRAPH_NODE_TYPE_MEMCPY' in found_values}} #: Memcpy node CU_GRAPH_NODE_TYPE_MEMCPY = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_MEMCPY{{endif}} {{if 'CU_GRAPH_NODE_TYPE_MEMSET' in found_values}} #: Memset node CU_GRAPH_NODE_TYPE_MEMSET = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_MEMSET{{endif}} {{if 'CU_GRAPH_NODE_TYPE_HOST' in found_values}} #: Host (executable) node CU_GRAPH_NODE_TYPE_HOST = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_HOST{{endif}} {{if 'CU_GRAPH_NODE_TYPE_GRAPH' in found_values}} #: Node which executes an embedded graph CU_GRAPH_NODE_TYPE_GRAPH = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_GRAPH{{endif}} {{if 'CU_GRAPH_NODE_TYPE_EMPTY' in found_values}} #: Empty (no-op) node CU_GRAPH_NODE_TYPE_EMPTY = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_EMPTY{{endif}} {{if 'CU_GRAPH_NODE_TYPE_WAIT_EVENT' in found_values}} #: External event wait node CU_GRAPH_NODE_TYPE_WAIT_EVENT = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_WAIT_EVENT{{endif}} {{if 'CU_GRAPH_NODE_TYPE_EVENT_RECORD' in found_values}} #: External event record node CU_GRAPH_NODE_TYPE_EVENT_RECORD = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_EVENT_RECORD{{endif}} {{if 'CU_GRAPH_NODE_TYPE_EXT_SEMAS_SIGNAL' in found_values}} #: External semaphore signal node CU_GRAPH_NODE_TYPE_EXT_SEMAS_SIGNAL = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_EXT_SEMAS_SIGNAL{{endif}} {{if 'CU_GRAPH_NODE_TYPE_EXT_SEMAS_WAIT' in found_values}} #: External semaphore wait node CU_GRAPH_NODE_TYPE_EXT_SEMAS_WAIT = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_EXT_SEMAS_WAIT{{endif}} {{if 'CU_GRAPH_NODE_TYPE_MEM_ALLOC' in found_values}} #: Memory Allocation Node CU_GRAPH_NODE_TYPE_MEM_ALLOC = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_MEM_ALLOC{{endif}} {{if 'CU_GRAPH_NODE_TYPE_MEM_FREE' in found_values}} #: Memory Free Node CU_GRAPH_NODE_TYPE_MEM_FREE = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_MEM_FREE{{endif}} {{if 'CU_GRAPH_NODE_TYPE_BATCH_MEM_OP' in found_values}} #: Batch MemOp Node CU_GRAPH_NODE_TYPE_BATCH_MEM_OP = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_BATCH_MEM_OP{{endif}} {{if 'CU_GRAPH_NODE_TYPE_CONDITIONAL' in found_values}} #: Conditional Node May be used #: to implement a conditional execution path or loop #: inside of a graph. The #: graph(s) contained within the body of the conditional node #: can be selectively executed #: or iterated upon based on the value of a conditional #: variable. #: #: Handles must be created in #: advance of creating the node #: using #: :py:obj:`~.cuGraphConditionalHandleCreate`. #: #: The following restrictions #: apply to graphs which contain conditional nodes: #: The graph cannot be used in #: a child node. #: Only one instantiation of #: the graph may exist at any point in time. #: The graph cannot be cloned. #: #: To set the control value, #: supply a default value when creating the handle and/or #: call #: :py:obj:`~.cudaGraphSetConditional` from device code. CU_GRAPH_NODE_TYPE_CONDITIONAL = ccuda.CUgraphNodeType_enum.CU_GRAPH_NODE_TYPE_CONDITIONAL{{endif}} {{endif}} {{if 'CUgraphDependencyType_enum' in found_types}} class CUgraphDependencyType(IntEnum): """ Type annotations that can be applied to graph edges as part of :py:obj:`~.CUgraphEdgeData`. """ {{if 'CU_GRAPH_DEPENDENCY_TYPE_DEFAULT' in found_values}} #: This is an ordinary dependency. CU_GRAPH_DEPENDENCY_TYPE_DEFAULT = ccuda.CUgraphDependencyType_enum.CU_GRAPH_DEPENDENCY_TYPE_DEFAULT{{endif}} {{if 'CU_GRAPH_DEPENDENCY_TYPE_PROGRAMMATIC' in found_values}} #: This dependency type allows the downstream node to use #: `cudaGridDependencySynchronize()`. It may only be used between #: kernel nodes, and must be used with either the #: :py:obj:`~.CU_GRAPH_KERNEL_NODE_PORT_PROGRAMMATIC` or #: :py:obj:`~.CU_GRAPH_KERNEL_NODE_PORT_LAUNCH_ORDER` outgoing port. CU_GRAPH_DEPENDENCY_TYPE_PROGRAMMATIC = ccuda.CUgraphDependencyType_enum.CU_GRAPH_DEPENDENCY_TYPE_PROGRAMMATIC{{endif}} {{endif}} {{if 'CUgraphInstantiateResult_enum' in found_types}} class CUgraphInstantiateResult(IntEnum): """ Graph instantiation results """ {{if 'CUDA_GRAPH_INSTANTIATE_SUCCESS' in found_values}} #: Instantiation succeeded CUDA_GRAPH_INSTANTIATE_SUCCESS = ccuda.CUgraphInstantiateResult_enum.CUDA_GRAPH_INSTANTIATE_SUCCESS{{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_ERROR' in found_values}} #: Instantiation failed for an unexpected reason which is described in #: the return value of the function CUDA_GRAPH_INSTANTIATE_ERROR = ccuda.CUgraphInstantiateResult_enum.CUDA_GRAPH_INSTANTIATE_ERROR{{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_INVALID_STRUCTURE' in found_values}} #: Instantiation failed due to invalid structure, such as cycles CUDA_GRAPH_INSTANTIATE_INVALID_STRUCTURE = ccuda.CUgraphInstantiateResult_enum.CUDA_GRAPH_INSTANTIATE_INVALID_STRUCTURE{{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_NODE_OPERATION_NOT_SUPPORTED' in found_values}} #: Instantiation for device launch failed because the graph contained #: an unsupported operation CUDA_GRAPH_INSTANTIATE_NODE_OPERATION_NOT_SUPPORTED = ccuda.CUgraphInstantiateResult_enum.CUDA_GRAPH_INSTANTIATE_NODE_OPERATION_NOT_SUPPORTED{{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_MULTIPLE_CTXS_NOT_SUPPORTED' in found_values}} #: Instantiation for device launch failed due to the nodes belonging to #: different contexts CUDA_GRAPH_INSTANTIATE_MULTIPLE_CTXS_NOT_SUPPORTED = ccuda.CUgraphInstantiateResult_enum.CUDA_GRAPH_INSTANTIATE_MULTIPLE_CTXS_NOT_SUPPORTED{{endif}} {{endif}} {{if 'CUsynchronizationPolicy_enum' in found_types}} class CUsynchronizationPolicy(IntEnum): """ """ {{if 'CU_SYNC_POLICY_AUTO' in found_values}} CU_SYNC_POLICY_AUTO = ccuda.CUsynchronizationPolicy_enum.CU_SYNC_POLICY_AUTO{{endif}} {{if 'CU_SYNC_POLICY_SPIN' in found_values}} CU_SYNC_POLICY_SPIN = ccuda.CUsynchronizationPolicy_enum.CU_SYNC_POLICY_SPIN{{endif}} {{if 'CU_SYNC_POLICY_YIELD' in found_values}} CU_SYNC_POLICY_YIELD = ccuda.CUsynchronizationPolicy_enum.CU_SYNC_POLICY_YIELD{{endif}} {{if 'CU_SYNC_POLICY_BLOCKING_SYNC' in found_values}} CU_SYNC_POLICY_BLOCKING_SYNC = ccuda.CUsynchronizationPolicy_enum.CU_SYNC_POLICY_BLOCKING_SYNC{{endif}} {{endif}} {{if 'CUclusterSchedulingPolicy_enum' in found_types}} class CUclusterSchedulingPolicy(IntEnum): """ Cluster scheduling policies. These may be passed to :py:obj:`~.cuFuncSetAttribute` or :py:obj:`~.cuKernelSetAttribute` """ {{if 'CU_CLUSTER_SCHEDULING_POLICY_DEFAULT' in found_values}} #: the default policy CU_CLUSTER_SCHEDULING_POLICY_DEFAULT = ccuda.CUclusterSchedulingPolicy_enum.CU_CLUSTER_SCHEDULING_POLICY_DEFAULT{{endif}} {{if 'CU_CLUSTER_SCHEDULING_POLICY_SPREAD' in found_values}} #: spread the blocks within a cluster to the SMs CU_CLUSTER_SCHEDULING_POLICY_SPREAD = ccuda.CUclusterSchedulingPolicy_enum.CU_CLUSTER_SCHEDULING_POLICY_SPREAD{{endif}} {{if 'CU_CLUSTER_SCHEDULING_POLICY_LOAD_BALANCING' in found_values}} #: allow the hardware to load-balance the blocks in a cluster to the #: SMs CU_CLUSTER_SCHEDULING_POLICY_LOAD_BALANCING = ccuda.CUclusterSchedulingPolicy_enum.CU_CLUSTER_SCHEDULING_POLICY_LOAD_BALANCING{{endif}} {{endif}} {{if 'CUlaunchMemSyncDomain_enum' in found_types}} class CUlaunchMemSyncDomain(IntEnum): """ Memory Synchronization Domain A kernel can be launched in a specified memory synchronization domain that affects all memory operations issued by that kernel. A memory barrier issued in one domain will only order memory operations in that domain, thus eliminating latency increase from memory barriers ordering unrelated traffic. By default, kernels are launched in domain 0. Kernel launched with :py:obj:`~.CU_LAUNCH_MEM_SYNC_DOMAIN_REMOTE` will have a different domain ID. User may also alter the domain ID with :py:obj:`~.CUlaunchMemSyncDomainMap` for a specific stream / graph node / kernel launch. See :py:obj:`~.CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN`, :py:obj:`~.cuStreamSetAttribute`, :py:obj:`~.cuLaunchKernelEx`, :py:obj:`~.cuGraphKernelNodeSetAttribute`. Memory operations done in kernels launched in different domains are considered system- scope distanced. In other words, a GPU scoped memory synchronization is not sufficient for memory order to be observed by kernels in another memory synchronization domain even if they are on the same GPU. """ {{if 'CU_LAUNCH_MEM_SYNC_DOMAIN_DEFAULT' in found_values}} #: Launch kernels in the default domain CU_LAUNCH_MEM_SYNC_DOMAIN_DEFAULT = ccuda.CUlaunchMemSyncDomain_enum.CU_LAUNCH_MEM_SYNC_DOMAIN_DEFAULT{{endif}} {{if 'CU_LAUNCH_MEM_SYNC_DOMAIN_REMOTE' in found_values}} #: Launch kernels in the remote domain CU_LAUNCH_MEM_SYNC_DOMAIN_REMOTE = ccuda.CUlaunchMemSyncDomain_enum.CU_LAUNCH_MEM_SYNC_DOMAIN_REMOTE{{endif}} {{endif}} {{if 'CUlaunchAttributeID_enum' in found_types}} class CUlaunchAttributeID(IntEnum): """ Launch attributes enum; used as id field of :py:obj:`~.CUlaunchAttribute` """ {{if 'CU_LAUNCH_ATTRIBUTE_IGNORE' in found_values}} #: Ignored entry, for convenient composition CU_LAUNCH_ATTRIBUTE_IGNORE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_IGNORE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.accessPolicyWindow`. CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_COOPERATIVE' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.cooperative`. CU_LAUNCH_ATTRIBUTE_COOPERATIVE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_COOPERATIVE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY' in found_values}} #: Valid for streams. See #: :py:obj:`~.CUlaunchAttributeValue.syncPolicy`. CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.clusterDim`. CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.clusterSchedulingPolicyPreference`. CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION' in found_values}} #: Valid for launches. Setting #: :py:obj:`~.CUlaunchAttributeValue.programmaticStreamSerializationAllowed` #: to non-0 signals that the kernel will use programmatic means to #: resolve its stream dependency, so that the CUDA runtime should #: opportunistically allow the grid's execution to overlap with the #: previous kernel in the stream, if that kernel requests the overlap. #: The dependent launches can choose to wait on the dependency using #: the programmatic sync (cudaGridDependencySynchronize() or equivalent #: PTX instructions). CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT' in found_values}} #: Valid for launches. Set #: :py:obj:`~.CUlaunchAttributeValue.programmaticEvent` to record the #: event. Event recorded through this launch attribute is guaranteed to #: only trigger after all block in the associated kernel trigger the #: event. A block can trigger the event through PTX launchdep.release #: or CUDA builtin function cudaTriggerProgrammaticLaunchCompletion(). #: A trigger can also be inserted at the beginning of each block's #: execution if triggerAtBlockStart is set to non-0. The dependent #: launches can choose to wait on the dependency using the programmatic #: sync (cudaGridDependencySynchronize() or equivalent PTX #: instructions). Note that dependents (including the CPU thread #: calling :py:obj:`~.cuEventSynchronize()`) are not guaranteed to #: observe the release precisely when it is released. For example, #: :py:obj:`~.cuEventSynchronize()` may only observe the event trigger #: long after the associated kernel has completed. This recording type #: is primarily meant for establishing programmatic dependency between #: device tasks. Note also this type of dependency allows, but does not #: guarantee, concurrent execution of tasks. #: The event supplied must not be an interprocess or interop event. #: The event must disable timing (i.e. must be created with the #: :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PRIORITY' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.priority`. CU_LAUNCH_ATTRIBUTE_PRIORITY = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PRIORITY{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.memSyncDomainMap`. CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.memSyncDomain`. CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT' in found_values}} #: Valid for launches. Set #: :py:obj:`~.CUlaunchAttributeValue.launchCompletionEvent` to record #: the event. #: Nominally, the event is triggered once all blocks of the kernel #: have begun execution. Currently this is a best effort. If a kernel B #: has a launch completion dependency on a kernel A, B may wait until A #: is complete. Alternatively, blocks of B may begin before all blocks #: of A have begun, for example if B can claim execution resources #: unavailable to A (e.g. they run on different GPUs) or if B is a #: higher priority than A. Exercise caution if such an ordering #: inversion could lead to deadlock. #: A launch completion event is nominally similar to a programmatic #: event with `triggerAtBlockStart` set except that it is not visible #: to `cudaGridDependencySynchronize()` and can be used with compute #: capability less than 9.0. #: The event supplied must not be an interprocess or interop event. #: The event must disable timing (i.e. must be created with the #: :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE' in found_values}} #: Valid for graph nodes, launches. This attribute is graphs-only, and #: passing it to a launch in a non-capturing stream will result in an #: error. #: :py:obj:`~.CUlaunchAttributeValue`::deviceUpdatableKernelNode::deviceUpdatable #: can only be set to 0 or 1. Setting the field to 1 indicates that the #: corresponding kernel node should be device-updatable. On success, a #: handle will be returned via #: :py:obj:`~.CUlaunchAttributeValue`::deviceUpdatableKernelNode::devNode #: which can be passed to the various device-side update functions to #: update the node's kernel parameters from within another kernel. For #: more information on the types of device updates that can be made, as #: well as the relevant limitations thereof, see #: :py:obj:`~.cudaGraphKernelNodeUpdatesApply`. #: Nodes which are device-updatable have additional restrictions #: compared to regular kernel nodes. Firstly, device-updatable nodes #: cannot be removed from their graph via #: :py:obj:`~.cuGraphDestroyNode`. Additionally, once opted-in to this #: functionality, a node cannot opt out, and any attempt to set the #: deviceUpdatable attribute to 0 will result in an error. Device- #: updatable kernel nodes also cannot have their attributes copied #: to/from another kernel node via #: :py:obj:`~.cuGraphKernelNodeCopyAttributes`. Graphs containing one #: or more device-updatable nodes also do not allow multiple #: instantiation, and neither the graph nor its instantiated version #: can be passed to :py:obj:`~.cuGraphExecUpdate`. #: If a graph contains device-updatable nodes and updates those nodes #: from the device from within the graph, the graph must be uploaded #: with :py:obj:`~.cuGraphUpload` before it is launched. For such a #: graph, if host-side executable graph updates are made to the device- #: updatable nodes, the graph must be uploaded before it is launched #: again. CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT' in found_values}} #: Valid for launches. On devices where the L1 cache and shared memory #: use the same hardware resources, setting #: :py:obj:`~.CUlaunchAttributeValue.sharedMemCarveout` to a percentage #: between 0-100 signals the CUDA driver to set the shared memory #: carveout preference, in percent of the total shared memory for that #: kernel launch. This attribute takes precedence over #: :py:obj:`~.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT`. This #: is only a hint, and the CUDA driver can choose a different #: configuration if required for the launch. CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT{{endif}} {{endif}} {{if 'CUstreamCaptureStatus_enum' in found_types}} class CUstreamCaptureStatus(IntEnum): """ Possible stream capture statuses returned by :py:obj:`~.cuStreamIsCapturing` """ {{if 'CU_STREAM_CAPTURE_STATUS_NONE' in found_values}} #: Stream is not capturing CU_STREAM_CAPTURE_STATUS_NONE = ccuda.CUstreamCaptureStatus_enum.CU_STREAM_CAPTURE_STATUS_NONE{{endif}} {{if 'CU_STREAM_CAPTURE_STATUS_ACTIVE' in found_values}} #: Stream is actively capturing CU_STREAM_CAPTURE_STATUS_ACTIVE = ccuda.CUstreamCaptureStatus_enum.CU_STREAM_CAPTURE_STATUS_ACTIVE{{endif}} {{if 'CU_STREAM_CAPTURE_STATUS_INVALIDATED' in found_values}} #: Stream is part of a capture sequence that has been invalidated, but #: not terminated CU_STREAM_CAPTURE_STATUS_INVALIDATED = ccuda.CUstreamCaptureStatus_enum.CU_STREAM_CAPTURE_STATUS_INVALIDATED{{endif}} {{endif}} {{if 'CUstreamCaptureMode_enum' in found_types}} class CUstreamCaptureMode(IntEnum): """ Possible modes for stream capture thread interactions. For more details see :py:obj:`~.cuStreamBeginCapture` and :py:obj:`~.cuThreadExchangeStreamCaptureMode` """ {{if 'CU_STREAM_CAPTURE_MODE_GLOBAL' in found_values}} CU_STREAM_CAPTURE_MODE_GLOBAL = ccuda.CUstreamCaptureMode_enum.CU_STREAM_CAPTURE_MODE_GLOBAL{{endif}} {{if 'CU_STREAM_CAPTURE_MODE_THREAD_LOCAL' in found_values}} CU_STREAM_CAPTURE_MODE_THREAD_LOCAL = ccuda.CUstreamCaptureMode_enum.CU_STREAM_CAPTURE_MODE_THREAD_LOCAL{{endif}} {{if 'CU_STREAM_CAPTURE_MODE_RELAXED' in found_values}} CU_STREAM_CAPTURE_MODE_RELAXED = ccuda.CUstreamCaptureMode_enum.CU_STREAM_CAPTURE_MODE_RELAXED{{endif}} {{endif}} {{if 'CUdriverProcAddress_flags_enum' in found_types}} class CUdriverProcAddress_flags(IntEnum): """ Flags to specify search options. For more details see :py:obj:`~.cuGetProcAddress` """ {{if 'CU_GET_PROC_ADDRESS_DEFAULT' in found_values}} #: Default search mode for driver symbols. CU_GET_PROC_ADDRESS_DEFAULT = ccuda.CUdriverProcAddress_flags_enum.CU_GET_PROC_ADDRESS_DEFAULT{{endif}} {{if 'CU_GET_PROC_ADDRESS_LEGACY_STREAM' in found_values}} #: Search for legacy versions of driver symbols. CU_GET_PROC_ADDRESS_LEGACY_STREAM = ccuda.CUdriverProcAddress_flags_enum.CU_GET_PROC_ADDRESS_LEGACY_STREAM{{endif}} {{if 'CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM' in found_values}} #: Search for per-thread versions of driver symbols. CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM = ccuda.CUdriverProcAddress_flags_enum.CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM{{endif}} {{endif}} {{if 'CUdriverProcAddressQueryResult_enum' in found_types}} class CUdriverProcAddressQueryResult(IntEnum): """ Flags to indicate search status. For more details see :py:obj:`~.cuGetProcAddress` """ {{if 'CU_GET_PROC_ADDRESS_SUCCESS' in found_values}} #: Symbol was succesfully found CU_GET_PROC_ADDRESS_SUCCESS = ccuda.CUdriverProcAddressQueryResult_enum.CU_GET_PROC_ADDRESS_SUCCESS{{endif}} {{if 'CU_GET_PROC_ADDRESS_SYMBOL_NOT_FOUND' in found_values}} #: Symbol was not found in search CU_GET_PROC_ADDRESS_SYMBOL_NOT_FOUND = ccuda.CUdriverProcAddressQueryResult_enum.CU_GET_PROC_ADDRESS_SYMBOL_NOT_FOUND{{endif}} {{if 'CU_GET_PROC_ADDRESS_VERSION_NOT_SUFFICIENT' in found_values}} #: Symbol was found but version supplied was not sufficient CU_GET_PROC_ADDRESS_VERSION_NOT_SUFFICIENT = ccuda.CUdriverProcAddressQueryResult_enum.CU_GET_PROC_ADDRESS_VERSION_NOT_SUFFICIENT{{endif}} {{endif}} {{if 'CUexecAffinityType_enum' in found_types}} class CUexecAffinityType(IntEnum): """ Execution Affinity Types """ {{if 'CU_EXEC_AFFINITY_TYPE_SM_COUNT' in found_values}} #: Create a context with limited SMs. CU_EXEC_AFFINITY_TYPE_SM_COUNT = ccuda.CUexecAffinityType_enum.CU_EXEC_AFFINITY_TYPE_SM_COUNT{{endif}} {{if 'CU_EXEC_AFFINITY_TYPE_MAX' in found_values}} CU_EXEC_AFFINITY_TYPE_MAX = ccuda.CUexecAffinityType_enum.CU_EXEC_AFFINITY_TYPE_MAX{{endif}} {{endif}} {{if 'CUcigDataType_enum' in found_types}} class CUcigDataType(IntEnum): """ """ {{if 'CIG_DATA_TYPE_D3D12_COMMAND_QUEUE' in found_values}} CIG_DATA_TYPE_D3D12_COMMAND_QUEUE = ccuda.CUcigDataType_enum.CIG_DATA_TYPE_D3D12_COMMAND_QUEUE{{endif}} {{endif}} {{if 'CUlibraryOption_enum' in found_types}} class CUlibraryOption(IntEnum): """ Library options to be specified with :py:obj:`~.cuLibraryLoadData()` or :py:obj:`~.cuLibraryLoadFromFile()` """ {{if 'CU_LIBRARY_HOST_UNIVERSAL_FUNCTION_AND_DATA_TABLE' in found_values}} CU_LIBRARY_HOST_UNIVERSAL_FUNCTION_AND_DATA_TABLE = ccuda.CUlibraryOption_enum.CU_LIBRARY_HOST_UNIVERSAL_FUNCTION_AND_DATA_TABLE{{endif}} {{if 'CU_LIBRARY_BINARY_IS_PRESERVED' in found_values}} #: Specifes that the argument `code` passed to #: :py:obj:`~.cuLibraryLoadData()` will be preserved. Specifying this #: option will let the driver know that `code` can be accessed at any #: point until :py:obj:`~.cuLibraryUnload()`. The default behavior is #: for the driver to allocate and maintain its own copy of `code`. Note #: that this is only a memory usage optimization hint and the driver #: can choose to ignore it if required. Specifying this option with #: :py:obj:`~.cuLibraryLoadFromFile()` is invalid and will return #: :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. CU_LIBRARY_BINARY_IS_PRESERVED = ccuda.CUlibraryOption_enum.CU_LIBRARY_BINARY_IS_PRESERVED{{endif}} {{if 'CU_LIBRARY_NUM_OPTIONS' in found_values}} CU_LIBRARY_NUM_OPTIONS = ccuda.CUlibraryOption_enum.CU_LIBRARY_NUM_OPTIONS{{endif}} {{endif}} {{if 'cudaError_enum' in found_types}} class CUresult(IntEnum): """ Error codes """ {{if 'CUDA_SUCCESS' in found_values}} #: The API call returned with no errors. In the case of query calls, #: this also means that the operation being queried is complete (see #: :py:obj:`~.cuEventQuery()` and :py:obj:`~.cuStreamQuery()`). CUDA_SUCCESS = ccuda.cudaError_enum.CUDA_SUCCESS{{endif}} {{if 'CUDA_ERROR_INVALID_VALUE' in found_values}} #: This indicates that one or more of the parameters passed to the API #: call is not within an acceptable range of values. CUDA_ERROR_INVALID_VALUE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_VALUE{{endif}} {{if 'CUDA_ERROR_OUT_OF_MEMORY' in found_values}} #: The API call failed because it was unable to allocate enough memory #: or other resources to perform the requested operation. CUDA_ERROR_OUT_OF_MEMORY = ccuda.cudaError_enum.CUDA_ERROR_OUT_OF_MEMORY{{endif}} {{if 'CUDA_ERROR_NOT_INITIALIZED' in found_values}} #: This indicates that the CUDA driver has not been initialized with #: :py:obj:`~.cuInit()` or that initialization has failed. CUDA_ERROR_NOT_INITIALIZED = ccuda.cudaError_enum.CUDA_ERROR_NOT_INITIALIZED{{endif}} {{if 'CUDA_ERROR_DEINITIALIZED' in found_values}} #: This indicates that the CUDA driver is in the process of shutting #: down. CUDA_ERROR_DEINITIALIZED = ccuda.cudaError_enum.CUDA_ERROR_DEINITIALIZED{{endif}} {{if 'CUDA_ERROR_PROFILER_DISABLED' in found_values}} #: This indicates profiler is not initialized for this run. This can #: happen when the application is running with external profiling tools #: like visual profiler. CUDA_ERROR_PROFILER_DISABLED = ccuda.cudaError_enum.CUDA_ERROR_PROFILER_DISABLED{{endif}} {{if 'CUDA_ERROR_PROFILER_NOT_INITIALIZED' in found_values}} #: [Deprecated] CUDA_ERROR_PROFILER_NOT_INITIALIZED = ccuda.cudaError_enum.CUDA_ERROR_PROFILER_NOT_INITIALIZED{{endif}} {{if 'CUDA_ERROR_PROFILER_ALREADY_STARTED' in found_values}} #: [Deprecated] CUDA_ERROR_PROFILER_ALREADY_STARTED = ccuda.cudaError_enum.CUDA_ERROR_PROFILER_ALREADY_STARTED{{endif}} {{if 'CUDA_ERROR_PROFILER_ALREADY_STOPPED' in found_values}} #: [Deprecated] CUDA_ERROR_PROFILER_ALREADY_STOPPED = ccuda.cudaError_enum.CUDA_ERROR_PROFILER_ALREADY_STOPPED{{endif}} {{if 'CUDA_ERROR_STUB_LIBRARY' in found_values}} #: This indicates that the CUDA driver that the application has loaded #: is a stub library. Applications that run with the stub rather than a #: real driver loaded will result in CUDA API returning this error. CUDA_ERROR_STUB_LIBRARY = ccuda.cudaError_enum.CUDA_ERROR_STUB_LIBRARY{{endif}} {{if 'CUDA_ERROR_DEVICE_UNAVAILABLE' in found_values}} #: This indicates that requested CUDA device is unavailable at the #: current time. Devices are often unavailable due to use of #: :py:obj:`~.CU_COMPUTEMODE_EXCLUSIVE_PROCESS` or #: :py:obj:`~.CU_COMPUTEMODE_PROHIBITED`. CUDA_ERROR_DEVICE_UNAVAILABLE = ccuda.cudaError_enum.CUDA_ERROR_DEVICE_UNAVAILABLE{{endif}} {{if 'CUDA_ERROR_NO_DEVICE' in found_values}} #: This indicates that no CUDA-capable devices were detected by the #: installed CUDA driver. CUDA_ERROR_NO_DEVICE = ccuda.cudaError_enum.CUDA_ERROR_NO_DEVICE{{endif}} {{if 'CUDA_ERROR_INVALID_DEVICE' in found_values}} #: This indicates that the device ordinal supplied by the user does not #: correspond to a valid CUDA device or that the action requested is #: invalid for the specified device. CUDA_ERROR_INVALID_DEVICE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_DEVICE{{endif}} {{if 'CUDA_ERROR_DEVICE_NOT_LICENSED' in found_values}} #: This error indicates that the Grid license is not applied. CUDA_ERROR_DEVICE_NOT_LICENSED = ccuda.cudaError_enum.CUDA_ERROR_DEVICE_NOT_LICENSED{{endif}} {{if 'CUDA_ERROR_INVALID_IMAGE' in found_values}} #: This indicates that the device kernel image is invalid. This can #: also indicate an invalid CUDA module. CUDA_ERROR_INVALID_IMAGE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_IMAGE{{endif}} {{if 'CUDA_ERROR_INVALID_CONTEXT' in found_values}} #: This most frequently indicates that there is no context bound to the #: current thread. This can also be returned if the context passed to #: an API call is not a valid handle (such as a context that has had #: :py:obj:`~.cuCtxDestroy()` invoked on it). This can also be returned #: if a user mixes different API versions (i.e. 3010 context with 3020 #: API calls). See :py:obj:`~.cuCtxGetApiVersion()` for more details. #: This can also be returned if the green context passed to an API call #: was not converted to a :py:obj:`~.CUcontext` using #: :py:obj:`~.cuCtxFromGreenCtx` API. CUDA_ERROR_INVALID_CONTEXT = ccuda.cudaError_enum.CUDA_ERROR_INVALID_CONTEXT{{endif}} {{if 'CUDA_ERROR_CONTEXT_ALREADY_CURRENT' in found_values}} #: This indicated that the context being supplied as a parameter to the #: API call was already the active context. [Deprecated] CUDA_ERROR_CONTEXT_ALREADY_CURRENT = ccuda.cudaError_enum.CUDA_ERROR_CONTEXT_ALREADY_CURRENT{{endif}} {{if 'CUDA_ERROR_MAP_FAILED' in found_values}} #: This indicates that a map or register operation has failed. CUDA_ERROR_MAP_FAILED = ccuda.cudaError_enum.CUDA_ERROR_MAP_FAILED{{endif}} {{if 'CUDA_ERROR_UNMAP_FAILED' in found_values}} #: This indicates that an unmap or unregister operation has failed. CUDA_ERROR_UNMAP_FAILED = ccuda.cudaError_enum.CUDA_ERROR_UNMAP_FAILED{{endif}} {{if 'CUDA_ERROR_ARRAY_IS_MAPPED' in found_values}} #: This indicates that the specified array is currently mapped and thus #: cannot be destroyed. CUDA_ERROR_ARRAY_IS_MAPPED = ccuda.cudaError_enum.CUDA_ERROR_ARRAY_IS_MAPPED{{endif}} {{if 'CUDA_ERROR_ALREADY_MAPPED' in found_values}} #: This indicates that the resource is already mapped. CUDA_ERROR_ALREADY_MAPPED = ccuda.cudaError_enum.CUDA_ERROR_ALREADY_MAPPED{{endif}} {{if 'CUDA_ERROR_NO_BINARY_FOR_GPU' in found_values}} #: This indicates that there is no kernel image available that is #: suitable for the device. This can occur when a user specifies code #: generation options for a particular CUDA source file that do not #: include the corresponding device configuration. CUDA_ERROR_NO_BINARY_FOR_GPU = ccuda.cudaError_enum.CUDA_ERROR_NO_BINARY_FOR_GPU{{endif}} {{if 'CUDA_ERROR_ALREADY_ACQUIRED' in found_values}} #: This indicates that a resource has already been acquired. CUDA_ERROR_ALREADY_ACQUIRED = ccuda.cudaError_enum.CUDA_ERROR_ALREADY_ACQUIRED{{endif}} {{if 'CUDA_ERROR_NOT_MAPPED' in found_values}} #: This indicates that a resource is not mapped. CUDA_ERROR_NOT_MAPPED = ccuda.cudaError_enum.CUDA_ERROR_NOT_MAPPED{{endif}} {{if 'CUDA_ERROR_NOT_MAPPED_AS_ARRAY' in found_values}} #: This indicates that a mapped resource is not available for access as #: an array. CUDA_ERROR_NOT_MAPPED_AS_ARRAY = ccuda.cudaError_enum.CUDA_ERROR_NOT_MAPPED_AS_ARRAY{{endif}} {{if 'CUDA_ERROR_NOT_MAPPED_AS_POINTER' in found_values}} #: This indicates that a mapped resource is not available for access as #: a pointer. CUDA_ERROR_NOT_MAPPED_AS_POINTER = ccuda.cudaError_enum.CUDA_ERROR_NOT_MAPPED_AS_POINTER{{endif}} {{if 'CUDA_ERROR_ECC_UNCORRECTABLE' in found_values}} #: This indicates that an uncorrectable ECC error was detected during #: execution. CUDA_ERROR_ECC_UNCORRECTABLE = ccuda.cudaError_enum.CUDA_ERROR_ECC_UNCORRECTABLE{{endif}} {{if 'CUDA_ERROR_UNSUPPORTED_LIMIT' in found_values}} #: This indicates that the :py:obj:`~.CUlimit` passed to the API call #: is not supported by the active device. CUDA_ERROR_UNSUPPORTED_LIMIT = ccuda.cudaError_enum.CUDA_ERROR_UNSUPPORTED_LIMIT{{endif}} {{if 'CUDA_ERROR_CONTEXT_ALREADY_IN_USE' in found_values}} #: This indicates that the :py:obj:`~.CUcontext` passed to the API call #: can only be bound to a single CPU thread at a time but is already #: bound to a CPU thread. CUDA_ERROR_CONTEXT_ALREADY_IN_USE = ccuda.cudaError_enum.CUDA_ERROR_CONTEXT_ALREADY_IN_USE{{endif}} {{if 'CUDA_ERROR_PEER_ACCESS_UNSUPPORTED' in found_values}} #: This indicates that peer access is not supported across the given #: devices. CUDA_ERROR_PEER_ACCESS_UNSUPPORTED = ccuda.cudaError_enum.CUDA_ERROR_PEER_ACCESS_UNSUPPORTED{{endif}} {{if 'CUDA_ERROR_INVALID_PTX' in found_values}} #: This indicates that a PTX JIT compilation failed. CUDA_ERROR_INVALID_PTX = ccuda.cudaError_enum.CUDA_ERROR_INVALID_PTX{{endif}} {{if 'CUDA_ERROR_INVALID_GRAPHICS_CONTEXT' in found_values}} #: This indicates an error with OpenGL or DirectX context. CUDA_ERROR_INVALID_GRAPHICS_CONTEXT = ccuda.cudaError_enum.CUDA_ERROR_INVALID_GRAPHICS_CONTEXT{{endif}} {{if 'CUDA_ERROR_NVLINK_UNCORRECTABLE' in found_values}} #: This indicates that an uncorrectable NVLink error was detected #: during the execution. CUDA_ERROR_NVLINK_UNCORRECTABLE = ccuda.cudaError_enum.CUDA_ERROR_NVLINK_UNCORRECTABLE{{endif}} {{if 'CUDA_ERROR_JIT_COMPILER_NOT_FOUND' in found_values}} #: This indicates that the PTX JIT compiler library was not found. CUDA_ERROR_JIT_COMPILER_NOT_FOUND = ccuda.cudaError_enum.CUDA_ERROR_JIT_COMPILER_NOT_FOUND{{endif}} {{if 'CUDA_ERROR_UNSUPPORTED_PTX_VERSION' in found_values}} #: This indicates that the provided PTX was compiled with an #: unsupported toolchain. CUDA_ERROR_UNSUPPORTED_PTX_VERSION = ccuda.cudaError_enum.CUDA_ERROR_UNSUPPORTED_PTX_VERSION{{endif}} {{if 'CUDA_ERROR_JIT_COMPILATION_DISABLED' in found_values}} #: This indicates that the PTX JIT compilation was disabled. CUDA_ERROR_JIT_COMPILATION_DISABLED = ccuda.cudaError_enum.CUDA_ERROR_JIT_COMPILATION_DISABLED{{endif}} {{if 'CUDA_ERROR_UNSUPPORTED_EXEC_AFFINITY' in found_values}} #: This indicates that the :py:obj:`~.CUexecAffinityType` passed to the #: API call is not supported by the active device. CUDA_ERROR_UNSUPPORTED_EXEC_AFFINITY = ccuda.cudaError_enum.CUDA_ERROR_UNSUPPORTED_EXEC_AFFINITY{{endif}} {{if 'CUDA_ERROR_UNSUPPORTED_DEVSIDE_SYNC' in found_values}} #: This indicates that the code to be compiled by the PTX JIT contains #: unsupported call to cudaDeviceSynchronize. CUDA_ERROR_UNSUPPORTED_DEVSIDE_SYNC = ccuda.cudaError_enum.CUDA_ERROR_UNSUPPORTED_DEVSIDE_SYNC{{endif}} {{if 'CUDA_ERROR_INVALID_SOURCE' in found_values}} #: This indicates that the device kernel source is invalid. This #: includes compilation/linker errors encountered in device code or #: user error. CUDA_ERROR_INVALID_SOURCE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_SOURCE{{endif}} {{if 'CUDA_ERROR_FILE_NOT_FOUND' in found_values}} #: This indicates that the file specified was not found. CUDA_ERROR_FILE_NOT_FOUND = ccuda.cudaError_enum.CUDA_ERROR_FILE_NOT_FOUND{{endif}} {{if 'CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND' in found_values}} #: This indicates that a link to a shared object failed to resolve. CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND = ccuda.cudaError_enum.CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND{{endif}} {{if 'CUDA_ERROR_SHARED_OBJECT_INIT_FAILED' in found_values}} #: This indicates that initialization of a shared object failed. CUDA_ERROR_SHARED_OBJECT_INIT_FAILED = ccuda.cudaError_enum.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED{{endif}} {{if 'CUDA_ERROR_OPERATING_SYSTEM' in found_values}} #: This indicates that an OS call failed. CUDA_ERROR_OPERATING_SYSTEM = ccuda.cudaError_enum.CUDA_ERROR_OPERATING_SYSTEM{{endif}} {{if 'CUDA_ERROR_INVALID_HANDLE' in found_values}} #: This indicates that a resource handle passed to the API call was not #: valid. Resource handles are opaque types like :py:obj:`~.CUstream` #: and :py:obj:`~.CUevent`. CUDA_ERROR_INVALID_HANDLE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_HANDLE{{endif}} {{if 'CUDA_ERROR_ILLEGAL_STATE' in found_values}} #: This indicates that a resource required by the API call is not in a #: valid state to perform the requested operation. CUDA_ERROR_ILLEGAL_STATE = ccuda.cudaError_enum.CUDA_ERROR_ILLEGAL_STATE{{endif}} {{if 'CUDA_ERROR_LOSSY_QUERY' in found_values}} #: This indicates an attempt was made to introspect an object in a way #: that would discard semantically important information. This is #: either due to the object using funtionality newer than the API #: version used to introspect it or omission of optional return #: arguments. CUDA_ERROR_LOSSY_QUERY = ccuda.cudaError_enum.CUDA_ERROR_LOSSY_QUERY{{endif}} {{if 'CUDA_ERROR_NOT_FOUND' in found_values}} #: This indicates that a named symbol was not found. Examples of #: symbols are global/constant variable names, driver function names, #: texture names, and surface names. CUDA_ERROR_NOT_FOUND = ccuda.cudaError_enum.CUDA_ERROR_NOT_FOUND{{endif}} {{if 'CUDA_ERROR_NOT_READY' in found_values}} #: This indicates that asynchronous operations issued previously have #: not completed yet. This result is not actually an error, but must be #: indicated differently than :py:obj:`~.CUDA_SUCCESS` (which indicates #: completion). Calls that may return this value include #: :py:obj:`~.cuEventQuery()` and :py:obj:`~.cuStreamQuery()`. CUDA_ERROR_NOT_READY = ccuda.cudaError_enum.CUDA_ERROR_NOT_READY{{endif}} {{if 'CUDA_ERROR_ILLEGAL_ADDRESS' in found_values}} #: While executing a kernel, the device encountered a load or store #: instruction on an invalid memory address. This leaves the process in #: an inconsistent state and any further CUDA work will return the same #: error. To continue using CUDA, the process must be terminated and #: relaunched. CUDA_ERROR_ILLEGAL_ADDRESS = ccuda.cudaError_enum.CUDA_ERROR_ILLEGAL_ADDRESS{{endif}} {{if 'CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES' in found_values}} #: This indicates that a launch did not occur because it did not have #: appropriate resources. This error usually indicates that the user #: has attempted to pass too many arguments to the device kernel, or #: the kernel launch specifies too many threads for the kernel's #: register count. Passing arguments of the wrong size (i.e. a 64-bit #: pointer when a 32-bit int is expected) is equivalent to passing too #: many arguments and can also result in this error. CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES = ccuda.cudaError_enum.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES{{endif}} {{if 'CUDA_ERROR_LAUNCH_TIMEOUT' in found_values}} #: This indicates that the device kernel took too long to execute. This #: can only occur if timeouts are enabled - see the device attribute #: :py:obj:`~.CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT` for more #: information. This leaves the process in an inconsistent state and #: any further CUDA work will return the same error. To continue using #: CUDA, the process must be terminated and relaunched. CUDA_ERROR_LAUNCH_TIMEOUT = ccuda.cudaError_enum.CUDA_ERROR_LAUNCH_TIMEOUT{{endif}} {{if 'CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING' in found_values}} #: This error indicates a kernel launch that uses an incompatible #: texturing mode. CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING = ccuda.cudaError_enum.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING{{endif}} {{if 'CUDA_ERROR_PEER_ACCESS_ALREADY_ENABLED' in found_values}} #: This error indicates that a call to #: :py:obj:`~.cuCtxEnablePeerAccess()` is trying to re-enable peer #: access to a context which has already had peer access to it enabled. CUDA_ERROR_PEER_ACCESS_ALREADY_ENABLED = ccuda.cudaError_enum.CUDA_ERROR_PEER_ACCESS_ALREADY_ENABLED{{endif}} {{if 'CUDA_ERROR_PEER_ACCESS_NOT_ENABLED' in found_values}} #: This error indicates that :py:obj:`~.cuCtxDisablePeerAccess()` is #: trying to disable peer access which has not been enabled yet via #: :py:obj:`~.cuCtxEnablePeerAccess()`. CUDA_ERROR_PEER_ACCESS_NOT_ENABLED = ccuda.cudaError_enum.CUDA_ERROR_PEER_ACCESS_NOT_ENABLED{{endif}} {{if 'CUDA_ERROR_PRIMARY_CONTEXT_ACTIVE' in found_values}} #: This error indicates that the primary context for the specified #: device has already been initialized. CUDA_ERROR_PRIMARY_CONTEXT_ACTIVE = ccuda.cudaError_enum.CUDA_ERROR_PRIMARY_CONTEXT_ACTIVE{{endif}} {{if 'CUDA_ERROR_CONTEXT_IS_DESTROYED' in found_values}} #: This error indicates that the context current to the calling thread #: has been destroyed using :py:obj:`~.cuCtxDestroy`, or is a primary #: context which has not yet been initialized. CUDA_ERROR_CONTEXT_IS_DESTROYED = ccuda.cudaError_enum.CUDA_ERROR_CONTEXT_IS_DESTROYED{{endif}} {{if 'CUDA_ERROR_ASSERT' in found_values}} #: A device-side assert triggered during kernel execution. The context #: cannot be used anymore, and must be destroyed. All existing device #: memory allocations from this context are invalid and must be #: reconstructed if the program is to continue using CUDA. CUDA_ERROR_ASSERT = ccuda.cudaError_enum.CUDA_ERROR_ASSERT{{endif}} {{if 'CUDA_ERROR_TOO_MANY_PEERS' in found_values}} #: This error indicates that the hardware resources required to enable #: peer access have been exhausted for one or more of the devices #: passed to :py:obj:`~.cuCtxEnablePeerAccess()`. CUDA_ERROR_TOO_MANY_PEERS = ccuda.cudaError_enum.CUDA_ERROR_TOO_MANY_PEERS{{endif}} {{if 'CUDA_ERROR_HOST_MEMORY_ALREADY_REGISTERED' in found_values}} #: This error indicates that the memory range passed to #: :py:obj:`~.cuMemHostRegister()` has already been registered. CUDA_ERROR_HOST_MEMORY_ALREADY_REGISTERED = ccuda.cudaError_enum.CUDA_ERROR_HOST_MEMORY_ALREADY_REGISTERED{{endif}} {{if 'CUDA_ERROR_HOST_MEMORY_NOT_REGISTERED' in found_values}} #: This error indicates that the pointer passed to #: :py:obj:`~.cuMemHostUnregister()` does not correspond to any #: currently registered memory region. CUDA_ERROR_HOST_MEMORY_NOT_REGISTERED = ccuda.cudaError_enum.CUDA_ERROR_HOST_MEMORY_NOT_REGISTERED{{endif}} {{if 'CUDA_ERROR_HARDWARE_STACK_ERROR' in found_values}} #: While executing a kernel, the device encountered a stack error. This #: can be due to stack corruption or exceeding the stack size limit. #: This leaves the process in an inconsistent state and any further #: CUDA work will return the same error. To continue using CUDA, the #: process must be terminated and relaunched. CUDA_ERROR_HARDWARE_STACK_ERROR = ccuda.cudaError_enum.CUDA_ERROR_HARDWARE_STACK_ERROR{{endif}} {{if 'CUDA_ERROR_ILLEGAL_INSTRUCTION' in found_values}} #: While executing a kernel, the device encountered an illegal #: instruction. This leaves the process in an inconsistent state and #: any further CUDA work will return the same error. To continue using #: CUDA, the process must be terminated and relaunched. CUDA_ERROR_ILLEGAL_INSTRUCTION = ccuda.cudaError_enum.CUDA_ERROR_ILLEGAL_INSTRUCTION{{endif}} {{if 'CUDA_ERROR_MISALIGNED_ADDRESS' in found_values}} #: While executing a kernel, the device encountered a load or store #: instruction on a memory address which is not aligned. This leaves #: the process in an inconsistent state and any further CUDA work will #: return the same error. To continue using CUDA, the process must be #: terminated and relaunched. CUDA_ERROR_MISALIGNED_ADDRESS = ccuda.cudaError_enum.CUDA_ERROR_MISALIGNED_ADDRESS{{endif}} {{if 'CUDA_ERROR_INVALID_ADDRESS_SPACE' in found_values}} #: While executing a kernel, the device encountered an instruction #: which can only operate on memory locations in certain address spaces #: (global, shared, or local), but was supplied a memory address not #: belonging to an allowed address space. This leaves the process in an #: inconsistent state and any further CUDA work will return the same #: error. To continue using CUDA, the process must be terminated and #: relaunched. CUDA_ERROR_INVALID_ADDRESS_SPACE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_ADDRESS_SPACE{{endif}} {{if 'CUDA_ERROR_INVALID_PC' in found_values}} #: While executing a kernel, the device program counter wrapped its #: address space. This leaves the process in an inconsistent state and #: any further CUDA work will return the same error. To continue using #: CUDA, the process must be terminated and relaunched. CUDA_ERROR_INVALID_PC = ccuda.cudaError_enum.CUDA_ERROR_INVALID_PC{{endif}} {{if 'CUDA_ERROR_LAUNCH_FAILED' in found_values}} #: An exception occurred on the device while executing a kernel. Common #: causes include dereferencing an invalid device pointer and accessing #: out of bounds shared memory. Less common cases can be system #: specific - more information about these cases can be found in the #: system specific user guide. This leaves the process in an #: inconsistent state and any further CUDA work will return the same #: error. To continue using CUDA, the process must be terminated and #: relaunched. CUDA_ERROR_LAUNCH_FAILED = ccuda.cudaError_enum.CUDA_ERROR_LAUNCH_FAILED{{endif}} {{if 'CUDA_ERROR_COOPERATIVE_LAUNCH_TOO_LARGE' in found_values}} #: This error indicates that the number of blocks launched per grid for #: a kernel that was launched via either #: :py:obj:`~.cuLaunchCooperativeKernel` or #: :py:obj:`~.cuLaunchCooperativeKernelMultiDevice` exceeds the maximum #: number of blocks as allowed by #: :py:obj:`~.cuOccupancyMaxActiveBlocksPerMultiprocessor` or #: :py:obj:`~.cuOccupancyMaxActiveBlocksPerMultiprocessorWithFlags` #: times the number of multiprocessors as specified by the device #: attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT`. CUDA_ERROR_COOPERATIVE_LAUNCH_TOO_LARGE = ccuda.cudaError_enum.CUDA_ERROR_COOPERATIVE_LAUNCH_TOO_LARGE{{endif}} {{if 'CUDA_ERROR_NOT_PERMITTED' in found_values}} #: This error indicates that the attempted operation is not permitted. CUDA_ERROR_NOT_PERMITTED = ccuda.cudaError_enum.CUDA_ERROR_NOT_PERMITTED{{endif}} {{if 'CUDA_ERROR_NOT_SUPPORTED' in found_values}} #: This error indicates that the attempted operation is not supported #: on the current system or device. CUDA_ERROR_NOT_SUPPORTED = ccuda.cudaError_enum.CUDA_ERROR_NOT_SUPPORTED{{endif}} {{if 'CUDA_ERROR_SYSTEM_NOT_READY' in found_values}} #: This error indicates that the system is not yet ready to start any #: CUDA work. To continue using CUDA, verify the system configuration #: is in a valid state and all required driver daemons are actively #: running. More information about this error can be found in the #: system specific user guide. CUDA_ERROR_SYSTEM_NOT_READY = ccuda.cudaError_enum.CUDA_ERROR_SYSTEM_NOT_READY{{endif}} {{if 'CUDA_ERROR_SYSTEM_DRIVER_MISMATCH' in found_values}} #: This error indicates that there is a mismatch between the versions #: of the display driver and the CUDA driver. Refer to the #: compatibility documentation for supported versions. CUDA_ERROR_SYSTEM_DRIVER_MISMATCH = ccuda.cudaError_enum.CUDA_ERROR_SYSTEM_DRIVER_MISMATCH{{endif}} {{if 'CUDA_ERROR_COMPAT_NOT_SUPPORTED_ON_DEVICE' in found_values}} #: This error indicates that the system was upgraded to run with #: forward compatibility but the visible hardware detected by CUDA does #: not support this configuration. Refer to the compatibility #: documentation for the supported hardware matrix or ensure that only #: supported hardware is visible during initialization via the #: CUDA_VISIBLE_DEVICES environment variable. CUDA_ERROR_COMPAT_NOT_SUPPORTED_ON_DEVICE = ccuda.cudaError_enum.CUDA_ERROR_COMPAT_NOT_SUPPORTED_ON_DEVICE{{endif}} {{if 'CUDA_ERROR_MPS_CONNECTION_FAILED' in found_values}} #: This error indicates that the MPS client failed to connect to the #: MPS control daemon or the MPS server. CUDA_ERROR_MPS_CONNECTION_FAILED = ccuda.cudaError_enum.CUDA_ERROR_MPS_CONNECTION_FAILED{{endif}} {{if 'CUDA_ERROR_MPS_RPC_FAILURE' in found_values}} #: This error indicates that the remote procedural call between the MPS #: server and the MPS client failed. CUDA_ERROR_MPS_RPC_FAILURE = ccuda.cudaError_enum.CUDA_ERROR_MPS_RPC_FAILURE{{endif}} {{if 'CUDA_ERROR_MPS_SERVER_NOT_READY' in found_values}} #: This error indicates that the MPS server is not ready to accept new #: MPS client requests. This error can be returned when the MPS server #: is in the process of recovering from a fatal failure. CUDA_ERROR_MPS_SERVER_NOT_READY = ccuda.cudaError_enum.CUDA_ERROR_MPS_SERVER_NOT_READY{{endif}} {{if 'CUDA_ERROR_MPS_MAX_CLIENTS_REACHED' in found_values}} #: This error indicates that the hardware resources required to create #: MPS client have been exhausted. CUDA_ERROR_MPS_MAX_CLIENTS_REACHED = ccuda.cudaError_enum.CUDA_ERROR_MPS_MAX_CLIENTS_REACHED{{endif}} {{if 'CUDA_ERROR_MPS_MAX_CONNECTIONS_REACHED' in found_values}} #: This error indicates the the hardware resources required to support #: device connections have been exhausted. CUDA_ERROR_MPS_MAX_CONNECTIONS_REACHED = ccuda.cudaError_enum.CUDA_ERROR_MPS_MAX_CONNECTIONS_REACHED{{endif}} {{if 'CUDA_ERROR_MPS_CLIENT_TERMINATED' in found_values}} #: This error indicates that the MPS client has been terminated by the #: server. To continue using CUDA, the process must be terminated and #: relaunched. CUDA_ERROR_MPS_CLIENT_TERMINATED = ccuda.cudaError_enum.CUDA_ERROR_MPS_CLIENT_TERMINATED{{endif}} {{if 'CUDA_ERROR_CDP_NOT_SUPPORTED' in found_values}} #: This error indicates that the module is using CUDA Dynamic #: Parallelism, but the current configuration, like MPS, does not #: support it. CUDA_ERROR_CDP_NOT_SUPPORTED = ccuda.cudaError_enum.CUDA_ERROR_CDP_NOT_SUPPORTED{{endif}} {{if 'CUDA_ERROR_CDP_VERSION_MISMATCH' in found_values}} #: This error indicates that a module contains an unsupported #: interaction between different versions of CUDA Dynamic Parallelism. CUDA_ERROR_CDP_VERSION_MISMATCH = ccuda.cudaError_enum.CUDA_ERROR_CDP_VERSION_MISMATCH{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED' in found_values}} #: This error indicates that the operation is not permitted when the #: stream is capturing. CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_INVALIDATED' in found_values}} #: This error indicates that the current capture sequence on the stream #: has been invalidated due to a previous error. CUDA_ERROR_STREAM_CAPTURE_INVALIDATED = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_INVALIDATED{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_MERGE' in found_values}} #: This error indicates that the operation would have resulted in a #: merge of two independent capture sequences. CUDA_ERROR_STREAM_CAPTURE_MERGE = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_MERGE{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_UNMATCHED' in found_values}} #: This error indicates that the capture was not initiated in this #: stream. CUDA_ERROR_STREAM_CAPTURE_UNMATCHED = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_UNMATCHED{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_UNJOINED' in found_values}} #: This error indicates that the capture sequence contains a fork that #: was not joined to the primary stream. CUDA_ERROR_STREAM_CAPTURE_UNJOINED = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_UNJOINED{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_ISOLATION' in found_values}} #: This error indicates that a dependency would have been created which #: crosses the capture sequence boundary. Only implicit in-stream #: ordering dependencies are allowed to cross the boundary. CUDA_ERROR_STREAM_CAPTURE_ISOLATION = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_ISOLATION{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_IMPLICIT' in found_values}} #: This error indicates a disallowed implicit dependency on a current #: capture sequence from cudaStreamLegacy. CUDA_ERROR_STREAM_CAPTURE_IMPLICIT = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_IMPLICIT{{endif}} {{if 'CUDA_ERROR_CAPTURED_EVENT' in found_values}} #: This error indicates that the operation is not permitted on an event #: which was last recorded in a capturing stream. CUDA_ERROR_CAPTURED_EVENT = ccuda.cudaError_enum.CUDA_ERROR_CAPTURED_EVENT{{endif}} {{if 'CUDA_ERROR_STREAM_CAPTURE_WRONG_THREAD' in found_values}} #: A stream capture sequence not initiated with the #: :py:obj:`~.CU_STREAM_CAPTURE_MODE_RELAXED` argument to #: :py:obj:`~.cuStreamBeginCapture` was passed to #: :py:obj:`~.cuStreamEndCapture` in a different thread. CUDA_ERROR_STREAM_CAPTURE_WRONG_THREAD = ccuda.cudaError_enum.CUDA_ERROR_STREAM_CAPTURE_WRONG_THREAD{{endif}} {{if 'CUDA_ERROR_TIMEOUT' in found_values}} #: This error indicates that the timeout specified for the wait #: operation has lapsed. CUDA_ERROR_TIMEOUT = ccuda.cudaError_enum.CUDA_ERROR_TIMEOUT{{endif}} {{if 'CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE' in found_values}} #: This error indicates that the graph update was not performed because #: it included changes which violated constraints specific to #: instantiated graph update. CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE = ccuda.cudaError_enum.CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE{{endif}} {{if 'CUDA_ERROR_EXTERNAL_DEVICE' in found_values}} #: This indicates that an async error has occurred in a device outside #: of CUDA. If CUDA was waiting for an external device's signal before #: consuming shared data, the external device signaled an error #: indicating that the data is not valid for consumption. This leaves #: the process in an inconsistent state and any further CUDA work will #: return the same error. To continue using CUDA, the process must be #: terminated and relaunched. CUDA_ERROR_EXTERNAL_DEVICE = ccuda.cudaError_enum.CUDA_ERROR_EXTERNAL_DEVICE{{endif}} {{if 'CUDA_ERROR_INVALID_CLUSTER_SIZE' in found_values}} #: Indicates a kernel launch error due to cluster misconfiguration. CUDA_ERROR_INVALID_CLUSTER_SIZE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_CLUSTER_SIZE{{endif}} {{if 'CUDA_ERROR_FUNCTION_NOT_LOADED' in found_values}} #: Indiciates a function handle is not loaded when calling an API that #: requires a loaded function. CUDA_ERROR_FUNCTION_NOT_LOADED = ccuda.cudaError_enum.CUDA_ERROR_FUNCTION_NOT_LOADED{{endif}} {{if 'CUDA_ERROR_INVALID_RESOURCE_TYPE' in found_values}} #: This error indicates one or more resources passed in are not valid #: resource types for the operation. CUDA_ERROR_INVALID_RESOURCE_TYPE = ccuda.cudaError_enum.CUDA_ERROR_INVALID_RESOURCE_TYPE{{endif}} {{if 'CUDA_ERROR_INVALID_RESOURCE_CONFIGURATION' in found_values}} #: This error indicates one or more resources are insufficient or non- #: applicable for the operation. CUDA_ERROR_INVALID_RESOURCE_CONFIGURATION = ccuda.cudaError_enum.CUDA_ERROR_INVALID_RESOURCE_CONFIGURATION{{endif}} {{if 'CUDA_ERROR_UNKNOWN' in found_values}} #: This indicates that an unknown internal error has occurred. CUDA_ERROR_UNKNOWN = ccuda.cudaError_enum.CUDA_ERROR_UNKNOWN{{endif}} {{endif}} {{if 'CUdevice_P2PAttribute_enum' in found_types}} class CUdevice_P2PAttribute(IntEnum): """ P2P Attributes """ {{if 'CU_DEVICE_P2P_ATTRIBUTE_PERFORMANCE_RANK' in found_values}} #: A relative value indicating the performance of the link between two #: devices CU_DEVICE_P2P_ATTRIBUTE_PERFORMANCE_RANK = ccuda.CUdevice_P2PAttribute_enum.CU_DEVICE_P2P_ATTRIBUTE_PERFORMANCE_RANK{{endif}} {{if 'CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED' in found_values}} #: P2P Access is enable CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED = ccuda.CUdevice_P2PAttribute_enum.CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED{{endif}} {{if 'CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED' in found_values}} #: Atomic operation over the link supported CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED = ccuda.CUdevice_P2PAttribute_enum.CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED{{endif}} {{if 'CU_DEVICE_P2P_ATTRIBUTE_ACCESS_ACCESS_SUPPORTED' in found_values}} #: [Deprecated] CU_DEVICE_P2P_ATTRIBUTE_ACCESS_ACCESS_SUPPORTED = ccuda.CUdevice_P2PAttribute_enum.CU_DEVICE_P2P_ATTRIBUTE_ACCESS_ACCESS_SUPPORTED{{endif}} {{if 'CU_DEVICE_P2P_ATTRIBUTE_CUDA_ARRAY_ACCESS_SUPPORTED' in found_values}} #: Accessing CUDA arrays over the link supported CU_DEVICE_P2P_ATTRIBUTE_CUDA_ARRAY_ACCESS_SUPPORTED = ccuda.CUdevice_P2PAttribute_enum.CU_DEVICE_P2P_ATTRIBUTE_CUDA_ARRAY_ACCESS_SUPPORTED{{endif}} {{endif}} {{if 'CUresourceViewFormat_enum' in found_types}} class CUresourceViewFormat(IntEnum): """ Resource view format """ {{if 'CU_RES_VIEW_FORMAT_NONE' in found_values}} #: No resource view format (use underlying resource format) CU_RES_VIEW_FORMAT_NONE = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_NONE{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_1X8' in found_values}} #: 1 channel unsigned 8-bit integers CU_RES_VIEW_FORMAT_UINT_1X8 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_1X8{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_2X8' in found_values}} #: 2 channel unsigned 8-bit integers CU_RES_VIEW_FORMAT_UINT_2X8 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_2X8{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_4X8' in found_values}} #: 4 channel unsigned 8-bit integers CU_RES_VIEW_FORMAT_UINT_4X8 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_4X8{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_1X8' in found_values}} #: 1 channel signed 8-bit integers CU_RES_VIEW_FORMAT_SINT_1X8 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_1X8{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_2X8' in found_values}} #: 2 channel signed 8-bit integers CU_RES_VIEW_FORMAT_SINT_2X8 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_2X8{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_4X8' in found_values}} #: 4 channel signed 8-bit integers CU_RES_VIEW_FORMAT_SINT_4X8 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_4X8{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_1X16' in found_values}} #: 1 channel unsigned 16-bit integers CU_RES_VIEW_FORMAT_UINT_1X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_1X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_2X16' in found_values}} #: 2 channel unsigned 16-bit integers CU_RES_VIEW_FORMAT_UINT_2X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_2X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_4X16' in found_values}} #: 4 channel unsigned 16-bit integers CU_RES_VIEW_FORMAT_UINT_4X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_4X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_1X16' in found_values}} #: 1 channel signed 16-bit integers CU_RES_VIEW_FORMAT_SINT_1X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_1X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_2X16' in found_values}} #: 2 channel signed 16-bit integers CU_RES_VIEW_FORMAT_SINT_2X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_2X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_4X16' in found_values}} #: 4 channel signed 16-bit integers CU_RES_VIEW_FORMAT_SINT_4X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_4X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_1X32' in found_values}} #: 1 channel unsigned 32-bit integers CU_RES_VIEW_FORMAT_UINT_1X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_1X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_2X32' in found_values}} #: 2 channel unsigned 32-bit integers CU_RES_VIEW_FORMAT_UINT_2X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_2X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_UINT_4X32' in found_values}} #: 4 channel unsigned 32-bit integers CU_RES_VIEW_FORMAT_UINT_4X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UINT_4X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_1X32' in found_values}} #: 1 channel signed 32-bit integers CU_RES_VIEW_FORMAT_SINT_1X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_1X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_2X32' in found_values}} #: 2 channel signed 32-bit integers CU_RES_VIEW_FORMAT_SINT_2X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_2X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_SINT_4X32' in found_values}} #: 4 channel signed 32-bit integers CU_RES_VIEW_FORMAT_SINT_4X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SINT_4X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_FLOAT_1X16' in found_values}} #: 1 channel 16-bit floating point CU_RES_VIEW_FORMAT_FLOAT_1X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_FLOAT_1X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_FLOAT_2X16' in found_values}} #: 2 channel 16-bit floating point CU_RES_VIEW_FORMAT_FLOAT_2X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_FLOAT_2X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_FLOAT_4X16' in found_values}} #: 4 channel 16-bit floating point CU_RES_VIEW_FORMAT_FLOAT_4X16 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_FLOAT_4X16{{endif}} {{if 'CU_RES_VIEW_FORMAT_FLOAT_1X32' in found_values}} #: 1 channel 32-bit floating point CU_RES_VIEW_FORMAT_FLOAT_1X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_FLOAT_1X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_FLOAT_2X32' in found_values}} #: 2 channel 32-bit floating point CU_RES_VIEW_FORMAT_FLOAT_2X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_FLOAT_2X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_FLOAT_4X32' in found_values}} #: 4 channel 32-bit floating point CU_RES_VIEW_FORMAT_FLOAT_4X32 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_FLOAT_4X32{{endif}} {{if 'CU_RES_VIEW_FORMAT_UNSIGNED_BC1' in found_values}} #: Block compressed 1 CU_RES_VIEW_FORMAT_UNSIGNED_BC1 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UNSIGNED_BC1{{endif}} {{if 'CU_RES_VIEW_FORMAT_UNSIGNED_BC2' in found_values}} #: Block compressed 2 CU_RES_VIEW_FORMAT_UNSIGNED_BC2 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UNSIGNED_BC2{{endif}} {{if 'CU_RES_VIEW_FORMAT_UNSIGNED_BC3' in found_values}} #: Block compressed 3 CU_RES_VIEW_FORMAT_UNSIGNED_BC3 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UNSIGNED_BC3{{endif}} {{if 'CU_RES_VIEW_FORMAT_UNSIGNED_BC4' in found_values}} #: Block compressed 4 unsigned CU_RES_VIEW_FORMAT_UNSIGNED_BC4 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UNSIGNED_BC4{{endif}} {{if 'CU_RES_VIEW_FORMAT_SIGNED_BC4' in found_values}} #: Block compressed 4 signed CU_RES_VIEW_FORMAT_SIGNED_BC4 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SIGNED_BC4{{endif}} {{if 'CU_RES_VIEW_FORMAT_UNSIGNED_BC5' in found_values}} #: Block compressed 5 unsigned CU_RES_VIEW_FORMAT_UNSIGNED_BC5 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UNSIGNED_BC5{{endif}} {{if 'CU_RES_VIEW_FORMAT_SIGNED_BC5' in found_values}} #: Block compressed 5 signed CU_RES_VIEW_FORMAT_SIGNED_BC5 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SIGNED_BC5{{endif}} {{if 'CU_RES_VIEW_FORMAT_UNSIGNED_BC6H' in found_values}} #: Block compressed 6 unsigned half-float CU_RES_VIEW_FORMAT_UNSIGNED_BC6H = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UNSIGNED_BC6H{{endif}} {{if 'CU_RES_VIEW_FORMAT_SIGNED_BC6H' in found_values}} #: Block compressed 6 signed half-float CU_RES_VIEW_FORMAT_SIGNED_BC6H = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_SIGNED_BC6H{{endif}} {{if 'CU_RES_VIEW_FORMAT_UNSIGNED_BC7' in found_values}} #: Block compressed 7 CU_RES_VIEW_FORMAT_UNSIGNED_BC7 = ccuda.CUresourceViewFormat_enum.CU_RES_VIEW_FORMAT_UNSIGNED_BC7{{endif}} {{endif}} {{if 'CUtensorMapDataType_enum' in found_types}} class CUtensorMapDataType(IntEnum): """ Tensor map data type """ {{if 'CU_TENSOR_MAP_DATA_TYPE_UINT8' in found_values}} CU_TENSOR_MAP_DATA_TYPE_UINT8 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_UINT8{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_UINT16' in found_values}} CU_TENSOR_MAP_DATA_TYPE_UINT16 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_UINT16{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_UINT32' in found_values}} CU_TENSOR_MAP_DATA_TYPE_UINT32 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_UINT32{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_INT32' in found_values}} CU_TENSOR_MAP_DATA_TYPE_INT32 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_INT32{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_UINT64' in found_values}} CU_TENSOR_MAP_DATA_TYPE_UINT64 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_UINT64{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_INT64' in found_values}} CU_TENSOR_MAP_DATA_TYPE_INT64 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_INT64{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_FLOAT16' in found_values}} CU_TENSOR_MAP_DATA_TYPE_FLOAT16 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_FLOAT16{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_FLOAT32' in found_values}} CU_TENSOR_MAP_DATA_TYPE_FLOAT32 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_FLOAT32{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_FLOAT64' in found_values}} CU_TENSOR_MAP_DATA_TYPE_FLOAT64 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_FLOAT64{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_BFLOAT16' in found_values}} CU_TENSOR_MAP_DATA_TYPE_BFLOAT16 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_BFLOAT16{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_FLOAT32_FTZ' in found_values}} CU_TENSOR_MAP_DATA_TYPE_FLOAT32_FTZ = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_FLOAT32_FTZ{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_TFLOAT32' in found_values}} CU_TENSOR_MAP_DATA_TYPE_TFLOAT32 = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_TFLOAT32{{endif}} {{if 'CU_TENSOR_MAP_DATA_TYPE_TFLOAT32_FTZ' in found_values}} CU_TENSOR_MAP_DATA_TYPE_TFLOAT32_FTZ = ccuda.CUtensorMapDataType_enum.CU_TENSOR_MAP_DATA_TYPE_TFLOAT32_FTZ{{endif}} {{endif}} {{if 'CUtensorMapInterleave_enum' in found_types}} class CUtensorMapInterleave(IntEnum): """ Tensor map interleave layout type """ {{if 'CU_TENSOR_MAP_INTERLEAVE_NONE' in found_values}} CU_TENSOR_MAP_INTERLEAVE_NONE = ccuda.CUtensorMapInterleave_enum.CU_TENSOR_MAP_INTERLEAVE_NONE{{endif}} {{if 'CU_TENSOR_MAP_INTERLEAVE_16B' in found_values}} CU_TENSOR_MAP_INTERLEAVE_16B = ccuda.CUtensorMapInterleave_enum.CU_TENSOR_MAP_INTERLEAVE_16B{{endif}} {{if 'CU_TENSOR_MAP_INTERLEAVE_32B' in found_values}} CU_TENSOR_MAP_INTERLEAVE_32B = ccuda.CUtensorMapInterleave_enum.CU_TENSOR_MAP_INTERLEAVE_32B{{endif}} {{endif}} {{if 'CUtensorMapSwizzle_enum' in found_types}} class CUtensorMapSwizzle(IntEnum): """ Tensor map swizzling mode of shared memory banks """ {{if 'CU_TENSOR_MAP_SWIZZLE_NONE' in found_values}} CU_TENSOR_MAP_SWIZZLE_NONE = ccuda.CUtensorMapSwizzle_enum.CU_TENSOR_MAP_SWIZZLE_NONE{{endif}} {{if 'CU_TENSOR_MAP_SWIZZLE_32B' in found_values}} CU_TENSOR_MAP_SWIZZLE_32B = ccuda.CUtensorMapSwizzle_enum.CU_TENSOR_MAP_SWIZZLE_32B{{endif}} {{if 'CU_TENSOR_MAP_SWIZZLE_64B' in found_values}} CU_TENSOR_MAP_SWIZZLE_64B = ccuda.CUtensorMapSwizzle_enum.CU_TENSOR_MAP_SWIZZLE_64B{{endif}} {{if 'CU_TENSOR_MAP_SWIZZLE_128B' in found_values}} CU_TENSOR_MAP_SWIZZLE_128B = ccuda.CUtensorMapSwizzle_enum.CU_TENSOR_MAP_SWIZZLE_128B{{endif}} {{endif}} {{if 'CUtensorMapL2promotion_enum' in found_types}} class CUtensorMapL2promotion(IntEnum): """ Tensor map L2 promotion type """ {{if 'CU_TENSOR_MAP_L2_PROMOTION_NONE' in found_values}} CU_TENSOR_MAP_L2_PROMOTION_NONE = ccuda.CUtensorMapL2promotion_enum.CU_TENSOR_MAP_L2_PROMOTION_NONE{{endif}} {{if 'CU_TENSOR_MAP_L2_PROMOTION_L2_64B' in found_values}} CU_TENSOR_MAP_L2_PROMOTION_L2_64B = ccuda.CUtensorMapL2promotion_enum.CU_TENSOR_MAP_L2_PROMOTION_L2_64B{{endif}} {{if 'CU_TENSOR_MAP_L2_PROMOTION_L2_128B' in found_values}} CU_TENSOR_MAP_L2_PROMOTION_L2_128B = ccuda.CUtensorMapL2promotion_enum.CU_TENSOR_MAP_L2_PROMOTION_L2_128B{{endif}} {{if 'CU_TENSOR_MAP_L2_PROMOTION_L2_256B' in found_values}} CU_TENSOR_MAP_L2_PROMOTION_L2_256B = ccuda.CUtensorMapL2promotion_enum.CU_TENSOR_MAP_L2_PROMOTION_L2_256B{{endif}} {{endif}} {{if 'CUtensorMapFloatOOBfill_enum' in found_types}} class CUtensorMapFloatOOBfill(IntEnum): """ Tensor map out-of-bounds fill type """ {{if 'CU_TENSOR_MAP_FLOAT_OOB_FILL_NONE' in found_values}} CU_TENSOR_MAP_FLOAT_OOB_FILL_NONE = ccuda.CUtensorMapFloatOOBfill_enum.CU_TENSOR_MAP_FLOAT_OOB_FILL_NONE{{endif}} {{if 'CU_TENSOR_MAP_FLOAT_OOB_FILL_NAN_REQUEST_ZERO_FMA' in found_values}} CU_TENSOR_MAP_FLOAT_OOB_FILL_NAN_REQUEST_ZERO_FMA = ccuda.CUtensorMapFloatOOBfill_enum.CU_TENSOR_MAP_FLOAT_OOB_FILL_NAN_REQUEST_ZERO_FMA{{endif}} {{endif}} {{if 'CUDA_POINTER_ATTRIBUTE_ACCESS_FLAGS_enum' in found_types}} class CUDA_POINTER_ATTRIBUTE_ACCESS_FLAGS(IntEnum): """ Access flags that specify the level of access the current context's device has on the memory referenced. """ {{if 'CU_POINTER_ATTRIBUTE_ACCESS_FLAG_NONE' in found_values}} #: No access, meaning the device cannot access this memory at all, thus #: must be staged through accessible memory in order to complete #: certain operations CU_POINTER_ATTRIBUTE_ACCESS_FLAG_NONE = ccuda.CUDA_POINTER_ATTRIBUTE_ACCESS_FLAGS_enum.CU_POINTER_ATTRIBUTE_ACCESS_FLAG_NONE{{endif}} {{if 'CU_POINTER_ATTRIBUTE_ACCESS_FLAG_READ' in found_values}} #: Read-only access, meaning writes to this memory are considered #: invalid accesses and thus return error in that case. CU_POINTER_ATTRIBUTE_ACCESS_FLAG_READ = ccuda.CUDA_POINTER_ATTRIBUTE_ACCESS_FLAGS_enum.CU_POINTER_ATTRIBUTE_ACCESS_FLAG_READ{{endif}} {{if 'CU_POINTER_ATTRIBUTE_ACCESS_FLAG_READWRITE' in found_values}} #: Read-write access, the device has full read-write access to the #: memory CU_POINTER_ATTRIBUTE_ACCESS_FLAG_READWRITE = ccuda.CUDA_POINTER_ATTRIBUTE_ACCESS_FLAGS_enum.CU_POINTER_ATTRIBUTE_ACCESS_FLAG_READWRITE{{endif}} {{endif}} {{if 'CUexternalMemoryHandleType_enum' in found_types}} class CUexternalMemoryHandleType(IntEnum): """ External memory handle types """ {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_FD' in found_values}} #: Handle is an opaque file descriptor CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_FD = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_FD{{endif}} {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32' in found_values}} #: Handle is an opaque shared NT handle CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32 = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32{{endif}} {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32_KMT' in found_values}} #: Handle is an opaque, globally shared handle CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32_KMT = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32_KMT{{endif}} {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_HEAP' in found_values}} #: Handle is a D3D12 heap object CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_HEAP = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_HEAP{{endif}} {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_RESOURCE' in found_values}} #: Handle is a D3D12 committed resource CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_RESOURCE = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_RESOURCE{{endif}} {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE' in found_values}} #: Handle is a shared NT handle to a D3D11 resource CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE{{endif}} {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE_KMT' in found_values}} #: Handle is a globally shared handle to a D3D11 resource CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE_KMT = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE_KMT{{endif}} {{if 'CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF' in found_values}} #: Handle is an NvSciBuf object CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF = ccuda.CUexternalMemoryHandleType_enum.CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF{{endif}} {{endif}} {{if 'CUexternalSemaphoreHandleType_enum' in found_types}} class CUexternalSemaphoreHandleType(IntEnum): """ External semaphore handle types """ {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_FD' in found_values}} #: Handle is an opaque file descriptor CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_FD = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_FD{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32' in found_values}} #: Handle is an opaque shared NT handle CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32 = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32_KMT' in found_values}} #: Handle is an opaque, globally shared handle CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32_KMT = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32_KMT{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D12_FENCE' in found_values}} #: Handle is a shared NT handle referencing a D3D12 fence object CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D12_FENCE = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D12_FENCE{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_FENCE' in found_values}} #: Handle is a shared NT handle referencing a D3D11 fence object CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_FENCE = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_FENCE{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC' in found_values}} #: Opaque handle to NvSciSync Object CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX' in found_values}} #: Handle is a shared NT handle referencing a D3D11 keyed mutex object CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX_KMT' in found_values}} #: Handle is a globally shared handle referencing a D3D11 keyed mutex #: object CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX_KMT = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX_KMT{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_FD' in found_values}} #: Handle is an opaque file descriptor referencing a timeline semaphore CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_FD = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_FD{{endif}} {{if 'CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_WIN32' in found_values}} #: Handle is an opaque shared NT handle referencing a timeline #: semaphore CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_WIN32 = ccuda.CUexternalSemaphoreHandleType_enum.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_WIN32{{endif}} {{endif}} {{if 'CUmemAllocationHandleType_enum' in found_types}} class CUmemAllocationHandleType(IntEnum): """ Flags for specifying particular handle types """ {{if 'CU_MEM_HANDLE_TYPE_NONE' in found_values}} #: Does not allow any export mechanism. > CU_MEM_HANDLE_TYPE_NONE = ccuda.CUmemAllocationHandleType_enum.CU_MEM_HANDLE_TYPE_NONE{{endif}} {{if 'CU_MEM_HANDLE_TYPE_POSIX_FILE_DESCRIPTOR' in found_values}} #: Allows a file descriptor to be used for exporting. Permitted only on #: POSIX systems. (int) CU_MEM_HANDLE_TYPE_POSIX_FILE_DESCRIPTOR = ccuda.CUmemAllocationHandleType_enum.CU_MEM_HANDLE_TYPE_POSIX_FILE_DESCRIPTOR{{endif}} {{if 'CU_MEM_HANDLE_TYPE_WIN32' in found_values}} #: Allows a Win32 NT handle to be used for exporting. (HANDLE) CU_MEM_HANDLE_TYPE_WIN32 = ccuda.CUmemAllocationHandleType_enum.CU_MEM_HANDLE_TYPE_WIN32{{endif}} {{if 'CU_MEM_HANDLE_TYPE_WIN32_KMT' in found_values}} #: Allows a Win32 KMT handle to be used for exporting. (D3DKMT_HANDLE) CU_MEM_HANDLE_TYPE_WIN32_KMT = ccuda.CUmemAllocationHandleType_enum.CU_MEM_HANDLE_TYPE_WIN32_KMT{{endif}} {{if 'CU_MEM_HANDLE_TYPE_FABRIC' in found_values}} #: Allows a fabric handle to be used for exporting. (CUmemFabricHandle) CU_MEM_HANDLE_TYPE_FABRIC = ccuda.CUmemAllocationHandleType_enum.CU_MEM_HANDLE_TYPE_FABRIC{{endif}} {{if 'CU_MEM_HANDLE_TYPE_MAX' in found_values}} CU_MEM_HANDLE_TYPE_MAX = ccuda.CUmemAllocationHandleType_enum.CU_MEM_HANDLE_TYPE_MAX{{endif}} {{endif}} {{if 'CUmemAccess_flags_enum' in found_types}} class CUmemAccess_flags(IntEnum): """ Specifies the memory protection flags for mapping. """ {{if 'CU_MEM_ACCESS_FLAGS_PROT_NONE' in found_values}} #: Default, make the address range not accessible CU_MEM_ACCESS_FLAGS_PROT_NONE = ccuda.CUmemAccess_flags_enum.CU_MEM_ACCESS_FLAGS_PROT_NONE{{endif}} {{if 'CU_MEM_ACCESS_FLAGS_PROT_READ' in found_values}} #: Make the address range read accessible CU_MEM_ACCESS_FLAGS_PROT_READ = ccuda.CUmemAccess_flags_enum.CU_MEM_ACCESS_FLAGS_PROT_READ{{endif}} {{if 'CU_MEM_ACCESS_FLAGS_PROT_READWRITE' in found_values}} #: Make the address range read-write accessible CU_MEM_ACCESS_FLAGS_PROT_READWRITE = ccuda.CUmemAccess_flags_enum.CU_MEM_ACCESS_FLAGS_PROT_READWRITE{{endif}} {{if 'CU_MEM_ACCESS_FLAGS_PROT_MAX' in found_values}} CU_MEM_ACCESS_FLAGS_PROT_MAX = ccuda.CUmemAccess_flags_enum.CU_MEM_ACCESS_FLAGS_PROT_MAX{{endif}} {{endif}} {{if 'CUmemLocationType_enum' in found_types}} class CUmemLocationType(IntEnum): """ Specifies the type of location """ {{if 'CU_MEM_LOCATION_TYPE_INVALID' in found_values}} CU_MEM_LOCATION_TYPE_INVALID = ccuda.CUmemLocationType_enum.CU_MEM_LOCATION_TYPE_INVALID{{endif}} {{if 'CU_MEM_LOCATION_TYPE_DEVICE' in found_values}} #: Location is a device location, thus id is a device ordinal CU_MEM_LOCATION_TYPE_DEVICE = ccuda.CUmemLocationType_enum.CU_MEM_LOCATION_TYPE_DEVICE{{endif}} {{if 'CU_MEM_LOCATION_TYPE_HOST' in found_values}} #: Location is host, id is ignored CU_MEM_LOCATION_TYPE_HOST = ccuda.CUmemLocationType_enum.CU_MEM_LOCATION_TYPE_HOST{{endif}} {{if 'CU_MEM_LOCATION_TYPE_HOST_NUMA' in found_values}} #: Location is a host NUMA node, thus id is a host NUMA node id CU_MEM_LOCATION_TYPE_HOST_NUMA = ccuda.CUmemLocationType_enum.CU_MEM_LOCATION_TYPE_HOST_NUMA{{endif}} {{if 'CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT' in found_values}} #: Location is a host NUMA node of the current thread, id is ignored CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT = ccuda.CUmemLocationType_enum.CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT{{endif}} {{if 'CU_MEM_LOCATION_TYPE_MAX' in found_values}} CU_MEM_LOCATION_TYPE_MAX = ccuda.CUmemLocationType_enum.CU_MEM_LOCATION_TYPE_MAX{{endif}} {{endif}} {{if 'CUmemAllocationType_enum' in found_types}} class CUmemAllocationType(IntEnum): """ Defines the allocation types available """ {{if 'CU_MEM_ALLOCATION_TYPE_INVALID' in found_values}} CU_MEM_ALLOCATION_TYPE_INVALID = ccuda.CUmemAllocationType_enum.CU_MEM_ALLOCATION_TYPE_INVALID{{endif}} {{if 'CU_MEM_ALLOCATION_TYPE_PINNED' in found_values}} #: This allocation type is 'pinned', i.e. cannot migrate from its #: current location while the application is actively using it CU_MEM_ALLOCATION_TYPE_PINNED = ccuda.CUmemAllocationType_enum.CU_MEM_ALLOCATION_TYPE_PINNED{{endif}} {{if 'CU_MEM_ALLOCATION_TYPE_MAX' in found_values}} CU_MEM_ALLOCATION_TYPE_MAX = ccuda.CUmemAllocationType_enum.CU_MEM_ALLOCATION_TYPE_MAX{{endif}} {{endif}} {{if 'CUmemAllocationGranularity_flags_enum' in found_types}} class CUmemAllocationGranularity_flags(IntEnum): """ Flag for requesting different optimal and required granularities for an allocation. """ {{if 'CU_MEM_ALLOC_GRANULARITY_MINIMUM' in found_values}} #: Minimum required granularity for allocation CU_MEM_ALLOC_GRANULARITY_MINIMUM = ccuda.CUmemAllocationGranularity_flags_enum.CU_MEM_ALLOC_GRANULARITY_MINIMUM{{endif}} {{if 'CU_MEM_ALLOC_GRANULARITY_RECOMMENDED' in found_values}} #: Recommended granularity for allocation for best performance CU_MEM_ALLOC_GRANULARITY_RECOMMENDED = ccuda.CUmemAllocationGranularity_flags_enum.CU_MEM_ALLOC_GRANULARITY_RECOMMENDED{{endif}} {{endif}} {{if 'CUmemRangeHandleType_enum' in found_types}} class CUmemRangeHandleType(IntEnum): """ Specifies the handle type for address range """ {{if 'CU_MEM_RANGE_HANDLE_TYPE_DMA_BUF_FD' in found_values}} CU_MEM_RANGE_HANDLE_TYPE_DMA_BUF_FD = ccuda.CUmemRangeHandleType_enum.CU_MEM_RANGE_HANDLE_TYPE_DMA_BUF_FD{{endif}} {{if 'CU_MEM_RANGE_HANDLE_TYPE_MAX' in found_values}} CU_MEM_RANGE_HANDLE_TYPE_MAX = ccuda.CUmemRangeHandleType_enum.CU_MEM_RANGE_HANDLE_TYPE_MAX{{endif}} {{endif}} {{if 'CUarraySparseSubresourceType_enum' in found_types}} class CUarraySparseSubresourceType(IntEnum): """ Sparse subresource types """ {{if 'CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_SPARSE_LEVEL' in found_values}} CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_SPARSE_LEVEL = ccuda.CUarraySparseSubresourceType_enum.CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_SPARSE_LEVEL{{endif}} {{if 'CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_MIPTAIL' in found_values}} CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_MIPTAIL = ccuda.CUarraySparseSubresourceType_enum.CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_MIPTAIL{{endif}} {{endif}} {{if 'CUmemOperationType_enum' in found_types}} class CUmemOperationType(IntEnum): """ Memory operation types """ {{if 'CU_MEM_OPERATION_TYPE_MAP' in found_values}} CU_MEM_OPERATION_TYPE_MAP = ccuda.CUmemOperationType_enum.CU_MEM_OPERATION_TYPE_MAP{{endif}} {{if 'CU_MEM_OPERATION_TYPE_UNMAP' in found_values}} CU_MEM_OPERATION_TYPE_UNMAP = ccuda.CUmemOperationType_enum.CU_MEM_OPERATION_TYPE_UNMAP{{endif}} {{endif}} {{if 'CUmemHandleType_enum' in found_types}} class CUmemHandleType(IntEnum): """ Memory handle types """ {{if 'CU_MEM_HANDLE_TYPE_GENERIC' in found_values}} CU_MEM_HANDLE_TYPE_GENERIC = ccuda.CUmemHandleType_enum.CU_MEM_HANDLE_TYPE_GENERIC{{endif}} {{endif}} {{if 'CUmemAllocationCompType_enum' in found_types}} class CUmemAllocationCompType(IntEnum): """ Specifies compression attribute for an allocation. """ {{if 'CU_MEM_ALLOCATION_COMP_NONE' in found_values}} #: Allocating non-compressible memory CU_MEM_ALLOCATION_COMP_NONE = ccuda.CUmemAllocationCompType_enum.CU_MEM_ALLOCATION_COMP_NONE{{endif}} {{if 'CU_MEM_ALLOCATION_COMP_GENERIC' in found_values}} #: Allocating compressible memory CU_MEM_ALLOCATION_COMP_GENERIC = ccuda.CUmemAllocationCompType_enum.CU_MEM_ALLOCATION_COMP_GENERIC{{endif}} {{endif}} {{if 'CUmulticastGranularity_flags_enum' in found_types}} class CUmulticastGranularity_flags(IntEnum): """ Flags for querying different granularities for a multicast object """ {{if 'CU_MULTICAST_GRANULARITY_MINIMUM' in found_values}} #: Minimum required granularity CU_MULTICAST_GRANULARITY_MINIMUM = ccuda.CUmulticastGranularity_flags_enum.CU_MULTICAST_GRANULARITY_MINIMUM{{endif}} {{if 'CU_MULTICAST_GRANULARITY_RECOMMENDED' in found_values}} #: Recommended granularity for best performance CU_MULTICAST_GRANULARITY_RECOMMENDED = ccuda.CUmulticastGranularity_flags_enum.CU_MULTICAST_GRANULARITY_RECOMMENDED{{endif}} {{endif}} {{if 'CUgraphExecUpdateResult_enum' in found_types}} class CUgraphExecUpdateResult(IntEnum): """ CUDA Graph Update error types """ {{if 'CU_GRAPH_EXEC_UPDATE_SUCCESS' in found_values}} #: The update succeeded CU_GRAPH_EXEC_UPDATE_SUCCESS = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_SUCCESS{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR' in found_values}} #: The update failed for an unexpected reason which is described in the #: return value of the function CU_GRAPH_EXEC_UPDATE_ERROR = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR_TOPOLOGY_CHANGED' in found_values}} #: The update failed because the topology changed CU_GRAPH_EXEC_UPDATE_ERROR_TOPOLOGY_CHANGED = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR_TOPOLOGY_CHANGED{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR_NODE_TYPE_CHANGED' in found_values}} #: The update failed because a node type changed CU_GRAPH_EXEC_UPDATE_ERROR_NODE_TYPE_CHANGED = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR_NODE_TYPE_CHANGED{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR_FUNCTION_CHANGED' in found_values}} #: The update failed because the function of a kernel node changed #: (CUDA driver < 11.2) CU_GRAPH_EXEC_UPDATE_ERROR_FUNCTION_CHANGED = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR_FUNCTION_CHANGED{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR_PARAMETERS_CHANGED' in found_values}} #: The update failed because the parameters changed in a way that is #: not supported CU_GRAPH_EXEC_UPDATE_ERROR_PARAMETERS_CHANGED = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR_PARAMETERS_CHANGED{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR_NOT_SUPPORTED' in found_values}} #: The update failed because something about the node is not supported CU_GRAPH_EXEC_UPDATE_ERROR_NOT_SUPPORTED = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR_NOT_SUPPORTED{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR_UNSUPPORTED_FUNCTION_CHANGE' in found_values}} #: The update failed because the function of a kernel node changed in #: an unsupported way CU_GRAPH_EXEC_UPDATE_ERROR_UNSUPPORTED_FUNCTION_CHANGE = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR_UNSUPPORTED_FUNCTION_CHANGE{{endif}} {{if 'CU_GRAPH_EXEC_UPDATE_ERROR_ATTRIBUTES_CHANGED' in found_values}} #: The update failed because the node attributes changed in a way that #: is not supported CU_GRAPH_EXEC_UPDATE_ERROR_ATTRIBUTES_CHANGED = ccuda.CUgraphExecUpdateResult_enum.CU_GRAPH_EXEC_UPDATE_ERROR_ATTRIBUTES_CHANGED{{endif}} {{endif}} {{if 'CUmemPool_attribute_enum' in found_types}} class CUmemPool_attribute(IntEnum): """ CUDA memory pool attributes """ {{if 'CU_MEMPOOL_ATTR_REUSE_FOLLOW_EVENT_DEPENDENCIES' in found_values}} #: (value type = int) Allow cuMemAllocAsync to use memory #: asynchronously freed in another streams as long as a stream ordering #: dependency of the allocating stream on the free action exists. Cuda #: events and null stream interactions can create the required stream #: ordered dependencies. (default enabled) CU_MEMPOOL_ATTR_REUSE_FOLLOW_EVENT_DEPENDENCIES = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_REUSE_FOLLOW_EVENT_DEPENDENCIES{{endif}} {{if 'CU_MEMPOOL_ATTR_REUSE_ALLOW_OPPORTUNISTIC' in found_values}} #: (value type = int) Allow reuse of already completed frees when there #: is no dependency between the free and allocation. (default enabled) CU_MEMPOOL_ATTR_REUSE_ALLOW_OPPORTUNISTIC = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_REUSE_ALLOW_OPPORTUNISTIC{{endif}} {{if 'CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES' in found_values}} #: (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). CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES{{endif}} {{if 'CU_MEMPOOL_ATTR_RELEASE_THRESHOLD' in found_values}} #: (value type = cuuint64_t) Amount of reserved memory in bytes to hold #: onto before trying to release memory back to the OS. When more than #: the release threshold bytes of memory are held by the memory pool, #: the allocator will try to release memory back to the OS on the next #: call to stream, event or context synchronize. (default 0) CU_MEMPOOL_ATTR_RELEASE_THRESHOLD = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_RELEASE_THRESHOLD{{endif}} {{if 'CU_MEMPOOL_ATTR_RESERVED_MEM_CURRENT' in found_values}} #: (value type = cuuint64_t) Amount of backing memory currently #: allocated for the mempool. CU_MEMPOOL_ATTR_RESERVED_MEM_CURRENT = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_RESERVED_MEM_CURRENT{{endif}} {{if 'CU_MEMPOOL_ATTR_RESERVED_MEM_HIGH' in found_values}} #: (value type = cuuint64_t) High watermark of backing memory allocated #: for the mempool since the last time it was reset. High watermark can #: only be reset to zero. CU_MEMPOOL_ATTR_RESERVED_MEM_HIGH = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_RESERVED_MEM_HIGH{{endif}} {{if 'CU_MEMPOOL_ATTR_USED_MEM_CURRENT' in found_values}} #: (value type = cuuint64_t) Amount of memory from the pool that is #: currently in use by the application. CU_MEMPOOL_ATTR_USED_MEM_CURRENT = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_USED_MEM_CURRENT{{endif}} {{if 'CU_MEMPOOL_ATTR_USED_MEM_HIGH' in found_values}} #: (value type = cuuint64_t) High watermark of the amount of memory #: from the pool that was in use by the application since the last time #: it was reset. High watermark can only be reset to zero. CU_MEMPOOL_ATTR_USED_MEM_HIGH = ccuda.CUmemPool_attribute_enum.CU_MEMPOOL_ATTR_USED_MEM_HIGH{{endif}} {{endif}} {{if 'CUgraphMem_attribute_enum' in found_types}} class CUgraphMem_attribute(IntEnum): """ """ {{if 'CU_GRAPH_MEM_ATTR_USED_MEM_CURRENT' in found_values}} #: (value type = cuuint64_t) Amount of memory, in bytes, currently #: associated with graphs CU_GRAPH_MEM_ATTR_USED_MEM_CURRENT = ccuda.CUgraphMem_attribute_enum.CU_GRAPH_MEM_ATTR_USED_MEM_CURRENT{{endif}} {{if 'CU_GRAPH_MEM_ATTR_USED_MEM_HIGH' in found_values}} #: (value type = cuuint64_t) High watermark of memory, in bytes, #: associated with graphs since the last time it was reset. High #: watermark can only be reset to zero. CU_GRAPH_MEM_ATTR_USED_MEM_HIGH = ccuda.CUgraphMem_attribute_enum.CU_GRAPH_MEM_ATTR_USED_MEM_HIGH{{endif}} {{if 'CU_GRAPH_MEM_ATTR_RESERVED_MEM_CURRENT' in found_values}} #: (value type = cuuint64_t) Amount of memory, in bytes, currently #: allocated for use by the CUDA graphs asynchronous allocator. CU_GRAPH_MEM_ATTR_RESERVED_MEM_CURRENT = ccuda.CUgraphMem_attribute_enum.CU_GRAPH_MEM_ATTR_RESERVED_MEM_CURRENT{{endif}} {{if 'CU_GRAPH_MEM_ATTR_RESERVED_MEM_HIGH' in found_values}} #: (value type = cuuint64_t) High watermark of memory, in bytes, #: currently allocated for use by the CUDA graphs asynchronous #: allocator. CU_GRAPH_MEM_ATTR_RESERVED_MEM_HIGH = ccuda.CUgraphMem_attribute_enum.CU_GRAPH_MEM_ATTR_RESERVED_MEM_HIGH{{endif}} {{endif}} {{if 'CUflushGPUDirectRDMAWritesOptions_enum' in found_types}} class CUflushGPUDirectRDMAWritesOptions(IntEnum): """ Bitmasks for :py:obj:`~.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_FLUSH_WRITES_OPTIONS` """ {{if 'CU_FLUSH_GPU_DIRECT_RDMA_WRITES_OPTION_HOST' in found_values}} #: :py:obj:`~.cuFlushGPUDirectRDMAWrites()` and its CUDA Runtime API #: counterpart are supported on the device. CU_FLUSH_GPU_DIRECT_RDMA_WRITES_OPTION_HOST = ccuda.CUflushGPUDirectRDMAWritesOptions_enum.CU_FLUSH_GPU_DIRECT_RDMA_WRITES_OPTION_HOST{{endif}} {{if 'CU_FLUSH_GPU_DIRECT_RDMA_WRITES_OPTION_MEMOPS' in found_values}} #: The :py:obj:`~.CU_STREAM_WAIT_VALUE_FLUSH` flag and the #: :py:obj:`~.CU_STREAM_MEM_OP_FLUSH_REMOTE_WRITES` MemOp are supported #: on the device. CU_FLUSH_GPU_DIRECT_RDMA_WRITES_OPTION_MEMOPS = ccuda.CUflushGPUDirectRDMAWritesOptions_enum.CU_FLUSH_GPU_DIRECT_RDMA_WRITES_OPTION_MEMOPS{{endif}} {{endif}} {{if 'CUGPUDirectRDMAWritesOrdering_enum' in found_types}} class CUGPUDirectRDMAWritesOrdering(IntEnum): """ Platform native ordering for GPUDirect RDMA writes """ {{if 'CU_GPU_DIRECT_RDMA_WRITES_ORDERING_NONE' in found_values}} #: The device does not natively support ordering of remote writes. #: :py:obj:`~.cuFlushGPUDirectRDMAWrites()` can be leveraged if #: supported. CU_GPU_DIRECT_RDMA_WRITES_ORDERING_NONE = ccuda.CUGPUDirectRDMAWritesOrdering_enum.CU_GPU_DIRECT_RDMA_WRITES_ORDERING_NONE{{endif}} {{if 'CU_GPU_DIRECT_RDMA_WRITES_ORDERING_OWNER' in found_values}} #: Natively, the device can consistently consume remote writes, #: although other CUDA devices may not. CU_GPU_DIRECT_RDMA_WRITES_ORDERING_OWNER = ccuda.CUGPUDirectRDMAWritesOrdering_enum.CU_GPU_DIRECT_RDMA_WRITES_ORDERING_OWNER{{endif}} {{if 'CU_GPU_DIRECT_RDMA_WRITES_ORDERING_ALL_DEVICES' in found_values}} #: Any CUDA device in the system can consistently consume remote writes #: to this device. CU_GPU_DIRECT_RDMA_WRITES_ORDERING_ALL_DEVICES = ccuda.CUGPUDirectRDMAWritesOrdering_enum.CU_GPU_DIRECT_RDMA_WRITES_ORDERING_ALL_DEVICES{{endif}} {{endif}} {{if 'CUflushGPUDirectRDMAWritesScope_enum' in found_types}} class CUflushGPUDirectRDMAWritesScope(IntEnum): """ The scopes for :py:obj:`~.cuFlushGPUDirectRDMAWrites` """ {{if 'CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TO_OWNER' in found_values}} #: Blocks until remote writes are visible to the CUDA device context #: owning the data. CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TO_OWNER = ccuda.CUflushGPUDirectRDMAWritesScope_enum.CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TO_OWNER{{endif}} {{if 'CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TO_ALL_DEVICES' in found_values}} #: Blocks until remote writes are visible to all CUDA device contexts. CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TO_ALL_DEVICES = ccuda.CUflushGPUDirectRDMAWritesScope_enum.CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TO_ALL_DEVICES{{endif}} {{endif}} {{if 'CUflushGPUDirectRDMAWritesTarget_enum' in found_types}} class CUflushGPUDirectRDMAWritesTarget(IntEnum): """ The targets for :py:obj:`~.cuFlushGPUDirectRDMAWrites` """ {{if 'CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TARGET_CURRENT_CTX' in found_values}} #: Sets the target for :py:obj:`~.cuFlushGPUDirectRDMAWrites()` to the #: currently active CUDA device context. CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TARGET_CURRENT_CTX = ccuda.CUflushGPUDirectRDMAWritesTarget_enum.CU_FLUSH_GPU_DIRECT_RDMA_WRITES_TARGET_CURRENT_CTX{{endif}} {{endif}} {{if 'CUgraphDebugDot_flags_enum' in found_types}} class CUgraphDebugDot_flags(IntEnum): """ The additional write options for :py:obj:`~.cuGraphDebugDotPrint` """ {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_VERBOSE' in found_values}} #: Output all debug data as if every debug flag is enabled CU_GRAPH_DEBUG_DOT_FLAGS_VERBOSE = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_VERBOSE{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_RUNTIME_TYPES' in found_values}} #: Use CUDA Runtime structures for output CU_GRAPH_DEBUG_DOT_FLAGS_RUNTIME_TYPES = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_RUNTIME_TYPES{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_KERNEL_NODE_PARAMS' in found_values}} #: Adds CUDA_KERNEL_NODE_PARAMS values to output CU_GRAPH_DEBUG_DOT_FLAGS_KERNEL_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_KERNEL_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_MEMCPY_NODE_PARAMS' in found_values}} #: Adds CUDA_MEMCPY3D values to output CU_GRAPH_DEBUG_DOT_FLAGS_MEMCPY_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_MEMCPY_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_MEMSET_NODE_PARAMS' in found_values}} #: Adds CUDA_MEMSET_NODE_PARAMS values to output CU_GRAPH_DEBUG_DOT_FLAGS_MEMSET_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_MEMSET_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_HOST_NODE_PARAMS' in found_values}} #: Adds CUDA_HOST_NODE_PARAMS values to output CU_GRAPH_DEBUG_DOT_FLAGS_HOST_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_HOST_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_EVENT_NODE_PARAMS' in found_values}} #: Adds CUevent handle from record and wait nodes to output CU_GRAPH_DEBUG_DOT_FLAGS_EVENT_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_EVENT_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_EXT_SEMAS_SIGNAL_NODE_PARAMS' in found_values}} #: Adds CUDA_EXT_SEM_SIGNAL_NODE_PARAMS values to output CU_GRAPH_DEBUG_DOT_FLAGS_EXT_SEMAS_SIGNAL_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_EXT_SEMAS_SIGNAL_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_EXT_SEMAS_WAIT_NODE_PARAMS' in found_values}} #: Adds CUDA_EXT_SEM_WAIT_NODE_PARAMS values to output CU_GRAPH_DEBUG_DOT_FLAGS_EXT_SEMAS_WAIT_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_EXT_SEMAS_WAIT_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_KERNEL_NODE_ATTRIBUTES' in found_values}} #: Adds CUkernelNodeAttrValue values to output CU_GRAPH_DEBUG_DOT_FLAGS_KERNEL_NODE_ATTRIBUTES = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_KERNEL_NODE_ATTRIBUTES{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_HANDLES' in found_values}} #: Adds node handles and every kernel function handle to output CU_GRAPH_DEBUG_DOT_FLAGS_HANDLES = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_HANDLES{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_MEM_ALLOC_NODE_PARAMS' in found_values}} #: Adds memory alloc node parameters to output CU_GRAPH_DEBUG_DOT_FLAGS_MEM_ALLOC_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_MEM_ALLOC_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_MEM_FREE_NODE_PARAMS' in found_values}} #: Adds memory free node parameters to output CU_GRAPH_DEBUG_DOT_FLAGS_MEM_FREE_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_MEM_FREE_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_BATCH_MEM_OP_NODE_PARAMS' in found_values}} #: Adds batch mem op node parameters to output CU_GRAPH_DEBUG_DOT_FLAGS_BATCH_MEM_OP_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_BATCH_MEM_OP_NODE_PARAMS{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_EXTRA_TOPO_INFO' in found_values}} #: Adds edge numbering information CU_GRAPH_DEBUG_DOT_FLAGS_EXTRA_TOPO_INFO = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_EXTRA_TOPO_INFO{{endif}} {{if 'CU_GRAPH_DEBUG_DOT_FLAGS_CONDITIONAL_NODE_PARAMS' in found_values}} #: Adds conditional node parameters to output CU_GRAPH_DEBUG_DOT_FLAGS_CONDITIONAL_NODE_PARAMS = ccuda.CUgraphDebugDot_flags_enum.CU_GRAPH_DEBUG_DOT_FLAGS_CONDITIONAL_NODE_PARAMS{{endif}} {{endif}} {{if 'CUuserObject_flags_enum' in found_types}} class CUuserObject_flags(IntEnum): """ Flags for user objects for graphs """ {{if 'CU_USER_OBJECT_NO_DESTRUCTOR_SYNC' in found_values}} #: Indicates the destructor execution is not synchronized by any CUDA #: handle. CU_USER_OBJECT_NO_DESTRUCTOR_SYNC = ccuda.CUuserObject_flags_enum.CU_USER_OBJECT_NO_DESTRUCTOR_SYNC{{endif}} {{endif}} {{if 'CUuserObjectRetain_flags_enum' in found_types}} class CUuserObjectRetain_flags(IntEnum): """ Flags for retaining user object references for graphs """ {{if 'CU_GRAPH_USER_OBJECT_MOVE' in found_values}} #: Transfer references from the caller rather than creating new #: references. CU_GRAPH_USER_OBJECT_MOVE = ccuda.CUuserObjectRetain_flags_enum.CU_GRAPH_USER_OBJECT_MOVE{{endif}} {{endif}} {{if 'CUgraphInstantiate_flags_enum' in found_types}} class CUgraphInstantiate_flags(IntEnum): """ Flags for instantiating a graph """ {{if 'CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH' in found_values}} #: Automatically free memory allocated in a graph before relaunching. CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH = ccuda.CUgraphInstantiate_flags_enum.CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH{{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_FLAG_UPLOAD' in found_values}} #: Automatically upload the graph after instantiation. Only supported #: by :py:obj:`~.cuGraphInstantiateWithParams`. The upload will be #: performed using the stream provided in `instantiateParams`. CUDA_GRAPH_INSTANTIATE_FLAG_UPLOAD = ccuda.CUgraphInstantiate_flags_enum.CUDA_GRAPH_INSTANTIATE_FLAG_UPLOAD{{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_FLAG_DEVICE_LAUNCH' in found_values}} #: Instantiate the graph to be launchable from the device. This flag #: can only be used on platforms which support unified addressing. This #: flag cannot be used in conjunction with #: CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH. CUDA_GRAPH_INSTANTIATE_FLAG_DEVICE_LAUNCH = ccuda.CUgraphInstantiate_flags_enum.CUDA_GRAPH_INSTANTIATE_FLAG_DEVICE_LAUNCH{{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_FLAG_USE_NODE_PRIORITY' in found_values}} #: Run the graph using the per-node priority attributes rather than the #: priority of the stream it is launched into. CUDA_GRAPH_INSTANTIATE_FLAG_USE_NODE_PRIORITY = ccuda.CUgraphInstantiate_flags_enum.CUDA_GRAPH_INSTANTIATE_FLAG_USE_NODE_PRIORITY{{endif}} {{endif}} {{if 'CUdeviceNumaConfig_enum' in found_types}} class CUdeviceNumaConfig(IntEnum): """ CUDA device NUMA configuration """ {{if 'CU_DEVICE_NUMA_CONFIG_NONE' in found_values}} #: The GPU is not a NUMA node CU_DEVICE_NUMA_CONFIG_NONE = ccuda.CUdeviceNumaConfig_enum.CU_DEVICE_NUMA_CONFIG_NONE{{endif}} {{if 'CU_DEVICE_NUMA_CONFIG_NUMA_NODE' in found_values}} #: The GPU is a NUMA node, CU_DEVICE_ATTRIBUTE_NUMA_ID contains its #: NUMA ID CU_DEVICE_NUMA_CONFIG_NUMA_NODE = ccuda.CUdeviceNumaConfig_enum.CU_DEVICE_NUMA_CONFIG_NUMA_NODE{{endif}} {{endif}} {{if 'CUmoduleLoadingMode_enum' in found_types}} class CUmoduleLoadingMode(IntEnum): """ CUDA Lazy Loading status """ {{if 'CU_MODULE_EAGER_LOADING' in found_values}} #: Lazy Kernel Loading is not enabled CU_MODULE_EAGER_LOADING = ccuda.CUmoduleLoadingMode_enum.CU_MODULE_EAGER_LOADING{{endif}} {{if 'CU_MODULE_LAZY_LOADING' in found_values}} #: Lazy Kernel Loading is enabled CU_MODULE_LAZY_LOADING = ccuda.CUmoduleLoadingMode_enum.CU_MODULE_LAZY_LOADING{{endif}} {{endif}} {{if 'CUfunctionLoadingState_enum' in found_types}} class CUfunctionLoadingState(IntEnum): """ """ {{if 'CU_FUNCTION_LOADING_STATE_UNLOADED' in found_values}} CU_FUNCTION_LOADING_STATE_UNLOADED = ccuda.CUfunctionLoadingState_enum.CU_FUNCTION_LOADING_STATE_UNLOADED{{endif}} {{if 'CU_FUNCTION_LOADING_STATE_LOADED' in found_values}} CU_FUNCTION_LOADING_STATE_LOADED = ccuda.CUfunctionLoadingState_enum.CU_FUNCTION_LOADING_STATE_LOADED{{endif}} {{if 'CU_FUNCTION_LOADING_STATE_MAX' in found_values}} CU_FUNCTION_LOADING_STATE_MAX = ccuda.CUfunctionLoadingState_enum.CU_FUNCTION_LOADING_STATE_MAX{{endif}} {{endif}} {{if 'CUcoredumpSettings_enum' in found_types}} class CUcoredumpSettings(IntEnum): """ Flags for choosing a coredump attribute to get/set """ {{if 'CU_COREDUMP_ENABLE_ON_EXCEPTION' in found_values}} CU_COREDUMP_ENABLE_ON_EXCEPTION = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_ENABLE_ON_EXCEPTION{{endif}} {{if 'CU_COREDUMP_TRIGGER_HOST' in found_values}} CU_COREDUMP_TRIGGER_HOST = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_TRIGGER_HOST{{endif}} {{if 'CU_COREDUMP_LIGHTWEIGHT' in found_values}} CU_COREDUMP_LIGHTWEIGHT = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_LIGHTWEIGHT{{endif}} {{if 'CU_COREDUMP_ENABLE_USER_TRIGGER' in found_values}} CU_COREDUMP_ENABLE_USER_TRIGGER = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_ENABLE_USER_TRIGGER{{endif}} {{if 'CU_COREDUMP_FILE' in found_values}} CU_COREDUMP_FILE = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_FILE{{endif}} {{if 'CU_COREDUMP_PIPE' in found_values}} CU_COREDUMP_PIPE = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_PIPE{{endif}} {{if 'CU_COREDUMP_GENERATION_FLAGS' in found_values}} CU_COREDUMP_GENERATION_FLAGS = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_GENERATION_FLAGS{{endif}} {{if 'CU_COREDUMP_MAX' in found_values}} CU_COREDUMP_MAX = ccuda.CUcoredumpSettings_enum.CU_COREDUMP_MAX{{endif}} {{endif}} {{if 'CUCoredumpGenerationFlags' in found_types}} class CUCoredumpGenerationFlags(IntEnum): """ Flags for controlling coredump contents """ {{if 'CU_COREDUMP_DEFAULT_FLAGS' in found_values}} CU_COREDUMP_DEFAULT_FLAGS = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_DEFAULT_FLAGS{{endif}} {{if 'CU_COREDUMP_SKIP_NONRELOCATED_ELF_IMAGES' in found_values}} CU_COREDUMP_SKIP_NONRELOCATED_ELF_IMAGES = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_SKIP_NONRELOCATED_ELF_IMAGES{{endif}} {{if 'CU_COREDUMP_SKIP_GLOBAL_MEMORY' in found_values}} CU_COREDUMP_SKIP_GLOBAL_MEMORY = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_SKIP_GLOBAL_MEMORY{{endif}} {{if 'CU_COREDUMP_SKIP_SHARED_MEMORY' in found_values}} CU_COREDUMP_SKIP_SHARED_MEMORY = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_SKIP_SHARED_MEMORY{{endif}} {{if 'CU_COREDUMP_SKIP_LOCAL_MEMORY' in found_values}} CU_COREDUMP_SKIP_LOCAL_MEMORY = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_SKIP_LOCAL_MEMORY{{endif}} {{if 'CU_COREDUMP_SKIP_ABORT' in found_values}} CU_COREDUMP_SKIP_ABORT = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_SKIP_ABORT{{endif}} {{if 'CU_COREDUMP_SKIP_CONSTBANK_MEMORY' in found_values}} CU_COREDUMP_SKIP_CONSTBANK_MEMORY = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_SKIP_CONSTBANK_MEMORY{{endif}} {{if 'CU_COREDUMP_LIGHTWEIGHT_FLAGS' in found_values}} CU_COREDUMP_LIGHTWEIGHT_FLAGS = ccuda.CUCoredumpGenerationFlags.CU_COREDUMP_LIGHTWEIGHT_FLAGS{{endif}} {{endif}} {{if 'CUgreenCtxCreate_flags' in found_types}} class CUgreenCtxCreate_flags(IntEnum): """ """ {{if 'CU_GREEN_CTX_DEFAULT_STREAM' in found_values}} #: Required. Creates a default stream to use inside the green context CU_GREEN_CTX_DEFAULT_STREAM = ccuda.CUgreenCtxCreate_flags.CU_GREEN_CTX_DEFAULT_STREAM{{endif}} {{endif}} {{if 'CUdevSmResourceSplit_flags' in found_types}} class CUdevSmResourceSplit_flags(IntEnum): """ """ {{if 'CU_DEV_SM_RESOURCE_SPLIT_IGNORE_SM_COSCHEDULING' in found_values}} CU_DEV_SM_RESOURCE_SPLIT_IGNORE_SM_COSCHEDULING = ccuda.CUdevSmResourceSplit_flags.CU_DEV_SM_RESOURCE_SPLIT_IGNORE_SM_COSCHEDULING{{endif}} {{if 'CU_DEV_SM_RESOURCE_SPLIT_MAX_POTENTIAL_CLUSTER_SIZE' in found_values}} CU_DEV_SM_RESOURCE_SPLIT_MAX_POTENTIAL_CLUSTER_SIZE = ccuda.CUdevSmResourceSplit_flags.CU_DEV_SM_RESOURCE_SPLIT_MAX_POTENTIAL_CLUSTER_SIZE{{endif}} {{endif}} {{if 'CUdevResourceType' in found_types}} class CUdevResourceType(IntEnum): """ Type of resource """ {{if 'CU_DEV_RESOURCE_TYPE_INVALID' in found_values}} CU_DEV_RESOURCE_TYPE_INVALID = ccuda.CUdevResourceType.CU_DEV_RESOURCE_TYPE_INVALID{{endif}} {{if 'CU_DEV_RESOURCE_TYPE_SM' in found_values}} #: Streaming multiprocessors related information CU_DEV_RESOURCE_TYPE_SM = ccuda.CUdevResourceType.CU_DEV_RESOURCE_TYPE_SM{{endif}} {{endif}} {{if 'CUoutput_mode_enum' in found_types}} class CUoutput_mode(IntEnum): """ Profiler Output Modes """ {{if 'CU_OUT_KEY_VALUE_PAIR' in found_values}} #: Output mode Key-Value pair format. CU_OUT_KEY_VALUE_PAIR = ccuda.CUoutput_mode_enum.CU_OUT_KEY_VALUE_PAIR{{endif}} {{if 'CU_OUT_CSV' in found_values}} #: Output mode Comma separated values format. CU_OUT_CSV = ccuda.CUoutput_mode_enum.CU_OUT_CSV{{endif}} {{endif}} {{if True}} class CUeglFrameType(IntEnum): """ CUDA EglFrame type - array or pointer """ {{if True}} #: Frame type CUDA array CU_EGL_FRAME_TYPE_ARRAY = ccuda.CUeglFrameType_enum.CU_EGL_FRAME_TYPE_ARRAY{{endif}} {{if True}} #: Frame type pointer CU_EGL_FRAME_TYPE_PITCH = ccuda.CUeglFrameType_enum.CU_EGL_FRAME_TYPE_PITCH{{endif}} {{endif}} {{if True}} class CUeglResourceLocationFlags(IntEnum): """ Resource location flags- sysmem or vidmem For CUDA context on iGPU, since video and system memory are equivalent - these flags will not have an effect on the execution. For CUDA context on dGPU, applications can use the flag :py:obj:`~.CUeglResourceLocationFlags` to give a hint about the desired location. :py:obj:`~.CU_EGL_RESOURCE_LOCATION_SYSMEM` - the frame data is made resident on the system memory to be accessed by CUDA. :py:obj:`~.CU_EGL_RESOURCE_LOCATION_VIDMEM` - the frame data is made resident on the dedicated video memory to be accessed by CUDA. There may be an additional latency due to new allocation and data migration, if the frame is produced on a different memory. """ {{if True}} #: Resource location sysmem CU_EGL_RESOURCE_LOCATION_SYSMEM = ccuda.CUeglResourceLocationFlags_enum.CU_EGL_RESOURCE_LOCATION_SYSMEM{{endif}} {{if True}} #: Resource location vidmem CU_EGL_RESOURCE_LOCATION_VIDMEM = ccuda.CUeglResourceLocationFlags_enum.CU_EGL_RESOURCE_LOCATION_VIDMEM{{endif}} {{endif}} {{if True}} class CUeglColorFormat(IntEnum): """ CUDA EGL Color Format - The different planar and multiplanar formats currently supported for CUDA_EGL interops. Three channel formats are currently not supported for :py:obj:`~.CU_EGL_FRAME_TYPE_ARRAY` """ {{if True}} #: Y, U, V in three surfaces, each in a separate surface, U/V width = #: 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YUV420_PLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_PLANAR{{endif}} {{if True}} #: Y, UV in two surfaces (UV as one surface) with VU byte ordering, #: width, height ratio same as YUV420Planar. CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR{{endif}} {{if True}} #: Y, U, V each in a separate surface, U/V width = 1/2 Y width, U/V #: height = Y height. CU_EGL_COLOR_FORMAT_YUV422_PLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV422_PLANAR{{endif}} {{if True}} #: Y, UV in two surfaces with VU byte ordering, width, height ratio #: same as YUV422Planar. CU_EGL_COLOR_FORMAT_YUV422_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV422_SEMIPLANAR{{endif}} {{if True}} #: R/G/B three channels in one surface with BGR byte ordering. Only #: pitch linear format supported. CU_EGL_COLOR_FORMAT_RGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_RGB{{endif}} {{if True}} #: R/G/B three channels in one surface with RGB byte ordering. Only #: pitch linear format supported. CU_EGL_COLOR_FORMAT_BGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BGR{{endif}} {{if True}} #: R/G/B/A four channels in one surface with BGRA byte ordering. CU_EGL_COLOR_FORMAT_ARGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_ARGB{{endif}} {{if True}} #: R/G/B/A four channels in one surface with ABGR byte ordering. CU_EGL_COLOR_FORMAT_RGBA = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_RGBA{{endif}} {{if True}} #: single luminance channel in one surface. CU_EGL_COLOR_FORMAT_L = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_L{{endif}} {{if True}} #: single color channel in one surface. CU_EGL_COLOR_FORMAT_R = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_R{{endif}} {{if True}} #: Y, U, V in three surfaces, each in a separate surface, U/V width = Y #: width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YUV444_PLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV444_PLANAR{{endif}} {{if True}} #: Y, UV in two surfaces (UV as one surface) with VU byte ordering, #: width, height ratio same as YUV444Planar. CU_EGL_COLOR_FORMAT_YUV444_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV444_SEMIPLANAR{{endif}} {{if True}} #: Y, U, V in one surface, interleaved as UYVY in one channel. CU_EGL_COLOR_FORMAT_YUYV_422 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUYV_422{{endif}} {{if True}} #: Y, U, V in one surface, interleaved as YUYV in one channel. CU_EGL_COLOR_FORMAT_UYVY_422 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_UYVY_422{{endif}} {{if True}} #: R/G/B/A four channels in one surface with RGBA byte ordering. CU_EGL_COLOR_FORMAT_ABGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_ABGR{{endif}} {{if True}} #: R/G/B/A four channels in one surface with ARGB byte ordering. CU_EGL_COLOR_FORMAT_BGRA = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BGRA{{endif}} {{if True}} #: Alpha color format - one channel in one surface. CU_EGL_COLOR_FORMAT_A = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_A{{endif}} {{if True}} #: R/G color format - two channels in one surface with GR byte ordering CU_EGL_COLOR_FORMAT_RG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_RG{{endif}} {{if True}} #: Y, U, V, A four channels in one surface, interleaved as VUYA. CU_EGL_COLOR_FORMAT_AYUV = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_AYUV{{endif}} {{if True}} #: Y, VU in two surfaces (VU as one surface) with UV byte ordering, U/V #: width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YVU444_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU444_SEMIPLANAR{{endif}} {{if True}} #: Y, VU in two surfaces (VU as one surface) with UV byte ordering, U/V #: width = 1/2 Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YVU422_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU422_SEMIPLANAR{{endif}} {{if True}} #: Y, VU in two surfaces (VU as one surface) with UV byte ordering, U/V #: width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR{{endif}} {{if True}} #: Y10, V10U10 in two surfaces (VU as one surface) with UV byte #: ordering, U/V width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_444_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_444_SEMIPLANAR{{endif}} {{if True}} #: Y10, V10U10 in two surfaces (VU as one surface) with UV byte #: ordering, U/V width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR{{endif}} {{if True}} #: Y12, V12U12 in two surfaces (VU as one surface) with UV byte #: ordering, U/V width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y12V12U12_444_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12V12U12_444_SEMIPLANAR{{endif}} {{if True}} #: Y12, V12U12 in two surfaces (VU as one surface) with UV byte #: ordering, U/V width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y12V12U12_420_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12V12U12_420_SEMIPLANAR{{endif}} {{if True}} #: Extended Range Y, U, V in one surface, interleaved as YVYU in one #: channel. CU_EGL_COLOR_FORMAT_VYUY_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_VYUY_ER{{endif}} {{if True}} #: Extended Range Y, U, V in one surface, interleaved as YUYV in one #: channel. CU_EGL_COLOR_FORMAT_UYVY_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_UYVY_ER{{endif}} {{if True}} #: Extended Range Y, U, V in one surface, interleaved as UYVY in one #: channel. CU_EGL_COLOR_FORMAT_YUYV_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUYV_ER{{endif}} {{if True}} #: Extended Range Y, U, V in one surface, interleaved as VYUY in one #: channel. CU_EGL_COLOR_FORMAT_YVYU_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVYU_ER{{endif}} {{if True}} #: Extended Range Y, U, V three channels in one surface, interleaved as #: VUY. Only pitch linear format supported. CU_EGL_COLOR_FORMAT_YUV_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV_ER{{endif}} {{if True}} #: Extended Range Y, U, V, A four channels in one surface, interleaved #: as AVUY. CU_EGL_COLOR_FORMAT_YUVA_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUVA_ER{{endif}} {{if True}} #: Extended Range Y, U, V, A four channels in one surface, interleaved #: as VUYA. CU_EGL_COLOR_FORMAT_AYUV_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_AYUV_ER{{endif}} {{if True}} #: Extended Range Y, U, V in three surfaces, U/V width = Y width, U/V #: height = Y height. CU_EGL_COLOR_FORMAT_YUV444_PLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV444_PLANAR_ER{{endif}} {{if True}} #: Extended Range Y, U, V in three surfaces, U/V width = 1/2 Y width, #: U/V height = Y height. CU_EGL_COLOR_FORMAT_YUV422_PLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV422_PLANAR_ER{{endif}} {{if True}} #: Extended Range Y, U, V in three surfaces, U/V width = 1/2 Y width, #: U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YUV420_PLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_PLANAR_ER{{endif}} {{if True}} #: Extended Range Y, UV in two surfaces (UV as one surface) with VU #: byte ordering, U/V width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YUV444_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV444_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y, UV in two surfaces (UV as one surface) with VU #: byte ordering, U/V width = 1/2 Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YUV422_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV422_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y, UV in two surfaces (UV as one surface) with VU #: byte ordering, U/V width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y, V, U in three surfaces, U/V width = Y width, U/V #: height = Y height. CU_EGL_COLOR_FORMAT_YVU444_PLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU444_PLANAR_ER{{endif}} {{if True}} #: Extended Range Y, V, U in three surfaces, U/V width = 1/2 Y width, #: U/V height = Y height. CU_EGL_COLOR_FORMAT_YVU422_PLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU422_PLANAR_ER{{endif}} {{if True}} #: Extended Range Y, V, U in three surfaces, U/V width = 1/2 Y width, #: U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_PLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_PLANAR_ER{{endif}} {{if True}} #: Extended Range Y, VU in two surfaces (VU as one surface) with UV #: byte ordering, U/V width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YVU444_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU444_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y, VU in two surfaces (VU as one surface) with UV #: byte ordering, U/V width = 1/2 Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YVU422_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU422_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y, VU in two surfaces (VU as one surface) with UV #: byte ordering, U/V width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR_ER{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved RGGB #: ordering. CU_EGL_COLOR_FORMAT_BAYER_RGGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_RGGB{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved BGGR #: ordering. CU_EGL_COLOR_FORMAT_BAYER_BGGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_BGGR{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved GRBG #: ordering. CU_EGL_COLOR_FORMAT_BAYER_GRBG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_GRBG{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved GBRG #: ordering. CU_EGL_COLOR_FORMAT_BAYER_GBRG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_GBRG{{endif}} {{if True}} #: Bayer10 format - one channel in one surface with interleaved RGGB #: ordering. Out of 16 bits, 10 bits used 6 bits No-op. CU_EGL_COLOR_FORMAT_BAYER10_RGGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER10_RGGB{{endif}} {{if True}} #: Bayer10 format - one channel in one surface with interleaved BGGR #: ordering. Out of 16 bits, 10 bits used 6 bits No-op. CU_EGL_COLOR_FORMAT_BAYER10_BGGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER10_BGGR{{endif}} {{if True}} #: Bayer10 format - one channel in one surface with interleaved GRBG #: ordering. Out of 16 bits, 10 bits used 6 bits No-op. CU_EGL_COLOR_FORMAT_BAYER10_GRBG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER10_GRBG{{endif}} {{if True}} #: Bayer10 format - one channel in one surface with interleaved GBRG #: ordering. Out of 16 bits, 10 bits used 6 bits No-op. CU_EGL_COLOR_FORMAT_BAYER10_GBRG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER10_GBRG{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved RGGB #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_RGGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_RGGB{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved BGGR #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_BGGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_BGGR{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved GRBG #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_GRBG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_GRBG{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved GBRG #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_GBRG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_GBRG{{endif}} {{if True}} #: Bayer14 format - one channel in one surface with interleaved RGGB #: ordering. Out of 16 bits, 14 bits used 2 bits No-op. CU_EGL_COLOR_FORMAT_BAYER14_RGGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER14_RGGB{{endif}} {{if True}} #: Bayer14 format - one channel in one surface with interleaved BGGR #: ordering. Out of 16 bits, 14 bits used 2 bits No-op. CU_EGL_COLOR_FORMAT_BAYER14_BGGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER14_BGGR{{endif}} {{if True}} #: Bayer14 format - one channel in one surface with interleaved GRBG #: ordering. Out of 16 bits, 14 bits used 2 bits No-op. CU_EGL_COLOR_FORMAT_BAYER14_GRBG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER14_GRBG{{endif}} {{if True}} #: Bayer14 format - one channel in one surface with interleaved GBRG #: ordering. Out of 16 bits, 14 bits used 2 bits No-op. CU_EGL_COLOR_FORMAT_BAYER14_GBRG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER14_GBRG{{endif}} {{if True}} #: Bayer20 format - one channel in one surface with interleaved RGGB #: ordering. Out of 32 bits, 20 bits used 12 bits No-op. CU_EGL_COLOR_FORMAT_BAYER20_RGGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER20_RGGB{{endif}} {{if True}} #: Bayer20 format - one channel in one surface with interleaved BGGR #: ordering. Out of 32 bits, 20 bits used 12 bits No-op. CU_EGL_COLOR_FORMAT_BAYER20_BGGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER20_BGGR{{endif}} {{if True}} #: Bayer20 format - one channel in one surface with interleaved GRBG #: ordering. Out of 32 bits, 20 bits used 12 bits No-op. CU_EGL_COLOR_FORMAT_BAYER20_GRBG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER20_GRBG{{endif}} {{if True}} #: Bayer20 format - one channel in one surface with interleaved GBRG #: ordering. Out of 32 bits, 20 bits used 12 bits No-op. CU_EGL_COLOR_FORMAT_BAYER20_GBRG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER20_GBRG{{endif}} {{if True}} #: Y, V, U in three surfaces, each in a separate surface, U/V width = Y #: width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YVU444_PLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU444_PLANAR{{endif}} {{if True}} #: Y, V, U in three surfaces, each in a separate surface, U/V width = #: 1/2 Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_YVU422_PLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU422_PLANAR{{endif}} {{if True}} #: Y, V, U in three surfaces, each in a separate surface, U/V width = #: 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_PLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_PLANAR{{endif}} {{if True}} #: Nvidia proprietary Bayer ISP format - one channel in one surface #: with interleaved RGGB ordering and mapped to opaque integer #: datatype. CU_EGL_COLOR_FORMAT_BAYER_ISP_RGGB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_ISP_RGGB{{endif}} {{if True}} #: Nvidia proprietary Bayer ISP format - one channel in one surface #: with interleaved BGGR ordering and mapped to opaque integer #: datatype. CU_EGL_COLOR_FORMAT_BAYER_ISP_BGGR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_ISP_BGGR{{endif}} {{if True}} #: Nvidia proprietary Bayer ISP format - one channel in one surface #: with interleaved GRBG ordering and mapped to opaque integer #: datatype. CU_EGL_COLOR_FORMAT_BAYER_ISP_GRBG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_ISP_GRBG{{endif}} {{if True}} #: Nvidia proprietary Bayer ISP format - one channel in one surface #: with interleaved GBRG ordering and mapped to opaque integer #: datatype. CU_EGL_COLOR_FORMAT_BAYER_ISP_GBRG = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_ISP_GBRG{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved BCCR #: ordering. CU_EGL_COLOR_FORMAT_BAYER_BCCR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_BCCR{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved RCCB #: ordering. CU_EGL_COLOR_FORMAT_BAYER_RCCB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_RCCB{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved CRBC #: ordering. CU_EGL_COLOR_FORMAT_BAYER_CRBC = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_CRBC{{endif}} {{if True}} #: Bayer format - one channel in one surface with interleaved CBRC #: ordering. CU_EGL_COLOR_FORMAT_BAYER_CBRC = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER_CBRC{{endif}} {{if True}} #: Bayer10 format - one channel in one surface with interleaved CCCC #: ordering. Out of 16 bits, 10 bits used 6 bits No-op. CU_EGL_COLOR_FORMAT_BAYER10_CCCC = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER10_CCCC{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved BCCR #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_BCCR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_BCCR{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved RCCB #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_RCCB = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_RCCB{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved CRBC #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_CRBC = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_CRBC{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved CBRC #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_CBRC = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_CBRC{{endif}} {{if True}} #: Bayer12 format - one channel in one surface with interleaved CCCC #: ordering. Out of 16 bits, 12 bits used 4 bits No-op. CU_EGL_COLOR_FORMAT_BAYER12_CCCC = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_BAYER12_CCCC{{endif}} {{if True}} #: Color format for single Y plane. CU_EGL_COLOR_FORMAT_Y = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y{{endif}} {{if True}} #: Y, UV in two surfaces (UV as one surface) U/V width = 1/2 Y width, #: U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR_2020 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR_2020{{endif}} {{if True}} #: Y, VU in two surfaces (VU as one surface) U/V width = 1/2 Y width, #: U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR_2020 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR_2020{{endif}} {{if True}} #: Y, U, V each in a separate surface, U/V width = 1/2 Y width, U/V #: height= 1/2 Y height. CU_EGL_COLOR_FORMAT_YUV420_PLANAR_2020 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_PLANAR_2020{{endif}} {{if True}} #: Y, V, U each in a separate surface, U/V width = 1/2 Y width, U/V #: height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_PLANAR_2020 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_PLANAR_2020{{endif}} {{if True}} #: Y, UV in two surfaces (UV as one surface) U/V width = 1/2 Y width, #: U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR_709 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_SEMIPLANAR_709{{endif}} {{if True}} #: Y, VU in two surfaces (VU as one surface) U/V width = 1/2 Y width, #: U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR_709 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_SEMIPLANAR_709{{endif}} {{if True}} #: Y, U, V each in a separate surface, U/V width = 1/2 Y width, U/V #: height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YUV420_PLANAR_709 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV420_PLANAR_709{{endif}} {{if True}} #: Y, V, U each in a separate surface, U/V width = 1/2 Y width, U/V #: height = 1/2 Y height. CU_EGL_COLOR_FORMAT_YVU420_PLANAR_709 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVU420_PLANAR_709{{endif}} {{if True}} #: Y10, V10U10 in two surfaces (VU as one surface), U/V width = 1/2 Y #: width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_709 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_709{{endif}} {{if True}} #: Y10, V10U10 in two surfaces (VU as one surface), U/V width = 1/2 Y #: width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_2020 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_2020{{endif}} {{if True}} #: Y10, V10U10 in two surfaces(VU as one surface) U/V width = 1/2 Y #: width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_422_SEMIPLANAR_2020 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_422_SEMIPLANAR_2020{{endif}} {{if True}} #: Y10, V10U10 in two surfaces(VU as one surface) U/V width = 1/2 Y #: width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_422_SEMIPLANAR = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_422_SEMIPLANAR{{endif}} {{if True}} #: Y10, V10U10 in two surfaces(VU as one surface) U/V width = 1/2 Y #: width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_422_SEMIPLANAR_709 = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_422_SEMIPLANAR_709{{endif}} {{if True}} #: Extended Range Color format for single Y plane. CU_EGL_COLOR_FORMAT_Y_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y_ER{{endif}} {{if True}} #: Extended Range Color format for single Y plane. CU_EGL_COLOR_FORMAT_Y_709_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y_709_ER{{endif}} {{if True}} #: Extended Range Color format for single Y10 plane. CU_EGL_COLOR_FORMAT_Y10_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10_ER{{endif}} {{if True}} #: Extended Range Color format for single Y10 plane. CU_EGL_COLOR_FORMAT_Y10_709_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10_709_ER{{endif}} {{if True}} #: Extended Range Color format for single Y12 plane. CU_EGL_COLOR_FORMAT_Y12_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12_ER{{endif}} {{if True}} #: Extended Range Color format for single Y12 plane. CU_EGL_COLOR_FORMAT_Y12_709_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12_709_ER{{endif}} {{if True}} #: Y, U, V, A four channels in one surface, interleaved as AVUY. CU_EGL_COLOR_FORMAT_YUVA = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUVA{{endif}} {{if True}} #: Y, U, V three channels in one surface, interleaved as VUY. Only #: pitch linear format supported. CU_EGL_COLOR_FORMAT_YUV = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YUV{{endif}} {{if True}} #: Y, U, V in one surface, interleaved as YVYU in one channel. CU_EGL_COLOR_FORMAT_YVYU = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_YVYU{{endif}} {{if True}} #: Y, U, V in one surface, interleaved as VYUY in one channel. CU_EGL_COLOR_FORMAT_VYUY = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_VYUY{{endif}} {{if True}} #: Extended Range Y10, V10U10 in two surfaces(VU as one surface) U/V #: width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y10, V10U10 in two surfaces(VU as one surface) U/V #: width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_709_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_420_SEMIPLANAR_709_ER{{endif}} {{if True}} #: Extended Range Y10, V10U10 in two surfaces (VU as one surface) U/V #: width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_444_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_444_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y10, V10U10 in two surfaces (VU as one surface) U/V #: width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y10V10U10_444_SEMIPLANAR_709_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y10V10U10_444_SEMIPLANAR_709_ER{{endif}} {{if True}} #: Extended Range Y12, V12U12 in two surfaces (VU as one surface) U/V #: width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y12V12U12_420_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12V12U12_420_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y12, V12U12 in two surfaces (VU as one surface) U/V #: width = 1/2 Y width, U/V height = 1/2 Y height. CU_EGL_COLOR_FORMAT_Y12V12U12_420_SEMIPLANAR_709_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12V12U12_420_SEMIPLANAR_709_ER{{endif}} {{if True}} #: Extended Range Y12, V12U12 in two surfaces (VU as one surface) U/V #: width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y12V12U12_444_SEMIPLANAR_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12V12U12_444_SEMIPLANAR_ER{{endif}} {{if True}} #: Extended Range Y12, V12U12 in two surfaces (VU as one surface) U/V #: width = Y width, U/V height = Y height. CU_EGL_COLOR_FORMAT_Y12V12U12_444_SEMIPLANAR_709_ER = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_Y12V12U12_444_SEMIPLANAR_709_ER{{endif}} {{if True}} CU_EGL_COLOR_FORMAT_MAX = ccuda.CUeglColorFormat_enum.CU_EGL_COLOR_FORMAT_MAX{{endif}} {{endif}} {{if True}} class CUGLDeviceList(IntEnum): """ CUDA devices corresponding to an OpenGL device """ {{if True}} #: The CUDA devices for all GPUs used by the current OpenGL context CU_GL_DEVICE_LIST_ALL = ccuda.CUGLDeviceList_enum.CU_GL_DEVICE_LIST_ALL{{endif}} {{if True}} #: The CUDA devices for the GPUs used by the current OpenGL context in #: its currently rendering frame CU_GL_DEVICE_LIST_CURRENT_FRAME = ccuda.CUGLDeviceList_enum.CU_GL_DEVICE_LIST_CURRENT_FRAME{{endif}} {{if True}} #: The CUDA devices for the GPUs to be used by the current OpenGL #: context in the next frame CU_GL_DEVICE_LIST_NEXT_FRAME = ccuda.CUGLDeviceList_enum.CU_GL_DEVICE_LIST_NEXT_FRAME{{endif}} {{endif}} {{if True}} class CUGLmap_flags(IntEnum): """ Flags to map or unmap a resource """ {{if True}} CU_GL_MAP_RESOURCE_FLAGS_NONE = ccuda.CUGLmap_flags_enum.CU_GL_MAP_RESOURCE_FLAGS_NONE{{endif}} {{if True}} CU_GL_MAP_RESOURCE_FLAGS_READ_ONLY = ccuda.CUGLmap_flags_enum.CU_GL_MAP_RESOURCE_FLAGS_READ_ONLY{{endif}} {{if True}} CU_GL_MAP_RESOURCE_FLAGS_WRITE_DISCARD = ccuda.CUGLmap_flags_enum.CU_GL_MAP_RESOURCE_FLAGS_WRITE_DISCARD{{endif}} {{endif}} {{if 'CUdeviceptr' in found_types}} cdef class CUdeviceptr: """ CUDA device pointer CUdeviceptr is defined as an unsigned integer type whose size matches the size of a pointer on the target platform. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUdevice' in found_types}} cdef class CUdevice: """ CUDA device Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, int init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUtexObject' in found_types}} cdef class CUtexObject: """ An opaque value that represents a CUDA texture object Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUsurfObject' in found_types}} cdef class CUsurfObject: """ An opaque value that represents a CUDA surface object Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUgraphConditionalHandle' in found_types}} cdef class CUgraphConditionalHandle: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, uint64_t init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUlaunchAttributeID_enum' in found_types}} class CUkernelNodeAttrID(IntEnum): """ Launch attributes enum; used as id field of :py:obj:`~.CUlaunchAttribute` """ {{if 'CU_LAUNCH_ATTRIBUTE_IGNORE' in found_values}} #: Ignored entry, for convenient composition CU_LAUNCH_ATTRIBUTE_IGNORE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_IGNORE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.accessPolicyWindow`. CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_COOPERATIVE' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.cooperative`. CU_LAUNCH_ATTRIBUTE_COOPERATIVE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_COOPERATIVE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY' in found_values}} #: Valid for streams. See #: :py:obj:`~.CUlaunchAttributeValue.syncPolicy`. CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.clusterDim`. CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.clusterSchedulingPolicyPreference`. CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION' in found_values}} #: Valid for launches. Setting #: :py:obj:`~.CUlaunchAttributeValue.programmaticStreamSerializationAllowed` #: to non-0 signals that the kernel will use programmatic means to #: resolve its stream dependency, so that the CUDA runtime should #: opportunistically allow the grid's execution to overlap with the #: previous kernel in the stream, if that kernel requests the overlap. #: The dependent launches can choose to wait on the dependency using #: the programmatic sync (cudaGridDependencySynchronize() or equivalent #: PTX instructions). CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT' in found_values}} #: Valid for launches. Set #: :py:obj:`~.CUlaunchAttributeValue.programmaticEvent` to record the #: event. Event recorded through this launch attribute is guaranteed to #: only trigger after all block in the associated kernel trigger the #: event. A block can trigger the event through PTX launchdep.release #: or CUDA builtin function cudaTriggerProgrammaticLaunchCompletion(). #: A trigger can also be inserted at the beginning of each block's #: execution if triggerAtBlockStart is set to non-0. The dependent #: launches can choose to wait on the dependency using the programmatic #: sync (cudaGridDependencySynchronize() or equivalent PTX #: instructions). Note that dependents (including the CPU thread #: calling :py:obj:`~.cuEventSynchronize()`) are not guaranteed to #: observe the release precisely when it is released. For example, #: :py:obj:`~.cuEventSynchronize()` may only observe the event trigger #: long after the associated kernel has completed. This recording type #: is primarily meant for establishing programmatic dependency between #: device tasks. Note also this type of dependency allows, but does not #: guarantee, concurrent execution of tasks. #: The event supplied must not be an interprocess or interop event. #: The event must disable timing (i.e. must be created with the #: :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PRIORITY' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.priority`. CU_LAUNCH_ATTRIBUTE_PRIORITY = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PRIORITY{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.memSyncDomainMap`. CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.memSyncDomain`. CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT' in found_values}} #: Valid for launches. Set #: :py:obj:`~.CUlaunchAttributeValue.launchCompletionEvent` to record #: the event. #: Nominally, the event is triggered once all blocks of the kernel #: have begun execution. Currently this is a best effort. If a kernel B #: has a launch completion dependency on a kernel A, B may wait until A #: is complete. Alternatively, blocks of B may begin before all blocks #: of A have begun, for example if B can claim execution resources #: unavailable to A (e.g. they run on different GPUs) or if B is a #: higher priority than A. Exercise caution if such an ordering #: inversion could lead to deadlock. #: A launch completion event is nominally similar to a programmatic #: event with `triggerAtBlockStart` set except that it is not visible #: to `cudaGridDependencySynchronize()` and can be used with compute #: capability less than 9.0. #: The event supplied must not be an interprocess or interop event. #: The event must disable timing (i.e. must be created with the #: :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE' in found_values}} #: Valid for graph nodes, launches. This attribute is graphs-only, and #: passing it to a launch in a non-capturing stream will result in an #: error. #: :py:obj:`~.CUlaunchAttributeValue`::deviceUpdatableKernelNode::deviceUpdatable #: can only be set to 0 or 1. Setting the field to 1 indicates that the #: corresponding kernel node should be device-updatable. On success, a #: handle will be returned via #: :py:obj:`~.CUlaunchAttributeValue`::deviceUpdatableKernelNode::devNode #: which can be passed to the various device-side update functions to #: update the node's kernel parameters from within another kernel. For #: more information on the types of device updates that can be made, as #: well as the relevant limitations thereof, see #: :py:obj:`~.cudaGraphKernelNodeUpdatesApply`. #: Nodes which are device-updatable have additional restrictions #: compared to regular kernel nodes. Firstly, device-updatable nodes #: cannot be removed from their graph via #: :py:obj:`~.cuGraphDestroyNode`. Additionally, once opted-in to this #: functionality, a node cannot opt out, and any attempt to set the #: deviceUpdatable attribute to 0 will result in an error. Device- #: updatable kernel nodes also cannot have their attributes copied #: to/from another kernel node via #: :py:obj:`~.cuGraphKernelNodeCopyAttributes`. Graphs containing one #: or more device-updatable nodes also do not allow multiple #: instantiation, and neither the graph nor its instantiated version #: can be passed to :py:obj:`~.cuGraphExecUpdate`. #: If a graph contains device-updatable nodes and updates those nodes #: from the device from within the graph, the graph must be uploaded #: with :py:obj:`~.cuGraphUpload` before it is launched. For such a #: graph, if host-side executable graph updates are made to the device- #: updatable nodes, the graph must be uploaded before it is launched #: again. CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT' in found_values}} #: Valid for launches. On devices where the L1 cache and shared memory #: use the same hardware resources, setting #: :py:obj:`~.CUlaunchAttributeValue.sharedMemCarveout` to a percentage #: between 0-100 signals the CUDA driver to set the shared memory #: carveout preference, in percent of the total shared memory for that #: kernel launch. This attribute takes precedence over #: :py:obj:`~.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT`. This #: is only a hint, and the CUDA driver can choose a different #: configuration if required for the launch. CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT{{endif}} {{endif}} {{if 'CUlaunchAttributeID_enum' in found_types}} class CUstreamAttrID(IntEnum): """ Launch attributes enum; used as id field of :py:obj:`~.CUlaunchAttribute` """ {{if 'CU_LAUNCH_ATTRIBUTE_IGNORE' in found_values}} #: Ignored entry, for convenient composition CU_LAUNCH_ATTRIBUTE_IGNORE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_IGNORE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.accessPolicyWindow`. CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_COOPERATIVE' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.cooperative`. CU_LAUNCH_ATTRIBUTE_COOPERATIVE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_COOPERATIVE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY' in found_values}} #: Valid for streams. See #: :py:obj:`~.CUlaunchAttributeValue.syncPolicy`. CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.clusterDim`. CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE' in found_values}} #: Valid for graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.clusterSchedulingPolicyPreference`. CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION' in found_values}} #: Valid for launches. Setting #: :py:obj:`~.CUlaunchAttributeValue.programmaticStreamSerializationAllowed` #: to non-0 signals that the kernel will use programmatic means to #: resolve its stream dependency, so that the CUDA runtime should #: opportunistically allow the grid's execution to overlap with the #: previous kernel in the stream, if that kernel requests the overlap. #: The dependent launches can choose to wait on the dependency using #: the programmatic sync (cudaGridDependencySynchronize() or equivalent #: PTX instructions). CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT' in found_values}} #: Valid for launches. Set #: :py:obj:`~.CUlaunchAttributeValue.programmaticEvent` to record the #: event. Event recorded through this launch attribute is guaranteed to #: only trigger after all block in the associated kernel trigger the #: event. A block can trigger the event through PTX launchdep.release #: or CUDA builtin function cudaTriggerProgrammaticLaunchCompletion(). #: A trigger can also be inserted at the beginning of each block's #: execution if triggerAtBlockStart is set to non-0. The dependent #: launches can choose to wait on the dependency using the programmatic #: sync (cudaGridDependencySynchronize() or equivalent PTX #: instructions). Note that dependents (including the CPU thread #: calling :py:obj:`~.cuEventSynchronize()`) are not guaranteed to #: observe the release precisely when it is released. For example, #: :py:obj:`~.cuEventSynchronize()` may only observe the event trigger #: long after the associated kernel has completed. This recording type #: is primarily meant for establishing programmatic dependency between #: device tasks. Note also this type of dependency allows, but does not #: guarantee, concurrent execution of tasks. #: The event supplied must not be an interprocess or interop event. #: The event must disable timing (i.e. must be created with the #: :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PRIORITY' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.priority`. CU_LAUNCH_ATTRIBUTE_PRIORITY = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PRIORITY{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.memSyncDomainMap`. CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN' in found_values}} #: Valid for streams, graph nodes, launches. See #: :py:obj:`~.CUlaunchAttributeValue.memSyncDomain`. CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT' in found_values}} #: Valid for launches. Set #: :py:obj:`~.CUlaunchAttributeValue.launchCompletionEvent` to record #: the event. #: Nominally, the event is triggered once all blocks of the kernel #: have begun execution. Currently this is a best effort. If a kernel B #: has a launch completion dependency on a kernel A, B may wait until A #: is complete. Alternatively, blocks of B may begin before all blocks #: of A have begun, for example if B can claim execution resources #: unavailable to A (e.g. they run on different GPUs) or if B is a #: higher priority than A. Exercise caution if such an ordering #: inversion could lead to deadlock. #: A launch completion event is nominally similar to a programmatic #: event with `triggerAtBlockStart` set except that it is not visible #: to `cudaGridDependencySynchronize()` and can be used with compute #: capability less than 9.0. #: The event supplied must not be an interprocess or interop event. #: The event must disable timing (i.e. must be created with the #: :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE' in found_values}} #: Valid for graph nodes, launches. This attribute is graphs-only, and #: passing it to a launch in a non-capturing stream will result in an #: error. #: :py:obj:`~.CUlaunchAttributeValue`::deviceUpdatableKernelNode::deviceUpdatable #: can only be set to 0 or 1. Setting the field to 1 indicates that the #: corresponding kernel node should be device-updatable. On success, a #: handle will be returned via #: :py:obj:`~.CUlaunchAttributeValue`::deviceUpdatableKernelNode::devNode #: which can be passed to the various device-side update functions to #: update the node's kernel parameters from within another kernel. For #: more information on the types of device updates that can be made, as #: well as the relevant limitations thereof, see #: :py:obj:`~.cudaGraphKernelNodeUpdatesApply`. #: Nodes which are device-updatable have additional restrictions #: compared to regular kernel nodes. Firstly, device-updatable nodes #: cannot be removed from their graph via #: :py:obj:`~.cuGraphDestroyNode`. Additionally, once opted-in to this #: functionality, a node cannot opt out, and any attempt to set the #: deviceUpdatable attribute to 0 will result in an error. Device- #: updatable kernel nodes also cannot have their attributes copied #: to/from another kernel node via #: :py:obj:`~.cuGraphKernelNodeCopyAttributes`. Graphs containing one #: or more device-updatable nodes also do not allow multiple #: instantiation, and neither the graph nor its instantiated version #: can be passed to :py:obj:`~.cuGraphExecUpdate`. #: If a graph contains device-updatable nodes and updates those nodes #: from the device from within the graph, the graph must be uploaded #: with :py:obj:`~.cuGraphUpload` before it is launched. For such a #: graph, if host-side executable graph updates are made to the device- #: updatable nodes, the graph must be uploaded before it is launched #: again. CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE{{endif}} {{if 'CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT' in found_values}} #: Valid for launches. On devices where the L1 cache and shared memory #: use the same hardware resources, setting #: :py:obj:`~.CUlaunchAttributeValue.sharedMemCarveout` to a percentage #: between 0-100 signals the CUDA driver to set the shared memory #: carveout preference, in percent of the total shared memory for that #: kernel launch. This attribute takes precedence over #: :py:obj:`~.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT`. This #: is only a hint, and the CUDA driver can choose a different #: configuration if required for the launch. CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT = ccuda.CUlaunchAttributeID_enum.CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT{{endif}} {{endif}} {{if 'CUmemGenericAllocationHandle' in found_types}} cdef class CUmemGenericAllocationHandle: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUcontext' in found_types}} cdef class CUcontext: """ A regular context handle Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUmodule' in found_types}} cdef class CUmodule: """ CUDA module Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUfunction' in found_types}} cdef class CUfunction: """ CUDA function Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUlibrary' in found_types}} cdef class CUlibrary: """ CUDA library Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUkernel' in found_types}} cdef class CUkernel: """ CUDA kernel Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUarray' in found_types}} cdef class CUarray: """ CUDA array Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUmipmappedArray' in found_types}} cdef class CUmipmappedArray: """ CUDA mipmapped array Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUtexref' in found_types}} cdef class CUtexref: """ CUDA texture reference Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUsurfref' in found_types}} cdef class CUsurfref: """ CUDA surface reference Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUevent' in found_types}} cdef class CUevent: """ CUDA event Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUstream' in found_types}} cdef class CUstream: """ CUDA stream Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUgraphicsResource' in found_types}} cdef class CUgraphicsResource: """ CUDA graphics interop resource Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUexternalMemory' in found_types}} cdef class CUexternalMemory: """ CUDA external memory Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUexternalSemaphore' in found_types}} cdef class CUexternalSemaphore: """ CUDA external semaphore Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUgraph' in found_types}} cdef class CUgraph: """ CUDA graph Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUgraphNode' in found_types}} cdef class CUgraphNode: """ CUDA graph node Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUgraphExec' in found_types}} cdef class CUgraphExec: """ CUDA executable graph Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUmemoryPool' in found_types}} cdef class CUmemoryPool: """ CUDA memory pool Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUuserObject' in found_types}} cdef class CUuserObject: """ CUDA user object for graphs Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUgraphDeviceNode' in found_types}} cdef class CUgraphDeviceNode: """ CUDA graph device node handle Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUasyncCallbackHandle' in found_types}} cdef class CUasyncCallbackHandle: """ CUDA async notification callback handle Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUgreenCtx' in found_types}} cdef class CUgreenCtx: """ A green context handle. This handle can be used safely from only one CPU thread at a time. Created via cuGreenCtxCreate Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUlinkState' in found_types}} cdef class CUlinkState: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): self._keepalive = [] def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUdevResourceDesc' in found_types}} cdef class CUdevResourceDesc: """ An opaque descriptor handle. The descriptor encapsulates multiple created and configured resources. Created via cuDevResourceGenerateDesc Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class CUeglStreamConnection: """ CUDA EGLSream Connection Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class EGLImageKHR: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class EGLStreamKHR: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class EGLSyncKHR: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUasyncCallback' in found_types}} cdef class CUasyncCallback: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUhostFn' in found_types}} cdef class CUhostFn: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUstreamCallback' in found_types}} cdef class CUstreamCallback: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUoccupancyB2DSize' in found_types}} cdef class CUoccupancyB2DSize: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val self._ptr[0] = init_value else: self._ptr = _ptr def __init__(self, *args, **kwargs): pass def __repr__(self): return '' def __index__(self): return self.__int__() def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'struct CUuuid_st' in found_types}} cdef class CUuuid_st: """ Attributes ---------- bytes : bytes < CUDA definition of UUID Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['bytes : ' + str(self.bytes.hex())] except ValueError: str_list += ['bytes : '] return '\n'.join(str_list) else: return '' @property def bytes(self): return PyBytes_FromStringAndSize(self._ptr[0].bytes, 16) {{endif}} {{if 'struct CUmemFabricHandle_st' in found_types}} cdef class CUmemFabricHandle_st: """ Fabric handle - An opaque handle representing a memory allocation that can be exported to processes in same or different nodes. For IPC between processes on different nodes they must be connected via the NVSwitch fabric. Attributes ---------- data : bytes Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['data : ' + str(self.data)] except ValueError: str_list += ['data : '] return '\n'.join(str_list) else: return '' @property def data(self): return PyBytes_FromStringAndSize(self._ptr[0].data, 64) @data.setter def data(self, data): if len(data) != 64: raise ValueError("data length must be 64, is " + str(len(data))) for i, b in enumerate(data): self._ptr[0].data[i] = b {{endif}} {{if 'struct CUipcEventHandle_st' in found_types}} cdef class CUipcEventHandle_st: """ CUDA IPC event handle Attributes ---------- reserved : bytes Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def reserved(self): return PyBytes_FromStringAndSize(self._ptr[0].reserved, 64) @reserved.setter def reserved(self, reserved): if len(reserved) != 64: raise ValueError("reserved length must be 64, is " + str(len(reserved))) if CHAR_MIN == 0: for i, b in enumerate(reserved): if b < 0 and b > -129: b = b + 256 self._ptr[0].reserved[i] = b else: for i, b in enumerate(reserved): if b > 127 and b < 256: b = b - 256 self._ptr[0].reserved[i] = b {{endif}} {{if 'struct CUipcMemHandle_st' in found_types}} cdef class CUipcMemHandle_st: """ CUDA IPC mem handle Attributes ---------- reserved : bytes Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def reserved(self): return PyBytes_FromStringAndSize(self._ptr[0].reserved, 64) @reserved.setter def reserved(self, reserved): if len(reserved) != 64: raise ValueError("reserved length must be 64, is " + str(len(reserved))) if CHAR_MIN == 0: for i, b in enumerate(reserved): if b < 0 and b > -129: b = b + 256 self._ptr[0].reserved[i] = b else: for i, b in enumerate(reserved): if b > 127 and b < 256: b = b - 256 self._ptr[0].reserved[i] = b {{endif}} {{if 'union CUstreamBatchMemOpParams_union' in found_types}} cdef class CUstreamMemOpWaitValueParams_st: """ Attributes ---------- operation : CUstreamBatchMemOpType address : CUdeviceptr value : cuuint32_t value64 : cuuint64_t flags : unsigned int alias : CUdeviceptr For driver internal use. Initial value is unimportant. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._address = CUdeviceptr(_ptr=&self._ptr[0].waitValue.address) self._value = cuuint32_t(_ptr=&self._ptr[0].waitValue.value) self._value64 = cuuint64_t(_ptr=&self._ptr[0].waitValue.value64) self._alias = CUdeviceptr(_ptr=&self._ptr[0].waitValue.alias) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].waitValue def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['operation : ' + str(self.operation)] except ValueError: str_list += ['operation : '] try: str_list += ['address : ' + str(self.address)] except ValueError: str_list += ['address : '] try: str_list += ['value : ' + str(self.value)] except ValueError: str_list += ['value : '] try: str_list += ['value64 : ' + str(self.value64)] except ValueError: str_list += ['value64 : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['alias : ' + str(self.alias)] except ValueError: str_list += ['alias : '] return '\n'.join(str_list) else: return '' @property def operation(self): return CUstreamBatchMemOpType(self._ptr[0].waitValue.operation) @operation.setter def operation(self, operation not None : CUstreamBatchMemOpType): self._ptr[0].waitValue.operation = operation.value @property def address(self): return self._address @address.setter def address(self, address): cdef ccuda.CUdeviceptr caddress if address is None: caddress = 0 elif isinstance(address, (CUdeviceptr)): paddress = int(address) caddress = paddress else: paddress = int(CUdeviceptr(address)) caddress = paddress self._address._ptr[0] = caddress @property def value(self): return self._value @value.setter def value(self, value): cdef ccuda.cuuint32_t cvalue if value is None: cvalue = 0 elif isinstance(value, (cuuint32_t)): pvalue = int(value) cvalue = pvalue else: pvalue = int(cuuint32_t(value)) cvalue = pvalue self._value._ptr[0] = cvalue @property def value64(self): return self._value64 @value64.setter def value64(self, value64): cdef ccuda.cuuint64_t cvalue64 if value64 is None: cvalue64 = 0 elif isinstance(value64, (cuuint64_t)): pvalue64 = int(value64) cvalue64 = pvalue64 else: pvalue64 = int(cuuint64_t(value64)) cvalue64 = pvalue64 self._value64._ptr[0] = cvalue64 @property def flags(self): return self._ptr[0].waitValue.flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].waitValue.flags = flags @property def alias(self): return self._alias @alias.setter def alias(self, alias): cdef ccuda.CUdeviceptr calias if alias is None: calias = 0 elif isinstance(alias, (CUdeviceptr)): palias = int(alias) calias = palias else: palias = int(CUdeviceptr(alias)) calias = palias self._alias._ptr[0] = calias {{endif}} {{if 'union CUstreamBatchMemOpParams_union' in found_types}} cdef class CUstreamMemOpWriteValueParams_st: """ Attributes ---------- operation : CUstreamBatchMemOpType address : CUdeviceptr value : cuuint32_t value64 : cuuint64_t flags : unsigned int alias : CUdeviceptr For driver internal use. Initial value is unimportant. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._address = CUdeviceptr(_ptr=&self._ptr[0].writeValue.address) self._value = cuuint32_t(_ptr=&self._ptr[0].writeValue.value) self._value64 = cuuint64_t(_ptr=&self._ptr[0].writeValue.value64) self._alias = CUdeviceptr(_ptr=&self._ptr[0].writeValue.alias) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].writeValue def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['operation : ' + str(self.operation)] except ValueError: str_list += ['operation : '] try: str_list += ['address : ' + str(self.address)] except ValueError: str_list += ['address : '] try: str_list += ['value : ' + str(self.value)] except ValueError: str_list += ['value : '] try: str_list += ['value64 : ' + str(self.value64)] except ValueError: str_list += ['value64 : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['alias : ' + str(self.alias)] except ValueError: str_list += ['alias : '] return '\n'.join(str_list) else: return '' @property def operation(self): return CUstreamBatchMemOpType(self._ptr[0].writeValue.operation) @operation.setter def operation(self, operation not None : CUstreamBatchMemOpType): self._ptr[0].writeValue.operation = operation.value @property def address(self): return self._address @address.setter def address(self, address): cdef ccuda.CUdeviceptr caddress if address is None: caddress = 0 elif isinstance(address, (CUdeviceptr)): paddress = int(address) caddress = paddress else: paddress = int(CUdeviceptr(address)) caddress = paddress self._address._ptr[0] = caddress @property def value(self): return self._value @value.setter def value(self, value): cdef ccuda.cuuint32_t cvalue if value is None: cvalue = 0 elif isinstance(value, (cuuint32_t)): pvalue = int(value) cvalue = pvalue else: pvalue = int(cuuint32_t(value)) cvalue = pvalue self._value._ptr[0] = cvalue @property def value64(self): return self._value64 @value64.setter def value64(self, value64): cdef ccuda.cuuint64_t cvalue64 if value64 is None: cvalue64 = 0 elif isinstance(value64, (cuuint64_t)): pvalue64 = int(value64) cvalue64 = pvalue64 else: pvalue64 = int(cuuint64_t(value64)) cvalue64 = pvalue64 self._value64._ptr[0] = cvalue64 @property def flags(self): return self._ptr[0].writeValue.flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].writeValue.flags = flags @property def alias(self): return self._alias @alias.setter def alias(self, alias): cdef ccuda.CUdeviceptr calias if alias is None: calias = 0 elif isinstance(alias, (CUdeviceptr)): palias = int(alias) calias = palias else: palias = int(CUdeviceptr(alias)) calias = palias self._alias._ptr[0] = calias {{endif}} {{if 'union CUstreamBatchMemOpParams_union' in found_types}} cdef class CUstreamMemOpFlushRemoteWritesParams_st: """ Attributes ---------- operation : CUstreamBatchMemOpType flags : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].flushRemoteWrites def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['operation : ' + str(self.operation)] except ValueError: str_list += ['operation : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def operation(self): return CUstreamBatchMemOpType(self._ptr[0].flushRemoteWrites.operation) @operation.setter def operation(self, operation not None : CUstreamBatchMemOpType): self._ptr[0].flushRemoteWrites.operation = operation.value @property def flags(self): return self._ptr[0].flushRemoteWrites.flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flushRemoteWrites.flags = flags {{endif}} {{if 'union CUstreamBatchMemOpParams_union' in found_types}} cdef class CUstreamMemOpMemoryBarrierParams_st: """ Attributes ---------- operation : CUstreamBatchMemOpType < Only supported in the _v2 API flags : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].memoryBarrier def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['operation : ' + str(self.operation)] except ValueError: str_list += ['operation : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def operation(self): return CUstreamBatchMemOpType(self._ptr[0].memoryBarrier.operation) @operation.setter def operation(self, operation not None : CUstreamBatchMemOpType): self._ptr[0].memoryBarrier.operation = operation.value @property def flags(self): return self._ptr[0].memoryBarrier.flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].memoryBarrier.flags = flags {{endif}} {{if 'union CUstreamBatchMemOpParams_union' in found_types}} cdef class CUstreamBatchMemOpParams_union: """ Per-operation parameters for cuStreamBatchMemOp Attributes ---------- operation : CUstreamBatchMemOpType waitValue : CUstreamMemOpWaitValueParams_st writeValue : CUstreamMemOpWriteValueParams_st flushRemoteWrites : CUstreamMemOpFlushRemoteWritesParams_st memoryBarrier : CUstreamMemOpMemoryBarrierParams_st pad : List[cuuint64_t] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._waitValue = CUstreamMemOpWaitValueParams_st(_ptr=self._ptr) self._writeValue = CUstreamMemOpWriteValueParams_st(_ptr=self._ptr) self._flushRemoteWrites = CUstreamMemOpFlushRemoteWritesParams_st(_ptr=self._ptr) self._memoryBarrier = CUstreamMemOpMemoryBarrierParams_st(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['operation : ' + str(self.operation)] except ValueError: str_list += ['operation : '] try: str_list += ['waitValue :\n' + '\n'.join([' ' + line for line in str(self.waitValue).splitlines()])] except ValueError: str_list += ['waitValue : '] try: str_list += ['writeValue :\n' + '\n'.join([' ' + line for line in str(self.writeValue).splitlines()])] except ValueError: str_list += ['writeValue : '] try: str_list += ['flushRemoteWrites :\n' + '\n'.join([' ' + line for line in str(self.flushRemoteWrites).splitlines()])] except ValueError: str_list += ['flushRemoteWrites : '] try: str_list += ['memoryBarrier :\n' + '\n'.join([' ' + line for line in str(self.memoryBarrier).splitlines()])] except ValueError: str_list += ['memoryBarrier : '] try: str_list += ['pad : ' + str(self.pad)] except ValueError: str_list += ['pad : '] return '\n'.join(str_list) else: return '' @property def operation(self): return CUstreamBatchMemOpType(self._ptr[0].operation) @operation.setter def operation(self, operation not None : CUstreamBatchMemOpType): self._ptr[0].operation = operation.value @property def waitValue(self): return self._waitValue @waitValue.setter def waitValue(self, waitValue not None : CUstreamMemOpWaitValueParams_st): string.memcpy(&self._ptr[0].waitValue, waitValue.getPtr(), sizeof(self._ptr[0].waitValue)) @property def writeValue(self): return self._writeValue @writeValue.setter def writeValue(self, writeValue not None : CUstreamMemOpWriteValueParams_st): string.memcpy(&self._ptr[0].writeValue, writeValue.getPtr(), sizeof(self._ptr[0].writeValue)) @property def flushRemoteWrites(self): return self._flushRemoteWrites @flushRemoteWrites.setter def flushRemoteWrites(self, flushRemoteWrites not None : CUstreamMemOpFlushRemoteWritesParams_st): string.memcpy(&self._ptr[0].flushRemoteWrites, flushRemoteWrites.getPtr(), sizeof(self._ptr[0].flushRemoteWrites)) @property def memoryBarrier(self): return self._memoryBarrier @memoryBarrier.setter def memoryBarrier(self, memoryBarrier not None : CUstreamMemOpMemoryBarrierParams_st): string.memcpy(&self._ptr[0].memoryBarrier, memoryBarrier.getPtr(), sizeof(self._ptr[0].memoryBarrier)) @property def pad(self): return [cuuint64_t(init_value=_pad) for _pad in self._ptr[0].pad] @pad.setter def pad(self, pad): self._ptr[0].pad = pad {{endif}} {{if 'struct CUDA_BATCH_MEM_OP_NODE_PARAMS_v1_st' in found_types}} cdef class CUDA_BATCH_MEM_OP_NODE_PARAMS_v1_st: """ Attributes ---------- ctx : CUcontext count : unsigned int paramArray : CUstreamBatchMemOpParams flags : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._ctx = CUcontext(_ptr=&self._ptr[0].ctx) def __dealloc__(self): if self._paramArray is not NULL: free(self._paramArray) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['ctx : ' + str(self.ctx)] except ValueError: str_list += ['ctx : '] try: str_list += ['count : ' + str(self.count)] except ValueError: str_list += ['count : '] try: str_list += ['paramArray : ' + str(self.paramArray)] except ValueError: str_list += ['paramArray : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def ctx(self): return self._ctx @ctx.setter def ctx(self, ctx): cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx self._ctx._ptr[0] = cctx @property def count(self): return self._ptr[0].count @count.setter def count(self, unsigned int count): self._ptr[0].count = count @property def paramArray(self): arrs = [self._ptr[0].paramArray + x*sizeof(ccuda.CUstreamBatchMemOpParams) for x in range(self._paramArray_length)] return [CUstreamBatchMemOpParams(_ptr=arr) for arr in arrs] @paramArray.setter def paramArray(self, val): if len(val) == 0: free(self._paramArray) self._paramArray_length = 0 self._ptr[0].paramArray = NULL else: if self._paramArray_length != len(val): free(self._paramArray) self._paramArray = calloc(len(val), sizeof(ccuda.CUstreamBatchMemOpParams)) if self._paramArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUstreamBatchMemOpParams))) self._paramArray_length = len(val) self._ptr[0].paramArray = self._paramArray for idx in range(len(val)): string.memcpy(&self._paramArray[idx], (val[idx])._ptr, sizeof(ccuda.CUstreamBatchMemOpParams)) @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags {{endif}} {{if 'struct CUDA_BATCH_MEM_OP_NODE_PARAMS_v2_st' in found_types}} cdef class CUDA_BATCH_MEM_OP_NODE_PARAMS_v2_st: """ Batch memory operation node parameters Attributes ---------- ctx : CUcontext Context to use for the operations. count : unsigned int Number of operations in paramArray. paramArray : CUstreamBatchMemOpParams Array of batch memory operations. flags : unsigned int Flags to control the node. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._ctx = CUcontext(_ptr=&self._ptr[0].ctx) def __dealloc__(self): if self._paramArray is not NULL: free(self._paramArray) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['ctx : ' + str(self.ctx)] except ValueError: str_list += ['ctx : '] try: str_list += ['count : ' + str(self.count)] except ValueError: str_list += ['count : '] try: str_list += ['paramArray : ' + str(self.paramArray)] except ValueError: str_list += ['paramArray : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def ctx(self): return self._ctx @ctx.setter def ctx(self, ctx): cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx self._ctx._ptr[0] = cctx @property def count(self): return self._ptr[0].count @count.setter def count(self, unsigned int count): self._ptr[0].count = count @property def paramArray(self): arrs = [self._ptr[0].paramArray + x*sizeof(ccuda.CUstreamBatchMemOpParams) for x in range(self._paramArray_length)] return [CUstreamBatchMemOpParams(_ptr=arr) for arr in arrs] @paramArray.setter def paramArray(self, val): if len(val) == 0: free(self._paramArray) self._paramArray_length = 0 self._ptr[0].paramArray = NULL else: if self._paramArray_length != len(val): free(self._paramArray) self._paramArray = calloc(len(val), sizeof(ccuda.CUstreamBatchMemOpParams)) if self._paramArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUstreamBatchMemOpParams))) self._paramArray_length = len(val) self._ptr[0].paramArray = self._paramArray for idx in range(len(val)): string.memcpy(&self._paramArray[idx], (val[idx])._ptr, sizeof(ccuda.CUstreamBatchMemOpParams)) @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags {{endif}} {{if 'struct CUasyncNotificationInfo_st' in found_types}} cdef class anon_struct0: """ Attributes ---------- bytesOverBudget : unsigned long long Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].info.overBudget def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['bytesOverBudget : ' + str(self.bytesOverBudget)] except ValueError: str_list += ['bytesOverBudget : '] return '\n'.join(str_list) else: return '' @property def bytesOverBudget(self): return self._ptr[0].info.overBudget.bytesOverBudget @bytesOverBudget.setter def bytesOverBudget(self, unsigned long long bytesOverBudget): self._ptr[0].info.overBudget.bytesOverBudget = bytesOverBudget {{endif}} {{if 'struct CUasyncNotificationInfo_st' in found_types}} cdef class anon_union2: """ Attributes ---------- overBudget : anon_struct0 Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._overBudget = anon_struct0(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].info def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['overBudget :\n' + '\n'.join([' ' + line for line in str(self.overBudget).splitlines()])] except ValueError: str_list += ['overBudget : '] return '\n'.join(str_list) else: return '' @property def overBudget(self): return self._overBudget @overBudget.setter def overBudget(self, overBudget not None : anon_struct0): string.memcpy(&self._ptr[0].info.overBudget, overBudget.getPtr(), sizeof(self._ptr[0].info.overBudget)) {{endif}} {{if 'struct CUasyncNotificationInfo_st' in found_types}} cdef class CUasyncNotificationInfo_st: """ Information passed to the user via the async notification callback Attributes ---------- type : CUasyncNotificationType info : anon_union2 Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUasyncNotificationInfo_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._info = anon_union2(_ptr=self._ptr) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['info :\n' + '\n'.join([' ' + line for line in str(self.info).splitlines()])] except ValueError: str_list += ['info : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUasyncNotificationType(self._ptr[0].type) @type.setter def type(self, type not None : CUasyncNotificationType): self._ptr[0].type = type.value @property def info(self): return self._info @info.setter def info(self, info not None : anon_union2): string.memcpy(&self._ptr[0].info, info.getPtr(), sizeof(self._ptr[0].info)) {{endif}} {{if 'struct CUdevprop_st' in found_types}} cdef class CUdevprop_st: """ Legacy device properties Attributes ---------- maxThreadsPerBlock : int Maximum number of threads per block maxThreadsDim : List[int] Maximum size of each dimension of a block maxGridSize : List[int] Maximum size of each dimension of a grid sharedMemPerBlock : int Shared memory available per block in bytes totalConstantMemory : int Constant memory available on device in bytes SIMDWidth : int Warp size in threads memPitch : int Maximum pitch in bytes allowed by memory copies regsPerBlock : int 32-bit registers available per block clockRate : int Clock frequency in kilohertz textureAlign : int Alignment requirement for textures Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['maxThreadsPerBlock : ' + str(self.maxThreadsPerBlock)] except ValueError: str_list += ['maxThreadsPerBlock : '] try: str_list += ['maxThreadsDim : ' + str(self.maxThreadsDim)] except ValueError: str_list += ['maxThreadsDim : '] try: str_list += ['maxGridSize : ' + str(self.maxGridSize)] except ValueError: str_list += ['maxGridSize : '] try: str_list += ['sharedMemPerBlock : ' + str(self.sharedMemPerBlock)] except ValueError: str_list += ['sharedMemPerBlock : '] try: str_list += ['totalConstantMemory : ' + str(self.totalConstantMemory)] except ValueError: str_list += ['totalConstantMemory : '] try: str_list += ['SIMDWidth : ' + str(self.SIMDWidth)] except ValueError: str_list += ['SIMDWidth : '] try: str_list += ['memPitch : ' + str(self.memPitch)] except ValueError: str_list += ['memPitch : '] try: str_list += ['regsPerBlock : ' + str(self.regsPerBlock)] except ValueError: str_list += ['regsPerBlock : '] try: str_list += ['clockRate : ' + str(self.clockRate)] except ValueError: str_list += ['clockRate : '] try: str_list += ['textureAlign : ' + str(self.textureAlign)] except ValueError: str_list += ['textureAlign : '] return '\n'.join(str_list) else: return '' @property def maxThreadsPerBlock(self): return self._ptr[0].maxThreadsPerBlock @maxThreadsPerBlock.setter def maxThreadsPerBlock(self, int maxThreadsPerBlock): self._ptr[0].maxThreadsPerBlock = maxThreadsPerBlock @property def maxThreadsDim(self): return self._ptr[0].maxThreadsDim @maxThreadsDim.setter def maxThreadsDim(self, maxThreadsDim): self._ptr[0].maxThreadsDim = maxThreadsDim @property def maxGridSize(self): return self._ptr[0].maxGridSize @maxGridSize.setter def maxGridSize(self, maxGridSize): self._ptr[0].maxGridSize = maxGridSize @property def sharedMemPerBlock(self): return self._ptr[0].sharedMemPerBlock @sharedMemPerBlock.setter def sharedMemPerBlock(self, int sharedMemPerBlock): self._ptr[0].sharedMemPerBlock = sharedMemPerBlock @property def totalConstantMemory(self): return self._ptr[0].totalConstantMemory @totalConstantMemory.setter def totalConstantMemory(self, int totalConstantMemory): self._ptr[0].totalConstantMemory = totalConstantMemory @property def SIMDWidth(self): return self._ptr[0].SIMDWidth @SIMDWidth.setter def SIMDWidth(self, int SIMDWidth): self._ptr[0].SIMDWidth = SIMDWidth @property def memPitch(self): return self._ptr[0].memPitch @memPitch.setter def memPitch(self, int memPitch): self._ptr[0].memPitch = memPitch @property def regsPerBlock(self): return self._ptr[0].regsPerBlock @regsPerBlock.setter def regsPerBlock(self, int regsPerBlock): self._ptr[0].regsPerBlock = regsPerBlock @property def clockRate(self): return self._ptr[0].clockRate @clockRate.setter def clockRate(self, int clockRate): self._ptr[0].clockRate = clockRate @property def textureAlign(self): return self._ptr[0].textureAlign @textureAlign.setter def textureAlign(self, int textureAlign): self._ptr[0].textureAlign = textureAlign {{endif}} {{if 'struct CUaccessPolicyWindow_st' in found_types}} cdef class CUaccessPolicyWindow_st: """ Specifies an access policy for a window, a contiguous extent of memory beginning at base_ptr and ending at base_ptr + num_bytes. num_bytes is limited by CU_DEVICE_ATTRIBUTE_MAX_ACCESS_POLICY_WINDOW_SIZE. Partition into many segments and assign segments such that: sum of "hit segments" / window == approx. ratio. sum of "miss segments" / window == approx 1-ratio. Segments and ratio specifications are fitted to the capabilities of the architecture. Accesses in a hit segment apply the hitProp access policy. Accesses in a miss segment apply the missProp access policy. Attributes ---------- base_ptr : Any Starting address of the access policy window. CUDA driver may align it. num_bytes : size_t Size in bytes of the window policy. CUDA driver may restrict the maximum size and alignment. hitRatio : float hitRatio specifies percentage of lines assigned hitProp, rest are assigned missProp. hitProp : CUaccessProperty CUaccessProperty set for hit. missProp : CUaccessProperty CUaccessProperty set for miss. Must be either NORMAL or STREAMING Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['base_ptr : ' + hex(self.base_ptr)] except ValueError: str_list += ['base_ptr : '] try: str_list += ['num_bytes : ' + str(self.num_bytes)] except ValueError: str_list += ['num_bytes : '] try: str_list += ['hitRatio : ' + str(self.hitRatio)] except ValueError: str_list += ['hitRatio : '] try: str_list += ['hitProp : ' + str(self.hitProp)] except ValueError: str_list += ['hitProp : '] try: str_list += ['missProp : ' + str(self.missProp)] except ValueError: str_list += ['missProp : '] return '\n'.join(str_list) else: return '' @property def base_ptr(self): return self._ptr[0].base_ptr @base_ptr.setter def base_ptr(self, base_ptr): _cbase_ptr = utils.HelperInputVoidPtr(base_ptr) self._ptr[0].base_ptr = _cbase_ptr.cptr @property def num_bytes(self): return self._ptr[0].num_bytes @num_bytes.setter def num_bytes(self, size_t num_bytes): self._ptr[0].num_bytes = num_bytes @property def hitRatio(self): return self._ptr[0].hitRatio @hitRatio.setter def hitRatio(self, float hitRatio): self._ptr[0].hitRatio = hitRatio @property def hitProp(self): return CUaccessProperty(self._ptr[0].hitProp) @hitProp.setter def hitProp(self, hitProp not None : CUaccessProperty): self._ptr[0].hitProp = hitProp.value @property def missProp(self): return CUaccessProperty(self._ptr[0].missProp) @missProp.setter def missProp(self, missProp not None : CUaccessProperty): self._ptr[0].missProp = missProp.value {{endif}} {{if 'struct CUDA_KERNEL_NODE_PARAMS_st' in found_types}} cdef class CUDA_KERNEL_NODE_PARAMS_st: """ GPU kernel node parameters Attributes ---------- func : CUfunction Kernel to launch gridDimX : unsigned int Width of grid in blocks gridDimY : unsigned int Height of grid in blocks gridDimZ : unsigned int Depth of grid in blocks blockDimX : unsigned int X dimension of each thread block blockDimY : unsigned int Y dimension of each thread block blockDimZ : unsigned int Z dimension of each thread block sharedMemBytes : unsigned int Dynamic shared-memory size per thread block in bytes kernelParams : Any Array of pointers to kernel parameters extra : Any Extra options Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._func = CUfunction(_ptr=&self._ptr[0].func) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['func : ' + str(self.func)] except ValueError: str_list += ['func : '] try: str_list += ['gridDimX : ' + str(self.gridDimX)] except ValueError: str_list += ['gridDimX : '] try: str_list += ['gridDimY : ' + str(self.gridDimY)] except ValueError: str_list += ['gridDimY : '] try: str_list += ['gridDimZ : ' + str(self.gridDimZ)] except ValueError: str_list += ['gridDimZ : '] try: str_list += ['blockDimX : ' + str(self.blockDimX)] except ValueError: str_list += ['blockDimX : '] try: str_list += ['blockDimY : ' + str(self.blockDimY)] except ValueError: str_list += ['blockDimY : '] try: str_list += ['blockDimZ : ' + str(self.blockDimZ)] except ValueError: str_list += ['blockDimZ : '] try: str_list += ['sharedMemBytes : ' + str(self.sharedMemBytes)] except ValueError: str_list += ['sharedMemBytes : '] try: str_list += ['kernelParams : ' + str(self.kernelParams)] except ValueError: str_list += ['kernelParams : '] try: str_list += ['extra : ' + str(self.extra)] except ValueError: str_list += ['extra : '] return '\n'.join(str_list) else: return '' @property def func(self): return self._func @func.setter def func(self, func): cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc self._func._ptr[0] = cfunc @property def gridDimX(self): return self._ptr[0].gridDimX @gridDimX.setter def gridDimX(self, unsigned int gridDimX): self._ptr[0].gridDimX = gridDimX @property def gridDimY(self): return self._ptr[0].gridDimY @gridDimY.setter def gridDimY(self, unsigned int gridDimY): self._ptr[0].gridDimY = gridDimY @property def gridDimZ(self): return self._ptr[0].gridDimZ @gridDimZ.setter def gridDimZ(self, unsigned int gridDimZ): self._ptr[0].gridDimZ = gridDimZ @property def blockDimX(self): return self._ptr[0].blockDimX @blockDimX.setter def blockDimX(self, unsigned int blockDimX): self._ptr[0].blockDimX = blockDimX @property def blockDimY(self): return self._ptr[0].blockDimY @blockDimY.setter def blockDimY(self, unsigned int blockDimY): self._ptr[0].blockDimY = blockDimY @property def blockDimZ(self): return self._ptr[0].blockDimZ @blockDimZ.setter def blockDimZ(self, unsigned int blockDimZ): self._ptr[0].blockDimZ = blockDimZ @property def sharedMemBytes(self): return self._ptr[0].sharedMemBytes @sharedMemBytes.setter def sharedMemBytes(self, unsigned int sharedMemBytes): self._ptr[0].sharedMemBytes = sharedMemBytes @property def kernelParams(self): return self._ptr[0].kernelParams @kernelParams.setter def kernelParams(self, kernelParams): self._ckernelParams = utils.HelperKernelParams(kernelParams) self._ptr[0].kernelParams = self._ckernelParams.ckernelParams @property def extra(self): return self._ptr[0].extra @extra.setter def extra(self, void_ptr extra): self._ptr[0].extra = extra {{endif}} {{if 'struct CUDA_KERNEL_NODE_PARAMS_v2_st' in found_types}} cdef class CUDA_KERNEL_NODE_PARAMS_v2_st: """ GPU kernel node parameters Attributes ---------- func : CUfunction Kernel to launch gridDimX : unsigned int Width of grid in blocks gridDimY : unsigned int Height of grid in blocks gridDimZ : unsigned int Depth of grid in blocks blockDimX : unsigned int X dimension of each thread block blockDimY : unsigned int Y dimension of each thread block blockDimZ : unsigned int Z dimension of each thread block sharedMemBytes : unsigned int Dynamic shared-memory size per thread block in bytes kernelParams : Any Array of pointers to kernel parameters extra : Any Extra options kern : CUkernel Kernel to launch, will only be referenced if func is NULL ctx : CUcontext Context for the kernel task to run in. The value NULL will indicate the current context should be used by the api. This field is ignored if func is set. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._func = CUfunction(_ptr=&self._ptr[0].func) self._kern = CUkernel(_ptr=&self._ptr[0].kern) self._ctx = CUcontext(_ptr=&self._ptr[0].ctx) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['func : ' + str(self.func)] except ValueError: str_list += ['func : '] try: str_list += ['gridDimX : ' + str(self.gridDimX)] except ValueError: str_list += ['gridDimX : '] try: str_list += ['gridDimY : ' + str(self.gridDimY)] except ValueError: str_list += ['gridDimY : '] try: str_list += ['gridDimZ : ' + str(self.gridDimZ)] except ValueError: str_list += ['gridDimZ : '] try: str_list += ['blockDimX : ' + str(self.blockDimX)] except ValueError: str_list += ['blockDimX : '] try: str_list += ['blockDimY : ' + str(self.blockDimY)] except ValueError: str_list += ['blockDimY : '] try: str_list += ['blockDimZ : ' + str(self.blockDimZ)] except ValueError: str_list += ['blockDimZ : '] try: str_list += ['sharedMemBytes : ' + str(self.sharedMemBytes)] except ValueError: str_list += ['sharedMemBytes : '] try: str_list += ['kernelParams : ' + str(self.kernelParams)] except ValueError: str_list += ['kernelParams : '] try: str_list += ['extra : ' + str(self.extra)] except ValueError: str_list += ['extra : '] try: str_list += ['kern : ' + str(self.kern)] except ValueError: str_list += ['kern : '] try: str_list += ['ctx : ' + str(self.ctx)] except ValueError: str_list += ['ctx : '] return '\n'.join(str_list) else: return '' @property def func(self): return self._func @func.setter def func(self, func): cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc self._func._ptr[0] = cfunc @property def gridDimX(self): return self._ptr[0].gridDimX @gridDimX.setter def gridDimX(self, unsigned int gridDimX): self._ptr[0].gridDimX = gridDimX @property def gridDimY(self): return self._ptr[0].gridDimY @gridDimY.setter def gridDimY(self, unsigned int gridDimY): self._ptr[0].gridDimY = gridDimY @property def gridDimZ(self): return self._ptr[0].gridDimZ @gridDimZ.setter def gridDimZ(self, unsigned int gridDimZ): self._ptr[0].gridDimZ = gridDimZ @property def blockDimX(self): return self._ptr[0].blockDimX @blockDimX.setter def blockDimX(self, unsigned int blockDimX): self._ptr[0].blockDimX = blockDimX @property def blockDimY(self): return self._ptr[0].blockDimY @blockDimY.setter def blockDimY(self, unsigned int blockDimY): self._ptr[0].blockDimY = blockDimY @property def blockDimZ(self): return self._ptr[0].blockDimZ @blockDimZ.setter def blockDimZ(self, unsigned int blockDimZ): self._ptr[0].blockDimZ = blockDimZ @property def sharedMemBytes(self): return self._ptr[0].sharedMemBytes @sharedMemBytes.setter def sharedMemBytes(self, unsigned int sharedMemBytes): self._ptr[0].sharedMemBytes = sharedMemBytes @property def kernelParams(self): return self._ptr[0].kernelParams @kernelParams.setter def kernelParams(self, kernelParams): self._ckernelParams = utils.HelperKernelParams(kernelParams) self._ptr[0].kernelParams = self._ckernelParams.ckernelParams @property def extra(self): return self._ptr[0].extra @extra.setter def extra(self, void_ptr extra): self._ptr[0].extra = extra @property def kern(self): return self._kern @kern.setter def kern(self, kern): cdef ccuda.CUkernel ckern if kern is None: ckern = 0 elif isinstance(kern, (CUkernel,)): pkern = int(kern) ckern = pkern else: pkern = int(CUkernel(kern)) ckern = pkern self._kern._ptr[0] = ckern @property def ctx(self): return self._ctx @ctx.setter def ctx(self, ctx): cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx self._ctx._ptr[0] = cctx {{endif}} {{if 'struct CUDA_KERNEL_NODE_PARAMS_v3_st' in found_types}} cdef class CUDA_KERNEL_NODE_PARAMS_v3_st: """ GPU kernel node parameters Attributes ---------- func : CUfunction Kernel to launch gridDimX : unsigned int Width of grid in blocks gridDimY : unsigned int Height of grid in blocks gridDimZ : unsigned int Depth of grid in blocks blockDimX : unsigned int X dimension of each thread block blockDimY : unsigned int Y dimension of each thread block blockDimZ : unsigned int Z dimension of each thread block sharedMemBytes : unsigned int Dynamic shared-memory size per thread block in bytes kernelParams : Any Array of pointers to kernel parameters extra : Any Extra options kern : CUkernel Kernel to launch, will only be referenced if func is NULL ctx : CUcontext Context for the kernel task to run in. The value NULL will indicate the current context should be used by the api. This field is ignored if func is set. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._func = CUfunction(_ptr=&self._ptr[0].func) self._kern = CUkernel(_ptr=&self._ptr[0].kern) self._ctx = CUcontext(_ptr=&self._ptr[0].ctx) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['func : ' + str(self.func)] except ValueError: str_list += ['func : '] try: str_list += ['gridDimX : ' + str(self.gridDimX)] except ValueError: str_list += ['gridDimX : '] try: str_list += ['gridDimY : ' + str(self.gridDimY)] except ValueError: str_list += ['gridDimY : '] try: str_list += ['gridDimZ : ' + str(self.gridDimZ)] except ValueError: str_list += ['gridDimZ : '] try: str_list += ['blockDimX : ' + str(self.blockDimX)] except ValueError: str_list += ['blockDimX : '] try: str_list += ['blockDimY : ' + str(self.blockDimY)] except ValueError: str_list += ['blockDimY : '] try: str_list += ['blockDimZ : ' + str(self.blockDimZ)] except ValueError: str_list += ['blockDimZ : '] try: str_list += ['sharedMemBytes : ' + str(self.sharedMemBytes)] except ValueError: str_list += ['sharedMemBytes : '] try: str_list += ['kernelParams : ' + str(self.kernelParams)] except ValueError: str_list += ['kernelParams : '] try: str_list += ['extra : ' + str(self.extra)] except ValueError: str_list += ['extra : '] try: str_list += ['kern : ' + str(self.kern)] except ValueError: str_list += ['kern : '] try: str_list += ['ctx : ' + str(self.ctx)] except ValueError: str_list += ['ctx : '] return '\n'.join(str_list) else: return '' @property def func(self): return self._func @func.setter def func(self, func): cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc self._func._ptr[0] = cfunc @property def gridDimX(self): return self._ptr[0].gridDimX @gridDimX.setter def gridDimX(self, unsigned int gridDimX): self._ptr[0].gridDimX = gridDimX @property def gridDimY(self): return self._ptr[0].gridDimY @gridDimY.setter def gridDimY(self, unsigned int gridDimY): self._ptr[0].gridDimY = gridDimY @property def gridDimZ(self): return self._ptr[0].gridDimZ @gridDimZ.setter def gridDimZ(self, unsigned int gridDimZ): self._ptr[0].gridDimZ = gridDimZ @property def blockDimX(self): return self._ptr[0].blockDimX @blockDimX.setter def blockDimX(self, unsigned int blockDimX): self._ptr[0].blockDimX = blockDimX @property def blockDimY(self): return self._ptr[0].blockDimY @blockDimY.setter def blockDimY(self, unsigned int blockDimY): self._ptr[0].blockDimY = blockDimY @property def blockDimZ(self): return self._ptr[0].blockDimZ @blockDimZ.setter def blockDimZ(self, unsigned int blockDimZ): self._ptr[0].blockDimZ = blockDimZ @property def sharedMemBytes(self): return self._ptr[0].sharedMemBytes @sharedMemBytes.setter def sharedMemBytes(self, unsigned int sharedMemBytes): self._ptr[0].sharedMemBytes = sharedMemBytes @property def kernelParams(self): return self._ptr[0].kernelParams @kernelParams.setter def kernelParams(self, kernelParams): self._ckernelParams = utils.HelperKernelParams(kernelParams) self._ptr[0].kernelParams = self._ckernelParams.ckernelParams @property def extra(self): return self._ptr[0].extra @extra.setter def extra(self, void_ptr extra): self._ptr[0].extra = extra @property def kern(self): return self._kern @kern.setter def kern(self, kern): cdef ccuda.CUkernel ckern if kern is None: ckern = 0 elif isinstance(kern, (CUkernel,)): pkern = int(kern) ckern = pkern else: pkern = int(CUkernel(kern)) ckern = pkern self._kern._ptr[0] = ckern @property def ctx(self): return self._ctx @ctx.setter def ctx(self, ctx): cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx self._ctx._ptr[0] = cctx {{endif}} {{if 'struct CUDA_MEMSET_NODE_PARAMS_st' in found_types}} cdef class CUDA_MEMSET_NODE_PARAMS_st: """ Memset node parameters Attributes ---------- dst : CUdeviceptr Destination device pointer pitch : size_t Pitch of destination device pointer. Unused if height is 1 value : unsigned int Value to be set elementSize : unsigned int Size of each element in bytes. Must be 1, 2, or 4. width : size_t Width of the row in elements height : size_t Number of rows Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._dst = CUdeviceptr(_ptr=&self._ptr[0].dst) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['dst : ' + str(self.dst)] except ValueError: str_list += ['dst : '] try: str_list += ['pitch : ' + str(self.pitch)] except ValueError: str_list += ['pitch : '] try: str_list += ['value : ' + str(self.value)] except ValueError: str_list += ['value : '] try: str_list += ['elementSize : ' + str(self.elementSize)] except ValueError: str_list += ['elementSize : '] try: str_list += ['width : ' + str(self.width)] except ValueError: str_list += ['width : '] try: str_list += ['height : ' + str(self.height)] except ValueError: str_list += ['height : '] return '\n'.join(str_list) else: return '' @property def dst(self): return self._dst @dst.setter def dst(self, dst): cdef ccuda.CUdeviceptr cdst if dst is None: cdst = 0 elif isinstance(dst, (CUdeviceptr)): pdst = int(dst) cdst = pdst else: pdst = int(CUdeviceptr(dst)) cdst = pdst self._dst._ptr[0] = cdst @property def pitch(self): return self._ptr[0].pitch @pitch.setter def pitch(self, size_t pitch): self._ptr[0].pitch = pitch @property def value(self): return self._ptr[0].value @value.setter def value(self, unsigned int value): self._ptr[0].value = value @property def elementSize(self): return self._ptr[0].elementSize @elementSize.setter def elementSize(self, unsigned int elementSize): self._ptr[0].elementSize = elementSize @property def width(self): return self._ptr[0].width @width.setter def width(self, size_t width): self._ptr[0].width = width @property def height(self): return self._ptr[0].height @height.setter def height(self, size_t height): self._ptr[0].height = height {{endif}} {{if 'struct CUDA_MEMSET_NODE_PARAMS_v2_st' in found_types}} cdef class CUDA_MEMSET_NODE_PARAMS_v2_st: """ Memset node parameters Attributes ---------- dst : CUdeviceptr Destination device pointer pitch : size_t Pitch of destination device pointer. Unused if height is 1 value : unsigned int Value to be set elementSize : unsigned int Size of each element in bytes. Must be 1, 2, or 4. width : size_t Width of the row in elements height : size_t Number of rows ctx : CUcontext Context on which to run the node Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._dst = CUdeviceptr(_ptr=&self._ptr[0].dst) self._ctx = CUcontext(_ptr=&self._ptr[0].ctx) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['dst : ' + str(self.dst)] except ValueError: str_list += ['dst : '] try: str_list += ['pitch : ' + str(self.pitch)] except ValueError: str_list += ['pitch : '] try: str_list += ['value : ' + str(self.value)] except ValueError: str_list += ['value : '] try: str_list += ['elementSize : ' + str(self.elementSize)] except ValueError: str_list += ['elementSize : '] try: str_list += ['width : ' + str(self.width)] except ValueError: str_list += ['width : '] try: str_list += ['height : ' + str(self.height)] except ValueError: str_list += ['height : '] try: str_list += ['ctx : ' + str(self.ctx)] except ValueError: str_list += ['ctx : '] return '\n'.join(str_list) else: return '' @property def dst(self): return self._dst @dst.setter def dst(self, dst): cdef ccuda.CUdeviceptr cdst if dst is None: cdst = 0 elif isinstance(dst, (CUdeviceptr)): pdst = int(dst) cdst = pdst else: pdst = int(CUdeviceptr(dst)) cdst = pdst self._dst._ptr[0] = cdst @property def pitch(self): return self._ptr[0].pitch @pitch.setter def pitch(self, size_t pitch): self._ptr[0].pitch = pitch @property def value(self): return self._ptr[0].value @value.setter def value(self, unsigned int value): self._ptr[0].value = value @property def elementSize(self): return self._ptr[0].elementSize @elementSize.setter def elementSize(self, unsigned int elementSize): self._ptr[0].elementSize = elementSize @property def width(self): return self._ptr[0].width @width.setter def width(self, size_t width): self._ptr[0].width = width @property def height(self): return self._ptr[0].height @height.setter def height(self, size_t height): self._ptr[0].height = height @property def ctx(self): return self._ctx @ctx.setter def ctx(self, ctx): cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx self._ctx._ptr[0] = cctx {{endif}} {{if 'struct CUDA_HOST_NODE_PARAMS_st' in found_types}} cdef class CUDA_HOST_NODE_PARAMS_st: """ Host node parameters Attributes ---------- fn : CUhostFn The function to call when the node executes userData : Any Argument to pass to the function Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._fn = CUhostFn(_ptr=&self._ptr[0].fn) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fn : ' + str(self.fn)] except ValueError: str_list += ['fn : '] try: str_list += ['userData : ' + hex(self.userData)] except ValueError: str_list += ['userData : '] return '\n'.join(str_list) else: return '' @property def fn(self): return self._fn @fn.setter def fn(self, fn): cdef ccuda.CUhostFn cfn if fn is None: cfn = 0 elif isinstance(fn, (CUhostFn)): pfn = int(fn) cfn = pfn else: pfn = int(CUhostFn(fn)) cfn = pfn self._fn._ptr[0] = cfn @property def userData(self): return self._ptr[0].userData @userData.setter def userData(self, userData): _cuserData = utils.HelperInputVoidPtr(userData) self._ptr[0].userData = _cuserData.cptr {{endif}} {{if 'struct CUDA_HOST_NODE_PARAMS_v2_st' in found_types}} cdef class CUDA_HOST_NODE_PARAMS_v2_st: """ Host node parameters Attributes ---------- fn : CUhostFn The function to call when the node executes userData : Any Argument to pass to the function Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._fn = CUhostFn(_ptr=&self._ptr[0].fn) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fn : ' + str(self.fn)] except ValueError: str_list += ['fn : '] try: str_list += ['userData : ' + hex(self.userData)] except ValueError: str_list += ['userData : '] return '\n'.join(str_list) else: return '' @property def fn(self): return self._fn @fn.setter def fn(self, fn): cdef ccuda.CUhostFn cfn if fn is None: cfn = 0 elif isinstance(fn, (CUhostFn)): pfn = int(fn) cfn = pfn else: pfn = int(CUhostFn(fn)) cfn = pfn self._fn._ptr[0] = cfn @property def userData(self): return self._ptr[0].userData @userData.setter def userData(self, userData): _cuserData = utils.HelperInputVoidPtr(userData) self._ptr[0].userData = _cuserData.cptr {{endif}} {{if 'struct CUDA_CONDITIONAL_NODE_PARAMS' in found_types}} cdef class CUDA_CONDITIONAL_NODE_PARAMS: """ Conditional node parameters Attributes ---------- handle : CUgraphConditionalHandle Conditional node handle. Handles must be created in advance of creating the node using cuGraphConditionalHandleCreate. type : CUgraphConditionalNodeType Type of conditional node. size : unsigned int Size of graph output array. Must be 1. phGraph_out : CUgraph CUDA-owned array populated with conditional node child graphs during creation of the node. Valid for the lifetime of the conditional node. The contents of the graph(s) are subject to the following constraints: - Allowed node types are kernel nodes, empty nodes, child graphs, memsets, memcopies, and conditionals. This applies recursively to child graphs and conditional bodies. - All kernels, including kernels in nested conditionals or child graphs at any level, must belong to the same CUDA context. These graphs may be populated using graph node creation APIs or cuStreamBeginCaptureToGraph. ctx : CUcontext Context on which to run the node. Must match context used to create the handle and all body nodes. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._handle = CUgraphConditionalHandle(_ptr=&self._ptr[0].handle) self._ctx = CUcontext(_ptr=&self._ptr[0].ctx) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['handle : ' + str(self.handle)] except ValueError: str_list += ['handle : '] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['size : ' + str(self.size)] except ValueError: str_list += ['size : '] try: str_list += ['phGraph_out : ' + str(self.phGraph_out)] except ValueError: str_list += ['phGraph_out : '] try: str_list += ['ctx : ' + str(self.ctx)] except ValueError: str_list += ['ctx : '] return '\n'.join(str_list) else: return '' @property def handle(self): return self._handle @handle.setter def handle(self, handle): cdef ccuda.CUgraphConditionalHandle chandle if handle is None: chandle = 0 elif isinstance(handle, (CUgraphConditionalHandle)): phandle = int(handle) chandle = phandle else: phandle = int(CUgraphConditionalHandle(handle)) chandle = phandle self._handle._ptr[0] = chandle @property def type(self): return CUgraphConditionalNodeType(self._ptr[0].type) @type.setter def type(self, type not None : CUgraphConditionalNodeType): self._ptr[0].type = type.value @property def size(self): return self._ptr[0].size @size.setter def size(self, unsigned int size): self._ptr[0].size = size @property def phGraph_out(self): arrs = [self._ptr[0].phGraph_out + x*sizeof(ccuda.CUgraph) for x in range(self.size)] return [CUgraph(_ptr=arr) for arr in arrs] @property def ctx(self): return self._ctx @ctx.setter def ctx(self, ctx): cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx self._ctx._ptr[0] = cctx {{endif}} {{if 'struct CUgraphEdgeData_st' in found_types}} cdef class CUgraphEdgeData_st: """ Optional annotation for edges in a CUDA graph. Note, all edges implicitly have annotations and default to a zero-initialized value if not specified. A zero-initialized struct indicates a standard full serialization of two nodes with memory visibility. Attributes ---------- from_port : bytes This indicates when the dependency is triggered from the upstream node on the edge. The meaning is specfic to the node type. A value of 0 in all cases means full completion of the upstream node, with memory visibility to the downstream node or portion thereof (indicated by `to_port`). Only kernel nodes define non-zero ports. A kernel node can use the following output port types: CU_GRAPH_KERNEL_NODE_PORT_DEFAULT, CU_GRAPH_KERNEL_NODE_PORT_PROGRAMMATIC, or CU_GRAPH_KERNEL_NODE_PORT_LAUNCH_ORDER. to_port : bytes This indicates what portion of the downstream node is dependent on the upstream node or portion thereof (indicated by `from_port`). The meaning is specific to the node type. A value of 0 in all cases means the entirety of the downstream node is dependent on the upstream work. Currently no node types define non-zero ports. Accordingly, this field must be set to zero. type : bytes This should be populated with a value from CUgraphDependencyType. (It is typed as char due to compiler-specific layout of bitfields.) See CUgraphDependencyType. reserved : bytes These bytes are unused and must be zeroed. This ensures compatibility if additional fields are added in the future. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['from_port : ' + str(self.from_port)] except ValueError: str_list += ['from_port : '] try: str_list += ['to_port : ' + str(self.to_port)] except ValueError: str_list += ['to_port : '] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def from_port(self): return self._ptr[0].from_port @from_port.setter def from_port(self, unsigned char from_port): self._ptr[0].from_port = from_port @property def to_port(self): return self._ptr[0].to_port @to_port.setter def to_port(self, unsigned char to_port): self._ptr[0].to_port = to_port @property def type(self): return self._ptr[0].type @type.setter def type(self, unsigned char type): self._ptr[0].type = type @property def reserved(self): return PyBytes_FromStringAndSize(self._ptr[0].reserved, 5) @reserved.setter def reserved(self, reserved): if len(reserved) != 5: raise ValueError("reserved length must be 5, is " + str(len(reserved))) for i, b in enumerate(reserved): self._ptr[0].reserved[i] = b {{endif}} {{if 'struct CUDA_GRAPH_INSTANTIATE_PARAMS_st' in found_types}} cdef class CUDA_GRAPH_INSTANTIATE_PARAMS_st: """ Graph instantiation parameters Attributes ---------- flags : cuuint64_t Instantiation flags hUploadStream : CUstream Upload stream hErrNode_out : CUgraphNode The node which caused instantiation to fail, if any result_out : CUgraphInstantiateResult Whether instantiation was successful. If it failed, the reason why Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._flags = cuuint64_t(_ptr=&self._ptr[0].flags) self._hUploadStream = CUstream(_ptr=&self._ptr[0].hUploadStream) self._hErrNode_out = CUgraphNode(_ptr=&self._ptr[0].hErrNode_out) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['hUploadStream : ' + str(self.hUploadStream)] except ValueError: str_list += ['hUploadStream : '] try: str_list += ['hErrNode_out : ' + str(self.hErrNode_out)] except ValueError: str_list += ['hErrNode_out : '] try: str_list += ['result_out : ' + str(self.result_out)] except ValueError: str_list += ['result_out : '] return '\n'.join(str_list) else: return '' @property def flags(self): return self._flags @flags.setter def flags(self, flags): cdef ccuda.cuuint64_t cflags if flags is None: cflags = 0 elif isinstance(flags, (cuuint64_t)): pflags = int(flags) cflags = pflags else: pflags = int(cuuint64_t(flags)) cflags = pflags self._flags._ptr[0] = cflags @property def hUploadStream(self): return self._hUploadStream @hUploadStream.setter def hUploadStream(self, hUploadStream): cdef ccuda.CUstream chUploadStream if hUploadStream is None: chUploadStream = 0 elif isinstance(hUploadStream, (CUstream,)): phUploadStream = int(hUploadStream) chUploadStream = phUploadStream else: phUploadStream = int(CUstream(hUploadStream)) chUploadStream = phUploadStream self._hUploadStream._ptr[0] = chUploadStream @property def hErrNode_out(self): return self._hErrNode_out @hErrNode_out.setter def hErrNode_out(self, hErrNode_out): cdef ccuda.CUgraphNode chErrNode_out if hErrNode_out is None: chErrNode_out = 0 elif isinstance(hErrNode_out, (CUgraphNode,)): phErrNode_out = int(hErrNode_out) chErrNode_out = phErrNode_out else: phErrNode_out = int(CUgraphNode(hErrNode_out)) chErrNode_out = phErrNode_out self._hErrNode_out._ptr[0] = chErrNode_out @property def result_out(self): return CUgraphInstantiateResult(self._ptr[0].result_out) @result_out.setter def result_out(self, result_out not None : CUgraphInstantiateResult): self._ptr[0].result_out = result_out.value {{endif}} {{if 'struct CUlaunchMemSyncDomainMap_st' in found_types}} cdef class CUlaunchMemSyncDomainMap_st: """ Memory Synchronization Domain map See ::cudaLaunchMemSyncDomain. By default, kernels are launched in domain 0. Kernel launched with CU_LAUNCH_MEM_SYNC_DOMAIN_REMOTE will have a different domain ID. User may also alter the domain ID with CUlaunchMemSyncDomainMap for a specific stream / graph node / kernel launch. See CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP. Domain ID range is available through CU_DEVICE_ATTRIBUTE_MEM_SYNC_DOMAIN_COUNT. Attributes ---------- default_ : bytes The default domain ID to use for designated kernels remote : bytes The remote domain ID to use for designated kernels Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['default_ : ' + str(self.default_)] except ValueError: str_list += ['default_ : '] try: str_list += ['remote : ' + str(self.remote)] except ValueError: str_list += ['remote : '] return '\n'.join(str_list) else: return '' @property def default_(self): return self._ptr[0].default_ @default_.setter def default_(self, unsigned char default_): self._ptr[0].default_ = default_ @property def remote(self): return self._ptr[0].remote @remote.setter def remote(self, unsigned char remote): self._ptr[0].remote = remote {{endif}} {{if 'union CUlaunchAttributeValue_union' in found_types}} cdef class anon_struct1: """ Attributes ---------- x : unsigned int y : unsigned int z : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].clusterDim def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['x : ' + str(self.x)] except ValueError: str_list += ['x : '] try: str_list += ['y : ' + str(self.y)] except ValueError: str_list += ['y : '] try: str_list += ['z : ' + str(self.z)] except ValueError: str_list += ['z : '] return '\n'.join(str_list) else: return '' @property def x(self): return self._ptr[0].clusterDim.x @x.setter def x(self, unsigned int x): self._ptr[0].clusterDim.x = x @property def y(self): return self._ptr[0].clusterDim.y @y.setter def y(self, unsigned int y): self._ptr[0].clusterDim.y = y @property def z(self): return self._ptr[0].clusterDim.z @z.setter def z(self, unsigned int z): self._ptr[0].clusterDim.z = z {{endif}} {{if 'union CUlaunchAttributeValue_union' in found_types}} cdef class anon_struct2: """ Attributes ---------- event : CUevent flags : int triggerAtBlockStart : int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._event = CUevent(_ptr=&self._ptr[0].programmaticEvent.event) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].programmaticEvent def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['event : ' + str(self.event)] except ValueError: str_list += ['event : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['triggerAtBlockStart : ' + str(self.triggerAtBlockStart)] except ValueError: str_list += ['triggerAtBlockStart : '] return '\n'.join(str_list) else: return '' @property def event(self): return self._event @event.setter def event(self, event): cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent self._event._ptr[0] = cevent @property def flags(self): return self._ptr[0].programmaticEvent.flags @flags.setter def flags(self, int flags): self._ptr[0].programmaticEvent.flags = flags @property def triggerAtBlockStart(self): return self._ptr[0].programmaticEvent.triggerAtBlockStart @triggerAtBlockStart.setter def triggerAtBlockStart(self, int triggerAtBlockStart): self._ptr[0].programmaticEvent.triggerAtBlockStart = triggerAtBlockStart {{endif}} {{if 'union CUlaunchAttributeValue_union' in found_types}} cdef class anon_struct3: """ Attributes ---------- event : CUevent flags : int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._event = CUevent(_ptr=&self._ptr[0].launchCompletionEvent.event) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].launchCompletionEvent def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['event : ' + str(self.event)] except ValueError: str_list += ['event : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def event(self): return self._event @event.setter def event(self, event): cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent self._event._ptr[0] = cevent @property def flags(self): return self._ptr[0].launchCompletionEvent.flags @flags.setter def flags(self, int flags): self._ptr[0].launchCompletionEvent.flags = flags {{endif}} {{if 'union CUlaunchAttributeValue_union' in found_types}} cdef class anon_struct4: """ Attributes ---------- deviceUpdatable : int devNode : CUgraphDeviceNode Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._devNode = CUgraphDeviceNode(_ptr=&self._ptr[0].deviceUpdatableKernelNode.devNode) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].deviceUpdatableKernelNode def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['deviceUpdatable : ' + str(self.deviceUpdatable)] except ValueError: str_list += ['deviceUpdatable : '] try: str_list += ['devNode : ' + str(self.devNode)] except ValueError: str_list += ['devNode : '] return '\n'.join(str_list) else: return '' @property def deviceUpdatable(self): return self._ptr[0].deviceUpdatableKernelNode.deviceUpdatable @deviceUpdatable.setter def deviceUpdatable(self, int deviceUpdatable): self._ptr[0].deviceUpdatableKernelNode.deviceUpdatable = deviceUpdatable @property def devNode(self): return self._devNode @devNode.setter def devNode(self, devNode): cdef ccuda.CUgraphDeviceNode cdevNode if devNode is None: cdevNode = 0 elif isinstance(devNode, (CUgraphDeviceNode,)): pdevNode = int(devNode) cdevNode = pdevNode else: pdevNode = int(CUgraphDeviceNode(devNode)) cdevNode = pdevNode self._devNode._ptr[0] = cdevNode {{endif}} {{if 'union CUlaunchAttributeValue_union' in found_types}} cdef class CUlaunchAttributeValue_union: """ Launch attributes union; used as value field of CUlaunchAttribute Attributes ---------- pad : bytes accessPolicyWindow : CUaccessPolicyWindow Value of launch attribute CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW. cooperative : int Value of launch attribute CU_LAUNCH_ATTRIBUTE_COOPERATIVE. Nonzero indicates a cooperative kernel (see cuLaunchCooperativeKernel). syncPolicy : CUsynchronizationPolicy Value of launch attribute CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY. ::CUsynchronizationPolicy for work queued up in this stream clusterDim : anon_struct1 Value of launch attribute CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION that represents the desired cluster dimensions for the kernel. Opaque type with the following fields: - `x` - The X dimension of the cluster, in blocks. Must be a divisor of the grid X dimension. - `y` - The Y dimension of the cluster, in blocks. Must be a divisor of the grid Y dimension. - `z` - The Z dimension of the cluster, in blocks. Must be a divisor of the grid Z dimension. clusterSchedulingPolicyPreference : CUclusterSchedulingPolicy Value of launch attribute CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE. Cluster scheduling policy preference for the kernel. programmaticStreamSerializationAllowed : int Value of launch attribute CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION. programmaticEvent : anon_struct2 Value of launch attribute CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT with the following fields: - `CUevent` event - Event to fire when all blocks trigger it. - `Event` record flags, see cuEventRecordWithFlags. Does not accept :CU_EVENT_RECORD_EXTERNAL. - `triggerAtBlockStart` - If this is set to non-0, each block launch will automatically trigger the event. launchCompletionEvent : anon_struct3 Value of launch attribute CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT with the following fields: - `CUevent` event - Event to fire when the last block launches - `int` flags; - Event record flags, see cuEventRecordWithFlags. Does not accept CU_EVENT_RECORD_EXTERNAL. priority : int Value of launch attribute CU_LAUNCH_ATTRIBUTE_PRIORITY. Execution priority of the kernel. memSyncDomainMap : CUlaunchMemSyncDomainMap Value of launch attribute CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN_MAP. See CUlaunchMemSyncDomainMap. memSyncDomain : CUlaunchMemSyncDomain Value of launch attribute CU_LAUNCH_ATTRIBUTE_MEM_SYNC_DOMAIN. See::CUlaunchMemSyncDomain deviceUpdatableKernelNode : anon_struct4 Value of launch attribute CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE. with the following fields: - `int` deviceUpdatable - Whether or not the resulting kernel node should be device-updatable. - `CUgraphDeviceNode` devNode - Returns a handle to pass to the various device-side update functions. sharedMemCarveout : unsigned int Value of launch attribute CU_LAUNCH_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._accessPolicyWindow = CUaccessPolicyWindow(_ptr=&self._ptr[0].accessPolicyWindow) self._clusterDim = anon_struct1(_ptr=self._ptr) self._programmaticEvent = anon_struct2(_ptr=self._ptr) self._launchCompletionEvent = anon_struct3(_ptr=self._ptr) self._memSyncDomainMap = CUlaunchMemSyncDomainMap(_ptr=&self._ptr[0].memSyncDomainMap) self._deviceUpdatableKernelNode = anon_struct4(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['pad : ' + str(self.pad)] except ValueError: str_list += ['pad : '] try: str_list += ['accessPolicyWindow :\n' + '\n'.join([' ' + line for line in str(self.accessPolicyWindow).splitlines()])] except ValueError: str_list += ['accessPolicyWindow : '] try: str_list += ['cooperative : ' + str(self.cooperative)] except ValueError: str_list += ['cooperative : '] try: str_list += ['syncPolicy : ' + str(self.syncPolicy)] except ValueError: str_list += ['syncPolicy : '] try: str_list += ['clusterDim :\n' + '\n'.join([' ' + line for line in str(self.clusterDim).splitlines()])] except ValueError: str_list += ['clusterDim : '] try: str_list += ['clusterSchedulingPolicyPreference : ' + str(self.clusterSchedulingPolicyPreference)] except ValueError: str_list += ['clusterSchedulingPolicyPreference : '] try: str_list += ['programmaticStreamSerializationAllowed : ' + str(self.programmaticStreamSerializationAllowed)] except ValueError: str_list += ['programmaticStreamSerializationAllowed : '] try: str_list += ['programmaticEvent :\n' + '\n'.join([' ' + line for line in str(self.programmaticEvent).splitlines()])] except ValueError: str_list += ['programmaticEvent : '] try: str_list += ['launchCompletionEvent :\n' + '\n'.join([' ' + line for line in str(self.launchCompletionEvent).splitlines()])] except ValueError: str_list += ['launchCompletionEvent : '] try: str_list += ['priority : ' + str(self.priority)] except ValueError: str_list += ['priority : '] try: str_list += ['memSyncDomainMap :\n' + '\n'.join([' ' + line for line in str(self.memSyncDomainMap).splitlines()])] except ValueError: str_list += ['memSyncDomainMap : '] try: str_list += ['memSyncDomain : ' + str(self.memSyncDomain)] except ValueError: str_list += ['memSyncDomain : '] try: str_list += ['deviceUpdatableKernelNode :\n' + '\n'.join([' ' + line for line in str(self.deviceUpdatableKernelNode).splitlines()])] except ValueError: str_list += ['deviceUpdatableKernelNode : '] try: str_list += ['sharedMemCarveout : ' + str(self.sharedMemCarveout)] except ValueError: str_list += ['sharedMemCarveout : '] return '\n'.join(str_list) else: return '' @property def pad(self): return PyBytes_FromStringAndSize(self._ptr[0].pad, 64) @pad.setter def pad(self, pad): if len(pad) != 64: raise ValueError("pad length must be 64, is " + str(len(pad))) if CHAR_MIN == 0: for i, b in enumerate(pad): if b < 0 and b > -129: b = b + 256 self._ptr[0].pad[i] = b else: for i, b in enumerate(pad): if b > 127 and b < 256: b = b - 256 self._ptr[0].pad[i] = b @property def accessPolicyWindow(self): return self._accessPolicyWindow @accessPolicyWindow.setter def accessPolicyWindow(self, accessPolicyWindow not None : CUaccessPolicyWindow): string.memcpy(&self._ptr[0].accessPolicyWindow, accessPolicyWindow.getPtr(), sizeof(self._ptr[0].accessPolicyWindow)) @property def cooperative(self): return self._ptr[0].cooperative @cooperative.setter def cooperative(self, int cooperative): self._ptr[0].cooperative = cooperative @property def syncPolicy(self): return CUsynchronizationPolicy(self._ptr[0].syncPolicy) @syncPolicy.setter def syncPolicy(self, syncPolicy not None : CUsynchronizationPolicy): self._ptr[0].syncPolicy = syncPolicy.value @property def clusterDim(self): return self._clusterDim @clusterDim.setter def clusterDim(self, clusterDim not None : anon_struct1): string.memcpy(&self._ptr[0].clusterDim, clusterDim.getPtr(), sizeof(self._ptr[0].clusterDim)) @property def clusterSchedulingPolicyPreference(self): return CUclusterSchedulingPolicy(self._ptr[0].clusterSchedulingPolicyPreference) @clusterSchedulingPolicyPreference.setter def clusterSchedulingPolicyPreference(self, clusterSchedulingPolicyPreference not None : CUclusterSchedulingPolicy): self._ptr[0].clusterSchedulingPolicyPreference = clusterSchedulingPolicyPreference.value @property def programmaticStreamSerializationAllowed(self): return self._ptr[0].programmaticStreamSerializationAllowed @programmaticStreamSerializationAllowed.setter def programmaticStreamSerializationAllowed(self, int programmaticStreamSerializationAllowed): self._ptr[0].programmaticStreamSerializationAllowed = programmaticStreamSerializationAllowed @property def programmaticEvent(self): return self._programmaticEvent @programmaticEvent.setter def programmaticEvent(self, programmaticEvent not None : anon_struct2): string.memcpy(&self._ptr[0].programmaticEvent, programmaticEvent.getPtr(), sizeof(self._ptr[0].programmaticEvent)) @property def launchCompletionEvent(self): return self._launchCompletionEvent @launchCompletionEvent.setter def launchCompletionEvent(self, launchCompletionEvent not None : anon_struct3): string.memcpy(&self._ptr[0].launchCompletionEvent, launchCompletionEvent.getPtr(), sizeof(self._ptr[0].launchCompletionEvent)) @property def priority(self): return self._ptr[0].priority @priority.setter def priority(self, int priority): self._ptr[0].priority = priority @property def memSyncDomainMap(self): return self._memSyncDomainMap @memSyncDomainMap.setter def memSyncDomainMap(self, memSyncDomainMap not None : CUlaunchMemSyncDomainMap): string.memcpy(&self._ptr[0].memSyncDomainMap, memSyncDomainMap.getPtr(), sizeof(self._ptr[0].memSyncDomainMap)) @property def memSyncDomain(self): return CUlaunchMemSyncDomain(self._ptr[0].memSyncDomain) @memSyncDomain.setter def memSyncDomain(self, memSyncDomain not None : CUlaunchMemSyncDomain): self._ptr[0].memSyncDomain = memSyncDomain.value @property def deviceUpdatableKernelNode(self): return self._deviceUpdatableKernelNode @deviceUpdatableKernelNode.setter def deviceUpdatableKernelNode(self, deviceUpdatableKernelNode not None : anon_struct4): string.memcpy(&self._ptr[0].deviceUpdatableKernelNode, deviceUpdatableKernelNode.getPtr(), sizeof(self._ptr[0].deviceUpdatableKernelNode)) @property def sharedMemCarveout(self): return self._ptr[0].sharedMemCarveout @sharedMemCarveout.setter def sharedMemCarveout(self, unsigned int sharedMemCarveout): self._ptr[0].sharedMemCarveout = sharedMemCarveout {{endif}} {{if 'struct CUlaunchAttribute_st' in found_types}} cdef class CUlaunchAttribute_st: """ Launch attribute Attributes ---------- id : CUlaunchAttributeID Attribute to set value : CUlaunchAttributeValue Value of the attribute Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._value = CUlaunchAttributeValue(_ptr=&self._ptr[0].value) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['id : ' + str(self.id)] except ValueError: str_list += ['id : '] try: str_list += ['value :\n' + '\n'.join([' ' + line for line in str(self.value).splitlines()])] except ValueError: str_list += ['value : '] return '\n'.join(str_list) else: return '' @property def id(self): return CUlaunchAttributeID(self._ptr[0].id) @id.setter def id(self, id not None : CUlaunchAttributeID): self._ptr[0].id = id.value @property def value(self): return self._value @value.setter def value(self, value not None : CUlaunchAttributeValue): string.memcpy(&self._ptr[0].value, value.getPtr(), sizeof(self._ptr[0].value)) {{endif}} {{if 'struct CUlaunchConfig_st' in found_types}} cdef class CUlaunchConfig_st: """ CUDA extensible launch configuration Attributes ---------- gridDimX : unsigned int Width of grid in blocks gridDimY : unsigned int Height of grid in blocks gridDimZ : unsigned int Depth of grid in blocks blockDimX : unsigned int X dimension of each thread block blockDimY : unsigned int Y dimension of each thread block blockDimZ : unsigned int Z dimension of each thread block sharedMemBytes : unsigned int Dynamic shared-memory size per thread block in bytes hStream : CUstream Stream identifier attrs : CUlaunchAttribute List of attributes; nullable if CUlaunchConfig::numAttrs == 0 numAttrs : unsigned int Number of attributes populated in CUlaunchConfig::attrs Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._hStream = CUstream(_ptr=&self._ptr[0].hStream) def __dealloc__(self): if self._attrs is not NULL: free(self._attrs) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['gridDimX : ' + str(self.gridDimX)] except ValueError: str_list += ['gridDimX : '] try: str_list += ['gridDimY : ' + str(self.gridDimY)] except ValueError: str_list += ['gridDimY : '] try: str_list += ['gridDimZ : ' + str(self.gridDimZ)] except ValueError: str_list += ['gridDimZ : '] try: str_list += ['blockDimX : ' + str(self.blockDimX)] except ValueError: str_list += ['blockDimX : '] try: str_list += ['blockDimY : ' + str(self.blockDimY)] except ValueError: str_list += ['blockDimY : '] try: str_list += ['blockDimZ : ' + str(self.blockDimZ)] except ValueError: str_list += ['blockDimZ : '] try: str_list += ['sharedMemBytes : ' + str(self.sharedMemBytes)] except ValueError: str_list += ['sharedMemBytes : '] try: str_list += ['hStream : ' + str(self.hStream)] except ValueError: str_list += ['hStream : '] try: str_list += ['attrs : ' + str(self.attrs)] except ValueError: str_list += ['attrs : '] try: str_list += ['numAttrs : ' + str(self.numAttrs)] except ValueError: str_list += ['numAttrs : '] return '\n'.join(str_list) else: return '' @property def gridDimX(self): return self._ptr[0].gridDimX @gridDimX.setter def gridDimX(self, unsigned int gridDimX): self._ptr[0].gridDimX = gridDimX @property def gridDimY(self): return self._ptr[0].gridDimY @gridDimY.setter def gridDimY(self, unsigned int gridDimY): self._ptr[0].gridDimY = gridDimY @property def gridDimZ(self): return self._ptr[0].gridDimZ @gridDimZ.setter def gridDimZ(self, unsigned int gridDimZ): self._ptr[0].gridDimZ = gridDimZ @property def blockDimX(self): return self._ptr[0].blockDimX @blockDimX.setter def blockDimX(self, unsigned int blockDimX): self._ptr[0].blockDimX = blockDimX @property def blockDimY(self): return self._ptr[0].blockDimY @blockDimY.setter def blockDimY(self, unsigned int blockDimY): self._ptr[0].blockDimY = blockDimY @property def blockDimZ(self): return self._ptr[0].blockDimZ @blockDimZ.setter def blockDimZ(self, unsigned int blockDimZ): self._ptr[0].blockDimZ = blockDimZ @property def sharedMemBytes(self): return self._ptr[0].sharedMemBytes @sharedMemBytes.setter def sharedMemBytes(self, unsigned int sharedMemBytes): self._ptr[0].sharedMemBytes = sharedMemBytes @property def hStream(self): return self._hStream @hStream.setter def hStream(self, hStream): cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream self._hStream._ptr[0] = chStream @property def attrs(self): arrs = [self._ptr[0].attrs + x*sizeof(ccuda.CUlaunchAttribute) for x in range(self._attrs_length)] return [CUlaunchAttribute(_ptr=arr) for arr in arrs] @attrs.setter def attrs(self, val): if len(val) == 0: free(self._attrs) self._attrs_length = 0 self._ptr[0].attrs = NULL else: if self._attrs_length != len(val): free(self._attrs) self._attrs = calloc(len(val), sizeof(ccuda.CUlaunchAttribute)) if self._attrs is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUlaunchAttribute))) self._attrs_length = len(val) self._ptr[0].attrs = self._attrs for idx in range(len(val)): string.memcpy(&self._attrs[idx], (val[idx])._ptr, sizeof(ccuda.CUlaunchAttribute)) @property def numAttrs(self): return self._ptr[0].numAttrs @numAttrs.setter def numAttrs(self, unsigned int numAttrs): self._ptr[0].numAttrs = numAttrs {{endif}} {{if 'struct CUexecAffinitySmCount_st' in found_types}} cdef class CUexecAffinitySmCount_st: """ Value for CU_EXEC_AFFINITY_TYPE_SM_COUNT Attributes ---------- val : unsigned int The number of SMs the context is limited to use. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['val : ' + str(self.val)] except ValueError: str_list += ['val : '] return '\n'.join(str_list) else: return '' @property def val(self): return self._ptr[0].val @val.setter def val(self, unsigned int val): self._ptr[0].val = val {{endif}} {{if 'struct CUexecAffinityParam_st' in found_types}} cdef class anon_union3: """ Attributes ---------- smCount : CUexecAffinitySmCount Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._smCount = CUexecAffinitySmCount(_ptr=&self._ptr[0].param.smCount) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].param def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['smCount :\n' + '\n'.join([' ' + line for line in str(self.smCount).splitlines()])] except ValueError: str_list += ['smCount : '] return '\n'.join(str_list) else: return '' @property def smCount(self): return self._smCount @smCount.setter def smCount(self, smCount not None : CUexecAffinitySmCount): string.memcpy(&self._ptr[0].param.smCount, smCount.getPtr(), sizeof(self._ptr[0].param.smCount)) {{endif}} {{if 'struct CUexecAffinityParam_st' in found_types}} cdef class CUexecAffinityParam_st: """ Execution Affinity Parameters Attributes ---------- type : CUexecAffinityType param : anon_union3 Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUexecAffinityParam_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._param = anon_union3(_ptr=self._ptr) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['param :\n' + '\n'.join([' ' + line for line in str(self.param).splitlines()])] except ValueError: str_list += ['param : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUexecAffinityType(self._ptr[0].type) @type.setter def type(self, type not None : CUexecAffinityType): self._ptr[0].type = type.value @property def param(self): return self._param @param.setter def param(self, param not None : anon_union3): string.memcpy(&self._ptr[0].param, param.getPtr(), sizeof(self._ptr[0].param)) {{endif}} {{if 'struct CUctxCigParam_st' in found_types}} cdef class CUctxCigParam_st: """ CIG Context Create Params Attributes ---------- sharedDataType : CUcigDataType sharedData : Any Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['sharedDataType : ' + str(self.sharedDataType)] except ValueError: str_list += ['sharedDataType : '] try: str_list += ['sharedData : ' + hex(self.sharedData)] except ValueError: str_list += ['sharedData : '] return '\n'.join(str_list) else: return '' @property def sharedDataType(self): return CUcigDataType(self._ptr[0].sharedDataType) @sharedDataType.setter def sharedDataType(self, sharedDataType not None : CUcigDataType): self._ptr[0].sharedDataType = sharedDataType.value @property def sharedData(self): return self._ptr[0].sharedData @sharedData.setter def sharedData(self, sharedData): _csharedData = utils.HelperInputVoidPtr(sharedData) self._ptr[0].sharedData = _csharedData.cptr {{endif}} {{if 'struct CUctxCreateParams_st' in found_types}} cdef class CUctxCreateParams_st: """ Params for creating CUDA context Exactly one of execAffinityParams and cigParams must be non-NULL. Attributes ---------- execAffinityParams : CUexecAffinityParam numExecAffinityParams : int cigParams : CUctxCigParam Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): if self._execAffinityParams is not NULL: free(self._execAffinityParams) if self._cigParams is not NULL: free(self._cigParams) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['execAffinityParams : ' + str(self.execAffinityParams)] except ValueError: str_list += ['execAffinityParams : '] try: str_list += ['numExecAffinityParams : ' + str(self.numExecAffinityParams)] except ValueError: str_list += ['numExecAffinityParams : '] try: str_list += ['cigParams : ' + str(self.cigParams)] except ValueError: str_list += ['cigParams : '] return '\n'.join(str_list) else: return '' @property def execAffinityParams(self): arrs = [self._ptr[0].execAffinityParams + x*sizeof(ccuda.CUexecAffinityParam) for x in range(self._execAffinityParams_length)] return [CUexecAffinityParam(_ptr=arr) for arr in arrs] @execAffinityParams.setter def execAffinityParams(self, val): if len(val) == 0: free(self._execAffinityParams) self._execAffinityParams_length = 0 self._ptr[0].execAffinityParams = NULL else: if self._execAffinityParams_length != len(val): free(self._execAffinityParams) self._execAffinityParams = calloc(len(val), sizeof(ccuda.CUexecAffinityParam)) if self._execAffinityParams is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUexecAffinityParam))) self._execAffinityParams_length = len(val) self._ptr[0].execAffinityParams = self._execAffinityParams for idx in range(len(val)): string.memcpy(&self._execAffinityParams[idx], (val[idx])._ptr, sizeof(ccuda.CUexecAffinityParam)) @property def numExecAffinityParams(self): return self._ptr[0].numExecAffinityParams @numExecAffinityParams.setter def numExecAffinityParams(self, int numExecAffinityParams): self._ptr[0].numExecAffinityParams = numExecAffinityParams @property def cigParams(self): arrs = [self._ptr[0].cigParams + x*sizeof(ccuda.CUctxCigParam) for x in range(self._cigParams_length)] return [CUctxCigParam(_ptr=arr) for arr in arrs] @cigParams.setter def cigParams(self, val): if len(val) == 0: free(self._cigParams) self._cigParams_length = 0 self._ptr[0].cigParams = NULL else: if self._cigParams_length != len(val): free(self._cigParams) self._cigParams = calloc(len(val), sizeof(ccuda.CUctxCigParam)) if self._cigParams is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUctxCigParam))) self._cigParams_length = len(val) self._ptr[0].cigParams = self._cigParams for idx in range(len(val)): string.memcpy(&self._cigParams[idx], (val[idx])._ptr, sizeof(ccuda.CUctxCigParam)) {{endif}} {{if 'struct CUlibraryHostUniversalFunctionAndDataTable_st' in found_types}} cdef class CUlibraryHostUniversalFunctionAndDataTable_st: """ Attributes ---------- functionTable : Any functionWindowSize : size_t dataTable : Any dataWindowSize : size_t Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['functionTable : ' + hex(self.functionTable)] except ValueError: str_list += ['functionTable : '] try: str_list += ['functionWindowSize : ' + str(self.functionWindowSize)] except ValueError: str_list += ['functionWindowSize : '] try: str_list += ['dataTable : ' + hex(self.dataTable)] except ValueError: str_list += ['dataTable : '] try: str_list += ['dataWindowSize : ' + str(self.dataWindowSize)] except ValueError: str_list += ['dataWindowSize : '] return '\n'.join(str_list) else: return '' @property def functionTable(self): return self._ptr[0].functionTable @functionTable.setter def functionTable(self, functionTable): _cfunctionTable = utils.HelperInputVoidPtr(functionTable) self._ptr[0].functionTable = _cfunctionTable.cptr @property def functionWindowSize(self): return self._ptr[0].functionWindowSize @functionWindowSize.setter def functionWindowSize(self, size_t functionWindowSize): self._ptr[0].functionWindowSize = functionWindowSize @property def dataTable(self): return self._ptr[0].dataTable @dataTable.setter def dataTable(self, dataTable): _cdataTable = utils.HelperInputVoidPtr(dataTable) self._ptr[0].dataTable = _cdataTable.cptr @property def dataWindowSize(self): return self._ptr[0].dataWindowSize @dataWindowSize.setter def dataWindowSize(self, size_t dataWindowSize): self._ptr[0].dataWindowSize = dataWindowSize {{endif}} {{if 'struct CUDA_MEMCPY2D_st' in found_types}} cdef class CUDA_MEMCPY2D_st: """ 2D memory copy parameters Attributes ---------- srcXInBytes : size_t Source X in bytes srcY : size_t Source Y srcMemoryType : CUmemorytype Source memory type (host, device, array) srcHost : Any Source host pointer srcDevice : CUdeviceptr Source device pointer srcArray : CUarray Source array reference srcPitch : size_t Source pitch (ignored when src is array) dstXInBytes : size_t Destination X in bytes dstY : size_t Destination Y dstMemoryType : CUmemorytype Destination memory type (host, device, array) dstHost : Any Destination host pointer dstDevice : CUdeviceptr Destination device pointer dstArray : CUarray Destination array reference dstPitch : size_t Destination pitch (ignored when dst is array) WidthInBytes : size_t Width of 2D memory copy in bytes Height : size_t Height of 2D memory copy Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._srcDevice = CUdeviceptr(_ptr=&self._ptr[0].srcDevice) self._srcArray = CUarray(_ptr=&self._ptr[0].srcArray) self._dstDevice = CUdeviceptr(_ptr=&self._ptr[0].dstDevice) self._dstArray = CUarray(_ptr=&self._ptr[0].dstArray) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['srcXInBytes : ' + str(self.srcXInBytes)] except ValueError: str_list += ['srcXInBytes : '] try: str_list += ['srcY : ' + str(self.srcY)] except ValueError: str_list += ['srcY : '] try: str_list += ['srcMemoryType : ' + str(self.srcMemoryType)] except ValueError: str_list += ['srcMemoryType : '] try: str_list += ['srcHost : ' + hex(self.srcHost)] except ValueError: str_list += ['srcHost : '] try: str_list += ['srcDevice : ' + str(self.srcDevice)] except ValueError: str_list += ['srcDevice : '] try: str_list += ['srcArray : ' + str(self.srcArray)] except ValueError: str_list += ['srcArray : '] try: str_list += ['srcPitch : ' + str(self.srcPitch)] except ValueError: str_list += ['srcPitch : '] try: str_list += ['dstXInBytes : ' + str(self.dstXInBytes)] except ValueError: str_list += ['dstXInBytes : '] try: str_list += ['dstY : ' + str(self.dstY)] except ValueError: str_list += ['dstY : '] try: str_list += ['dstMemoryType : ' + str(self.dstMemoryType)] except ValueError: str_list += ['dstMemoryType : '] try: str_list += ['dstHost : ' + hex(self.dstHost)] except ValueError: str_list += ['dstHost : '] try: str_list += ['dstDevice : ' + str(self.dstDevice)] except ValueError: str_list += ['dstDevice : '] try: str_list += ['dstArray : ' + str(self.dstArray)] except ValueError: str_list += ['dstArray : '] try: str_list += ['dstPitch : ' + str(self.dstPitch)] except ValueError: str_list += ['dstPitch : '] try: str_list += ['WidthInBytes : ' + str(self.WidthInBytes)] except ValueError: str_list += ['WidthInBytes : '] try: str_list += ['Height : ' + str(self.Height)] except ValueError: str_list += ['Height : '] return '\n'.join(str_list) else: return '' @property def srcXInBytes(self): return self._ptr[0].srcXInBytes @srcXInBytes.setter def srcXInBytes(self, size_t srcXInBytes): self._ptr[0].srcXInBytes = srcXInBytes @property def srcY(self): return self._ptr[0].srcY @srcY.setter def srcY(self, size_t srcY): self._ptr[0].srcY = srcY @property def srcMemoryType(self): return CUmemorytype(self._ptr[0].srcMemoryType) @srcMemoryType.setter def srcMemoryType(self, srcMemoryType not None : CUmemorytype): self._ptr[0].srcMemoryType = srcMemoryType.value @property def srcHost(self): return self._ptr[0].srcHost @srcHost.setter def srcHost(self, srcHost): _csrcHost = utils.HelperInputVoidPtr(srcHost) self._ptr[0].srcHost = _csrcHost.cptr @property def srcDevice(self): return self._srcDevice @srcDevice.setter def srcDevice(self, srcDevice): cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice self._srcDevice._ptr[0] = csrcDevice @property def srcArray(self): return self._srcArray @srcArray.setter def srcArray(self, srcArray): cdef ccuda.CUarray csrcArray if srcArray is None: csrcArray = 0 elif isinstance(srcArray, (CUarray,)): psrcArray = int(srcArray) csrcArray = psrcArray else: psrcArray = int(CUarray(srcArray)) csrcArray = psrcArray self._srcArray._ptr[0] = csrcArray @property def srcPitch(self): return self._ptr[0].srcPitch @srcPitch.setter def srcPitch(self, size_t srcPitch): self._ptr[0].srcPitch = srcPitch @property def dstXInBytes(self): return self._ptr[0].dstXInBytes @dstXInBytes.setter def dstXInBytes(self, size_t dstXInBytes): self._ptr[0].dstXInBytes = dstXInBytes @property def dstY(self): return self._ptr[0].dstY @dstY.setter def dstY(self, size_t dstY): self._ptr[0].dstY = dstY @property def dstMemoryType(self): return CUmemorytype(self._ptr[0].dstMemoryType) @dstMemoryType.setter def dstMemoryType(self, dstMemoryType not None : CUmemorytype): self._ptr[0].dstMemoryType = dstMemoryType.value @property def dstHost(self): return self._ptr[0].dstHost @dstHost.setter def dstHost(self, dstHost): _cdstHost = utils.HelperInputVoidPtr(dstHost) self._ptr[0].dstHost = _cdstHost.cptr @property def dstDevice(self): return self._dstDevice @dstDevice.setter def dstDevice(self, dstDevice): cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice self._dstDevice._ptr[0] = cdstDevice @property def dstArray(self): return self._dstArray @dstArray.setter def dstArray(self, dstArray): cdef ccuda.CUarray cdstArray if dstArray is None: cdstArray = 0 elif isinstance(dstArray, (CUarray,)): pdstArray = int(dstArray) cdstArray = pdstArray else: pdstArray = int(CUarray(dstArray)) cdstArray = pdstArray self._dstArray._ptr[0] = cdstArray @property def dstPitch(self): return self._ptr[0].dstPitch @dstPitch.setter def dstPitch(self, size_t dstPitch): self._ptr[0].dstPitch = dstPitch @property def WidthInBytes(self): return self._ptr[0].WidthInBytes @WidthInBytes.setter def WidthInBytes(self, size_t WidthInBytes): self._ptr[0].WidthInBytes = WidthInBytes @property def Height(self): return self._ptr[0].Height @Height.setter def Height(self, size_t Height): self._ptr[0].Height = Height {{endif}} {{if 'struct CUDA_MEMCPY3D_st' in found_types}} cdef class CUDA_MEMCPY3D_st: """ 3D memory copy parameters Attributes ---------- srcXInBytes : size_t Source X in bytes srcY : size_t Source Y srcZ : size_t Source Z srcLOD : size_t Source LOD srcMemoryType : CUmemorytype Source memory type (host, device, array) srcHost : Any Source host pointer srcDevice : CUdeviceptr Source device pointer srcArray : CUarray Source array reference reserved0 : Any Must be NULL srcPitch : size_t Source pitch (ignored when src is array) srcHeight : size_t Source height (ignored when src is array; may be 0 if Depth==1) dstXInBytes : size_t Destination X in bytes dstY : size_t Destination Y dstZ : size_t Destination Z dstLOD : size_t Destination LOD dstMemoryType : CUmemorytype Destination memory type (host, device, array) dstHost : Any Destination host pointer dstDevice : CUdeviceptr Destination device pointer dstArray : CUarray Destination array reference reserved1 : Any Must be NULL dstPitch : size_t Destination pitch (ignored when dst is array) dstHeight : size_t Destination height (ignored when dst is array; may be 0 if Depth==1) WidthInBytes : size_t Width of 3D memory copy in bytes Height : size_t Height of 3D memory copy Depth : size_t Depth of 3D memory copy Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._srcDevice = CUdeviceptr(_ptr=&self._ptr[0].srcDevice) self._srcArray = CUarray(_ptr=&self._ptr[0].srcArray) self._dstDevice = CUdeviceptr(_ptr=&self._ptr[0].dstDevice) self._dstArray = CUarray(_ptr=&self._ptr[0].dstArray) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['srcXInBytes : ' + str(self.srcXInBytes)] except ValueError: str_list += ['srcXInBytes : '] try: str_list += ['srcY : ' + str(self.srcY)] except ValueError: str_list += ['srcY : '] try: str_list += ['srcZ : ' + str(self.srcZ)] except ValueError: str_list += ['srcZ : '] try: str_list += ['srcLOD : ' + str(self.srcLOD)] except ValueError: str_list += ['srcLOD : '] try: str_list += ['srcMemoryType : ' + str(self.srcMemoryType)] except ValueError: str_list += ['srcMemoryType : '] try: str_list += ['srcHost : ' + hex(self.srcHost)] except ValueError: str_list += ['srcHost : '] try: str_list += ['srcDevice : ' + str(self.srcDevice)] except ValueError: str_list += ['srcDevice : '] try: str_list += ['srcArray : ' + str(self.srcArray)] except ValueError: str_list += ['srcArray : '] try: str_list += ['reserved0 : ' + hex(self.reserved0)] except ValueError: str_list += ['reserved0 : '] try: str_list += ['srcPitch : ' + str(self.srcPitch)] except ValueError: str_list += ['srcPitch : '] try: str_list += ['srcHeight : ' + str(self.srcHeight)] except ValueError: str_list += ['srcHeight : '] try: str_list += ['dstXInBytes : ' + str(self.dstXInBytes)] except ValueError: str_list += ['dstXInBytes : '] try: str_list += ['dstY : ' + str(self.dstY)] except ValueError: str_list += ['dstY : '] try: str_list += ['dstZ : ' + str(self.dstZ)] except ValueError: str_list += ['dstZ : '] try: str_list += ['dstLOD : ' + str(self.dstLOD)] except ValueError: str_list += ['dstLOD : '] try: str_list += ['dstMemoryType : ' + str(self.dstMemoryType)] except ValueError: str_list += ['dstMemoryType : '] try: str_list += ['dstHost : ' + hex(self.dstHost)] except ValueError: str_list += ['dstHost : '] try: str_list += ['dstDevice : ' + str(self.dstDevice)] except ValueError: str_list += ['dstDevice : '] try: str_list += ['dstArray : ' + str(self.dstArray)] except ValueError: str_list += ['dstArray : '] try: str_list += ['reserved1 : ' + hex(self.reserved1)] except ValueError: str_list += ['reserved1 : '] try: str_list += ['dstPitch : ' + str(self.dstPitch)] except ValueError: str_list += ['dstPitch : '] try: str_list += ['dstHeight : ' + str(self.dstHeight)] except ValueError: str_list += ['dstHeight : '] try: str_list += ['WidthInBytes : ' + str(self.WidthInBytes)] except ValueError: str_list += ['WidthInBytes : '] try: str_list += ['Height : ' + str(self.Height)] except ValueError: str_list += ['Height : '] try: str_list += ['Depth : ' + str(self.Depth)] except ValueError: str_list += ['Depth : '] return '\n'.join(str_list) else: return '' @property def srcXInBytes(self): return self._ptr[0].srcXInBytes @srcXInBytes.setter def srcXInBytes(self, size_t srcXInBytes): self._ptr[0].srcXInBytes = srcXInBytes @property def srcY(self): return self._ptr[0].srcY @srcY.setter def srcY(self, size_t srcY): self._ptr[0].srcY = srcY @property def srcZ(self): return self._ptr[0].srcZ @srcZ.setter def srcZ(self, size_t srcZ): self._ptr[0].srcZ = srcZ @property def srcLOD(self): return self._ptr[0].srcLOD @srcLOD.setter def srcLOD(self, size_t srcLOD): self._ptr[0].srcLOD = srcLOD @property def srcMemoryType(self): return CUmemorytype(self._ptr[0].srcMemoryType) @srcMemoryType.setter def srcMemoryType(self, srcMemoryType not None : CUmemorytype): self._ptr[0].srcMemoryType = srcMemoryType.value @property def srcHost(self): return self._ptr[0].srcHost @srcHost.setter def srcHost(self, srcHost): _csrcHost = utils.HelperInputVoidPtr(srcHost) self._ptr[0].srcHost = _csrcHost.cptr @property def srcDevice(self): return self._srcDevice @srcDevice.setter def srcDevice(self, srcDevice): cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice self._srcDevice._ptr[0] = csrcDevice @property def srcArray(self): return self._srcArray @srcArray.setter def srcArray(self, srcArray): cdef ccuda.CUarray csrcArray if srcArray is None: csrcArray = 0 elif isinstance(srcArray, (CUarray,)): psrcArray = int(srcArray) csrcArray = psrcArray else: psrcArray = int(CUarray(srcArray)) csrcArray = psrcArray self._srcArray._ptr[0] = csrcArray @property def reserved0(self): return self._ptr[0].reserved0 @reserved0.setter def reserved0(self, reserved0): _creserved0 = utils.HelperInputVoidPtr(reserved0) self._ptr[0].reserved0 = _creserved0.cptr @property def srcPitch(self): return self._ptr[0].srcPitch @srcPitch.setter def srcPitch(self, size_t srcPitch): self._ptr[0].srcPitch = srcPitch @property def srcHeight(self): return self._ptr[0].srcHeight @srcHeight.setter def srcHeight(self, size_t srcHeight): self._ptr[0].srcHeight = srcHeight @property def dstXInBytes(self): return self._ptr[0].dstXInBytes @dstXInBytes.setter def dstXInBytes(self, size_t dstXInBytes): self._ptr[0].dstXInBytes = dstXInBytes @property def dstY(self): return self._ptr[0].dstY @dstY.setter def dstY(self, size_t dstY): self._ptr[0].dstY = dstY @property def dstZ(self): return self._ptr[0].dstZ @dstZ.setter def dstZ(self, size_t dstZ): self._ptr[0].dstZ = dstZ @property def dstLOD(self): return self._ptr[0].dstLOD @dstLOD.setter def dstLOD(self, size_t dstLOD): self._ptr[0].dstLOD = dstLOD @property def dstMemoryType(self): return CUmemorytype(self._ptr[0].dstMemoryType) @dstMemoryType.setter def dstMemoryType(self, dstMemoryType not None : CUmemorytype): self._ptr[0].dstMemoryType = dstMemoryType.value @property def dstHost(self): return self._ptr[0].dstHost @dstHost.setter def dstHost(self, dstHost): _cdstHost = utils.HelperInputVoidPtr(dstHost) self._ptr[0].dstHost = _cdstHost.cptr @property def dstDevice(self): return self._dstDevice @dstDevice.setter def dstDevice(self, dstDevice): cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice self._dstDevice._ptr[0] = cdstDevice @property def dstArray(self): return self._dstArray @dstArray.setter def dstArray(self, dstArray): cdef ccuda.CUarray cdstArray if dstArray is None: cdstArray = 0 elif isinstance(dstArray, (CUarray,)): pdstArray = int(dstArray) cdstArray = pdstArray else: pdstArray = int(CUarray(dstArray)) cdstArray = pdstArray self._dstArray._ptr[0] = cdstArray @property def reserved1(self): return self._ptr[0].reserved1 @reserved1.setter def reserved1(self, reserved1): _creserved1 = utils.HelperInputVoidPtr(reserved1) self._ptr[0].reserved1 = _creserved1.cptr @property def dstPitch(self): return self._ptr[0].dstPitch @dstPitch.setter def dstPitch(self, size_t dstPitch): self._ptr[0].dstPitch = dstPitch @property def dstHeight(self): return self._ptr[0].dstHeight @dstHeight.setter def dstHeight(self, size_t dstHeight): self._ptr[0].dstHeight = dstHeight @property def WidthInBytes(self): return self._ptr[0].WidthInBytes @WidthInBytes.setter def WidthInBytes(self, size_t WidthInBytes): self._ptr[0].WidthInBytes = WidthInBytes @property def Height(self): return self._ptr[0].Height @Height.setter def Height(self, size_t Height): self._ptr[0].Height = Height @property def Depth(self): return self._ptr[0].Depth @Depth.setter def Depth(self, size_t Depth): self._ptr[0].Depth = Depth {{endif}} {{if 'struct CUDA_MEMCPY3D_PEER_st' in found_types}} cdef class CUDA_MEMCPY3D_PEER_st: """ 3D memory cross-context copy parameters Attributes ---------- srcXInBytes : size_t Source X in bytes srcY : size_t Source Y srcZ : size_t Source Z srcLOD : size_t Source LOD srcMemoryType : CUmemorytype Source memory type (host, device, array) srcHost : Any Source host pointer srcDevice : CUdeviceptr Source device pointer srcArray : CUarray Source array reference srcContext : CUcontext Source context (ignored with srcMemoryType is CU_MEMORYTYPE_ARRAY) srcPitch : size_t Source pitch (ignored when src is array) srcHeight : size_t Source height (ignored when src is array; may be 0 if Depth==1) dstXInBytes : size_t Destination X in bytes dstY : size_t Destination Y dstZ : size_t Destination Z dstLOD : size_t Destination LOD dstMemoryType : CUmemorytype Destination memory type (host, device, array) dstHost : Any Destination host pointer dstDevice : CUdeviceptr Destination device pointer dstArray : CUarray Destination array reference dstContext : CUcontext Destination context (ignored with dstMemoryType is CU_MEMORYTYPE_ARRAY) dstPitch : size_t Destination pitch (ignored when dst is array) dstHeight : size_t Destination height (ignored when dst is array; may be 0 if Depth==1) WidthInBytes : size_t Width of 3D memory copy in bytes Height : size_t Height of 3D memory copy Depth : size_t Depth of 3D memory copy Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._srcDevice = CUdeviceptr(_ptr=&self._ptr[0].srcDevice) self._srcArray = CUarray(_ptr=&self._ptr[0].srcArray) self._srcContext = CUcontext(_ptr=&self._ptr[0].srcContext) self._dstDevice = CUdeviceptr(_ptr=&self._ptr[0].dstDevice) self._dstArray = CUarray(_ptr=&self._ptr[0].dstArray) self._dstContext = CUcontext(_ptr=&self._ptr[0].dstContext) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['srcXInBytes : ' + str(self.srcXInBytes)] except ValueError: str_list += ['srcXInBytes : '] try: str_list += ['srcY : ' + str(self.srcY)] except ValueError: str_list += ['srcY : '] try: str_list += ['srcZ : ' + str(self.srcZ)] except ValueError: str_list += ['srcZ : '] try: str_list += ['srcLOD : ' + str(self.srcLOD)] except ValueError: str_list += ['srcLOD : '] try: str_list += ['srcMemoryType : ' + str(self.srcMemoryType)] except ValueError: str_list += ['srcMemoryType : '] try: str_list += ['srcHost : ' + hex(self.srcHost)] except ValueError: str_list += ['srcHost : '] try: str_list += ['srcDevice : ' + str(self.srcDevice)] except ValueError: str_list += ['srcDevice : '] try: str_list += ['srcArray : ' + str(self.srcArray)] except ValueError: str_list += ['srcArray : '] try: str_list += ['srcContext : ' + str(self.srcContext)] except ValueError: str_list += ['srcContext : '] try: str_list += ['srcPitch : ' + str(self.srcPitch)] except ValueError: str_list += ['srcPitch : '] try: str_list += ['srcHeight : ' + str(self.srcHeight)] except ValueError: str_list += ['srcHeight : '] try: str_list += ['dstXInBytes : ' + str(self.dstXInBytes)] except ValueError: str_list += ['dstXInBytes : '] try: str_list += ['dstY : ' + str(self.dstY)] except ValueError: str_list += ['dstY : '] try: str_list += ['dstZ : ' + str(self.dstZ)] except ValueError: str_list += ['dstZ : '] try: str_list += ['dstLOD : ' + str(self.dstLOD)] except ValueError: str_list += ['dstLOD : '] try: str_list += ['dstMemoryType : ' + str(self.dstMemoryType)] except ValueError: str_list += ['dstMemoryType : '] try: str_list += ['dstHost : ' + hex(self.dstHost)] except ValueError: str_list += ['dstHost : '] try: str_list += ['dstDevice : ' + str(self.dstDevice)] except ValueError: str_list += ['dstDevice : '] try: str_list += ['dstArray : ' + str(self.dstArray)] except ValueError: str_list += ['dstArray : '] try: str_list += ['dstContext : ' + str(self.dstContext)] except ValueError: str_list += ['dstContext : '] try: str_list += ['dstPitch : ' + str(self.dstPitch)] except ValueError: str_list += ['dstPitch : '] try: str_list += ['dstHeight : ' + str(self.dstHeight)] except ValueError: str_list += ['dstHeight : '] try: str_list += ['WidthInBytes : ' + str(self.WidthInBytes)] except ValueError: str_list += ['WidthInBytes : '] try: str_list += ['Height : ' + str(self.Height)] except ValueError: str_list += ['Height : '] try: str_list += ['Depth : ' + str(self.Depth)] except ValueError: str_list += ['Depth : '] return '\n'.join(str_list) else: return '' @property def srcXInBytes(self): return self._ptr[0].srcXInBytes @srcXInBytes.setter def srcXInBytes(self, size_t srcXInBytes): self._ptr[0].srcXInBytes = srcXInBytes @property def srcY(self): return self._ptr[0].srcY @srcY.setter def srcY(self, size_t srcY): self._ptr[0].srcY = srcY @property def srcZ(self): return self._ptr[0].srcZ @srcZ.setter def srcZ(self, size_t srcZ): self._ptr[0].srcZ = srcZ @property def srcLOD(self): return self._ptr[0].srcLOD @srcLOD.setter def srcLOD(self, size_t srcLOD): self._ptr[0].srcLOD = srcLOD @property def srcMemoryType(self): return CUmemorytype(self._ptr[0].srcMemoryType) @srcMemoryType.setter def srcMemoryType(self, srcMemoryType not None : CUmemorytype): self._ptr[0].srcMemoryType = srcMemoryType.value @property def srcHost(self): return self._ptr[0].srcHost @srcHost.setter def srcHost(self, srcHost): _csrcHost = utils.HelperInputVoidPtr(srcHost) self._ptr[0].srcHost = _csrcHost.cptr @property def srcDevice(self): return self._srcDevice @srcDevice.setter def srcDevice(self, srcDevice): cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice self._srcDevice._ptr[0] = csrcDevice @property def srcArray(self): return self._srcArray @srcArray.setter def srcArray(self, srcArray): cdef ccuda.CUarray csrcArray if srcArray is None: csrcArray = 0 elif isinstance(srcArray, (CUarray,)): psrcArray = int(srcArray) csrcArray = psrcArray else: psrcArray = int(CUarray(srcArray)) csrcArray = psrcArray self._srcArray._ptr[0] = csrcArray @property def srcContext(self): return self._srcContext @srcContext.setter def srcContext(self, srcContext): cdef ccuda.CUcontext csrcContext if srcContext is None: csrcContext = 0 elif isinstance(srcContext, (CUcontext,)): psrcContext = int(srcContext) csrcContext = psrcContext else: psrcContext = int(CUcontext(srcContext)) csrcContext = psrcContext self._srcContext._ptr[0] = csrcContext @property def srcPitch(self): return self._ptr[0].srcPitch @srcPitch.setter def srcPitch(self, size_t srcPitch): self._ptr[0].srcPitch = srcPitch @property def srcHeight(self): return self._ptr[0].srcHeight @srcHeight.setter def srcHeight(self, size_t srcHeight): self._ptr[0].srcHeight = srcHeight @property def dstXInBytes(self): return self._ptr[0].dstXInBytes @dstXInBytes.setter def dstXInBytes(self, size_t dstXInBytes): self._ptr[0].dstXInBytes = dstXInBytes @property def dstY(self): return self._ptr[0].dstY @dstY.setter def dstY(self, size_t dstY): self._ptr[0].dstY = dstY @property def dstZ(self): return self._ptr[0].dstZ @dstZ.setter def dstZ(self, size_t dstZ): self._ptr[0].dstZ = dstZ @property def dstLOD(self): return self._ptr[0].dstLOD @dstLOD.setter def dstLOD(self, size_t dstLOD): self._ptr[0].dstLOD = dstLOD @property def dstMemoryType(self): return CUmemorytype(self._ptr[0].dstMemoryType) @dstMemoryType.setter def dstMemoryType(self, dstMemoryType not None : CUmemorytype): self._ptr[0].dstMemoryType = dstMemoryType.value @property def dstHost(self): return self._ptr[0].dstHost @dstHost.setter def dstHost(self, dstHost): _cdstHost = utils.HelperInputVoidPtr(dstHost) self._ptr[0].dstHost = _cdstHost.cptr @property def dstDevice(self): return self._dstDevice @dstDevice.setter def dstDevice(self, dstDevice): cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice self._dstDevice._ptr[0] = cdstDevice @property def dstArray(self): return self._dstArray @dstArray.setter def dstArray(self, dstArray): cdef ccuda.CUarray cdstArray if dstArray is None: cdstArray = 0 elif isinstance(dstArray, (CUarray,)): pdstArray = int(dstArray) cdstArray = pdstArray else: pdstArray = int(CUarray(dstArray)) cdstArray = pdstArray self._dstArray._ptr[0] = cdstArray @property def dstContext(self): return self._dstContext @dstContext.setter def dstContext(self, dstContext): cdef ccuda.CUcontext cdstContext if dstContext is None: cdstContext = 0 elif isinstance(dstContext, (CUcontext,)): pdstContext = int(dstContext) cdstContext = pdstContext else: pdstContext = int(CUcontext(dstContext)) cdstContext = pdstContext self._dstContext._ptr[0] = cdstContext @property def dstPitch(self): return self._ptr[0].dstPitch @dstPitch.setter def dstPitch(self, size_t dstPitch): self._ptr[0].dstPitch = dstPitch @property def dstHeight(self): return self._ptr[0].dstHeight @dstHeight.setter def dstHeight(self, size_t dstHeight): self._ptr[0].dstHeight = dstHeight @property def WidthInBytes(self): return self._ptr[0].WidthInBytes @WidthInBytes.setter def WidthInBytes(self, size_t WidthInBytes): self._ptr[0].WidthInBytes = WidthInBytes @property def Height(self): return self._ptr[0].Height @Height.setter def Height(self, size_t Height): self._ptr[0].Height = Height @property def Depth(self): return self._ptr[0].Depth @Depth.setter def Depth(self, size_t Depth): self._ptr[0].Depth = Depth {{endif}} {{if 'struct CUDA_MEMCPY_NODE_PARAMS_st' in found_types}} cdef class CUDA_MEMCPY_NODE_PARAMS_st: """ Memcpy node parameters Attributes ---------- flags : int Must be zero reserved : int Must be zero copyCtx : CUcontext Context on which to run the node copyParams : CUDA_MEMCPY3D Parameters for the memory copy Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._copyCtx = CUcontext(_ptr=&self._ptr[0].copyCtx) self._copyParams = CUDA_MEMCPY3D(_ptr=&self._ptr[0].copyParams) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] try: str_list += ['copyCtx : ' + str(self.copyCtx)] except ValueError: str_list += ['copyCtx : '] try: str_list += ['copyParams :\n' + '\n'.join([' ' + line for line in str(self.copyParams).splitlines()])] except ValueError: str_list += ['copyParams : '] return '\n'.join(str_list) else: return '' @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, int reserved): self._ptr[0].reserved = reserved @property def copyCtx(self): return self._copyCtx @copyCtx.setter def copyCtx(self, copyCtx): cdef ccuda.CUcontext ccopyCtx if copyCtx is None: ccopyCtx = 0 elif isinstance(copyCtx, (CUcontext,)): pcopyCtx = int(copyCtx) ccopyCtx = pcopyCtx else: pcopyCtx = int(CUcontext(copyCtx)) ccopyCtx = pcopyCtx self._copyCtx._ptr[0] = ccopyCtx @property def copyParams(self): return self._copyParams @copyParams.setter def copyParams(self, copyParams not None : CUDA_MEMCPY3D): string.memcpy(&self._ptr[0].copyParams, copyParams.getPtr(), sizeof(self._ptr[0].copyParams)) {{endif}} {{if 'struct CUDA_ARRAY_DESCRIPTOR_st' in found_types}} cdef class CUDA_ARRAY_DESCRIPTOR_st: """ Array descriptor Attributes ---------- Width : size_t Width of array Height : size_t Height of array Format : CUarray_format Array format NumChannels : unsigned int Channels per array element Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['Width : ' + str(self.Width)] except ValueError: str_list += ['Width : '] try: str_list += ['Height : ' + str(self.Height)] except ValueError: str_list += ['Height : '] try: str_list += ['Format : ' + str(self.Format)] except ValueError: str_list += ['Format : '] try: str_list += ['NumChannels : ' + str(self.NumChannels)] except ValueError: str_list += ['NumChannels : '] return '\n'.join(str_list) else: return '' @property def Width(self): return self._ptr[0].Width @Width.setter def Width(self, size_t Width): self._ptr[0].Width = Width @property def Height(self): return self._ptr[0].Height @Height.setter def Height(self, size_t Height): self._ptr[0].Height = Height @property def Format(self): return CUarray_format(self._ptr[0].Format) @Format.setter def Format(self, Format not None : CUarray_format): self._ptr[0].Format = Format.value @property def NumChannels(self): return self._ptr[0].NumChannels @NumChannels.setter def NumChannels(self, unsigned int NumChannels): self._ptr[0].NumChannels = NumChannels {{endif}} {{if 'struct CUDA_ARRAY3D_DESCRIPTOR_st' in found_types}} cdef class CUDA_ARRAY3D_DESCRIPTOR_st: """ 3D array descriptor Attributes ---------- Width : size_t Width of 3D array Height : size_t Height of 3D array Depth : size_t Depth of 3D array Format : CUarray_format Array format NumChannels : unsigned int Channels per array element Flags : unsigned int Flags Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['Width : ' + str(self.Width)] except ValueError: str_list += ['Width : '] try: str_list += ['Height : ' + str(self.Height)] except ValueError: str_list += ['Height : '] try: str_list += ['Depth : ' + str(self.Depth)] except ValueError: str_list += ['Depth : '] try: str_list += ['Format : ' + str(self.Format)] except ValueError: str_list += ['Format : '] try: str_list += ['NumChannels : ' + str(self.NumChannels)] except ValueError: str_list += ['NumChannels : '] try: str_list += ['Flags : ' + str(self.Flags)] except ValueError: str_list += ['Flags : '] return '\n'.join(str_list) else: return '' @property def Width(self): return self._ptr[0].Width @Width.setter def Width(self, size_t Width): self._ptr[0].Width = Width @property def Height(self): return self._ptr[0].Height @Height.setter def Height(self, size_t Height): self._ptr[0].Height = Height @property def Depth(self): return self._ptr[0].Depth @Depth.setter def Depth(self, size_t Depth): self._ptr[0].Depth = Depth @property def Format(self): return CUarray_format(self._ptr[0].Format) @Format.setter def Format(self, Format not None : CUarray_format): self._ptr[0].Format = Format.value @property def NumChannels(self): return self._ptr[0].NumChannels @NumChannels.setter def NumChannels(self, unsigned int NumChannels): self._ptr[0].NumChannels = NumChannels @property def Flags(self): return self._ptr[0].Flags @Flags.setter def Flags(self, unsigned int Flags): self._ptr[0].Flags = Flags {{endif}} {{if 'struct CUDA_ARRAY_SPARSE_PROPERTIES_st' in found_types}} cdef class anon_struct5: """ Attributes ---------- width : unsigned int height : unsigned int depth : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].tileExtent def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['width : ' + str(self.width)] except ValueError: str_list += ['width : '] try: str_list += ['height : ' + str(self.height)] except ValueError: str_list += ['height : '] try: str_list += ['depth : ' + str(self.depth)] except ValueError: str_list += ['depth : '] return '\n'.join(str_list) else: return '' @property def width(self): return self._ptr[0].tileExtent.width @width.setter def width(self, unsigned int width): self._ptr[0].tileExtent.width = width @property def height(self): return self._ptr[0].tileExtent.height @height.setter def height(self, unsigned int height): self._ptr[0].tileExtent.height = height @property def depth(self): return self._ptr[0].tileExtent.depth @depth.setter def depth(self, unsigned int depth): self._ptr[0].tileExtent.depth = depth {{endif}} {{if 'struct CUDA_ARRAY_SPARSE_PROPERTIES_st' in found_types}} cdef class CUDA_ARRAY_SPARSE_PROPERTIES_st: """ CUDA array sparse properties Attributes ---------- tileExtent : anon_struct5 miptailFirstLevel : unsigned int First mip level at which the mip tail begins. miptailSize : unsigned long long Total size of the mip tail. flags : unsigned int Flags will either be zero or CU_ARRAY_SPARSE_PROPERTIES_SINGLE_MIPTAIL reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._tileExtent = anon_struct5(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['tileExtent :\n' + '\n'.join([' ' + line for line in str(self.tileExtent).splitlines()])] except ValueError: str_list += ['tileExtent : '] try: str_list += ['miptailFirstLevel : ' + str(self.miptailFirstLevel)] except ValueError: str_list += ['miptailFirstLevel : '] try: str_list += ['miptailSize : ' + str(self.miptailSize)] except ValueError: str_list += ['miptailSize : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def tileExtent(self): return self._tileExtent @tileExtent.setter def tileExtent(self, tileExtent not None : anon_struct5): string.memcpy(&self._ptr[0].tileExtent, tileExtent.getPtr(), sizeof(self._ptr[0].tileExtent)) @property def miptailFirstLevel(self): return self._ptr[0].miptailFirstLevel @miptailFirstLevel.setter def miptailFirstLevel(self, unsigned int miptailFirstLevel): self._ptr[0].miptailFirstLevel = miptailFirstLevel @property def miptailSize(self): return self._ptr[0].miptailSize @miptailSize.setter def miptailSize(self, unsigned long long miptailSize): self._ptr[0].miptailSize = miptailSize @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_ARRAY_MEMORY_REQUIREMENTS_st' in found_types}} cdef class CUDA_ARRAY_MEMORY_REQUIREMENTS_st: """ CUDA array memory requirements Attributes ---------- size : size_t Total required memory size alignment : size_t alignment requirement reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['size : ' + str(self.size)] except ValueError: str_list += ['size : '] try: str_list += ['alignment : ' + str(self.alignment)] except ValueError: str_list += ['alignment : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def size(self): return self._ptr[0].size @size.setter def size(self, size_t size): self._ptr[0].size = size @property def alignment(self): return self._ptr[0].alignment @alignment.setter def alignment(self, size_t alignment): self._ptr[0].alignment = alignment @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} cdef class anon_struct6: """ Attributes ---------- hArray : CUarray Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._hArray = CUarray(_ptr=&self._ptr[0].res.array.hArray) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].res.array def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['hArray : ' + str(self.hArray)] except ValueError: str_list += ['hArray : '] return '\n'.join(str_list) else: return '' @property def hArray(self): return self._hArray @hArray.setter def hArray(self, hArray): cdef ccuda.CUarray chArray if hArray is None: chArray = 0 elif isinstance(hArray, (CUarray,)): phArray = int(hArray) chArray = phArray else: phArray = int(CUarray(hArray)) chArray = phArray self._hArray._ptr[0] = chArray {{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} cdef class anon_struct7: """ Attributes ---------- hMipmappedArray : CUmipmappedArray Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._hMipmappedArray = CUmipmappedArray(_ptr=&self._ptr[0].res.mipmap.hMipmappedArray) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].res.mipmap def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['hMipmappedArray : ' + str(self.hMipmappedArray)] except ValueError: str_list += ['hMipmappedArray : '] return '\n'.join(str_list) else: return '' @property def hMipmappedArray(self): return self._hMipmappedArray @hMipmappedArray.setter def hMipmappedArray(self, hMipmappedArray): cdef ccuda.CUmipmappedArray chMipmappedArray if hMipmappedArray is None: chMipmappedArray = 0 elif isinstance(hMipmappedArray, (CUmipmappedArray,)): phMipmappedArray = int(hMipmappedArray) chMipmappedArray = phMipmappedArray else: phMipmappedArray = int(CUmipmappedArray(hMipmappedArray)) chMipmappedArray = phMipmappedArray self._hMipmappedArray._ptr[0] = chMipmappedArray {{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} cdef class anon_struct8: """ Attributes ---------- devPtr : CUdeviceptr format : CUarray_format numChannels : unsigned int sizeInBytes : size_t Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._devPtr = CUdeviceptr(_ptr=&self._ptr[0].res.linear.devPtr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].res.linear def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['devPtr : ' + str(self.devPtr)] except ValueError: str_list += ['devPtr : '] try: str_list += ['format : ' + str(self.format)] except ValueError: str_list += ['format : '] try: str_list += ['numChannels : ' + str(self.numChannels)] except ValueError: str_list += ['numChannels : '] try: str_list += ['sizeInBytes : ' + str(self.sizeInBytes)] except ValueError: str_list += ['sizeInBytes : '] return '\n'.join(str_list) else: return '' @property def devPtr(self): return self._devPtr @devPtr.setter def devPtr(self, devPtr): cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr self._devPtr._ptr[0] = cdevPtr @property def format(self): return CUarray_format(self._ptr[0].res.linear.format) @format.setter def format(self, format not None : CUarray_format): self._ptr[0].res.linear.format = format.value @property def numChannels(self): return self._ptr[0].res.linear.numChannels @numChannels.setter def numChannels(self, unsigned int numChannels): self._ptr[0].res.linear.numChannels = numChannels @property def sizeInBytes(self): return self._ptr[0].res.linear.sizeInBytes @sizeInBytes.setter def sizeInBytes(self, size_t sizeInBytes): self._ptr[0].res.linear.sizeInBytes = sizeInBytes {{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} cdef class anon_struct9: """ Attributes ---------- devPtr : CUdeviceptr format : CUarray_format numChannels : unsigned int width : size_t height : size_t pitchInBytes : size_t Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._devPtr = CUdeviceptr(_ptr=&self._ptr[0].res.pitch2D.devPtr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].res.pitch2D def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['devPtr : ' + str(self.devPtr)] except ValueError: str_list += ['devPtr : '] try: str_list += ['format : ' + str(self.format)] except ValueError: str_list += ['format : '] try: str_list += ['numChannels : ' + str(self.numChannels)] except ValueError: str_list += ['numChannels : '] try: str_list += ['width : ' + str(self.width)] except ValueError: str_list += ['width : '] try: str_list += ['height : ' + str(self.height)] except ValueError: str_list += ['height : '] try: str_list += ['pitchInBytes : ' + str(self.pitchInBytes)] except ValueError: str_list += ['pitchInBytes : '] return '\n'.join(str_list) else: return '' @property def devPtr(self): return self._devPtr @devPtr.setter def devPtr(self, devPtr): cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr self._devPtr._ptr[0] = cdevPtr @property def format(self): return CUarray_format(self._ptr[0].res.pitch2D.format) @format.setter def format(self, format not None : CUarray_format): self._ptr[0].res.pitch2D.format = format.value @property def numChannels(self): return self._ptr[0].res.pitch2D.numChannels @numChannels.setter def numChannels(self, unsigned int numChannels): self._ptr[0].res.pitch2D.numChannels = numChannels @property def width(self): return self._ptr[0].res.pitch2D.width @width.setter def width(self, size_t width): self._ptr[0].res.pitch2D.width = width @property def height(self): return self._ptr[0].res.pitch2D.height @height.setter def height(self, size_t height): self._ptr[0].res.pitch2D.height = height @property def pitchInBytes(self): return self._ptr[0].res.pitch2D.pitchInBytes @pitchInBytes.setter def pitchInBytes(self, size_t pitchInBytes): self._ptr[0].res.pitch2D.pitchInBytes = pitchInBytes {{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} cdef class anon_struct10: """ Attributes ---------- reserved : List[int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].res.reserved def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def reserved(self): return self._ptr[0].res.reserved.reserved @reserved.setter def reserved(self, reserved): self._ptr[0].res.reserved.reserved = reserved {{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} cdef class anon_union4: """ Attributes ---------- array : anon_struct6 mipmap : anon_struct7 linear : anon_struct8 pitch2D : anon_struct9 reserved : anon_struct10 Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._array = anon_struct6(_ptr=self._ptr) self._mipmap = anon_struct7(_ptr=self._ptr) self._linear = anon_struct8(_ptr=self._ptr) self._pitch2D = anon_struct9(_ptr=self._ptr) self._reserved = anon_struct10(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].res def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['array :\n' + '\n'.join([' ' + line for line in str(self.array).splitlines()])] except ValueError: str_list += ['array : '] try: str_list += ['mipmap :\n' + '\n'.join([' ' + line for line in str(self.mipmap).splitlines()])] except ValueError: str_list += ['mipmap : '] try: str_list += ['linear :\n' + '\n'.join([' ' + line for line in str(self.linear).splitlines()])] except ValueError: str_list += ['linear : '] try: str_list += ['pitch2D :\n' + '\n'.join([' ' + line for line in str(self.pitch2D).splitlines()])] except ValueError: str_list += ['pitch2D : '] try: str_list += ['reserved :\n' + '\n'.join([' ' + line for line in str(self.reserved).splitlines()])] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def array(self): return self._array @array.setter def array(self, array not None : anon_struct6): string.memcpy(&self._ptr[0].res.array, array.getPtr(), sizeof(self._ptr[0].res.array)) @property def mipmap(self): return self._mipmap @mipmap.setter def mipmap(self, mipmap not None : anon_struct7): string.memcpy(&self._ptr[0].res.mipmap, mipmap.getPtr(), sizeof(self._ptr[0].res.mipmap)) @property def linear(self): return self._linear @linear.setter def linear(self, linear not None : anon_struct8): string.memcpy(&self._ptr[0].res.linear, linear.getPtr(), sizeof(self._ptr[0].res.linear)) @property def pitch2D(self): return self._pitch2D @pitch2D.setter def pitch2D(self, pitch2D not None : anon_struct9): string.memcpy(&self._ptr[0].res.pitch2D, pitch2D.getPtr(), sizeof(self._ptr[0].res.pitch2D)) @property def reserved(self): return self._reserved @reserved.setter def reserved(self, reserved not None : anon_struct10): string.memcpy(&self._ptr[0].res.reserved, reserved.getPtr(), sizeof(self._ptr[0].res.reserved)) {{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} cdef class CUDA_RESOURCE_DESC_st: """ CUDA Resource descriptor Attributes ---------- resType : CUresourcetype Resource type res : anon_union4 flags : unsigned int Flags (must be zero) Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUDA_RESOURCE_DESC_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._res = anon_union4(_ptr=self._ptr) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['resType : ' + str(self.resType)] except ValueError: str_list += ['resType : '] try: str_list += ['res :\n' + '\n'.join([' ' + line for line in str(self.res).splitlines()])] except ValueError: str_list += ['res : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def resType(self): return CUresourcetype(self._ptr[0].resType) @resType.setter def resType(self, resType not None : CUresourcetype): self._ptr[0].resType = resType.value @property def res(self): return self._res @res.setter def res(self, res not None : anon_union4): string.memcpy(&self._ptr[0].res, res.getPtr(), sizeof(self._ptr[0].res)) @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags {{endif}} {{if 'struct CUDA_TEXTURE_DESC_st' in found_types}} cdef class CUDA_TEXTURE_DESC_st: """ Texture descriptor Attributes ---------- addressMode : List[CUaddress_mode] Address modes filterMode : CUfilter_mode Filter mode flags : unsigned int Flags maxAnisotropy : unsigned int Maximum anisotropy ratio mipmapFilterMode : CUfilter_mode Mipmap filter mode mipmapLevelBias : float Mipmap level bias minMipmapLevelClamp : float Mipmap minimum level clamp maxMipmapLevelClamp : float Mipmap maximum level clamp borderColor : List[float] Border Color reserved : List[int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['addressMode : ' + str(self.addressMode)] except ValueError: str_list += ['addressMode : '] try: str_list += ['filterMode : ' + str(self.filterMode)] except ValueError: str_list += ['filterMode : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['maxAnisotropy : ' + str(self.maxAnisotropy)] except ValueError: str_list += ['maxAnisotropy : '] try: str_list += ['mipmapFilterMode : ' + str(self.mipmapFilterMode)] except ValueError: str_list += ['mipmapFilterMode : '] try: str_list += ['mipmapLevelBias : ' + str(self.mipmapLevelBias)] except ValueError: str_list += ['mipmapLevelBias : '] try: str_list += ['minMipmapLevelClamp : ' + str(self.minMipmapLevelClamp)] except ValueError: str_list += ['minMipmapLevelClamp : '] try: str_list += ['maxMipmapLevelClamp : ' + str(self.maxMipmapLevelClamp)] except ValueError: str_list += ['maxMipmapLevelClamp : '] try: str_list += ['borderColor : ' + str(self.borderColor)] except ValueError: str_list += ['borderColor : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def addressMode(self): return [CUaddress_mode(_x) for _x in list(self._ptr[0].addressMode)] @addressMode.setter def addressMode(self, addressMode): self._ptr[0].addressMode = [_x.value for _x in addressMode] @property def filterMode(self): return CUfilter_mode(self._ptr[0].filterMode) @filterMode.setter def filterMode(self, filterMode not None : CUfilter_mode): self._ptr[0].filterMode = filterMode.value @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def maxAnisotropy(self): return self._ptr[0].maxAnisotropy @maxAnisotropy.setter def maxAnisotropy(self, unsigned int maxAnisotropy): self._ptr[0].maxAnisotropy = maxAnisotropy @property def mipmapFilterMode(self): return CUfilter_mode(self._ptr[0].mipmapFilterMode) @mipmapFilterMode.setter def mipmapFilterMode(self, mipmapFilterMode not None : CUfilter_mode): self._ptr[0].mipmapFilterMode = mipmapFilterMode.value @property def mipmapLevelBias(self): return self._ptr[0].mipmapLevelBias @mipmapLevelBias.setter def mipmapLevelBias(self, float mipmapLevelBias): self._ptr[0].mipmapLevelBias = mipmapLevelBias @property def minMipmapLevelClamp(self): return self._ptr[0].minMipmapLevelClamp @minMipmapLevelClamp.setter def minMipmapLevelClamp(self, float minMipmapLevelClamp): self._ptr[0].minMipmapLevelClamp = minMipmapLevelClamp @property def maxMipmapLevelClamp(self): return self._ptr[0].maxMipmapLevelClamp @maxMipmapLevelClamp.setter def maxMipmapLevelClamp(self, float maxMipmapLevelClamp): self._ptr[0].maxMipmapLevelClamp = maxMipmapLevelClamp @property def borderColor(self): return self._ptr[0].borderColor @borderColor.setter def borderColor(self, borderColor): self._ptr[0].borderColor = borderColor @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_RESOURCE_VIEW_DESC_st' in found_types}} cdef class CUDA_RESOURCE_VIEW_DESC_st: """ Resource view descriptor Attributes ---------- format : CUresourceViewFormat Resource view format width : size_t Width of the resource view height : size_t Height of the resource view depth : size_t Depth of the resource view firstMipmapLevel : unsigned int First defined mipmap level lastMipmapLevel : unsigned int Last defined mipmap level firstLayer : unsigned int First layer index lastLayer : unsigned int Last layer index reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['format : ' + str(self.format)] except ValueError: str_list += ['format : '] try: str_list += ['width : ' + str(self.width)] except ValueError: str_list += ['width : '] try: str_list += ['height : ' + str(self.height)] except ValueError: str_list += ['height : '] try: str_list += ['depth : ' + str(self.depth)] except ValueError: str_list += ['depth : '] try: str_list += ['firstMipmapLevel : ' + str(self.firstMipmapLevel)] except ValueError: str_list += ['firstMipmapLevel : '] try: str_list += ['lastMipmapLevel : ' + str(self.lastMipmapLevel)] except ValueError: str_list += ['lastMipmapLevel : '] try: str_list += ['firstLayer : ' + str(self.firstLayer)] except ValueError: str_list += ['firstLayer : '] try: str_list += ['lastLayer : ' + str(self.lastLayer)] except ValueError: str_list += ['lastLayer : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def format(self): return CUresourceViewFormat(self._ptr[0].format) @format.setter def format(self, format not None : CUresourceViewFormat): self._ptr[0].format = format.value @property def width(self): return self._ptr[0].width @width.setter def width(self, size_t width): self._ptr[0].width = width @property def height(self): return self._ptr[0].height @height.setter def height(self, size_t height): self._ptr[0].height = height @property def depth(self): return self._ptr[0].depth @depth.setter def depth(self, size_t depth): self._ptr[0].depth = depth @property def firstMipmapLevel(self): return self._ptr[0].firstMipmapLevel @firstMipmapLevel.setter def firstMipmapLevel(self, unsigned int firstMipmapLevel): self._ptr[0].firstMipmapLevel = firstMipmapLevel @property def lastMipmapLevel(self): return self._ptr[0].lastMipmapLevel @lastMipmapLevel.setter def lastMipmapLevel(self, unsigned int lastMipmapLevel): self._ptr[0].lastMipmapLevel = lastMipmapLevel @property def firstLayer(self): return self._ptr[0].firstLayer @firstLayer.setter def firstLayer(self, unsigned int firstLayer): self._ptr[0].firstLayer = firstLayer @property def lastLayer(self): return self._ptr[0].lastLayer @lastLayer.setter def lastLayer(self, unsigned int lastLayer): self._ptr[0].lastLayer = lastLayer @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUtensorMap_st' in found_types}} cdef class CUtensorMap_st: """ Tensor map descriptor. Requires compiler support for aligning to 64 bytes. Attributes ---------- opaque : List[cuuint64_t] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['opaque : ' + str(self.opaque)] except ValueError: str_list += ['opaque : '] return '\n'.join(str_list) else: return '' @property def opaque(self): return [cuuint64_t(init_value=_opaque) for _opaque in self._ptr[0].opaque] @opaque.setter def opaque(self, opaque): self._ptr[0].opaque = opaque {{endif}} {{if 'struct CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_st' in found_types}} cdef class CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_st: """ GPU Direct v3 tokens Attributes ---------- p2pToken : unsigned long long vaSpaceToken : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['p2pToken : ' + str(self.p2pToken)] except ValueError: str_list += ['p2pToken : '] try: str_list += ['vaSpaceToken : ' + str(self.vaSpaceToken)] except ValueError: str_list += ['vaSpaceToken : '] return '\n'.join(str_list) else: return '' @property def p2pToken(self): return self._ptr[0].p2pToken @p2pToken.setter def p2pToken(self, unsigned long long p2pToken): self._ptr[0].p2pToken = p2pToken @property def vaSpaceToken(self): return self._ptr[0].vaSpaceToken @vaSpaceToken.setter def vaSpaceToken(self, unsigned int vaSpaceToken): self._ptr[0].vaSpaceToken = vaSpaceToken {{endif}} {{if 'struct CUDA_LAUNCH_PARAMS_st' in found_types}} cdef class CUDA_LAUNCH_PARAMS_st: """ Kernel launch parameters Attributes ---------- function : CUfunction Kernel to launch gridDimX : unsigned int Width of grid in blocks gridDimY : unsigned int Height of grid in blocks gridDimZ : unsigned int Depth of grid in blocks blockDimX : unsigned int X dimension of each thread block blockDimY : unsigned int Y dimension of each thread block blockDimZ : unsigned int Z dimension of each thread block sharedMemBytes : unsigned int Dynamic shared-memory size per thread block in bytes hStream : CUstream Stream identifier kernelParams : Any Array of pointers to kernel parameters Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._function = CUfunction(_ptr=&self._ptr[0].function) self._hStream = CUstream(_ptr=&self._ptr[0].hStream) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['function : ' + str(self.function)] except ValueError: str_list += ['function : '] try: str_list += ['gridDimX : ' + str(self.gridDimX)] except ValueError: str_list += ['gridDimX : '] try: str_list += ['gridDimY : ' + str(self.gridDimY)] except ValueError: str_list += ['gridDimY : '] try: str_list += ['gridDimZ : ' + str(self.gridDimZ)] except ValueError: str_list += ['gridDimZ : '] try: str_list += ['blockDimX : ' + str(self.blockDimX)] except ValueError: str_list += ['blockDimX : '] try: str_list += ['blockDimY : ' + str(self.blockDimY)] except ValueError: str_list += ['blockDimY : '] try: str_list += ['blockDimZ : ' + str(self.blockDimZ)] except ValueError: str_list += ['blockDimZ : '] try: str_list += ['sharedMemBytes : ' + str(self.sharedMemBytes)] except ValueError: str_list += ['sharedMemBytes : '] try: str_list += ['hStream : ' + str(self.hStream)] except ValueError: str_list += ['hStream : '] try: str_list += ['kernelParams : ' + str(self.kernelParams)] except ValueError: str_list += ['kernelParams : '] return '\n'.join(str_list) else: return '' @property def function(self): return self._function @function.setter def function(self, function): cdef ccuda.CUfunction cfunction if function is None: cfunction = 0 elif isinstance(function, (CUfunction,)): pfunction = int(function) cfunction = pfunction else: pfunction = int(CUfunction(function)) cfunction = pfunction self._function._ptr[0] = cfunction @property def gridDimX(self): return self._ptr[0].gridDimX @gridDimX.setter def gridDimX(self, unsigned int gridDimX): self._ptr[0].gridDimX = gridDimX @property def gridDimY(self): return self._ptr[0].gridDimY @gridDimY.setter def gridDimY(self, unsigned int gridDimY): self._ptr[0].gridDimY = gridDimY @property def gridDimZ(self): return self._ptr[0].gridDimZ @gridDimZ.setter def gridDimZ(self, unsigned int gridDimZ): self._ptr[0].gridDimZ = gridDimZ @property def blockDimX(self): return self._ptr[0].blockDimX @blockDimX.setter def blockDimX(self, unsigned int blockDimX): self._ptr[0].blockDimX = blockDimX @property def blockDimY(self): return self._ptr[0].blockDimY @blockDimY.setter def blockDimY(self, unsigned int blockDimY): self._ptr[0].blockDimY = blockDimY @property def blockDimZ(self): return self._ptr[0].blockDimZ @blockDimZ.setter def blockDimZ(self, unsigned int blockDimZ): self._ptr[0].blockDimZ = blockDimZ @property def sharedMemBytes(self): return self._ptr[0].sharedMemBytes @sharedMemBytes.setter def sharedMemBytes(self, unsigned int sharedMemBytes): self._ptr[0].sharedMemBytes = sharedMemBytes @property def hStream(self): return self._hStream @hStream.setter def hStream(self, hStream): cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream self._hStream._ptr[0] = chStream @property def kernelParams(self): return self._ptr[0].kernelParams @kernelParams.setter def kernelParams(self, kernelParams): self._ckernelParams = utils.HelperKernelParams(kernelParams) self._ptr[0].kernelParams = self._ckernelParams.ckernelParams {{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st' in found_types}} cdef class anon_struct11: """ Attributes ---------- handle : Any name : Any Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].handle.win32 def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['handle : ' + hex(self.handle)] except ValueError: str_list += ['handle : '] try: str_list += ['name : ' + hex(self.name)] except ValueError: str_list += ['name : '] return '\n'.join(str_list) else: return '' @property def handle(self): return self._ptr[0].handle.win32.handle @handle.setter def handle(self, handle): _chandle = utils.HelperInputVoidPtr(handle) self._ptr[0].handle.win32.handle = _chandle.cptr @property def name(self): return self._ptr[0].handle.win32.name @name.setter def name(self, name): _cname = utils.HelperInputVoidPtr(name) self._ptr[0].handle.win32.name = _cname.cptr {{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st' in found_types}} cdef class anon_union5: """ Attributes ---------- fd : int win32 : anon_struct11 nvSciBufObject : Any Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._win32 = anon_struct11(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].handle def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fd : ' + str(self.fd)] except ValueError: str_list += ['fd : '] try: str_list += ['win32 :\n' + '\n'.join([' ' + line for line in str(self.win32).splitlines()])] except ValueError: str_list += ['win32 : '] try: str_list += ['nvSciBufObject : ' + hex(self.nvSciBufObject)] except ValueError: str_list += ['nvSciBufObject : '] return '\n'.join(str_list) else: return '' @property def fd(self): return self._ptr[0].handle.fd @fd.setter def fd(self, int fd): self._ptr[0].handle.fd = fd @property def win32(self): return self._win32 @win32.setter def win32(self, win32 not None : anon_struct11): string.memcpy(&self._ptr[0].handle.win32, win32.getPtr(), sizeof(self._ptr[0].handle.win32)) @property def nvSciBufObject(self): return self._ptr[0].handle.nvSciBufObject @nvSciBufObject.setter def nvSciBufObject(self, nvSciBufObject): _cnvSciBufObject = utils.HelperInputVoidPtr(nvSciBufObject) self._ptr[0].handle.nvSciBufObject = _cnvSciBufObject.cptr {{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st' in found_types}} cdef class CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st: """ External memory handle descriptor Attributes ---------- type : CUexternalMemoryHandleType Type of the handle handle : anon_union5 size : unsigned long long Size of the memory allocation flags : unsigned int Flags must either be zero or CUDA_EXTERNAL_MEMORY_DEDICATED reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._handle = anon_union5(_ptr=self._ptr) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['handle :\n' + '\n'.join([' ' + line for line in str(self.handle).splitlines()])] except ValueError: str_list += ['handle : '] try: str_list += ['size : ' + str(self.size)] except ValueError: str_list += ['size : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUexternalMemoryHandleType(self._ptr[0].type) @type.setter def type(self, type not None : CUexternalMemoryHandleType): self._ptr[0].type = type.value @property def handle(self): return self._handle @handle.setter def handle(self, handle not None : anon_union5): string.memcpy(&self._ptr[0].handle, handle.getPtr(), sizeof(self._ptr[0].handle)) @property def size(self): return self._ptr[0].size @size.setter def size(self, unsigned long long size): self._ptr[0].size = size @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_BUFFER_DESC_st' in found_types}} cdef class CUDA_EXTERNAL_MEMORY_BUFFER_DESC_st: """ External memory buffer descriptor Attributes ---------- offset : unsigned long long Offset into the memory object where the buffer's base is size : unsigned long long Size of the buffer flags : unsigned int Flags reserved for future use. Must be zero. reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['offset : ' + str(self.offset)] except ValueError: str_list += ['offset : '] try: str_list += ['size : ' + str(self.size)] except ValueError: str_list += ['size : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def offset(self): return self._ptr[0].offset @offset.setter def offset(self, unsigned long long offset): self._ptr[0].offset = offset @property def size(self): return self._ptr[0].size @size.setter def size(self, unsigned long long size): self._ptr[0].size = size @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_st' in found_types}} cdef class CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_st: """ External memory mipmap descriptor Attributes ---------- offset : unsigned long long Offset into the memory object where the base level of the mipmap chain is. arrayDesc : CUDA_ARRAY3D_DESCRIPTOR Format, dimension and type of base level of the mipmap chain numLevels : unsigned int Total number of levels in the mipmap chain reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._arrayDesc = CUDA_ARRAY3D_DESCRIPTOR(_ptr=&self._ptr[0].arrayDesc) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['offset : ' + str(self.offset)] except ValueError: str_list += ['offset : '] try: str_list += ['arrayDesc :\n' + '\n'.join([' ' + line for line in str(self.arrayDesc).splitlines()])] except ValueError: str_list += ['arrayDesc : '] try: str_list += ['numLevels : ' + str(self.numLevels)] except ValueError: str_list += ['numLevels : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def offset(self): return self._ptr[0].offset @offset.setter def offset(self, unsigned long long offset): self._ptr[0].offset = offset @property def arrayDesc(self): return self._arrayDesc @arrayDesc.setter def arrayDesc(self, arrayDesc not None : CUDA_ARRAY3D_DESCRIPTOR): string.memcpy(&self._ptr[0].arrayDesc, arrayDesc.getPtr(), sizeof(self._ptr[0].arrayDesc)) @property def numLevels(self): return self._ptr[0].numLevels @numLevels.setter def numLevels(self, unsigned int numLevels): self._ptr[0].numLevels = numLevels @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st' in found_types}} cdef class anon_struct12: """ Attributes ---------- handle : Any name : Any Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].handle.win32 def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['handle : ' + hex(self.handle)] except ValueError: str_list += ['handle : '] try: str_list += ['name : ' + hex(self.name)] except ValueError: str_list += ['name : '] return '\n'.join(str_list) else: return '' @property def handle(self): return self._ptr[0].handle.win32.handle @handle.setter def handle(self, handle): _chandle = utils.HelperInputVoidPtr(handle) self._ptr[0].handle.win32.handle = _chandle.cptr @property def name(self): return self._ptr[0].handle.win32.name @name.setter def name(self, name): _cname = utils.HelperInputVoidPtr(name) self._ptr[0].handle.win32.name = _cname.cptr {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st' in found_types}} cdef class anon_union6: """ Attributes ---------- fd : int win32 : anon_struct12 nvSciSyncObj : Any Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._win32 = anon_struct12(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].handle def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fd : ' + str(self.fd)] except ValueError: str_list += ['fd : '] try: str_list += ['win32 :\n' + '\n'.join([' ' + line for line in str(self.win32).splitlines()])] except ValueError: str_list += ['win32 : '] try: str_list += ['nvSciSyncObj : ' + hex(self.nvSciSyncObj)] except ValueError: str_list += ['nvSciSyncObj : '] return '\n'.join(str_list) else: return '' @property def fd(self): return self._ptr[0].handle.fd @fd.setter def fd(self, int fd): self._ptr[0].handle.fd = fd @property def win32(self): return self._win32 @win32.setter def win32(self, win32 not None : anon_struct12): string.memcpy(&self._ptr[0].handle.win32, win32.getPtr(), sizeof(self._ptr[0].handle.win32)) @property def nvSciSyncObj(self): return self._ptr[0].handle.nvSciSyncObj @nvSciSyncObj.setter def nvSciSyncObj(self, nvSciSyncObj): _cnvSciSyncObj = utils.HelperInputVoidPtr(nvSciSyncObj) self._ptr[0].handle.nvSciSyncObj = _cnvSciSyncObj.cptr {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st' in found_types}} cdef class CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st: """ External semaphore handle descriptor Attributes ---------- type : CUexternalSemaphoreHandleType Type of the handle handle : anon_union6 flags : unsigned int Flags reserved for the future. Must be zero. reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._handle = anon_union6(_ptr=self._ptr) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['handle :\n' + '\n'.join([' ' + line for line in str(self.handle).splitlines()])] except ValueError: str_list += ['handle : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUexternalSemaphoreHandleType(self._ptr[0].type) @type.setter def type(self, type not None : CUexternalSemaphoreHandleType): self._ptr[0].type = type.value @property def handle(self): return self._handle @handle.setter def handle(self, handle not None : anon_union6): string.memcpy(&self._ptr[0].handle, handle.getPtr(), sizeof(self._ptr[0].handle)) @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st' in found_types}} cdef class anon_struct13: """ Attributes ---------- value : unsigned long long Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params.fence def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['value : ' + str(self.value)] except ValueError: str_list += ['value : '] return '\n'.join(str_list) else: return '' @property def value(self): return self._ptr[0].params.fence.value @value.setter def value(self, unsigned long long value): self._ptr[0].params.fence.value = value {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st' in found_types}} cdef class anon_union7: """ Attributes ---------- fence : Any reserved : unsigned long long Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params.nvSciSync def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fence : ' + hex(self.fence)] except ValueError: str_list += ['fence : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def fence(self): return self._ptr[0].params.nvSciSync.fence @fence.setter def fence(self, fence): _cfence = utils.HelperInputVoidPtr(fence) self._ptr[0].params.nvSciSync.fence = _cfence.cptr @property def reserved(self): return self._ptr[0].params.nvSciSync.reserved @reserved.setter def reserved(self, unsigned long long reserved): self._ptr[0].params.nvSciSync.reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st' in found_types}} cdef class anon_struct14: """ Attributes ---------- key : unsigned long long Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params.keyedMutex def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['key : ' + str(self.key)] except ValueError: str_list += ['key : '] return '\n'.join(str_list) else: return '' @property def key(self): return self._ptr[0].params.keyedMutex.key @key.setter def key(self, unsigned long long key): self._ptr[0].params.keyedMutex.key = key {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st' in found_types}} cdef class anon_struct15: """ Attributes ---------- fence : anon_struct13 nvSciSync : anon_union7 keyedMutex : anon_struct14 reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._fence = anon_struct13(_ptr=self._ptr) self._nvSciSync = anon_union7(_ptr=self._ptr) self._keyedMutex = anon_struct14(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fence :\n' + '\n'.join([' ' + line for line in str(self.fence).splitlines()])] except ValueError: str_list += ['fence : '] try: str_list += ['nvSciSync :\n' + '\n'.join([' ' + line for line in str(self.nvSciSync).splitlines()])] except ValueError: str_list += ['nvSciSync : '] try: str_list += ['keyedMutex :\n' + '\n'.join([' ' + line for line in str(self.keyedMutex).splitlines()])] except ValueError: str_list += ['keyedMutex : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def fence(self): return self._fence @fence.setter def fence(self, fence not None : anon_struct13): string.memcpy(&self._ptr[0].params.fence, fence.getPtr(), sizeof(self._ptr[0].params.fence)) @property def nvSciSync(self): return self._nvSciSync @nvSciSync.setter def nvSciSync(self, nvSciSync not None : anon_union7): string.memcpy(&self._ptr[0].params.nvSciSync, nvSciSync.getPtr(), sizeof(self._ptr[0].params.nvSciSync)) @property def keyedMutex(self): return self._keyedMutex @keyedMutex.setter def keyedMutex(self, keyedMutex not None : anon_struct14): string.memcpy(&self._ptr[0].params.keyedMutex, keyedMutex.getPtr(), sizeof(self._ptr[0].params.keyedMutex)) @property def reserved(self): return self._ptr[0].params.reserved @reserved.setter def reserved(self, reserved): self._ptr[0].params.reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st' in found_types}} cdef class CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st: """ External semaphore signal parameters Attributes ---------- params : anon_struct15 flags : unsigned int Only when ::CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS is used to signal a CUexternalSemaphore of type CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC, the valid flag is CUDA_EXTERNAL_SEMAPHORE_SIGNAL_SKIP_NVSCIBUF_MEMSYNC which indicates that while signaling the CUexternalSemaphore, no memory synchronization operations should be performed for any external memory object imported as CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF. For all other types of CUexternalSemaphore, flags must be zero. reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._params = anon_struct15(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['params :\n' + '\n'.join([' ' + line for line in str(self.params).splitlines()])] except ValueError: str_list += ['params : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def params(self): return self._params @params.setter def params(self, params not None : anon_struct15): string.memcpy(&self._ptr[0].params, params.getPtr(), sizeof(self._ptr[0].params)) @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st' in found_types}} cdef class anon_struct16: """ Attributes ---------- value : unsigned long long Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params.fence def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['value : ' + str(self.value)] except ValueError: str_list += ['value : '] return '\n'.join(str_list) else: return '' @property def value(self): return self._ptr[0].params.fence.value @value.setter def value(self, unsigned long long value): self._ptr[0].params.fence.value = value {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st' in found_types}} cdef class anon_union8: """ Attributes ---------- fence : Any reserved : unsigned long long Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params.nvSciSync def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fence : ' + hex(self.fence)] except ValueError: str_list += ['fence : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def fence(self): return self._ptr[0].params.nvSciSync.fence @fence.setter def fence(self, fence): _cfence = utils.HelperInputVoidPtr(fence) self._ptr[0].params.nvSciSync.fence = _cfence.cptr @property def reserved(self): return self._ptr[0].params.nvSciSync.reserved @reserved.setter def reserved(self, unsigned long long reserved): self._ptr[0].params.nvSciSync.reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st' in found_types}} cdef class anon_struct17: """ Attributes ---------- key : unsigned long long timeoutMs : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params.keyedMutex def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['key : ' + str(self.key)] except ValueError: str_list += ['key : '] try: str_list += ['timeoutMs : ' + str(self.timeoutMs)] except ValueError: str_list += ['timeoutMs : '] return '\n'.join(str_list) else: return '' @property def key(self): return self._ptr[0].params.keyedMutex.key @key.setter def key(self, unsigned long long key): self._ptr[0].params.keyedMutex.key = key @property def timeoutMs(self): return self._ptr[0].params.keyedMutex.timeoutMs @timeoutMs.setter def timeoutMs(self, unsigned int timeoutMs): self._ptr[0].params.keyedMutex.timeoutMs = timeoutMs {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st' in found_types}} cdef class anon_struct18: """ Attributes ---------- fence : anon_struct16 nvSciSync : anon_union8 keyedMutex : anon_struct17 reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._fence = anon_struct16(_ptr=self._ptr) self._nvSciSync = anon_union8(_ptr=self._ptr) self._keyedMutex = anon_struct17(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].params def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['fence :\n' + '\n'.join([' ' + line for line in str(self.fence).splitlines()])] except ValueError: str_list += ['fence : '] try: str_list += ['nvSciSync :\n' + '\n'.join([' ' + line for line in str(self.nvSciSync).splitlines()])] except ValueError: str_list += ['nvSciSync : '] try: str_list += ['keyedMutex :\n' + '\n'.join([' ' + line for line in str(self.keyedMutex).splitlines()])] except ValueError: str_list += ['keyedMutex : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def fence(self): return self._fence @fence.setter def fence(self, fence not None : anon_struct16): string.memcpy(&self._ptr[0].params.fence, fence.getPtr(), sizeof(self._ptr[0].params.fence)) @property def nvSciSync(self): return self._nvSciSync @nvSciSync.setter def nvSciSync(self, nvSciSync not None : anon_union8): string.memcpy(&self._ptr[0].params.nvSciSync, nvSciSync.getPtr(), sizeof(self._ptr[0].params.nvSciSync)) @property def keyedMutex(self): return self._keyedMutex @keyedMutex.setter def keyedMutex(self, keyedMutex not None : anon_struct17): string.memcpy(&self._ptr[0].params.keyedMutex, keyedMutex.getPtr(), sizeof(self._ptr[0].params.keyedMutex)) @property def reserved(self): return self._ptr[0].params.reserved @reserved.setter def reserved(self, reserved): self._ptr[0].params.reserved = reserved {{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st' in found_types}} cdef class CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st: """ External semaphore wait parameters Attributes ---------- params : anon_struct18 flags : unsigned int Only when ::CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS is used to wait on a CUexternalSemaphore of type CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC, the valid flag is CUDA_EXTERNAL_SEMAPHORE_WAIT_SKIP_NVSCIBUF_MEMSYNC which indicates that while waiting for the CUexternalSemaphore, no memory synchronization operations should be performed for any external memory object imported as CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF. For all other types of CUexternalSemaphore, flags must be zero. reserved : List[unsigned int] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._params = anon_struct18(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['params :\n' + '\n'.join([' ' + line for line in str(self.params).splitlines()])] except ValueError: str_list += ['params : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def params(self): return self._params @params.setter def params(self, params not None : anon_struct18): string.memcpy(&self._ptr[0].params, params.getPtr(), sizeof(self._ptr[0].params)) @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_st' in found_types}} cdef class CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_st: """ Semaphore signal node parameters Attributes ---------- extSemArray : CUexternalSemaphore Array of external semaphore handles. paramsArray : CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS Array of external semaphore signal parameters. numExtSems : unsigned int Number of handles and parameters supplied in extSemArray and paramsArray. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): if self._extSemArray is not NULL: free(self._extSemArray) if self._paramsArray is not NULL: free(self._paramsArray) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['extSemArray : ' + str(self.extSemArray)] except ValueError: str_list += ['extSemArray : '] try: str_list += ['paramsArray : ' + str(self.paramsArray)] except ValueError: str_list += ['paramsArray : '] try: str_list += ['numExtSems : ' + str(self.numExtSems)] except ValueError: str_list += ['numExtSems : '] return '\n'.join(str_list) else: return '' @property def extSemArray(self): arrs = [self._ptr[0].extSemArray + x*sizeof(ccuda.CUexternalSemaphore) for x in range(self._extSemArray_length)] return [CUexternalSemaphore(_ptr=arr) for arr in arrs] @extSemArray.setter def extSemArray(self, val): if len(val) == 0: free(self._extSemArray) self._extSemArray_length = 0 self._ptr[0].extSemArray = NULL else: if self._extSemArray_length != len(val): free(self._extSemArray) self._extSemArray = calloc(len(val), sizeof(ccuda.CUexternalSemaphore)) if self._extSemArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUexternalSemaphore))) self._extSemArray_length = len(val) self._ptr[0].extSemArray = self._extSemArray for idx in range(len(val)): self._extSemArray[idx] = (val[idx])._ptr[0] @property def paramsArray(self): arrs = [self._ptr[0].paramsArray + x*sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS) for x in range(self._paramsArray_length)] return [CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS(_ptr=arr) for arr in arrs] @paramsArray.setter def paramsArray(self, val): if len(val) == 0: free(self._paramsArray) self._paramsArray_length = 0 self._ptr[0].paramsArray = NULL else: if self._paramsArray_length != len(val): free(self._paramsArray) self._paramsArray = calloc(len(val), sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS)) if self._paramsArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS))) self._paramsArray_length = len(val) self._ptr[0].paramsArray = self._paramsArray for idx in range(len(val)): string.memcpy(&self._paramsArray[idx], (val[idx])._ptr, sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS)) @property def numExtSems(self): return self._ptr[0].numExtSems @numExtSems.setter def numExtSems(self, unsigned int numExtSems): self._ptr[0].numExtSems = numExtSems {{endif}} {{if 'struct CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2_st' in found_types}} cdef class CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2_st: """ Semaphore signal node parameters Attributes ---------- extSemArray : CUexternalSemaphore Array of external semaphore handles. paramsArray : CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS Array of external semaphore signal parameters. numExtSems : unsigned int Number of handles and parameters supplied in extSemArray and paramsArray. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): if self._extSemArray is not NULL: free(self._extSemArray) if self._paramsArray is not NULL: free(self._paramsArray) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['extSemArray : ' + str(self.extSemArray)] except ValueError: str_list += ['extSemArray : '] try: str_list += ['paramsArray : ' + str(self.paramsArray)] except ValueError: str_list += ['paramsArray : '] try: str_list += ['numExtSems : ' + str(self.numExtSems)] except ValueError: str_list += ['numExtSems : '] return '\n'.join(str_list) else: return '' @property def extSemArray(self): arrs = [self._ptr[0].extSemArray + x*sizeof(ccuda.CUexternalSemaphore) for x in range(self._extSemArray_length)] return [CUexternalSemaphore(_ptr=arr) for arr in arrs] @extSemArray.setter def extSemArray(self, val): if len(val) == 0: free(self._extSemArray) self._extSemArray_length = 0 self._ptr[0].extSemArray = NULL else: if self._extSemArray_length != len(val): free(self._extSemArray) self._extSemArray = calloc(len(val), sizeof(ccuda.CUexternalSemaphore)) if self._extSemArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUexternalSemaphore))) self._extSemArray_length = len(val) self._ptr[0].extSemArray = self._extSemArray for idx in range(len(val)): self._extSemArray[idx] = (val[idx])._ptr[0] @property def paramsArray(self): arrs = [self._ptr[0].paramsArray + x*sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS) for x in range(self._paramsArray_length)] return [CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS(_ptr=arr) for arr in arrs] @paramsArray.setter def paramsArray(self, val): if len(val) == 0: free(self._paramsArray) self._paramsArray_length = 0 self._ptr[0].paramsArray = NULL else: if self._paramsArray_length != len(val): free(self._paramsArray) self._paramsArray = calloc(len(val), sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS)) if self._paramsArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS))) self._paramsArray_length = len(val) self._ptr[0].paramsArray = self._paramsArray for idx in range(len(val)): string.memcpy(&self._paramsArray[idx], (val[idx])._ptr, sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS)) @property def numExtSems(self): return self._ptr[0].numExtSems @numExtSems.setter def numExtSems(self, unsigned int numExtSems): self._ptr[0].numExtSems = numExtSems {{endif}} {{if 'struct CUDA_EXT_SEM_WAIT_NODE_PARAMS_st' in found_types}} cdef class CUDA_EXT_SEM_WAIT_NODE_PARAMS_st: """ Semaphore wait node parameters Attributes ---------- extSemArray : CUexternalSemaphore Array of external semaphore handles. paramsArray : CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS Array of external semaphore wait parameters. numExtSems : unsigned int Number of handles and parameters supplied in extSemArray and paramsArray. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): if self._extSemArray is not NULL: free(self._extSemArray) if self._paramsArray is not NULL: free(self._paramsArray) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['extSemArray : ' + str(self.extSemArray)] except ValueError: str_list += ['extSemArray : '] try: str_list += ['paramsArray : ' + str(self.paramsArray)] except ValueError: str_list += ['paramsArray : '] try: str_list += ['numExtSems : ' + str(self.numExtSems)] except ValueError: str_list += ['numExtSems : '] return '\n'.join(str_list) else: return '' @property def extSemArray(self): arrs = [self._ptr[0].extSemArray + x*sizeof(ccuda.CUexternalSemaphore) for x in range(self._extSemArray_length)] return [CUexternalSemaphore(_ptr=arr) for arr in arrs] @extSemArray.setter def extSemArray(self, val): if len(val) == 0: free(self._extSemArray) self._extSemArray_length = 0 self._ptr[0].extSemArray = NULL else: if self._extSemArray_length != len(val): free(self._extSemArray) self._extSemArray = calloc(len(val), sizeof(ccuda.CUexternalSemaphore)) if self._extSemArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUexternalSemaphore))) self._extSemArray_length = len(val) self._ptr[0].extSemArray = self._extSemArray for idx in range(len(val)): self._extSemArray[idx] = (val[idx])._ptr[0] @property def paramsArray(self): arrs = [self._ptr[0].paramsArray + x*sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS) for x in range(self._paramsArray_length)] return [CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS(_ptr=arr) for arr in arrs] @paramsArray.setter def paramsArray(self, val): if len(val) == 0: free(self._paramsArray) self._paramsArray_length = 0 self._ptr[0].paramsArray = NULL else: if self._paramsArray_length != len(val): free(self._paramsArray) self._paramsArray = calloc(len(val), sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS)) if self._paramsArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS))) self._paramsArray_length = len(val) self._ptr[0].paramsArray = self._paramsArray for idx in range(len(val)): string.memcpy(&self._paramsArray[idx], (val[idx])._ptr, sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS)) @property def numExtSems(self): return self._ptr[0].numExtSems @numExtSems.setter def numExtSems(self, unsigned int numExtSems): self._ptr[0].numExtSems = numExtSems {{endif}} {{if 'struct CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2_st' in found_types}} cdef class CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2_st: """ Semaphore wait node parameters Attributes ---------- extSemArray : CUexternalSemaphore Array of external semaphore handles. paramsArray : CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS Array of external semaphore wait parameters. numExtSems : unsigned int Number of handles and parameters supplied in extSemArray and paramsArray. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): if self._extSemArray is not NULL: free(self._extSemArray) if self._paramsArray is not NULL: free(self._paramsArray) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['extSemArray : ' + str(self.extSemArray)] except ValueError: str_list += ['extSemArray : '] try: str_list += ['paramsArray : ' + str(self.paramsArray)] except ValueError: str_list += ['paramsArray : '] try: str_list += ['numExtSems : ' + str(self.numExtSems)] except ValueError: str_list += ['numExtSems : '] return '\n'.join(str_list) else: return '' @property def extSemArray(self): arrs = [self._ptr[0].extSemArray + x*sizeof(ccuda.CUexternalSemaphore) for x in range(self._extSemArray_length)] return [CUexternalSemaphore(_ptr=arr) for arr in arrs] @extSemArray.setter def extSemArray(self, val): if len(val) == 0: free(self._extSemArray) self._extSemArray_length = 0 self._ptr[0].extSemArray = NULL else: if self._extSemArray_length != len(val): free(self._extSemArray) self._extSemArray = calloc(len(val), sizeof(ccuda.CUexternalSemaphore)) if self._extSemArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUexternalSemaphore))) self._extSemArray_length = len(val) self._ptr[0].extSemArray = self._extSemArray for idx in range(len(val)): self._extSemArray[idx] = (val[idx])._ptr[0] @property def paramsArray(self): arrs = [self._ptr[0].paramsArray + x*sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS) for x in range(self._paramsArray_length)] return [CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS(_ptr=arr) for arr in arrs] @paramsArray.setter def paramsArray(self, val): if len(val) == 0: free(self._paramsArray) self._paramsArray_length = 0 self._ptr[0].paramsArray = NULL else: if self._paramsArray_length != len(val): free(self._paramsArray) self._paramsArray = calloc(len(val), sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS)) if self._paramsArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS))) self._paramsArray_length = len(val) self._ptr[0].paramsArray = self._paramsArray for idx in range(len(val)): string.memcpy(&self._paramsArray[idx], (val[idx])._ptr, sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS)) @property def numExtSems(self): return self._ptr[0].numExtSems @numExtSems.setter def numExtSems(self, unsigned int numExtSems): self._ptr[0].numExtSems = numExtSems {{endif}} {{if 'struct CUarrayMapInfo_st' in found_types}} cdef class anon_union9: """ Attributes ---------- mipmap : CUmipmappedArray array : CUarray Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._mipmap = CUmipmappedArray(_ptr=&self._ptr[0].resource.mipmap) self._array = CUarray(_ptr=&self._ptr[0].resource.array) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].resource def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['mipmap : ' + str(self.mipmap)] except ValueError: str_list += ['mipmap : '] try: str_list += ['array : ' + str(self.array)] except ValueError: str_list += ['array : '] return '\n'.join(str_list) else: return '' @property def mipmap(self): return self._mipmap @mipmap.setter def mipmap(self, mipmap): cdef ccuda.CUmipmappedArray cmipmap if mipmap is None: cmipmap = 0 elif isinstance(mipmap, (CUmipmappedArray,)): pmipmap = int(mipmap) cmipmap = pmipmap else: pmipmap = int(CUmipmappedArray(mipmap)) cmipmap = pmipmap self._mipmap._ptr[0] = cmipmap @property def array(self): return self._array @array.setter def array(self, array): cdef ccuda.CUarray carray if array is None: carray = 0 elif isinstance(array, (CUarray,)): parray = int(array) carray = parray else: parray = int(CUarray(array)) carray = parray self._array._ptr[0] = carray {{endif}} {{if 'struct CUarrayMapInfo_st' in found_types}} cdef class anon_struct19: """ Attributes ---------- level : unsigned int layer : unsigned int offsetX : unsigned int offsetY : unsigned int offsetZ : unsigned int extentWidth : unsigned int extentHeight : unsigned int extentDepth : unsigned int Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].subresource.sparseLevel def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['level : ' + str(self.level)] except ValueError: str_list += ['level : '] try: str_list += ['layer : ' + str(self.layer)] except ValueError: str_list += ['layer : '] try: str_list += ['offsetX : ' + str(self.offsetX)] except ValueError: str_list += ['offsetX : '] try: str_list += ['offsetY : ' + str(self.offsetY)] except ValueError: str_list += ['offsetY : '] try: str_list += ['offsetZ : ' + str(self.offsetZ)] except ValueError: str_list += ['offsetZ : '] try: str_list += ['extentWidth : ' + str(self.extentWidth)] except ValueError: str_list += ['extentWidth : '] try: str_list += ['extentHeight : ' + str(self.extentHeight)] except ValueError: str_list += ['extentHeight : '] try: str_list += ['extentDepth : ' + str(self.extentDepth)] except ValueError: str_list += ['extentDepth : '] return '\n'.join(str_list) else: return '' @property def level(self): return self._ptr[0].subresource.sparseLevel.level @level.setter def level(self, unsigned int level): self._ptr[0].subresource.sparseLevel.level = level @property def layer(self): return self._ptr[0].subresource.sparseLevel.layer @layer.setter def layer(self, unsigned int layer): self._ptr[0].subresource.sparseLevel.layer = layer @property def offsetX(self): return self._ptr[0].subresource.sparseLevel.offsetX @offsetX.setter def offsetX(self, unsigned int offsetX): self._ptr[0].subresource.sparseLevel.offsetX = offsetX @property def offsetY(self): return self._ptr[0].subresource.sparseLevel.offsetY @offsetY.setter def offsetY(self, unsigned int offsetY): self._ptr[0].subresource.sparseLevel.offsetY = offsetY @property def offsetZ(self): return self._ptr[0].subresource.sparseLevel.offsetZ @offsetZ.setter def offsetZ(self, unsigned int offsetZ): self._ptr[0].subresource.sparseLevel.offsetZ = offsetZ @property def extentWidth(self): return self._ptr[0].subresource.sparseLevel.extentWidth @extentWidth.setter def extentWidth(self, unsigned int extentWidth): self._ptr[0].subresource.sparseLevel.extentWidth = extentWidth @property def extentHeight(self): return self._ptr[0].subresource.sparseLevel.extentHeight @extentHeight.setter def extentHeight(self, unsigned int extentHeight): self._ptr[0].subresource.sparseLevel.extentHeight = extentHeight @property def extentDepth(self): return self._ptr[0].subresource.sparseLevel.extentDepth @extentDepth.setter def extentDepth(self, unsigned int extentDepth): self._ptr[0].subresource.sparseLevel.extentDepth = extentDepth {{endif}} {{if 'struct CUarrayMapInfo_st' in found_types}} cdef class anon_struct20: """ Attributes ---------- layer : unsigned int offset : unsigned long long size : unsigned long long Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].subresource.miptail def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['layer : ' + str(self.layer)] except ValueError: str_list += ['layer : '] try: str_list += ['offset : ' + str(self.offset)] except ValueError: str_list += ['offset : '] try: str_list += ['size : ' + str(self.size)] except ValueError: str_list += ['size : '] return '\n'.join(str_list) else: return '' @property def layer(self): return self._ptr[0].subresource.miptail.layer @layer.setter def layer(self, unsigned int layer): self._ptr[0].subresource.miptail.layer = layer @property def offset(self): return self._ptr[0].subresource.miptail.offset @offset.setter def offset(self, unsigned long long offset): self._ptr[0].subresource.miptail.offset = offset @property def size(self): return self._ptr[0].subresource.miptail.size @size.setter def size(self, unsigned long long size): self._ptr[0].subresource.miptail.size = size {{endif}} {{if 'struct CUarrayMapInfo_st' in found_types}} cdef class anon_union10: """ Attributes ---------- sparseLevel : anon_struct19 miptail : anon_struct20 Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._sparseLevel = anon_struct19(_ptr=self._ptr) self._miptail = anon_struct20(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].subresource def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['sparseLevel :\n' + '\n'.join([' ' + line for line in str(self.sparseLevel).splitlines()])] except ValueError: str_list += ['sparseLevel : '] try: str_list += ['miptail :\n' + '\n'.join([' ' + line for line in str(self.miptail).splitlines()])] except ValueError: str_list += ['miptail : '] return '\n'.join(str_list) else: return '' @property def sparseLevel(self): return self._sparseLevel @sparseLevel.setter def sparseLevel(self, sparseLevel not None : anon_struct19): string.memcpy(&self._ptr[0].subresource.sparseLevel, sparseLevel.getPtr(), sizeof(self._ptr[0].subresource.sparseLevel)) @property def miptail(self): return self._miptail @miptail.setter def miptail(self, miptail not None : anon_struct20): string.memcpy(&self._ptr[0].subresource.miptail, miptail.getPtr(), sizeof(self._ptr[0].subresource.miptail)) {{endif}} {{if 'struct CUarrayMapInfo_st' in found_types}} cdef class anon_union11: """ Attributes ---------- memHandle : CUmemGenericAllocationHandle Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): self._memHandle = CUmemGenericAllocationHandle(_ptr=&self._ptr[0].memHandle.memHandle) def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].memHandle def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['memHandle : ' + str(self.memHandle)] except ValueError: str_list += ['memHandle : '] return '\n'.join(str_list) else: return '' @property def memHandle(self): return self._memHandle @memHandle.setter def memHandle(self, memHandle): cdef ccuda.CUmemGenericAllocationHandle cmemHandle if memHandle is None: cmemHandle = 0 elif isinstance(memHandle, (CUmemGenericAllocationHandle)): pmemHandle = int(memHandle) cmemHandle = pmemHandle else: pmemHandle = int(CUmemGenericAllocationHandle(memHandle)) cmemHandle = pmemHandle self._memHandle._ptr[0] = cmemHandle {{endif}} {{if 'struct CUarrayMapInfo_st' in found_types}} cdef class CUarrayMapInfo_st: """ Specifies the CUDA array or CUDA mipmapped array memory mapping information Attributes ---------- resourceType : CUresourcetype Resource type resource : anon_union9 subresourceType : CUarraySparseSubresourceType Sparse subresource type subresource : anon_union10 memOperationType : CUmemOperationType Memory operation type memHandleType : CUmemHandleType Memory handle type memHandle : anon_union11 offset : unsigned long long Offset within mip tail Offset within the memory deviceBitMask : unsigned int Device ordinal bit mask flags : unsigned int flags for future use, must be zero now. reserved : List[unsigned int] Reserved for future use, must be zero now. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUarrayMapInfo_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._resource = anon_union9(_ptr=self._ptr) self._subresource = anon_union10(_ptr=self._ptr) self._memHandle = anon_union11(_ptr=self._ptr) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['resourceType : ' + str(self.resourceType)] except ValueError: str_list += ['resourceType : '] try: str_list += ['resource :\n' + '\n'.join([' ' + line for line in str(self.resource).splitlines()])] except ValueError: str_list += ['resource : '] try: str_list += ['subresourceType : ' + str(self.subresourceType)] except ValueError: str_list += ['subresourceType : '] try: str_list += ['subresource :\n' + '\n'.join([' ' + line for line in str(self.subresource).splitlines()])] except ValueError: str_list += ['subresource : '] try: str_list += ['memOperationType : ' + str(self.memOperationType)] except ValueError: str_list += ['memOperationType : '] try: str_list += ['memHandleType : ' + str(self.memHandleType)] except ValueError: str_list += ['memHandleType : '] try: str_list += ['memHandle :\n' + '\n'.join([' ' + line for line in str(self.memHandle).splitlines()])] except ValueError: str_list += ['memHandle : '] try: str_list += ['offset : ' + str(self.offset)] except ValueError: str_list += ['offset : '] try: str_list += ['deviceBitMask : ' + str(self.deviceBitMask)] except ValueError: str_list += ['deviceBitMask : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def resourceType(self): return CUresourcetype(self._ptr[0].resourceType) @resourceType.setter def resourceType(self, resourceType not None : CUresourcetype): self._ptr[0].resourceType = resourceType.value @property def resource(self): return self._resource @resource.setter def resource(self, resource not None : anon_union9): string.memcpy(&self._ptr[0].resource, resource.getPtr(), sizeof(self._ptr[0].resource)) @property def subresourceType(self): return CUarraySparseSubresourceType(self._ptr[0].subresourceType) @subresourceType.setter def subresourceType(self, subresourceType not None : CUarraySparseSubresourceType): self._ptr[0].subresourceType = subresourceType.value @property def subresource(self): return self._subresource @subresource.setter def subresource(self, subresource not None : anon_union10): string.memcpy(&self._ptr[0].subresource, subresource.getPtr(), sizeof(self._ptr[0].subresource)) @property def memOperationType(self): return CUmemOperationType(self._ptr[0].memOperationType) @memOperationType.setter def memOperationType(self, memOperationType not None : CUmemOperationType): self._ptr[0].memOperationType = memOperationType.value @property def memHandleType(self): return CUmemHandleType(self._ptr[0].memHandleType) @memHandleType.setter def memHandleType(self, memHandleType not None : CUmemHandleType): self._ptr[0].memHandleType = memHandleType.value @property def memHandle(self): return self._memHandle @memHandle.setter def memHandle(self, memHandle not None : anon_union11): string.memcpy(&self._ptr[0].memHandle, memHandle.getPtr(), sizeof(self._ptr[0].memHandle)) @property def offset(self): return self._ptr[0].offset @offset.setter def offset(self, unsigned long long offset): self._ptr[0].offset = offset @property def deviceBitMask(self): return self._ptr[0].deviceBitMask @deviceBitMask.setter def deviceBitMask(self, unsigned int deviceBitMask): self._ptr[0].deviceBitMask = deviceBitMask @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned int flags): self._ptr[0].flags = flags @property def reserved(self): return self._ptr[0].reserved @reserved.setter def reserved(self, reserved): self._ptr[0].reserved = reserved {{endif}} {{if 'struct CUmemLocation_st' in found_types}} cdef class CUmemLocation_st: """ Specifies a memory location. Attributes ---------- type : CUmemLocationType Specifies the location type, which modifies the meaning of id. id : int identifier for a given this location's CUmemLocationType. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['id : ' + str(self.id)] except ValueError: str_list += ['id : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUmemLocationType(self._ptr[0].type) @type.setter def type(self, type not None : CUmemLocationType): self._ptr[0].type = type.value @property def id(self): return self._ptr[0].id @id.setter def id(self, int id): self._ptr[0].id = id {{endif}} {{if 'struct CUmemAllocationProp_st' in found_types}} cdef class anon_struct21: """ Attributes ---------- compressionType : bytes gpuDirectRDMACapable : bytes usage : unsigned short reserved : bytes Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].allocFlags def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['compressionType : ' + str(self.compressionType)] except ValueError: str_list += ['compressionType : '] try: str_list += ['gpuDirectRDMACapable : ' + str(self.gpuDirectRDMACapable)] except ValueError: str_list += ['gpuDirectRDMACapable : '] try: str_list += ['usage : ' + str(self.usage)] except ValueError: str_list += ['usage : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def compressionType(self): return self._ptr[0].allocFlags.compressionType @compressionType.setter def compressionType(self, unsigned char compressionType): self._ptr[0].allocFlags.compressionType = compressionType @property def gpuDirectRDMACapable(self): return self._ptr[0].allocFlags.gpuDirectRDMACapable @gpuDirectRDMACapable.setter def gpuDirectRDMACapable(self, unsigned char gpuDirectRDMACapable): self._ptr[0].allocFlags.gpuDirectRDMACapable = gpuDirectRDMACapable @property def usage(self): return self._ptr[0].allocFlags.usage @usage.setter def usage(self, unsigned short usage): self._ptr[0].allocFlags.usage = usage @property def reserved(self): return PyBytes_FromStringAndSize(self._ptr[0].allocFlags.reserved, 4) @reserved.setter def reserved(self, reserved): if len(reserved) != 4: raise ValueError("reserved length must be 4, is " + str(len(reserved))) for i, b in enumerate(reserved): self._ptr[0].allocFlags.reserved[i] = b {{endif}} {{if 'struct CUmemAllocationProp_st' in found_types}} cdef class CUmemAllocationProp_st: """ Specifies the allocation properties for a allocation. Attributes ---------- type : CUmemAllocationType Allocation type requestedHandleTypes : CUmemAllocationHandleType requested CUmemAllocationHandleType location : CUmemLocation Location of allocation win32HandleMetaData : Any Windows-specific POBJECT_ATTRIBUTES required when CU_MEM_HANDLE_TYPE_WIN32 is specified. This object attributes structure includes security attributes that define the scope of which exported allocations may be transferred to other processes. In all other cases, this field is required to be zero. allocFlags : anon_struct21 Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._location = CUmemLocation(_ptr=&self._ptr[0].location) self._allocFlags = anon_struct21(_ptr=self._ptr) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['requestedHandleTypes : ' + str(self.requestedHandleTypes)] except ValueError: str_list += ['requestedHandleTypes : '] try: str_list += ['location :\n' + '\n'.join([' ' + line for line in str(self.location).splitlines()])] except ValueError: str_list += ['location : '] try: str_list += ['win32HandleMetaData : ' + hex(self.win32HandleMetaData)] except ValueError: str_list += ['win32HandleMetaData : '] try: str_list += ['allocFlags :\n' + '\n'.join([' ' + line for line in str(self.allocFlags).splitlines()])] except ValueError: str_list += ['allocFlags : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUmemAllocationType(self._ptr[0].type) @type.setter def type(self, type not None : CUmemAllocationType): self._ptr[0].type = type.value @property def requestedHandleTypes(self): return CUmemAllocationHandleType(self._ptr[0].requestedHandleTypes) @requestedHandleTypes.setter def requestedHandleTypes(self, requestedHandleTypes not None : CUmemAllocationHandleType): self._ptr[0].requestedHandleTypes = requestedHandleTypes.value @property def location(self): return self._location @location.setter def location(self, location not None : CUmemLocation): string.memcpy(&self._ptr[0].location, location.getPtr(), sizeof(self._ptr[0].location)) @property def win32HandleMetaData(self): return self._ptr[0].win32HandleMetaData @win32HandleMetaData.setter def win32HandleMetaData(self, win32HandleMetaData): _cwin32HandleMetaData = utils.HelperInputVoidPtr(win32HandleMetaData) self._ptr[0].win32HandleMetaData = _cwin32HandleMetaData.cptr @property def allocFlags(self): return self._allocFlags @allocFlags.setter def allocFlags(self, allocFlags not None : anon_struct21): string.memcpy(&self._ptr[0].allocFlags, allocFlags.getPtr(), sizeof(self._ptr[0].allocFlags)) {{endif}} {{if 'struct CUmulticastObjectProp_st' in found_types}} cdef class CUmulticastObjectProp_st: """ Specifies the properties for a multicast object. Attributes ---------- numDevices : unsigned int The number of devices in the multicast team that will bind memory to this object size : size_t The maximum amount of memory that can be bound to this multicast object per device handleTypes : unsigned long long Bitmask of exportable handle types (see CUmemAllocationHandleType) for this object flags : unsigned long long Flags for future use, must be zero now Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['numDevices : ' + str(self.numDevices)] except ValueError: str_list += ['numDevices : '] try: str_list += ['size : ' + str(self.size)] except ValueError: str_list += ['size : '] try: str_list += ['handleTypes : ' + str(self.handleTypes)] except ValueError: str_list += ['handleTypes : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def numDevices(self): return self._ptr[0].numDevices @numDevices.setter def numDevices(self, unsigned int numDevices): self._ptr[0].numDevices = numDevices @property def size(self): return self._ptr[0].size @size.setter def size(self, size_t size): self._ptr[0].size = size @property def handleTypes(self): return self._ptr[0].handleTypes @handleTypes.setter def handleTypes(self, unsigned long long handleTypes): self._ptr[0].handleTypes = handleTypes @property def flags(self): return self._ptr[0].flags @flags.setter def flags(self, unsigned long long flags): self._ptr[0].flags = flags {{endif}} {{if 'struct CUmemAccessDesc_st' in found_types}} cdef class CUmemAccessDesc_st: """ Memory access descriptor Attributes ---------- location : CUmemLocation Location on which the request is to change it's accessibility flags : CUmemAccess_flags ::CUmemProt accessibility flags to set on the request Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._location = CUmemLocation(_ptr=&self._ptr[0].location) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['location :\n' + '\n'.join([' ' + line for line in str(self.location).splitlines()])] except ValueError: str_list += ['location : '] try: str_list += ['flags : ' + str(self.flags)] except ValueError: str_list += ['flags : '] return '\n'.join(str_list) else: return '' @property def location(self): return self._location @location.setter def location(self, location not None : CUmemLocation): string.memcpy(&self._ptr[0].location, location.getPtr(), sizeof(self._ptr[0].location)) @property def flags(self): return CUmemAccess_flags(self._ptr[0].flags) @flags.setter def flags(self, flags not None : CUmemAccess_flags): self._ptr[0].flags = flags.value {{endif}} {{if 'struct CUgraphExecUpdateResultInfo_st' in found_types}} cdef class CUgraphExecUpdateResultInfo_st: """ Result information returned by cuGraphExecUpdate Attributes ---------- result : CUgraphExecUpdateResult Gives more specific detail when a cuda graph update fails. errorNode : CUgraphNode The "to node" of the error edge when the topologies do not match. The error node when the error is associated with a specific node. NULL when the error is generic. errorFromNode : CUgraphNode The from node of error edge when the topologies do not match. Otherwise NULL. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._errorNode = CUgraphNode(_ptr=&self._ptr[0].errorNode) self._errorFromNode = CUgraphNode(_ptr=&self._ptr[0].errorFromNode) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['result : ' + str(self.result)] except ValueError: str_list += ['result : '] try: str_list += ['errorNode : ' + str(self.errorNode)] except ValueError: str_list += ['errorNode : '] try: str_list += ['errorFromNode : ' + str(self.errorFromNode)] except ValueError: str_list += ['errorFromNode : '] return '\n'.join(str_list) else: return '' @property def result(self): return CUgraphExecUpdateResult(self._ptr[0].result) @result.setter def result(self, result not None : CUgraphExecUpdateResult): self._ptr[0].result = result.value @property def errorNode(self): return self._errorNode @errorNode.setter def errorNode(self, errorNode): cdef ccuda.CUgraphNode cerrorNode if errorNode is None: cerrorNode = 0 elif isinstance(errorNode, (CUgraphNode,)): perrorNode = int(errorNode) cerrorNode = perrorNode else: perrorNode = int(CUgraphNode(errorNode)) cerrorNode = perrorNode self._errorNode._ptr[0] = cerrorNode @property def errorFromNode(self): return self._errorFromNode @errorFromNode.setter def errorFromNode(self, errorFromNode): cdef ccuda.CUgraphNode cerrorFromNode if errorFromNode is None: cerrorFromNode = 0 elif isinstance(errorFromNode, (CUgraphNode,)): perrorFromNode = int(errorFromNode) cerrorFromNode = perrorFromNode else: perrorFromNode = int(CUgraphNode(errorFromNode)) cerrorFromNode = perrorFromNode self._errorFromNode._ptr[0] = cerrorFromNode {{endif}} {{if 'struct CUmemPoolProps_st' in found_types}} cdef class CUmemPoolProps_st: """ Specifies the properties of allocations made from the pool. Attributes ---------- allocType : CUmemAllocationType Allocation type. Currently must be specified as CU_MEM_ALLOCATION_TYPE_PINNED handleTypes : CUmemAllocationHandleType Handle types that will be supported by allocations from the pool. location : CUmemLocation Location where allocations should reside. win32SecurityAttributes : Any Windows-specific LPSECURITYATTRIBUTES required when CU_MEM_HANDLE_TYPE_WIN32 is specified. This security attribute defines the scope of which exported allocations may be transferred to other processes. In all other cases, this field is required to be zero. maxSize : size_t Maximum pool size. When set to 0, defaults to a system dependent value. usage : unsigned short Bitmask indicating intended usage for the pool. reserved : bytes reserved for future use, must be 0 Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._location = CUmemLocation(_ptr=&self._ptr[0].location) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['allocType : ' + str(self.allocType)] except ValueError: str_list += ['allocType : '] try: str_list += ['handleTypes : ' + str(self.handleTypes)] except ValueError: str_list += ['handleTypes : '] try: str_list += ['location :\n' + '\n'.join([' ' + line for line in str(self.location).splitlines()])] except ValueError: str_list += ['location : '] try: str_list += ['win32SecurityAttributes : ' + hex(self.win32SecurityAttributes)] except ValueError: str_list += ['win32SecurityAttributes : '] try: str_list += ['maxSize : ' + str(self.maxSize)] except ValueError: str_list += ['maxSize : '] try: str_list += ['usage : ' + str(self.usage)] except ValueError: str_list += ['usage : '] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def allocType(self): return CUmemAllocationType(self._ptr[0].allocType) @allocType.setter def allocType(self, allocType not None : CUmemAllocationType): self._ptr[0].allocType = allocType.value @property def handleTypes(self): return CUmemAllocationHandleType(self._ptr[0].handleTypes) @handleTypes.setter def handleTypes(self, handleTypes not None : CUmemAllocationHandleType): self._ptr[0].handleTypes = handleTypes.value @property def location(self): return self._location @location.setter def location(self, location not None : CUmemLocation): string.memcpy(&self._ptr[0].location, location.getPtr(), sizeof(self._ptr[0].location)) @property def win32SecurityAttributes(self): return self._ptr[0].win32SecurityAttributes @win32SecurityAttributes.setter def win32SecurityAttributes(self, win32SecurityAttributes): _cwin32SecurityAttributes = utils.HelperInputVoidPtr(win32SecurityAttributes) self._ptr[0].win32SecurityAttributes = _cwin32SecurityAttributes.cptr @property def maxSize(self): return self._ptr[0].maxSize @maxSize.setter def maxSize(self, size_t maxSize): self._ptr[0].maxSize = maxSize @property def usage(self): return self._ptr[0].usage @usage.setter def usage(self, unsigned short usage): self._ptr[0].usage = usage @property def reserved(self): return PyBytes_FromStringAndSize(self._ptr[0].reserved, 54) @reserved.setter def reserved(self, reserved): if len(reserved) != 54: raise ValueError("reserved length must be 54, is " + str(len(reserved))) for i, b in enumerate(reserved): self._ptr[0].reserved[i] = b {{endif}} {{if 'struct CUmemPoolPtrExportData_st' in found_types}} cdef class CUmemPoolPtrExportData_st: """ Opaque data for exporting a pool allocation Attributes ---------- reserved : bytes Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['reserved : ' + str(self.reserved)] except ValueError: str_list += ['reserved : '] return '\n'.join(str_list) else: return '' @property def reserved(self): return PyBytes_FromStringAndSize(self._ptr[0].reserved, 64) @reserved.setter def reserved(self, reserved): if len(reserved) != 64: raise ValueError("reserved length must be 64, is " + str(len(reserved))) for i, b in enumerate(reserved): self._ptr[0].reserved[i] = b {{endif}} {{if 'struct CUDA_MEM_ALLOC_NODE_PARAMS_v1_st' in found_types}} cdef class CUDA_MEM_ALLOC_NODE_PARAMS_v1_st: """ Memory allocation node parameters Attributes ---------- poolProps : CUmemPoolProps in: location where the allocation should reside (specified in ::location). ::handleTypes must be CU_MEM_HANDLE_TYPE_NONE. IPC is not supported. accessDescs : CUmemAccessDesc in: array of memory access descriptors. Used to describe peer GPU access accessDescCount : size_t in: number of memory access descriptors. Must not exceed the number of GPUs. bytesize : size_t in: size in bytes of the requested allocation dptr : CUdeviceptr out: address of the allocation returned by CUDA Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._poolProps = CUmemPoolProps(_ptr=&self._ptr[0].poolProps) self._dptr = CUdeviceptr(_ptr=&self._ptr[0].dptr) def __dealloc__(self): if self._accessDescs is not NULL: free(self._accessDescs) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['poolProps :\n' + '\n'.join([' ' + line for line in str(self.poolProps).splitlines()])] except ValueError: str_list += ['poolProps : '] try: str_list += ['accessDescs : ' + str(self.accessDescs)] except ValueError: str_list += ['accessDescs : '] try: str_list += ['accessDescCount : ' + str(self.accessDescCount)] except ValueError: str_list += ['accessDescCount : '] try: str_list += ['bytesize : ' + str(self.bytesize)] except ValueError: str_list += ['bytesize : '] try: str_list += ['dptr : ' + str(self.dptr)] except ValueError: str_list += ['dptr : '] return '\n'.join(str_list) else: return '' @property def poolProps(self): return self._poolProps @poolProps.setter def poolProps(self, poolProps not None : CUmemPoolProps): string.memcpy(&self._ptr[0].poolProps, poolProps.getPtr(), sizeof(self._ptr[0].poolProps)) @property def accessDescs(self): arrs = [self._ptr[0].accessDescs + x*sizeof(ccuda.CUmemAccessDesc) for x in range(self._accessDescs_length)] return [CUmemAccessDesc(_ptr=arr) for arr in arrs] @accessDescs.setter def accessDescs(self, val): if len(val) == 0: free(self._accessDescs) self._accessDescs_length = 0 self._ptr[0].accessDescs = NULL else: if self._accessDescs_length != len(val): free(self._accessDescs) self._accessDescs = calloc(len(val), sizeof(ccuda.CUmemAccessDesc)) if self._accessDescs is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUmemAccessDesc))) self._accessDescs_length = len(val) self._ptr[0].accessDescs = self._accessDescs for idx in range(len(val)): string.memcpy(&self._accessDescs[idx], (val[idx])._ptr, sizeof(ccuda.CUmemAccessDesc)) @property def accessDescCount(self): return self._ptr[0].accessDescCount @accessDescCount.setter def accessDescCount(self, size_t accessDescCount): self._ptr[0].accessDescCount = accessDescCount @property def bytesize(self): return self._ptr[0].bytesize @bytesize.setter def bytesize(self, size_t bytesize): self._ptr[0].bytesize = bytesize @property def dptr(self): return self._dptr @dptr.setter def dptr(self, dptr): cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr self._dptr._ptr[0] = cdptr {{endif}} {{if 'struct CUDA_MEM_ALLOC_NODE_PARAMS_v2_st' in found_types}} cdef class CUDA_MEM_ALLOC_NODE_PARAMS_v2_st: """ Memory allocation node parameters Attributes ---------- poolProps : CUmemPoolProps in: location where the allocation should reside (specified in ::location). ::handleTypes must be CU_MEM_HANDLE_TYPE_NONE. IPC is not supported. accessDescs : CUmemAccessDesc in: array of memory access descriptors. Used to describe peer GPU access accessDescCount : size_t in: number of memory access descriptors. Must not exceed the number of GPUs. bytesize : size_t in: size in bytes of the requested allocation dptr : CUdeviceptr out: address of the allocation returned by CUDA Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._poolProps = CUmemPoolProps(_ptr=&self._ptr[0].poolProps) self._dptr = CUdeviceptr(_ptr=&self._ptr[0].dptr) def __dealloc__(self): if self._accessDescs is not NULL: free(self._accessDescs) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['poolProps :\n' + '\n'.join([' ' + line for line in str(self.poolProps).splitlines()])] except ValueError: str_list += ['poolProps : '] try: str_list += ['accessDescs : ' + str(self.accessDescs)] except ValueError: str_list += ['accessDescs : '] try: str_list += ['accessDescCount : ' + str(self.accessDescCount)] except ValueError: str_list += ['accessDescCount : '] try: str_list += ['bytesize : ' + str(self.bytesize)] except ValueError: str_list += ['bytesize : '] try: str_list += ['dptr : ' + str(self.dptr)] except ValueError: str_list += ['dptr : '] return '\n'.join(str_list) else: return '' @property def poolProps(self): return self._poolProps @poolProps.setter def poolProps(self, poolProps not None : CUmemPoolProps): string.memcpy(&self._ptr[0].poolProps, poolProps.getPtr(), sizeof(self._ptr[0].poolProps)) @property def accessDescs(self): arrs = [self._ptr[0].accessDescs + x*sizeof(ccuda.CUmemAccessDesc) for x in range(self._accessDescs_length)] return [CUmemAccessDesc(_ptr=arr) for arr in arrs] @accessDescs.setter def accessDescs(self, val): if len(val) == 0: free(self._accessDescs) self._accessDescs_length = 0 self._ptr[0].accessDescs = NULL else: if self._accessDescs_length != len(val): free(self._accessDescs) self._accessDescs = calloc(len(val), sizeof(ccuda.CUmemAccessDesc)) if self._accessDescs is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(val)) + 'x' + str(sizeof(ccuda.CUmemAccessDesc))) self._accessDescs_length = len(val) self._ptr[0].accessDescs = self._accessDescs for idx in range(len(val)): string.memcpy(&self._accessDescs[idx], (val[idx])._ptr, sizeof(ccuda.CUmemAccessDesc)) @property def accessDescCount(self): return self._ptr[0].accessDescCount @accessDescCount.setter def accessDescCount(self, size_t accessDescCount): self._ptr[0].accessDescCount = accessDescCount @property def bytesize(self): return self._ptr[0].bytesize @bytesize.setter def bytesize(self, size_t bytesize): self._ptr[0].bytesize = bytesize @property def dptr(self): return self._dptr @dptr.setter def dptr(self, dptr): cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr self._dptr._ptr[0] = cdptr {{endif}} {{if 'struct CUDA_MEM_FREE_NODE_PARAMS_st' in found_types}} cdef class CUDA_MEM_FREE_NODE_PARAMS_st: """ Memory free node parameters Attributes ---------- dptr : CUdeviceptr in: the pointer to free Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._dptr = CUdeviceptr(_ptr=&self._ptr[0].dptr) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['dptr : ' + str(self.dptr)] except ValueError: str_list += ['dptr : '] return '\n'.join(str_list) else: return '' @property def dptr(self): return self._dptr @dptr.setter def dptr(self, dptr): cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr self._dptr._ptr[0] = cdptr {{endif}} {{if 'struct CUDA_CHILD_GRAPH_NODE_PARAMS_st' in found_types}} cdef class CUDA_CHILD_GRAPH_NODE_PARAMS_st: """ Child graph node parameters Attributes ---------- graph : CUgraph The child graph to clone into the node for node creation, or a handle to the graph owned by the node for node query Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._graph = CUgraph(_ptr=&self._ptr[0].graph) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['graph : ' + str(self.graph)] except ValueError: str_list += ['graph : '] return '\n'.join(str_list) else: return '' @property def graph(self): return self._graph @graph.setter def graph(self, graph): cdef ccuda.CUgraph cgraph if graph is None: cgraph = 0 elif isinstance(graph, (CUgraph,)): pgraph = int(graph) cgraph = pgraph else: pgraph = int(CUgraph(graph)) cgraph = pgraph self._graph._ptr[0] = cgraph {{endif}} {{if 'struct CUDA_EVENT_RECORD_NODE_PARAMS_st' in found_types}} cdef class CUDA_EVENT_RECORD_NODE_PARAMS_st: """ Event record node parameters Attributes ---------- event : CUevent The event to record when the node executes Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._event = CUevent(_ptr=&self._ptr[0].event) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['event : ' + str(self.event)] except ValueError: str_list += ['event : '] return '\n'.join(str_list) else: return '' @property def event(self): return self._event @event.setter def event(self, event): cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent self._event._ptr[0] = cevent {{endif}} {{if 'struct CUDA_EVENT_WAIT_NODE_PARAMS_st' in found_types}} cdef class CUDA_EVENT_WAIT_NODE_PARAMS_st: """ Event wait node parameters Attributes ---------- event : CUevent The event to wait on from the node Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._event = CUevent(_ptr=&self._ptr[0].event) def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['event : ' + str(self.event)] except ValueError: str_list += ['event : '] return '\n'.join(str_list) else: return '' @property def event(self): return self._event @event.setter def event(self, event): cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent self._event._ptr[0] = cevent {{endif}} {{if 'struct CUgraphNodeParams_st' in found_types}} cdef class CUgraphNodeParams_st: """ Graph node parameters. See cuGraphAddNode. Attributes ---------- type : CUgraphNodeType Type of the node reserved0 : List[int] Reserved. Must be zero. reserved1 : List[long long] Padding. Unused bytes must be zero. kernel : CUDA_KERNEL_NODE_PARAMS_v3 Kernel node parameters. memcpy : CUDA_MEMCPY_NODE_PARAMS Memcpy node parameters. memset : CUDA_MEMSET_NODE_PARAMS_v2 Memset node parameters. host : CUDA_HOST_NODE_PARAMS_v2 Host node parameters. graph : CUDA_CHILD_GRAPH_NODE_PARAMS Child graph node parameters. eventWait : CUDA_EVENT_WAIT_NODE_PARAMS Event wait node parameters. eventRecord : CUDA_EVENT_RECORD_NODE_PARAMS Event record node parameters. extSemSignal : CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2 External semaphore signal node parameters. extSemWait : CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2 External semaphore wait node parameters. alloc : CUDA_MEM_ALLOC_NODE_PARAMS_v2 Memory allocation node parameters. free : CUDA_MEM_FREE_NODE_PARAMS Memory free node parameters. memOp : CUDA_BATCH_MEM_OP_NODE_PARAMS_v2 MemOp node parameters. conditional : CUDA_CONDITIONAL_NODE_PARAMS Conditional node parameters. reserved2 : long long Reserved bytes. Must be zero. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUgraphNodeParams_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._kernel = CUDA_KERNEL_NODE_PARAMS_v3(_ptr=&self._ptr[0].kernel) self._memcpy = CUDA_MEMCPY_NODE_PARAMS(_ptr=&self._ptr[0].memcpy) self._memset = CUDA_MEMSET_NODE_PARAMS_v2(_ptr=&self._ptr[0].memset) self._host = CUDA_HOST_NODE_PARAMS_v2(_ptr=&self._ptr[0].host) self._graph = CUDA_CHILD_GRAPH_NODE_PARAMS(_ptr=&self._ptr[0].graph) self._eventWait = CUDA_EVENT_WAIT_NODE_PARAMS(_ptr=&self._ptr[0].eventWait) self._eventRecord = CUDA_EVENT_RECORD_NODE_PARAMS(_ptr=&self._ptr[0].eventRecord) self._extSemSignal = CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2(_ptr=&self._ptr[0].extSemSignal) self._extSemWait = CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2(_ptr=&self._ptr[0].extSemWait) self._alloc = CUDA_MEM_ALLOC_NODE_PARAMS_v2(_ptr=&self._ptr[0].alloc) self._free = CUDA_MEM_FREE_NODE_PARAMS(_ptr=&self._ptr[0].free) self._memOp = CUDA_BATCH_MEM_OP_NODE_PARAMS_v2(_ptr=&self._ptr[0].memOp) self._conditional = CUDA_CONDITIONAL_NODE_PARAMS(_ptr=&self._ptr[0].conditional) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['reserved0 : ' + str(self.reserved0)] except ValueError: str_list += ['reserved0 : '] try: str_list += ['reserved1 : ' + str(self.reserved1)] except ValueError: str_list += ['reserved1 : '] try: str_list += ['kernel :\n' + '\n'.join([' ' + line for line in str(self.kernel).splitlines()])] except ValueError: str_list += ['kernel : '] try: str_list += ['memcpy :\n' + '\n'.join([' ' + line for line in str(self.memcpy).splitlines()])] except ValueError: str_list += ['memcpy : '] try: str_list += ['memset :\n' + '\n'.join([' ' + line for line in str(self.memset).splitlines()])] except ValueError: str_list += ['memset : '] try: str_list += ['host :\n' + '\n'.join([' ' + line for line in str(self.host).splitlines()])] except ValueError: str_list += ['host : '] try: str_list += ['graph :\n' + '\n'.join([' ' + line for line in str(self.graph).splitlines()])] except ValueError: str_list += ['graph : '] try: str_list += ['eventWait :\n' + '\n'.join([' ' + line for line in str(self.eventWait).splitlines()])] except ValueError: str_list += ['eventWait : '] try: str_list += ['eventRecord :\n' + '\n'.join([' ' + line for line in str(self.eventRecord).splitlines()])] except ValueError: str_list += ['eventRecord : '] try: str_list += ['extSemSignal :\n' + '\n'.join([' ' + line for line in str(self.extSemSignal).splitlines()])] except ValueError: str_list += ['extSemSignal : '] try: str_list += ['extSemWait :\n' + '\n'.join([' ' + line for line in str(self.extSemWait).splitlines()])] except ValueError: str_list += ['extSemWait : '] try: str_list += ['alloc :\n' + '\n'.join([' ' + line for line in str(self.alloc).splitlines()])] except ValueError: str_list += ['alloc : '] try: str_list += ['free :\n' + '\n'.join([' ' + line for line in str(self.free).splitlines()])] except ValueError: str_list += ['free : '] try: str_list += ['memOp :\n' + '\n'.join([' ' + line for line in str(self.memOp).splitlines()])] except ValueError: str_list += ['memOp : '] try: str_list += ['conditional :\n' + '\n'.join([' ' + line for line in str(self.conditional).splitlines()])] except ValueError: str_list += ['conditional : '] try: str_list += ['reserved2 : ' + str(self.reserved2)] except ValueError: str_list += ['reserved2 : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUgraphNodeType(self._ptr[0].type) @type.setter def type(self, type not None : CUgraphNodeType): self._ptr[0].type = type.value @property def reserved0(self): return self._ptr[0].reserved0 @reserved0.setter def reserved0(self, reserved0): self._ptr[0].reserved0 = reserved0 @property def reserved1(self): return self._ptr[0].reserved1 @reserved1.setter def reserved1(self, reserved1): self._ptr[0].reserved1 = reserved1 @property def kernel(self): return self._kernel @kernel.setter def kernel(self, kernel not None : CUDA_KERNEL_NODE_PARAMS_v3): string.memcpy(&self._ptr[0].kernel, kernel.getPtr(), sizeof(self._ptr[0].kernel)) @property def memcpy(self): return self._memcpy @memcpy.setter def memcpy(self, memcpy not None : CUDA_MEMCPY_NODE_PARAMS): string.memcpy(&self._ptr[0].memcpy, memcpy.getPtr(), sizeof(self._ptr[0].memcpy)) @property def memset(self): return self._memset @memset.setter def memset(self, memset not None : CUDA_MEMSET_NODE_PARAMS_v2): string.memcpy(&self._ptr[0].memset, memset.getPtr(), sizeof(self._ptr[0].memset)) @property def host(self): return self._host @host.setter def host(self, host not None : CUDA_HOST_NODE_PARAMS_v2): string.memcpy(&self._ptr[0].host, host.getPtr(), sizeof(self._ptr[0].host)) @property def graph(self): return self._graph @graph.setter def graph(self, graph not None : CUDA_CHILD_GRAPH_NODE_PARAMS): string.memcpy(&self._ptr[0].graph, graph.getPtr(), sizeof(self._ptr[0].graph)) @property def eventWait(self): return self._eventWait @eventWait.setter def eventWait(self, eventWait not None : CUDA_EVENT_WAIT_NODE_PARAMS): string.memcpy(&self._ptr[0].eventWait, eventWait.getPtr(), sizeof(self._ptr[0].eventWait)) @property def eventRecord(self): return self._eventRecord @eventRecord.setter def eventRecord(self, eventRecord not None : CUDA_EVENT_RECORD_NODE_PARAMS): string.memcpy(&self._ptr[0].eventRecord, eventRecord.getPtr(), sizeof(self._ptr[0].eventRecord)) @property def extSemSignal(self): return self._extSemSignal @extSemSignal.setter def extSemSignal(self, extSemSignal not None : CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2): string.memcpy(&self._ptr[0].extSemSignal, extSemSignal.getPtr(), sizeof(self._ptr[0].extSemSignal)) @property def extSemWait(self): return self._extSemWait @extSemWait.setter def extSemWait(self, extSemWait not None : CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2): string.memcpy(&self._ptr[0].extSemWait, extSemWait.getPtr(), sizeof(self._ptr[0].extSemWait)) @property def alloc(self): return self._alloc @alloc.setter def alloc(self, alloc not None : CUDA_MEM_ALLOC_NODE_PARAMS_v2): string.memcpy(&self._ptr[0].alloc, alloc.getPtr(), sizeof(self._ptr[0].alloc)) @property def free(self): return self._free @free.setter def free(self, free not None : CUDA_MEM_FREE_NODE_PARAMS): string.memcpy(&self._ptr[0].free, free.getPtr(), sizeof(self._ptr[0].free)) @property def memOp(self): return self._memOp @memOp.setter def memOp(self, memOp not None : CUDA_BATCH_MEM_OP_NODE_PARAMS_v2): string.memcpy(&self._ptr[0].memOp, memOp.getPtr(), sizeof(self._ptr[0].memOp)) @property def conditional(self): return self._conditional @conditional.setter def conditional(self, conditional not None : CUDA_CONDITIONAL_NODE_PARAMS): string.memcpy(&self._ptr[0].conditional, conditional.getPtr(), sizeof(self._ptr[0].conditional)) @property def reserved2(self): return self._ptr[0].reserved2 @reserved2.setter def reserved2(self, long long reserved2): self._ptr[0].reserved2 = reserved2 {{endif}} {{if 'struct CUdevSmResource_st' in found_types}} cdef class CUdevSmResource_st: """ Attributes ---------- smCount : unsigned int The amount of streaming multiprocessors available in this resource. This is an output parameter only, do not write to this field. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): pass def __dealloc__(self): pass def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['smCount : ' + str(self.smCount)] except ValueError: str_list += ['smCount : '] return '\n'.join(str_list) else: return '' @property def smCount(self): return self._ptr[0].smCount @smCount.setter def smCount(self, unsigned int smCount): self._ptr[0].smCount = smCount {{endif}} {{if 'struct CUdevResource_st' in found_types}} cdef class CUdevResource_st: """ Attributes ---------- type : CUdevResourceType Type of resource, dictates which union field was last set _internal_padding : bytes sm : CUdevSmResource Resource corresponding to CU_DEV_RESOURCE_TYPE_SM ``. type. _oversize : bytes Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUdevResource_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._sm = CUdevSmResource(_ptr=&self._ptr[0].sm) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['type : ' + str(self.type)] except ValueError: str_list += ['type : '] try: str_list += ['_internal_padding : ' + str(self._internal_padding)] except ValueError: str_list += ['_internal_padding : '] try: str_list += ['sm :\n' + '\n'.join([' ' + line for line in str(self.sm).splitlines()])] except ValueError: str_list += ['sm : '] try: str_list += ['_oversize : ' + str(self._oversize)] except ValueError: str_list += ['_oversize : '] return '\n'.join(str_list) else: return '' @property def type(self): return CUdevResourceType(self._ptr[0].type) @type.setter def type(self, type not None : CUdevResourceType): self._ptr[0].type = type.value @property def _internal_padding(self): return PyBytes_FromStringAndSize(self._ptr[0]._internal_padding, 92) @_internal_padding.setter def _internal_padding(self, _internal_padding): if len(_internal_padding) != 92: raise ValueError("_internal_padding length must be 92, is " + str(len(_internal_padding))) for i, b in enumerate(_internal_padding): self._ptr[0]._internal_padding[i] = b @property def sm(self): return self._sm @sm.setter def sm(self, sm not None : CUdevSmResource): string.memcpy(&self._ptr[0].sm, sm.getPtr(), sizeof(self._ptr[0].sm)) @property def _oversize(self): return PyBytes_FromStringAndSize(self._ptr[0]._oversize, 48) @_oversize.setter def _oversize(self, _oversize): if len(_oversize) != 48: raise ValueError("_oversize length must be 48, is " + str(len(_oversize))) for i, b in enumerate(_oversize): self._ptr[0]._oversize[i] = b {{endif}} {{if True}} cdef class anon_union14: """ Attributes ---------- pArray : List[CUarray] pPitch : List[Any] Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr): self._ptr = _ptr def __init__(self, void_ptr _ptr): pass def __dealloc__(self): pass def getPtr(self): return &self._ptr[0].frame def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['pArray : ' + str(self.pArray)] except ValueError: str_list += ['pArray : '] try: str_list += ['pPitch : ' + hex(self.pPitch)] except ValueError: str_list += ['pPitch : '] return '\n'.join(str_list) else: return '' @property def pArray(self): return [CUarray(init_value=_pArray) for _pArray in self._ptr[0].frame.pArray] @pArray.setter def pArray(self, pArray : List[CUarray]): if len(pArray) != 3: raise IndexError('not enough values found during array assignment, expected 3, got', len(pArray)) pArray = [int(_pArray) for _pArray in pArray] for _idx, _pArray in enumerate(pArray): self._ptr[0].frame.pArray[_idx] = _pArray @property def pPitch(self): return [_pPitch for _pPitch in self._ptr[0].frame.pPitch] @pPitch.setter def pPitch(self, pPitch : List[int]): if len(pPitch) != 3: raise IndexError('not enough values found during array assignment, expected 3, got', len(pPitch)) pPitch = [_pPitch for _pPitch in pPitch] for _idx, _pPitch in enumerate(pPitch): self._ptr[0].frame.pPitch[_idx] = _pPitch {{endif}} {{if True}} cdef class CUeglFrame_st: """ CUDA EGLFrame structure Descriptor - structure defining one frame of EGL. Each frame may contain one or more planes depending on whether the surface * is Multiplanar or not. Attributes ---------- frame : anon_union14 width : unsigned int Width of first plane height : unsigned int Height of first plane depth : unsigned int Depth of first plane pitch : unsigned int Pitch of first plane planeCount : unsigned int Number of planes numChannels : unsigned int Number of channels for the plane frameType : CUeglFrameType Array or Pitch eglColorFormat : CUeglColorFormat CUDA EGL Color Format cuFormat : CUarray_format CUDA Array Format Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, void_ptr _ptr = 0): if _ptr == 0: self._val_ptr = calloc(1, sizeof(ccuda.CUeglFrame_st)) self._ptr = self._val_ptr else: self._ptr = _ptr def __init__(self, void_ptr _ptr = 0): self._frame = anon_union14(_ptr=self._ptr) def __dealloc__(self): if self._val_ptr is not NULL: free(self._val_ptr) def getPtr(self): return self._ptr def __repr__(self): if self._ptr is not NULL: str_list = [] try: str_list += ['frame :\n' + '\n'.join([' ' + line for line in str(self.frame).splitlines()])] except ValueError: str_list += ['frame : '] try: str_list += ['width : ' + str(self.width)] except ValueError: str_list += ['width : '] try: str_list += ['height : ' + str(self.height)] except ValueError: str_list += ['height : '] try: str_list += ['depth : ' + str(self.depth)] except ValueError: str_list += ['depth : '] try: str_list += ['pitch : ' + str(self.pitch)] except ValueError: str_list += ['pitch : '] try: str_list += ['planeCount : ' + str(self.planeCount)] except ValueError: str_list += ['planeCount : '] try: str_list += ['numChannels : ' + str(self.numChannels)] except ValueError: str_list += ['numChannels : '] try: str_list += ['frameType : ' + str(self.frameType)] except ValueError: str_list += ['frameType : '] try: str_list += ['eglColorFormat : ' + str(self.eglColorFormat)] except ValueError: str_list += ['eglColorFormat : '] try: str_list += ['cuFormat : ' + str(self.cuFormat)] except ValueError: str_list += ['cuFormat : '] return '\n'.join(str_list) else: return '' @property def frame(self): return self._frame @frame.setter def frame(self, frame not None : anon_union14): string.memcpy(&self._ptr[0].frame, frame.getPtr(), sizeof(self._ptr[0].frame)) @property def width(self): return self._ptr[0].width @width.setter def width(self, unsigned int width): self._ptr[0].width = width @property def height(self): return self._ptr[0].height @height.setter def height(self, unsigned int height): self._ptr[0].height = height @property def depth(self): return self._ptr[0].depth @depth.setter def depth(self, unsigned int depth): self._ptr[0].depth = depth @property def pitch(self): return self._ptr[0].pitch @pitch.setter def pitch(self, unsigned int pitch): self._ptr[0].pitch = pitch @property def planeCount(self): return self._ptr[0].planeCount @planeCount.setter def planeCount(self, unsigned int planeCount): self._ptr[0].planeCount = planeCount @property def numChannels(self): return self._ptr[0].numChannels @numChannels.setter def numChannels(self, unsigned int numChannels): self._ptr[0].numChannels = numChannels @property def frameType(self): return CUeglFrameType(self._ptr[0].frameType) @frameType.setter def frameType(self, frameType not None : CUeglFrameType): self._ptr[0].frameType = frameType.value @property def eglColorFormat(self): return CUeglColorFormat(self._ptr[0].eglColorFormat) @eglColorFormat.setter def eglColorFormat(self, eglColorFormat not None : CUeglColorFormat): self._ptr[0].eglColorFormat = eglColorFormat.value @property def cuFormat(self): return CUarray_format(self._ptr[0].cuFormat) @cuFormat.setter def cuFormat(self, cuFormat not None : CUarray_format): self._ptr[0].cuFormat = cuFormat.value {{endif}} {{if 'cuuint32_t' in found_types}} cdef class cuuint32_t: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, uint32_t init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'cuuint64_t' in found_types}} cdef class cuuint64_t: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, uint64_t init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUdeviceptr_v2' in found_types}} cdef class CUdeviceptr_v2: """ CUDA device pointer CUdeviceptr is defined as an unsigned integer type whose size matches the size of a pointer on the target platform. Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUdevice_v1' in found_types}} cdef class CUdevice_v1: """ CUDA device Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, int init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUtexObject_v1' in found_types}} cdef class CUtexObject_v1: """ An opaque value that represents a CUDA texture object Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUsurfObject_v1' in found_types}} cdef class CUsurfObject_v1: """ An opaque value that represents a CUDA surface object Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'CUmemGenericAllocationHandle_v1' in found_types}} cdef class CUmemGenericAllocationHandle_v1: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class GLenum: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned int init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class GLuint: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned int init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class EGLint: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned int init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class VdpDevice: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, uint32_t init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class VdpGetProcAddress: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, unsigned long long init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class VdpVideoSurface: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, uint32_t init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if True}} cdef class VdpOutputSurface: """ Methods ------- getPtr() Get memory address of class instance """ def __cinit__(self, uint32_t init_value = 0, void_ptr _ptr = 0): if _ptr == 0: self._ptr = &self.__val else: self._ptr = _ptr if init_value: self._ptr[0] = init_value def __dealloc__(self): pass def __repr__(self): return '' def __int__(self): return self._ptr[0] def getPtr(self): return self._ptr {{endif}} {{if 'cuGetErrorString' in found_functions}} @cython.embedsignature(True) def cuGetErrorString(error not None : CUresult): """ Gets the string description of an error code. Sets `*pStr` to the address of a NULL-terminated string description of the error code `error`. If the error code is not recognized, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned and `*pStr` will be set to the NULL address. Parameters ---------- error : :py:obj:`~.CUresult` Error code to convert to string Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pStr : bytes Address of the string pointer. See Also -------- :py:obj:`~.CUresult`, :py:obj:`~.cudaGetErrorString` """ cdef ccuda.CUresult cerror = error.value cdef const char* pStr = NULL err = ccuda.cuGetErrorString(cerror, &pStr) return (CUresult(err), pStr) {{endif}} {{if 'cuGetErrorName' in found_functions}} @cython.embedsignature(True) def cuGetErrorName(error not None : CUresult): """ Gets the string representation of an error code enum name. Sets `*pStr` to the address of a NULL-terminated string representation of the name of the enum error code `error`. If the error code is not recognized, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned and `*pStr` will be set to the NULL address. Parameters ---------- error : :py:obj:`~.CUresult` Error code to convert to string Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pStr : bytes Address of the string pointer. See Also -------- :py:obj:`~.CUresult`, :py:obj:`~.cudaGetErrorName` """ cdef ccuda.CUresult cerror = error.value cdef const char* pStr = NULL err = ccuda.cuGetErrorName(cerror, &pStr) return (CUresult(err), pStr) {{endif}} {{if 'cuInit' in found_functions}} @cython.embedsignature(True) def cuInit(unsigned int Flags): """ Initialize the CUDA driver API Initializes the driver API and must be called before any other function from the driver API in the current process. Currently, the `Flags` parameter must be 0. If :py:obj:`~.cuInit()` has not been called, any function from the driver API will return :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`. Parameters ---------- Flags : unsigned int Initialization flag for CUDA. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_SYSTEM_DRIVER_MISMATCH`, :py:obj:`~.CUDA_ERROR_COMPAT_NOT_SUPPORTED_ON_DEVICE` """ err = ccuda.cuInit(Flags) return (CUresult(err),) {{endif}} {{if 'cuDriverGetVersion' in found_functions}} @cython.embedsignature(True) def cuDriverGetVersion(): """ Returns the latest CUDA version supported by driver. Returns in `*driverVersion` the version of CUDA supported by the driver. The version is returned as (1000 * major + 10 * minor). For example, CUDA 9.2 would be represented by 9020. This function automatically returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if `driverVersion` is NULL. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` driverVersion : int Returns the CUDA driver version See Also -------- :py:obj:`~.cudaDriverGetVersion`, :py:obj:`~.cudaRuntimeGetVersion` """ cdef int driverVersion = 0 err = ccuda.cuDriverGetVersion(&driverVersion) return (CUresult(err), driverVersion) {{endif}} {{if 'cuDeviceGet' in found_functions}} @cython.embedsignature(True) def cuDeviceGet(int ordinal): """ Returns a handle to a compute device. Returns in `*device` a device handle given an ordinal in the range [0, :py:obj:`~.cuDeviceGetCount()`-1]. Parameters ---------- ordinal : int Device number to get handle for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` device : :py:obj:`~.CUdevice` Returned device handle See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGetLuid`, :py:obj:`~.cuDeviceTotalMem`, :py:obj:`~.cuDeviceGetExecAffinitySupport` """ cdef CUdevice device = CUdevice() err = ccuda.cuDeviceGet(device._ptr, ordinal) return (CUresult(err), device) {{endif}} {{if 'cuDeviceGetCount' in found_functions}} @cython.embedsignature(True) def cuDeviceGetCount(): """ Returns the number of compute-capable devices. Returns in `*count` the number of devices with compute capability greater than or equal to 2.0 that are available for execution. If there is no such device, :py:obj:`~.cuDeviceGetCount()` returns 0. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` count : int Returned number of compute-capable devices See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGetLuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem`, :py:obj:`~.cuDeviceGetExecAffinitySupport`, :py:obj:`~.cudaGetDeviceCount` """ cdef int count = 0 err = ccuda.cuDeviceGetCount(&count) return (CUresult(err), count) {{endif}} {{if 'cuDeviceGetName' in found_functions}} @cython.embedsignature(True) def cuDeviceGetName(int length, dev): """ Returns an identifier string for the device. Returns an ASCII string identifying the device `dev` in the NULL- terminated string pointed to by `name`. `length` specifies the maximum length of the string that may be returned. Parameters ---------- length : int Maximum length of string to store in `name` dev : :py:obj:`~.CUdevice` Device to get identifier string for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` name : bytes Returned identifier string for the device See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGetLuid`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem`, :py:obj:`~.cuDeviceGetExecAffinitySupport`, :py:obj:`~.cudaGetDeviceProperties` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev pyname = b" " * length cdef char* name = pyname err = ccuda.cuDeviceGetName(name, length, cdev) return (CUresult(err), pyname) {{endif}} {{if 'cuDeviceGetUuid' in found_functions}} @cython.embedsignature(True) def cuDeviceGetUuid(dev): """ Return an UUID for the device. Note there is a later version of this API, :py:obj:`~.cuDeviceGetUuid_v2`. It will supplant this version in 12.0, which is retained for minor version compatibility. Returns 16-octets identifying the device `dev` in the structure pointed by the `uuid`. Parameters ---------- dev : :py:obj:`~.CUdevice` Device to get identifier string for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` uuid : :py:obj:`~.CUuuid` Returned UUID See Also -------- :py:obj:`~.cuDeviceGetUuid_v2` :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetLuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem`, :py:obj:`~.cuDeviceGetExecAffinitySupport`, :py:obj:`~.cudaGetDeviceProperties` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUuuid uuid = CUuuid() err = ccuda.cuDeviceGetUuid(uuid._ptr, cdev) return (CUresult(err), uuid) {{endif}} {{if 'cuDeviceGetUuid_v2' in found_functions}} @cython.embedsignature(True) def cuDeviceGetUuid_v2(dev): """ Return an UUID for the device (11.4+) Returns 16-octets identifying the device `dev` in the structure pointed by the `uuid`. If the device is in MIG mode, returns its MIG UUID which uniquely identifies the subscribed MIG compute instance. Parameters ---------- dev : :py:obj:`~.CUdevice` Device to get identifier string for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` uuid : :py:obj:`~.CUuuid` Returned UUID See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetLuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem`, :py:obj:`~.cudaGetDeviceProperties` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUuuid uuid = CUuuid() err = ccuda.cuDeviceGetUuid_v2(uuid._ptr, cdev) return (CUresult(err), uuid) {{endif}} {{if 'cuDeviceGetLuid' in found_functions}} @cython.embedsignature(True) def cuDeviceGetLuid(dev): """ Return an LUID and device node mask for the device. Return identifying information (`luid` and `deviceNodeMask`) to allow matching device with graphics APIs. Parameters ---------- dev : :py:obj:`~.CUdevice` Device to get identifier string for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` luid : bytes Returned LUID deviceNodeMask : unsigned int Returned device node mask See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem`, :py:obj:`~.cuDeviceGetExecAffinitySupport`, :py:obj:`~.cudaGetDeviceProperties` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef char luid[8] cdef unsigned int deviceNodeMask = 0 err = ccuda.cuDeviceGetLuid(luid, &deviceNodeMask, cdev) return (CUresult(err), luid, deviceNodeMask) {{endif}} {{if 'cuDeviceTotalMem_v2' in found_functions}} @cython.embedsignature(True) def cuDeviceTotalMem(dev): """ Returns the total amount of memory on the device. Returns in `*bytes` the total amount of memory available on the device `dev` in bytes. Parameters ---------- dev : :py:obj:`~.CUdevice` Device handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` numbytes : int Returned memory available on device in bytes See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceGetExecAffinitySupport`, :py:obj:`~.cudaMemGetInfo` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef size_t numbytes = 0 err = ccuda.cuDeviceTotalMem(&numbytes, cdev) return (CUresult(err), numbytes) {{endif}} {{if 'cuDeviceGetTexture1DLinearMaxWidth' in found_functions}} @cython.embedsignature(True) def cuDeviceGetTexture1DLinearMaxWidth(pformat not None : CUarray_format, unsigned numChannels, dev): """ Returns the maximum number of elements allocatable in a 1D linear texture for a given texture element size. Returns in `maxWidthInElements` the maximum number of texture elements allocatable in a 1D linear texture for given `pformat` and `numChannels`. Parameters ---------- pformat : :py:obj:`~.CUarray_format` Texture format. numChannels : unsigned Number of channels per texture element. dev : :py:obj:`~.CUdevice` Device handle. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` maxWidthInElements : int Returned maximum number of texture elements allocatable for given `pformat` and `numChannels`. See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cudaMemGetInfo`, :py:obj:`~.cuDeviceTotalMem` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef size_t maxWidthInElements = 0 cdef ccuda.CUarray_format cpformat = pformat.value err = ccuda.cuDeviceGetTexture1DLinearMaxWidth(&maxWidthInElements, cpformat, numChannels, cdev) return (CUresult(err), maxWidthInElements) {{endif}} {{if 'cuDeviceGetAttribute' in found_functions}} @cython.embedsignature(True) def cuDeviceGetAttribute(attrib not None : CUdevice_attribute, dev): """ Returns information about the device. Returns in `*pi` the integer value of the attribute `attrib` on device `dev`. The supported attributes are: - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_BLOCK`: Maximum number of threads per block; - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_X`: Maximum x-dimension of a block - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Y`: Maximum y-dimension of a block - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_BLOCK_DIM_Z`: Maximum z-dimension of a block - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_X`: Maximum x-dimension of a grid - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Y`: Maximum y-dimension of a grid - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_GRID_DIM_Z`: Maximum z-dimension of a grid - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK`: Maximum amount of shared memory available to a thread block in bytes - :py:obj:`~.CU_DEVICE_ATTRIBUTE_TOTAL_CONSTANT_MEMORY`: Memory available on device for constant variables in a CUDA C kernel in bytes - :py:obj:`~.CU_DEVICE_ATTRIBUTE_WARP_SIZE`: Warp size in threads - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_PITCH`: Maximum pitch in bytes allowed by the memory copy functions that involve memory regions allocated through :py:obj:`~.cuMemAllocPitch()` - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_WIDTH`: Maximum 1D texture width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LINEAR_WIDTH`: Maximum width for a 1D texture bound to linear memory - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_MIPMAPPED_WIDTH`: Maximum mipmapped 1D texture width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_WIDTH`: Maximum 2D texture width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_HEIGHT`: Maximum 2D texture height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_WIDTH`: Maximum width for a 2D texture bound to linear memory - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_HEIGHT`: Maximum height for a 2D texture bound to linear memory - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_PITCH`: Maximum pitch in bytes for a 2D texture bound to linear memory - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_WIDTH`: Maximum mipmapped 2D texture width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_MIPMAPPED_HEIGHT`: Maximum mipmapped 2D texture height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH`: Maximum 3D texture width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT`: Maximum 3D texture height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH`: Maximum 3D texture depth - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_WIDTH_ALTERNATE`: Alternate maximum 3D texture width, 0 if no alternate maximum 3D texture size is supported - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_HEIGHT_ALTERNATE`: Alternate maximum 3D texture height, 0 if no alternate maximum 3D texture size is supported - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE3D_DEPTH_ALTERNATE`: Alternate maximum 3D texture depth, 0 if no alternate maximum 3D texture size is supported - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_WIDTH`: Maximum cubemap texture width or height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_WIDTH`: Maximum 1D layered texture width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LAYERED_LAYERS`: Maximum layers in a 1D layered texture - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_WIDTH`: Maximum 2D layered texture width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_HEIGHT`: Maximum 2D layered texture height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LAYERED_LAYERS`: Maximum layers in a 2D layered texture - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_WIDTH`: Maximum cubemap layered texture width or height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURECUBEMAP_LAYERED_LAYERS`: Maximum layers in a cubemap layered texture - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_WIDTH`: Maximum 1D surface width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_WIDTH`: Maximum 2D surface width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_HEIGHT`: Maximum 2D surface height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_WIDTH`: Maximum 3D surface width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_HEIGHT`: Maximum 3D surface height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE3D_DEPTH`: Maximum 3D surface depth - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_WIDTH`: Maximum 1D layered surface width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE1D_LAYERED_LAYERS`: Maximum layers in a 1D layered surface - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_WIDTH`: Maximum 2D layered surface width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_HEIGHT`: Maximum 2D layered surface height - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACE2D_LAYERED_LAYERS`: Maximum layers in a 2D layered surface - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_WIDTH`: Maximum cubemap surface width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_WIDTH`: Maximum cubemap layered surface width - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_SURFACECUBEMAP_LAYERED_LAYERS`: Maximum layers in a cubemap layered surface - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_BLOCK`: Maximum number of 32-bit registers available to a thread block - :py:obj:`~.CU_DEVICE_ATTRIBUTE_CLOCK_RATE`: The typical clock frequency in kilohertz - :py:obj:`~.CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT`: Alignment requirement; texture base addresses aligned to :py:obj:`~.textureAlign` bytes do not need an offset applied to texture fetches - :py:obj:`~.CU_DEVICE_ATTRIBUTE_TEXTURE_PITCH_ALIGNMENT`: Pitch alignment requirement for 2D texture references bound to pitched memory - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GPU_OVERLAP`: 1 if the device can concurrently copy memory between host and device while executing a kernel, or 0 if not - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT`: Number of multiprocessors on the device - :py:obj:`~.CU_DEVICE_ATTRIBUTE_KERNEL_EXEC_TIMEOUT`: 1 if there is a run time limit for kernels executed on the device, or 0 if not - :py:obj:`~.CU_DEVICE_ATTRIBUTE_INTEGRATED`: 1 if the device is integrated with the memory subsystem, or 0 if not - :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_MAP_HOST_MEMORY`: 1 if the device can map host memory into the CUDA address space, or 0 if not - :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_MODE`: Compute mode that device is currently in. Available modes are as follows: - :py:obj:`~.CU_COMPUTEMODE_DEFAULT`: Default mode - Device is not restricted and can have multiple CUDA contexts present at a single time. - :py:obj:`~.CU_COMPUTEMODE_PROHIBITED`: Compute-prohibited mode - Device is prohibited from creating new CUDA contexts. - :py:obj:`~.CU_COMPUTEMODE_EXCLUSIVE_PROCESS`: Compute-exclusive- process mode - Device can have only one context used by a single process at a time. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_KERNELS`: 1 if the device supports executing multiple kernels within the same context simultaneously, or 0 if not. It is not guaranteed that multiple kernels will be resident on the device concurrently so this feature should not be relied upon for correctness. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_ECC_ENABLED`: 1 if error correction is enabled on the device, 0 if error correction is disabled or not supported by the device - :py:obj:`~.CU_DEVICE_ATTRIBUTE_PCI_BUS_ID`: PCI bus identifier of the device - :py:obj:`~.CU_DEVICE_ATTRIBUTE_PCI_DEVICE_ID`: PCI device (also known as slot) identifier of the device - :py:obj:`~.CU_DEVICE_ATTRIBUTE_PCI_DOMAIN_ID`: PCI domain identifier of the device - :py:obj:`~.CU_DEVICE_ATTRIBUTE_TCC_DRIVER`: 1 if the device is using a TCC driver. TCC is only available on Tesla hardware running Windows Vista or later - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MEMORY_CLOCK_RATE`: Peak memory clock frequency in kilohertz - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GLOBAL_MEMORY_BUS_WIDTH`: Global memory bus width in bits - :py:obj:`~.CU_DEVICE_ATTRIBUTE_L2_CACHE_SIZE`: Size of L2 cache in bytes. 0 if the device doesn't have L2 cache - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_THREADS_PER_MULTIPROCESSOR`: Maximum resident threads per multiprocessor - :py:obj:`~.CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING`: 1 if the device shares a unified address space with the host, or 0 if not - :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR`: Major compute capability version number - :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR`: Minor compute capability version number - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GLOBAL_L1_CACHE_SUPPORTED`: 1 if device supports caching globals in L1 cache, 0 if caching globals in L1 cache is not supported by the device - :py:obj:`~.CU_DEVICE_ATTRIBUTE_LOCAL_L1_CACHE_SUPPORTED`: 1 if device supports caching locals in L1 cache, 0 if caching locals in L1 cache is not supported by the device - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_MULTIPROCESSOR`: Maximum amount of shared memory available to a multiprocessor in bytes; this amount is shared by all thread blocks simultaneously resident on a multiprocessor - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_REGISTERS_PER_MULTIPROCESSOR`: Maximum number of 32-bit registers available to a multiprocessor; this number is shared by all thread blocks simultaneously resident on a multiprocessor - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY`: 1 if device supports allocating managed memory on this system, 0 if allocating managed memory is not supported by the device on this system. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD`: 1 if device is on a multi-GPU board, 0 if not. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MULTI_GPU_BOARD_GROUP_ID`: Unique identifier for a group of devices associated with the same board. Devices on the same multi-GPU board will share the same identifier. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_HOST_NATIVE_ATOMIC_SUPPORTED`: 1 if Link between the device and the host supports native atomic operations. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_SINGLE_TO_DOUBLE_PRECISION_PERF_RATIO`: Ratio of single precision performance (in floating-point operations per second) to double precision performance. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`: Device supports coherently accessing pageable memory without calling cudaHostRegister on it. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`: Device can coherently access managed memory concurrently with the CPU. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_PREEMPTION_SUPPORTED`: Device supports Compute Preemption. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_USE_HOST_POINTER_FOR_REGISTERED_MEM`: Device can access host registered memory at the same virtual address as the CPU. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK_OPTIN`: The maximum per block shared memory size supported on this device. This is the maximum value that can be opted into when using the :py:obj:`~.cuFuncSetAttribute()` or :py:obj:`~.cuKernelSetAttribute()` call. For more details see :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES` - :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`: Device accesses pageable memory via the host's page tables. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_DIRECT_MANAGED_MEM_ACCESS_FROM_HOST`: The host can directly access managed memory on the device without migration. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_VIRTUAL_MEMORY_MANAGEMENT_SUPPORTED`: Device supports virtual memory management APIs like :py:obj:`~.cuMemAddressReserve`, :py:obj:`~.cuMemCreate`, :py:obj:`~.cuMemMap` and related APIs - :py:obj:`~.CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_POSIX_FILE_DESCRIPTOR_SUPPORTED`: Device supports exporting memory to a posix file descriptor with :py:obj:`~.cuMemExportToShareableHandle`, if requested via :py:obj:`~.cuMemCreate` - :py:obj:`~.CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_HANDLE_SUPPORTED`: Device supports exporting memory to a Win32 NT handle with :py:obj:`~.cuMemExportToShareableHandle`, if requested via :py:obj:`~.cuMemCreate` - :py:obj:`~.CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_WIN32_KMT_HANDLE_SUPPORTED`: Device supports exporting memory to a Win32 KMT handle with :py:obj:`~.cuMemExportToShareableHandle`, if requested via :py:obj:`~.cuMemCreate` - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_BLOCKS_PER_MULTIPROCESSOR`: Maximum number of thread blocks that can reside on a multiprocessor - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GENERIC_COMPRESSION_SUPPORTED`: Device supports compressible memory allocation via :py:obj:`~.cuMemCreate` - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_PERSISTING_L2_CACHE_SIZE`: Maximum L2 persisting lines capacity setting in bytes - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_ACCESS_POLICY_WINDOW_SIZE`: Maximum value of :py:obj:`~.CUaccessPolicyWindow.num_bytes` - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WITH_CUDA_VMM_SUPPORTED`: Device supports specifying the GPUDirect RDMA flag with :py:obj:`~.cuMemCreate`. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_RESERVED_SHARED_MEMORY_PER_BLOCK`: Amount of shared memory per block reserved by CUDA driver in bytes - :py:obj:`~.CU_DEVICE_ATTRIBUTE_SPARSE_CUDA_ARRAY_SUPPORTED`: Device supports sparse CUDA arrays and sparse CUDA mipmapped arrays. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_READ_ONLY_HOST_REGISTER_SUPPORTED`: Device supports using the :py:obj:`~.cuMemHostRegister` flag :py:obj:`~.CU_MEMHOSTERGISTER_READ_ONLY` to register memory that must be mapped as read-only to the GPU - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MEMORY_POOLS_SUPPORTED`: Device supports using the :py:obj:`~.cuMemAllocAsync` and :py:obj:`~.cuMemPool` family of APIs - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_SUPPORTED`: Device supports GPUDirect RDMA APIs, like nvidia_p2p_get_pages (see https://docs.nvidia.com/cuda/gpudirect-rdma for more information) - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_FLUSH_WRITES_OPTIONS`: The returned attribute shall be interpreted as a bitmask, where the individual bits are described by the :py:obj:`~.CUflushGPUDirectRDMAWritesOptions` enum - :py:obj:`~.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WRITES_ORDERING`: GPUDirect RDMA writes to the device do not need to be flushed for consumers within the scope indicated by the returned attribute. See :py:obj:`~.CUGPUDirectRDMAWritesOrdering` for the numerical values returned here. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MEMPOOL_SUPPORTED_HANDLE_TYPES`: Bitmask of handle types supported with mempool based IPC - :py:obj:`~.CU_DEVICE_ATTRIBUTE_DEFERRED_MAPPING_CUDA_ARRAY_SUPPORTED`: Device supports deferred mapping CUDA arrays and CUDA mipmapped arrays. - :py:obj:`~.CU_DEVICE_ATTRIBUTE_NUMA_CONFIG`: NUMA configuration of a device: value is of type :py:obj:`~.CUdeviceNumaConfig` enum - :py:obj:`~.CU_DEVICE_ATTRIBUTE_NUMA_ID`: NUMA node ID of the GPU memory - :py:obj:`~.CU_DEVICE_ATTRIBUTE_MULTICAST_SUPPORTED`: Device supports switch multicast and reduction operations. Parameters ---------- attrib : :py:obj:`~.CUdevice_attribute` Device attribute to query dev : :py:obj:`~.CUdevice` Device handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` pi : int Returned device attribute value See Also -------- :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem`, :py:obj:`~.cuDeviceGetExecAffinitySupport`, :py:obj:`~.cudaDeviceGetAttribute`, :py:obj:`~.cudaGetDeviceProperties` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef int pi = 0 cdef ccuda.CUdevice_attribute cattrib = attrib.value err = ccuda.cuDeviceGetAttribute(&pi, cattrib, cdev) return (CUresult(err), pi) {{endif}} {{if 'cuDeviceGetNvSciSyncAttributes' in found_functions}} @cython.embedsignature(True) def cuDeviceGetNvSciSyncAttributes(nvSciSyncAttrList, dev, int flags): """ Return NvSciSync attributes that this device can support. Returns in `nvSciSyncAttrList`, the properties of NvSciSync that this CUDA device, `dev` can support. The returned `nvSciSyncAttrList` can be used to create an NvSciSync object that matches this device's capabilities. If NvSciSyncAttrKey_RequiredPerm field in `nvSciSyncAttrList` is already set this API will return :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. The applications should set `nvSciSyncAttrList` to a valid NvSciSyncAttrList failing which this API will return :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`. The `flags` controls how applications intends to use the NvSciSync created from the `nvSciSyncAttrList`. The valid flags are: - :py:obj:`~.CUDA_NVSCISYNC_ATTR_SIGNAL`, specifies that the applications intends to signal an NvSciSync on this CUDA device. - :py:obj:`~.CUDA_NVSCISYNC_ATTR_WAIT`, specifies that the applications intends to wait on an NvSciSync on this CUDA device. At least one of these flags must be set, failing which the API returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. Both the flags are orthogonal to one another: a developer may set both these flags that allows to set both wait and signal specific attributes in the same `nvSciSyncAttrList`. Note that this API updates the input `nvSciSyncAttrList` with values equivalent to the following public attribute key-values: NvSciSyncAttrKey_RequiredPerm is set to - NvSciSyncAccessPerm_SignalOnly if :py:obj:`~.CUDA_NVSCISYNC_ATTR_SIGNAL` is set in `flags`. - NvSciSyncAccessPerm_WaitOnly if :py:obj:`~.CUDA_NVSCISYNC_ATTR_WAIT` is set in `flags`. - NvSciSyncAccessPerm_WaitSignal if both :py:obj:`~.CUDA_NVSCISYNC_ATTR_WAIT` and :py:obj:`~.CUDA_NVSCISYNC_ATTR_SIGNAL` are set in `flags`. NvSciSyncAttrKey_PrimitiveInfo is set to - NvSciSyncAttrValPrimitiveType_SysmemSemaphore on any valid `device`. - NvSciSyncAttrValPrimitiveType_Syncpoint if `device` is a Tegra device. - NvSciSyncAttrValPrimitiveType_SysmemSemaphorePayload64b if `device` is GA10X+. NvSciSyncAttrKey_GpuId is set to the same UUID that is returned for this `device` from :py:obj:`~.cuDeviceGetUuid`. :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` Parameters ---------- nvSciSyncAttrList : Any Return NvSciSync attributes supported. dev : :py:obj:`~.CUdevice` Valid Cuda Device to get NvSciSync attributes for. flags : int flags describing NvSciSync usage. Returns ------- CUresult See Also -------- :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuDestroyExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cnvSciSyncAttrList = utils.HelperInputVoidPtr(nvSciSyncAttrList) cdef void* cnvSciSyncAttrList_ptr = cnvSciSyncAttrList.cptr err = ccuda.cuDeviceGetNvSciSyncAttributes(cnvSciSyncAttrList_ptr, cdev, flags) return (CUresult(err),) {{endif}} {{if 'cuDeviceSetMemPool' in found_functions}} @cython.embedsignature(True) def cuDeviceSetMemPool(dev, pool): """ Sets the current memory pool of a device. The memory pool must be local to the specified device. :py:obj:`~.cuMemAllocAsync` allocates from the current mempool of the provided stream's device. By default, a device's current memory pool is its default memory pool. Parameters ---------- dev : :py:obj:`~.CUdevice` None pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` None Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate`, :py:obj:`~.cuMemPoolDestroy`, :py:obj:`~.cuMemAllocFromPoolAsync` Notes ----- Use :py:obj:`~.cuMemAllocFromPoolAsync` to specify asynchronous allocations from a device different than the one the stream runs on. """ cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev err = ccuda.cuDeviceSetMemPool(cdev, cpool) return (CUresult(err),) {{endif}} {{if 'cuDeviceGetMemPool' in found_functions}} @cython.embedsignature(True) def cuDeviceGetMemPool(dev): """ Gets the current mempool for a device. Returns the last pool provided to :py:obj:`~.cuDeviceSetMemPool` for this device or the device's default memory pool if :py:obj:`~.cuDeviceSetMemPool` has never been called. By default the current mempool is the default mempool for a device. Otherwise the returned pool must have been set with :py:obj:`~.cuDeviceSetMemPool`. Parameters ---------- dev : :py:obj:`~.CUdevice` None Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pool : :py:obj:`~.CUmemoryPool` None See Also -------- :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuMemPoolCreate`, :py:obj:`~.cuDeviceSetMemPool` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUmemoryPool pool = CUmemoryPool() err = ccuda.cuDeviceGetMemPool(pool._ptr, cdev) return (CUresult(err), pool) {{endif}} {{if 'cuDeviceGetDefaultMemPool' in found_functions}} @cython.embedsignature(True) def cuDeviceGetDefaultMemPool(dev): """ Returns the default mempool of a device. The default mempool of a device contains device memory from that device. Parameters ---------- dev : :py:obj:`~.CUdevice` None Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED` :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` pool_out : :py:obj:`~.CUmemoryPool` None See Also -------- :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemPoolTrimTo`, :py:obj:`~.cuMemPoolGetAttribute`, :py:obj:`~.cuMemPoolSetAttribute`, :py:obj:`~.cuMemPoolSetAccess`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUmemoryPool pool_out = CUmemoryPool() err = ccuda.cuDeviceGetDefaultMemPool(pool_out._ptr, cdev) return (CUresult(err), pool_out) {{endif}} {{if 'cuDeviceGetExecAffinitySupport' in found_functions}} @cython.embedsignature(True) def cuDeviceGetExecAffinitySupport(typename not None : CUexecAffinityType, dev): """ Returns information about the execution affinity support of the device. Returns in `*pi` whether execution affinity type `typename` is supported by device `dev`. The supported types are: - :py:obj:`~.CU_EXEC_AFFINITY_TYPE_SM_COUNT`: 1 if context with limited SMs is supported by the device, or 0 if not; Parameters ---------- typename : :py:obj:`~.CUexecAffinityType` Execution affinity type to query dev : :py:obj:`~.CUdevice` Device handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` pi : int 1 if the execution affinity type `typename` is supported by the device, or 0 if not See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef int pi = 0 cdef ccuda.CUexecAffinityType ctypename = typename.value err = ccuda.cuDeviceGetExecAffinitySupport(&pi, ctypename, cdev) return (CUresult(err), pi) {{endif}} {{if 'cuFlushGPUDirectRDMAWrites' in found_functions}} @cython.embedsignature(True) def cuFlushGPUDirectRDMAWrites(target not None : CUflushGPUDirectRDMAWritesTarget, scope not None : CUflushGPUDirectRDMAWritesScope): """ Blocks until remote writes are visible to the specified scope. Blocks until GPUDirect RDMA writes to the target context via mappings created through APIs like nvidia_p2p_get_pages (see https://docs.nvidia.com/cuda/gpudirect-rdma for more information), are visible to the specified scope. If the scope equals or lies within the scope indicated by :py:obj:`~.CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WRITES_ORDERING`, the call will be a no-op and can be safely omitted for performance. This can be determined by comparing the numerical values between the two enums, with smaller scopes having smaller values. Users may query support for this API via :py:obj:`~.CU_DEVICE_ATTRIBUTE_FLUSH_FLUSH_GPU_DIRECT_RDMA_OPTIONS`. Parameters ---------- target : :py:obj:`~.CUflushGPUDirectRDMAWritesTarget` The target of the operation, see :py:obj:`~.CUflushGPUDirectRDMAWritesTarget` scope : :py:obj:`~.CUflushGPUDirectRDMAWritesScope` The scope of the operation, see :py:obj:`~.CUflushGPUDirectRDMAWritesScope` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, """ cdef ccuda.CUflushGPUDirectRDMAWritesTarget ctarget = target.value cdef ccuda.CUflushGPUDirectRDMAWritesScope cscope = scope.value err = ccuda.cuFlushGPUDirectRDMAWrites(ctarget, cscope) return (CUresult(err),) {{endif}} {{if 'cuDeviceGetProperties' in found_functions}} @cython.embedsignature(True) def cuDeviceGetProperties(dev): """ Returns properties for a selected device. [Deprecated] This function was deprecated as of CUDA 5.0 and replaced by :py:obj:`~.cuDeviceGetAttribute()`. Returns in `*prop` the properties of device `dev`. The :py:obj:`~.CUdevprop` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.maxThreadsPerBlock` is the maximum number of threads per block; - :py:obj:`~.maxThreadsDim`[3] is the maximum sizes of each dimension of a block; - :py:obj:`~.maxGridSize`[3] is the maximum sizes of each dimension of a grid; - :py:obj:`~.sharedMemPerBlock` is the total amount of shared memory available per block in bytes; - :py:obj:`~.totalConstantMemory` is the total amount of constant memory available on the device in bytes; - :py:obj:`~.SIMDWidth` is the warp size; - :py:obj:`~.memPitch` is the maximum pitch allowed by the memory copy functions that involve memory regions allocated through :py:obj:`~.cuMemAllocPitch()`; - :py:obj:`~.regsPerBlock` is the total number of registers available per block; - :py:obj:`~.clockRate` is the clock frequency in kilohertz; - :py:obj:`~.textureAlign` is the alignment requirement; texture base addresses that are aligned to :py:obj:`~.textureAlign` bytes do not need an offset applied to texture fetches. Parameters ---------- dev : :py:obj:`~.CUdevice` Device to get properties for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` prop : :py:obj:`~.CUdevprop` Returned properties of device See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUdevprop prop = CUdevprop() err = ccuda.cuDeviceGetProperties(prop._ptr, cdev) return (CUresult(err), prop) {{endif}} {{if 'cuDeviceComputeCapability' in found_functions}} @cython.embedsignature(True) def cuDeviceComputeCapability(dev): """ Returns the compute capability of the device. [Deprecated] This function was deprecated as of CUDA 5.0 and its functionality superseded by :py:obj:`~.cuDeviceGetAttribute()`. Returns in `*major` and `*minor` the major and minor revision numbers that define the compute capability of the device `dev`. Parameters ---------- dev : :py:obj:`~.CUdevice` Device handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` major : int Major revision number minor : int Minor revision number See Also -------- :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetCount`, :py:obj:`~.cuDeviceGetName`, :py:obj:`~.cuDeviceGetUuid`, :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceTotalMem` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef int major = 0 cdef int minor = 0 err = ccuda.cuDeviceComputeCapability(&major, &minor, cdev) return (CUresult(err), major, minor) {{endif}} {{if 'cuDevicePrimaryCtxRetain' in found_functions}} @cython.embedsignature(True) def cuDevicePrimaryCtxRetain(dev): """ Retain the primary context on the GPU. Retains the primary context on the device. Once the user successfully retains the primary context, the primary context will be active and available to the user until the user releases it with :py:obj:`~.cuDevicePrimaryCtxRelease()` or resets it with :py:obj:`~.cuDevicePrimaryCtxReset()`. Unlike :py:obj:`~.cuCtxCreate()` the newly retained context is not pushed onto the stack. Retaining the primary context for the first time will fail with :py:obj:`~.CUDA_ERROR_UNKNOWN` if the compute mode of the device is :py:obj:`~.CU_COMPUTEMODE_PROHIBITED`. The function :py:obj:`~.cuDeviceGetAttribute()` can be used with :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_MODE` to determine the compute mode of the device. The `nvidia-smi` tool can be used to set the compute mode for devices. Documentation for `nvidia-smi` can be obtained by passing a -h option to it. Please note that the primary context always supports pinned allocations. Other flags can be specified by :py:obj:`~.cuDevicePrimaryCtxSetFlags()`. Parameters ---------- dev : :py:obj:`~.CUdevice` Device for which primary context is requested Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pctx : :py:obj:`~.CUcontext` Returned context handle of the new context See Also -------- :py:obj:`~.cuDevicePrimaryCtxRelease`, :py:obj:`~.cuDevicePrimaryCtxSetFlags`, :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUcontext pctx = CUcontext() err = ccuda.cuDevicePrimaryCtxRetain(pctx._ptr, cdev) return (CUresult(err), pctx) {{endif}} {{if 'cuDevicePrimaryCtxRelease_v2' in found_functions}} @cython.embedsignature(True) def cuDevicePrimaryCtxRelease(dev): """ Release the primary context on the GPU. Releases the primary context interop on the device. A retained context should always be released once the user is done using it. The context is automatically reset once the last reference to it is released. This behavior is different when the primary context was retained by the CUDA runtime from CUDA 4.0 and earlier. In this case, the primary context remains always active. Releasing a primary context that has not been previously retained will fail with :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`. Please note that unlike :py:obj:`~.cuCtxDestroy()` this method does not pop the context from stack in any circumstances. Parameters ---------- dev : :py:obj:`~.CUdevice` Device which primary context is released Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuDevicePrimaryCtxRetain`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev err = ccuda.cuDevicePrimaryCtxRelease(cdev) return (CUresult(err),) {{endif}} {{if 'cuDevicePrimaryCtxSetFlags_v2' in found_functions}} @cython.embedsignature(True) def cuDevicePrimaryCtxSetFlags(dev, unsigned int flags): """ Set flags for the primary context. Sets the flags for the primary context on the device overwriting perviously set ones. The three LSBs of the `flags` parameter can be used to control how the OS thread, which owns the CUDA context at the time of an API call, interacts with the OS scheduler when waiting for results from the GPU. Only one of the scheduling flags can be set when creating a context. - :py:obj:`~.CU_CTX_SCHED_SPIN`: Instruct CUDA to actively spin when waiting for results from the GPU. This can decrease latency when waiting for the GPU, but may lower the performance of CPU threads if they are performing work in parallel with the CUDA thread. - :py:obj:`~.CU_CTX_SCHED_YIELD`: Instruct CUDA to yield its thread when waiting for results from the GPU. This can increase latency when waiting for the GPU, but can increase the performance of CPU threads performing work in parallel with the GPU. - :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. - :py:obj:`~.CU_CTX_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. Deprecated: This flag was deprecated as of CUDA 4.0 and was replaced with :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`. - :py:obj:`~.CU_CTX_SCHED_AUTO`: The default value if the `flags` parameter is zero, uses a heuristic based on the number of active CUDA contexts in the process `C` and the number of logical processors in the system `P`. If `C` > `P`, then CUDA will yield to other OS threads when waiting for the GPU (:py:obj:`~.CU_CTX_SCHED_YIELD`), otherwise CUDA will not yield while waiting for results and actively spin on the processor (:py:obj:`~.CU_CTX_SCHED_SPIN`). Additionally, on Tegra devices, :py:obj:`~.CU_CTX_SCHED_AUTO` uses a heuristic based on the power profile of the platform and may choose :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC` for low-powered devices. - :py:obj:`~.CU_CTX_LMEM_RESIZE_TO_MAX`: Instruct CUDA to not reduce local memory after resizing local memory for a kernel. This can prevent thrashing by local memory allocations when launching many kernels with high local memory usage at the cost of potentially increased memory usage. Deprecated: This flag is deprecated and the behavior enabled by this flag is now the default and cannot be disabled. - :py:obj:`~.CU_CTX_COREDUMP_ENABLE`: If GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if this context raises an exception during execution. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. The initial settings will be taken from the global settings at the time of context creation. The other settings that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. - :py:obj:`~.CU_CTX_USER_COREDUMP_ENABLE`: If user-triggered GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if data is written to a certain pipe that is present in the OS space. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. It is important to note that the pipe name `must` be set with :py:obj:`~.cuCoredumpSetAttributeGlobal` before creating the context if this flag is used. Setting this flag implies that :py:obj:`~.CU_CTX_COREDUMP_ENABLE` is set. The initial settings will be taken from the global settings at the time of context creation. The other settings that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. - :py:obj:`~.CU_CTX_SYNC_MEMOPS`: Ensures that synchronous memory operations initiated on this context will always synchronize. See further documentation in the section titled "API Synchronization behavior" to learn more about cases when synchronous memory operations can exhibit asynchronous behavior. Parameters ---------- dev : :py:obj:`~.CUdevice` Device for which the primary context flags are set flags : unsigned int New flags for the device Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuDevicePrimaryCtxRetain`, :py:obj:`~.cuDevicePrimaryCtxGetState`, :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxSetFlags`, :py:obj:`~.cudaSetDeviceFlags` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev err = ccuda.cuDevicePrimaryCtxSetFlags(cdev, flags) return (CUresult(err),) {{endif}} {{if 'cuDevicePrimaryCtxGetState' in found_functions}} @cython.embedsignature(True) def cuDevicePrimaryCtxGetState(dev): """ Get the state of the primary context. Returns in `*flags` the flags for the primary context of `dev`, and in `*active` whether it is active. See :py:obj:`~.cuDevicePrimaryCtxSetFlags` for flag values. Parameters ---------- dev : :py:obj:`~.CUdevice` Device to get primary context flags for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, flags : unsigned int Pointer to store flags active : int Pointer to store context state; 0 = inactive, 1 = active See Also -------- :py:obj:`~.cuDevicePrimaryCtxSetFlags`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxSetFlags`, :py:obj:`~.cudaGetDeviceFlags` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef unsigned int flags = 0 cdef int active = 0 err = ccuda.cuDevicePrimaryCtxGetState(cdev, &flags, &active) return (CUresult(err), flags, active) {{endif}} {{if 'cuDevicePrimaryCtxReset_v2' in found_functions}} @cython.embedsignature(True) def cuDevicePrimaryCtxReset(dev): """ Destroy all allocations and reset all state on the primary context. Explicitly destroys and cleans up all resources associated with the current device in the current process. Note that it is responsibility of the calling function to ensure that no other module in the process is using the device any more. For that reason it is recommended to use :py:obj:`~.cuDevicePrimaryCtxRelease()` in most cases. However it is safe for other modules to call :py:obj:`~.cuDevicePrimaryCtxRelease()` even after resetting the device. Resetting the primary context does not release it, an application that has retained the primary context should explicitly release its usage. Parameters ---------- dev : :py:obj:`~.CUdevice` Device for which primary context is destroyed Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_PRIMARY_CONTEXT_ACTIVE` See Also -------- :py:obj:`~.cuDevicePrimaryCtxRetain`, :py:obj:`~.cuDevicePrimaryCtxRelease`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cudaDeviceReset` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev err = ccuda.cuDevicePrimaryCtxReset(cdev) return (CUresult(err),) {{endif}} {{if 'cuCtxCreate_v2' in found_functions}} @cython.embedsignature(True) def cuCtxCreate(unsigned int flags, dev): """ Create a CUDA context. Creates a new CUDA context and associates it with the calling thread. The `flags` parameter is described below. The context is created with a usage count of 1 and the caller of :py:obj:`~.cuCtxCreate()` must call :py:obj:`~.cuCtxDestroy()` when done using the context. If a context is already current to the thread, it is supplanted by the newly created context and may be restored by a subsequent call to :py:obj:`~.cuCtxPopCurrent()`. The three LSBs of the `flags` parameter can be used to control how the OS thread, which owns the CUDA context at the time of an API call, interacts with the OS scheduler when waiting for results from the GPU. Only one of the scheduling flags can be set when creating a context. - :py:obj:`~.CU_CTX_SCHED_SPIN`: Instruct CUDA to actively spin when waiting for results from the GPU. This can decrease latency when waiting for the GPU, but may lower the performance of CPU threads if they are performing work in parallel with the CUDA thread. - :py:obj:`~.CU_CTX_SCHED_YIELD`: Instruct CUDA to yield its thread when waiting for results from the GPU. This can increase latency when waiting for the GPU, but can increase the performance of CPU threads performing work in parallel with the GPU. - :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. - :py:obj:`~.CU_CTX_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. Deprecated: This flag was deprecated as of CUDA 4.0 and was replaced with :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`. - :py:obj:`~.CU_CTX_SCHED_AUTO`: The default value if the `flags` parameter is zero, uses a heuristic based on the number of active CUDA contexts in the process `C` and the number of logical processors in the system `P`. If `C` > `P`, then CUDA will yield to other OS threads when waiting for the GPU (:py:obj:`~.CU_CTX_SCHED_YIELD`), otherwise CUDA will not yield while waiting for results and actively spin on the processor (:py:obj:`~.CU_CTX_SCHED_SPIN`). Additionally, on Tegra devices, :py:obj:`~.CU_CTX_SCHED_AUTO` uses a heuristic based on the power profile of the platform and may choose :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC` for low-powered devices. - :py:obj:`~.CU_CTX_MAP_HOST`: Instruct CUDA to support mapped pinned allocations. This flag must be set in order to allocate pinned host memory that is accessible to the GPU. - :py:obj:`~.CU_CTX_LMEM_RESIZE_TO_MAX`: Instruct CUDA to not reduce local memory after resizing local memory for a kernel. This can prevent thrashing by local memory allocations when launching many kernels with high local memory usage at the cost of potentially increased memory usage. Deprecated: This flag is deprecated and the behavior enabled by this flag is now the default and cannot be disabled. Instead, the per-thread stack size can be controlled with :py:obj:`~.cuCtxSetLimit()`. - :py:obj:`~.CU_CTX_COREDUMP_ENABLE`: If GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if this context raises an exception during execution. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. The initial attributes will be taken from the global attributes at the time of context creation. The other attributes that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. - :py:obj:`~.CU_CTX_USER_COREDUMP_ENABLE`: If user-triggered GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if data is written to a certain pipe that is present in the OS space. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. It is important to note that the pipe name `must` be set with :py:obj:`~.cuCoredumpSetAttributeGlobal` before creating the context if this flag is used. Setting this flag implies that :py:obj:`~.CU_CTX_COREDUMP_ENABLE` is set. The initial attributes will be taken from the global attributes at the time of context creation. The other attributes that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. Setting this flag on any context creation is equivalent to setting the :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER` attribute to `true` globally. - :py:obj:`~.CU_CTX_SYNC_MEMOPS`: Ensures that synchronous memory operations initiated on this context will always synchronize. See further documentation in the section titled "API Synchronization behavior" to learn more about cases when synchronous memory operations can exhibit asynchronous behavior. Context creation will fail with :py:obj:`~.CUDA_ERROR_UNKNOWN` if the compute mode of the device is :py:obj:`~.CU_COMPUTEMODE_PROHIBITED`. The function :py:obj:`~.cuDeviceGetAttribute()` can be used with :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_MODE` to determine the compute mode of the device. The `nvidia-smi` tool can be used to set the compute mode for * devices. Documentation for `nvidia-smi` can be obtained by passing a -h option to it. Parameters ---------- flags : unsigned int Context creation flags dev : :py:obj:`~.CUdevice` Device to create context on Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pctx : :py:obj:`~.CUcontext` Returned context handle of the new context See Also -------- :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCoredumpSetAttributeGlobal`, :py:obj:`~.cuCoredumpSetAttribute`, :py:obj:`~.cuCtxSynchronize` Notes ----- In most cases it is recommended to use :py:obj:`~.cuDevicePrimaryCtxRetain`. """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUcontext pctx = CUcontext() err = ccuda.cuCtxCreate(pctx._ptr, flags, cdev) return (CUresult(err), pctx) {{endif}} {{if 'cuCtxCreate_v3' in found_functions}} @cython.embedsignature(True) def cuCtxCreate_v3(paramsArray : Optional[Tuple[CUexecAffinityParam] | List[CUexecAffinityParam]], int numParams, unsigned int flags, dev): """ Create a CUDA context with execution affinity. Creates a new CUDA context with execution affinity and associates it with the calling thread. The `paramsArray` and `flags` parameter are described below. The context is created with a usage count of 1 and the caller of :py:obj:`~.cuCtxCreate()` must call :py:obj:`~.cuCtxDestroy()` when done using the context. If a context is already current to the thread, it is supplanted by the newly created context and may be restored by a subsequent call to :py:obj:`~.cuCtxPopCurrent()`. The type and the amount of execution resource the context can use is limited by `paramsArray` and `numParams`. The `paramsArray` is an array of `CUexecAffinityParam` and the `numParams` describes the size of the array. If two `CUexecAffinityParam` in the array have the same type, the latter execution affinity parameter overrides the former execution affinity parameter. The supported execution affinity types are: - :py:obj:`~.CU_EXEC_AFFINITY_TYPE_SM_COUNT` limits the portion of SMs that the context can use. The portion of SMs is specified as the number of SMs via `CUexecAffinitySmCount`. This limit will be internally rounded up to the next hardware-supported amount. Hence, it is imperative to query the actual execution affinity of the context via `cuCtxGetExecAffinity` after context creation. Currently, this attribute is only supported under Volta+ MPS. The three LSBs of the `flags` parameter can be used to control how the OS thread, which owns the CUDA context at the time of an API call, interacts with the OS scheduler when waiting for results from the GPU. Only one of the scheduling flags can be set when creating a context. - :py:obj:`~.CU_CTX_SCHED_SPIN`: Instruct CUDA to actively spin when waiting for results from the GPU. This can decrease latency when waiting for the GPU, but may lower the performance of CPU threads if they are performing work in parallel with the CUDA thread. - :py:obj:`~.CU_CTX_SCHED_YIELD`: Instruct CUDA to yield its thread when waiting for results from the GPU. This can increase latency when waiting for the GPU, but can increase the performance of CPU threads performing work in parallel with the GPU. - :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. - :py:obj:`~.CU_CTX_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. Deprecated: This flag was deprecated as of CUDA 4.0 and was replaced with :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`. - :py:obj:`~.CU_CTX_SCHED_AUTO`: The default value if the `flags` parameter is zero, uses a heuristic based on the number of active CUDA contexts in the process `C` and the number of logical processors in the system `P`. If `C` > `P`, then CUDA will yield to other OS threads when waiting for the GPU (:py:obj:`~.CU_CTX_SCHED_YIELD`), otherwise CUDA will not yield while waiting for results and actively spin on the processor (:py:obj:`~.CU_CTX_SCHED_SPIN`). Additionally, on Tegra devices, :py:obj:`~.CU_CTX_SCHED_AUTO` uses a heuristic based on the power profile of the platform and may choose :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC` for low-powered devices. - :py:obj:`~.CU_CTX_MAP_HOST`: Instruct CUDA to support mapped pinned allocations. This flag must be set in order to allocate pinned host memory that is accessible to the GPU. - :py:obj:`~.CU_CTX_LMEM_RESIZE_TO_MAX`: Instruct CUDA to not reduce local memory after resizing local memory for a kernel. This can prevent thrashing by local memory allocations when launching many kernels with high local memory usage at the cost of potentially increased memory usage. Deprecated: This flag is deprecated and the behavior enabled by this flag is now the default and cannot be disabled. Instead, the per-thread stack size can be controlled with :py:obj:`~.cuCtxSetLimit()`. - :py:obj:`~.CU_CTX_COREDUMP_ENABLE`: If GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if this context raises an exception during execution. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. The initial attributes will be taken from the global attributes at the time of context creation. The other attributes that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. - :py:obj:`~.CU_CTX_USER_COREDUMP_ENABLE`: If user-triggered GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if data is written to a certain pipe that is present in the OS space. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. It is important to note that the pipe name `must` be set with :py:obj:`~.cuCoredumpSetAttributeGlobal` before creating the context if this flag is used. Setting this flag implies that :py:obj:`~.CU_CTX_COREDUMP_ENABLE` is set. The initial attributes will be taken from the global attributes at the time of context creation. The other attributes that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. Setting this flag on any context creation is equivalent to setting the :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER` attribute to `true` globally. Context creation will fail with :py:obj:`~.CUDA_ERROR_UNKNOWN` if the compute mode of the device is :py:obj:`~.CU_COMPUTEMODE_PROHIBITED`. The function :py:obj:`~.cuDeviceGetAttribute()` can be used with :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_MODE` to determine the compute mode of the device. The `nvidia-smi` tool can be used to set the compute mode for * devices. Documentation for `nvidia-smi` can be obtained by passing a -h option to it. Parameters ---------- paramsArray : List[:py:obj:`~.CUexecAffinityParam`] Execution affinity parameters numParams : int Number of execution affinity parameters flags : unsigned int Context creation flags dev : :py:obj:`~.CUdevice` Device to create context on Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_EXEC_AFFINITY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pctx : :py:obj:`~.CUcontext` Returned context handle of the new context See Also -------- :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cuCoredumpSetAttributeGlobal`, :py:obj:`~.cuCoredumpSetAttribute`, :py:obj:`~.CUexecAffinityParam` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev paramsArray = [] if paramsArray is None else paramsArray if not all(isinstance(_x, (CUexecAffinityParam,)) for _x in paramsArray): raise TypeError("Argument 'paramsArray' is not instance of type (expected Tuple[ccuda.CUexecAffinityParam,] or List[ccuda.CUexecAffinityParam,]") cdef CUcontext pctx = CUcontext() cdef ccuda.CUexecAffinityParam* cparamsArray = NULL if len(paramsArray) > 0: cparamsArray = calloc(len(paramsArray), sizeof(ccuda.CUexecAffinityParam)) if cparamsArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(paramsArray)) + 'x' + str(sizeof(ccuda.CUexecAffinityParam))) for idx in range(len(paramsArray)): string.memcpy(&cparamsArray[idx], (paramsArray[idx])._ptr, sizeof(ccuda.CUexecAffinityParam)) err = ccuda.cuCtxCreate_v3(pctx._ptr, (paramsArray[0])._ptr if len(paramsArray) == 1 else cparamsArray, numParams, flags, cdev) if cparamsArray is not NULL: free(cparamsArray) return (CUresult(err), pctx) {{endif}} {{if 'cuCtxCreate_v4' in found_functions}} @cython.embedsignature(True) def cuCtxCreate_v4(ctxCreateParams : Optional[CUctxCreateParams], unsigned int flags, dev): """ Create a CUDA context. Creates a new CUDA context and associates it with the calling thread. The `flags` parameter is described below. The context is created with a usage count of 1 and the caller of :py:obj:`~.cuCtxCreate()` must call :py:obj:`~.cuCtxDestroy()` when done using the context. If a context is already current to the thread, it is supplanted by the newly created context and may be restored by a subsequent call to :py:obj:`~.cuCtxPopCurrent()`. CUDA context can be created with execution affinity. The type and the amount of execution resource the context can use is limited by `paramsArray` and `numExecAffinityParams` in `execAffinity`. The `paramsArray` is an array of `CUexecAffinityParam` and the `numExecAffinityParams` describes the size of the paramsArray. If two `CUexecAffinityParam` in the array have the same type, the latter execution affinity parameter overrides the former execution affinity parameter. The supported execution affinity types are: - :py:obj:`~.CU_EXEC_AFFINITY_TYPE_SM_COUNT` limits the portion of SMs that the context can use. The portion of SMs is specified as the number of SMs via `CUexecAffinitySmCount`. This limit will be internally rounded up to the next hardware-supported amount. Hence, it is imperative to query the actual execution affinity of the context via `cuCtxGetExecAffinity` after context creation. Currently, this attribute is only supported under Volta+ MPS. CUDA context can be created in CIG(CUDA in Graphics) mode by setting /p cigParams. Hardware support and software support for graphics clients can be determined using :py:obj:`~.cuDeviceGetAttribute()` with :py:obj:`~.CU_DEVICE_ATTRIBUTE_D3D12_CIG_SUPPORTED`. Data from graphics client is shared with CUDA via the /p sharedData in /pcigParams. For D3D12, /p sharedData is a ID3D12CommandQueue handle. Either /p execAffinityParams or /p cigParams can be set to a non-null value. Setting both to a non-null value will result in an undefined behavior. The three LSBs of the `flags` parameter can be used to control how the OS thread, which owns the CUDA context at the time of an API call, interacts with the OS scheduler when waiting for results from the GPU. Only one of the scheduling flags can be set when creating a context. - :py:obj:`~.CU_CTX_SCHED_SPIN`: Instruct CUDA to actively spin when waiting for results from the GPU. This can decrease latency when waiting for the GPU, but may lower the performance of CPU threads if they are performing work in parallel with the CUDA thread. - :py:obj:`~.CU_CTX_SCHED_YIELD`: Instruct CUDA to yield its thread when waiting for results from the GPU. This can increase latency when waiting for the GPU, but can increase the performance of CPU threads performing work in parallel with the GPU. - :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. - :py:obj:`~.CU_CTX_BLOCKING_SYNC`: Instruct CUDA to block the CPU thread on a synchronization primitive when waiting for the GPU to finish work. Deprecated: This flag was deprecated as of CUDA 4.0 and was replaced with :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC`. - :py:obj:`~.CU_CTX_SCHED_AUTO`: The default value if the `flags` parameter is zero, uses a heuristic based on the number of active CUDA contexts in the process `C` and the number of logical processors in the system `P`. If `C` > `P`, then CUDA will yield to other OS threads when waiting for the GPU (:py:obj:`~.CU_CTX_SCHED_YIELD`), otherwise CUDA will not yield while waiting for results and actively spin on the processor (:py:obj:`~.CU_CTX_SCHED_SPIN`). Additionally, on Tegra devices, :py:obj:`~.CU_CTX_SCHED_AUTO` uses a heuristic based on the power profile of the platform and may choose :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC` for low-powered devices. - :py:obj:`~.CU_CTX_MAP_HOST`: Instruct CUDA to support mapped pinned allocations. This flag must be set in order to allocate pinned host memory that is accessible to the GPU. - :py:obj:`~.CU_CTX_LMEM_RESIZE_TO_MAX`: Instruct CUDA to not reduce local memory after resizing local memory for a kernel. This can prevent thrashing by local memory allocations when launching many kernels with high local memory usage at the cost of potentially increased memory usage. Deprecated: This flag is deprecated and the behavior enabled by this flag is now the default and cannot be disabled. Instead, the per-thread stack size can be controlled with :py:obj:`~.cuCtxSetLimit()`. - :py:obj:`~.CU_CTX_COREDUMP_ENABLE`: If GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if this context raises an exception during execution. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. The initial attributes will be taken from the global attributes at the time of context creation. The other attributes that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. This flag is not supported when CUDA context is created in CIG(CUDA in Graphics) mode. - :py:obj:`~.CU_CTX_USER_COREDUMP_ENABLE`: If user-triggered GPU coredumps have not been enabled globally with :py:obj:`~.cuCoredumpSetAttributeGlobal` or environment variables, this flag can be set during context creation to instruct CUDA to create a coredump if data is written to a certain pipe that is present in the OS space. These environment variables are described in the CUDA-GDB user guide under the "GPU core dump support" section. It is important to note that the pipe name `must` be set with :py:obj:`~.cuCoredumpSetAttributeGlobal` before creating the context if this flag is used. Setting this flag implies that :py:obj:`~.CU_CTX_COREDUMP_ENABLE` is set. The initial attributes will be taken from the global attributes at the time of context creation. The other attributes that control coredump output can be modified by calling :py:obj:`~.cuCoredumpSetAttribute` from the created context after it becomes current. Setting this flag on any context creation is equivalent to setting the :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER` attribute to `true` globally. This flag is not supported when CUDA context is created in CIG(CUDA in Graphics) mode. - :py:obj:`~.CU_CTX_SYNC_MEMOPS`: Ensures that synchronous memory operations initiated on this context will always synchronize. See further documentation in the section titled "API Synchronization behavior" to learn more about cases when synchronous memory operations can exhibit asynchronous behavior. Context creation will fail with :py:obj:`~.CUDA_ERROR_UNKNOWN` if the compute mode of the device is :py:obj:`~.CU_COMPUTEMODE_PROHIBITED`. The function :py:obj:`~.cuDeviceGetAttribute()` can be used with :py:obj:`~.CU_DEVICE_ATTRIBUTE_COMPUTE_MODE` to determine the compute mode of the device. The `nvidia-smi` tool can be used to set the compute mode for * devices. Documentation for `nvidia-smi` can be obtained by passing a -h option to it. Context creation will fail with :: CUDA_ERROR_INVALID_VALUE if invalid parameter was passed by client to create the CUDA context. Context creation in CIG mode will fail with :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` if CIG is not supported by the device or the driver. Parameters ---------- ctxCreateParams : :py:obj:`~.CUctxCreateParams` Context creation parameters flags : unsigned int Context creation flags dev : :py:obj:`~.CUdevice` Device to create context on Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pctx : :py:obj:`~.CUcontext` Returned context handle of the new context See Also -------- :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCoredumpSetAttributeGlobal`, :py:obj:`~.cuCoredumpSetAttribute`, :py:obj:`~.cuCtxSynchronize` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef CUcontext pctx = CUcontext() cdef ccuda.CUctxCreateParams* cctxCreateParams_ptr = ctxCreateParams._ptr if ctxCreateParams != None else NULL err = ccuda.cuCtxCreate_v4(pctx._ptr, cctxCreateParams_ptr, flags, cdev) return (CUresult(err), pctx) {{endif}} {{if 'cuCtxDestroy_v2' in found_functions}} @cython.embedsignature(True) def cuCtxDestroy(ctx): """ Destroy a CUDA context. Destroys the CUDA context specified by `ctx`. The context `ctx` will be destroyed regardless of how many threads it is current to. It is the responsibility of the calling function to ensure that no API call issues using `ctx` while :py:obj:`~.cuCtxDestroy()` is executing. Destroys and cleans up all resources associated with the context. It is the caller's responsibility to ensure that the context or its resources are not accessed or passed in subsequent API calls and doing so will result in undefined behavior. These resources include CUDA types :py:obj:`~.CUmodule`, :py:obj:`~.CUfunction`, :py:obj:`~.CUstream`, :py:obj:`~.CUevent`, :py:obj:`~.CUarray`, :py:obj:`~.CUmipmappedArray`, :py:obj:`~.CUtexObject`, :py:obj:`~.CUsurfObject`, :py:obj:`~.CUtexref`, :py:obj:`~.CUsurfref`, :py:obj:`~.CUgraphicsResource`, :py:obj:`~.CUlinkState`, :py:obj:`~.CUexternalMemory` and :py:obj:`~.CUexternalSemaphore`. These resources also include memory allocations by :py:obj:`~.cuMemAlloc()`, :py:obj:`~.cuMemAllocHost()`, :py:obj:`~.cuMemAllocManaged()` and :py:obj:`~.cuMemAllocPitch()`. If `ctx` is current to the calling thread then `ctx` will also be popped from the current thread's context stack (as though :py:obj:`~.cuCtxPopCurrent()` were called). If `ctx` is current to other threads, then `ctx` will remain current to those threads, and attempting to access `ctx` from those threads will result in the error :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED`. Parameters ---------- ctx : :py:obj:`~.CUcontext` Context to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` Notes ----- :py:obj:`~.cuCtxDestroy()` will not destroy memory allocations by :py:obj:`~.cuMemCreate()`, :py:obj:`~.cuMemAllocAsync()` and :py:obj:`~.cuMemAllocFromPoolAsync()`. These memory allocations are not associated with any CUDA context and need to be destroyed explicitly. """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx err = ccuda.cuCtxDestroy(cctx) return (CUresult(err),) {{endif}} {{if 'cuCtxPushCurrent_v2' in found_functions}} @cython.embedsignature(True) def cuCtxPushCurrent(ctx): """ Pushes a context on the current CPU thread. Pushes the given context `ctx` onto the CPU thread's stack of current contexts. The specified context becomes the CPU thread's current context, so all CUDA functions that operate on the current context are affected. The previous current context may be made current again by calling :py:obj:`~.cuCtxDestroy()` or :py:obj:`~.cuCtxPopCurrent()`. Parameters ---------- ctx : :py:obj:`~.CUcontext` Context to push Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx err = ccuda.cuCtxPushCurrent(cctx) return (CUresult(err),) {{endif}} {{if 'cuCtxPopCurrent_v2' in found_functions}} @cython.embedsignature(True) def cuCtxPopCurrent(): """ Pops the current CUDA context from the current CPU thread. Pops the current CUDA context from the CPU thread and passes back the old context handle in `*pctx`. That context may then be made current to a different CPU thread by calling :py:obj:`~.cuCtxPushCurrent()`. If a context was current to the CPU thread before :py:obj:`~.cuCtxCreate()` or :py:obj:`~.cuCtxPushCurrent()` was called, this function makes that context current to the CPU thread again. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` pctx : :py:obj:`~.CUcontext` Returned popped context handle See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` """ cdef CUcontext pctx = CUcontext() err = ccuda.cuCtxPopCurrent(pctx._ptr) return (CUresult(err), pctx) {{endif}} {{if 'cuCtxSetCurrent' in found_functions}} @cython.embedsignature(True) def cuCtxSetCurrent(ctx): """ Binds the specified CUDA context to the calling CPU thread. Binds the specified CUDA context to the calling CPU thread. If `ctx` is NULL then the CUDA context previously bound to the calling CPU thread is unbound and :py:obj:`~.CUDA_SUCCESS` is returned. If there exists a CUDA context stack on the calling CPU thread, this will replace the top of that stack with `ctx`. If `ctx` is NULL then this will be equivalent to popping the top of the calling CPU thread's CUDA context stack (or a no-op if the calling CPU thread's CUDA context stack is empty). Parameters ---------- ctx : :py:obj:`~.CUcontext` Context to bind to the calling CPU thread Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuCtxGetCurrent`, :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cudaSetDevice` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx err = ccuda.cuCtxSetCurrent(cctx) return (CUresult(err),) {{endif}} {{if 'cuCtxGetCurrent' in found_functions}} @cython.embedsignature(True) def cuCtxGetCurrent(): """ Returns the CUDA context bound to the calling CPU thread. Returns in `*pctx` the CUDA context bound to the calling CPU thread. If no context is bound to the calling CPU thread then `*pctx` is set to NULL and :py:obj:`~.CUDA_SUCCESS` is returned. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, pctx : :py:obj:`~.CUcontext` Returned context handle See Also -------- :py:obj:`~.cuCtxSetCurrent`, :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cudaGetDevice` """ cdef CUcontext pctx = CUcontext() err = ccuda.cuCtxGetCurrent(pctx._ptr) return (CUresult(err), pctx) {{endif}} {{if 'cuCtxGetDevice' in found_functions}} @cython.embedsignature(True) def cuCtxGetDevice(): """ Returns the device ID for the current context. Returns in `*device` the ordinal of the current context's device. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, device : :py:obj:`~.CUdevice` Returned device ID for the current context See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cudaGetDevice` """ cdef CUdevice device = CUdevice() err = ccuda.cuCtxGetDevice(device._ptr) return (CUresult(err), device) {{endif}} {{if 'cuCtxGetFlags' in found_functions}} @cython.embedsignature(True) def cuCtxGetFlags(): """ Returns the flags for the current context. Returns in `*flags` the flags of the current context. See :py:obj:`~.cuCtxCreate` for flag values. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, flags : unsigned int Pointer to store flags of current context See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetCurrent`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxGetSharedMemConfig`, :py:obj:`~.cuCtxGetStreamPriorityRange`, :py:obj:`~.cuCtxSetFlags`, :py:obj:`~.cudaGetDeviceFlags` """ cdef unsigned int flags = 0 err = ccuda.cuCtxGetFlags(&flags) return (CUresult(err), flags) {{endif}} {{if 'cuCtxSetFlags' in found_functions}} @cython.embedsignature(True) def cuCtxSetFlags(unsigned int flags): """ Sets the flags for the current context. Sets the flags for the current context overwriting previously set ones. See :py:obj:`~.cuDevicePrimaryCtxSetFlags` for flag values. Parameters ---------- flags : unsigned int Flags to set on the current context Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetCurrent`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxGetSharedMemConfig`, :py:obj:`~.cuCtxGetStreamPriorityRange`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cudaGetDeviceFlags`, :py:obj:`~.cuDevicePrimaryCtxSetFlags`, """ err = ccuda.cuCtxSetFlags(flags) return (CUresult(err),) {{endif}} {{if 'cuCtxGetId' in found_functions}} @cython.embedsignature(True) def cuCtxGetId(ctx): """ Returns the unique Id associated with the context supplied. Returns in `ctxId` the unique Id which is associated with a given context. The Id is unique for the life of the program for this instance of CUDA. If context is supplied as NULL and there is one current, the Id of the current context is returned. Parameters ---------- ctx : :py:obj:`~.CUcontext` Context for which to obtain the Id Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` ctxId : unsigned long long Pointer to store the Id of the context See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPushCurrent` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx cdef unsigned long long ctxId = 0 err = ccuda.cuCtxGetId(cctx, &ctxId) return (CUresult(err), ctxId) {{endif}} {{if 'cuCtxSynchronize' in found_functions}} @cython.embedsignature(True) def cuCtxSynchronize(): """ Block for the current context's tasks to complete. Blocks until the current context has completed all preceding requested tasks. If the current context is the primary context, green contexts that have been created will also be synchronized. :py:obj:`~.cuCtxSynchronize()` returns an error if one of the preceding tasks failed. If the context was created with the :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC` flag, the CPU thread will block until the GPU context has finished its work. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cudaDeviceSynchronize` """ err = ccuda.cuCtxSynchronize() return (CUresult(err),) {{endif}} {{if 'cuCtxSetLimit' in found_functions}} @cython.embedsignature(True) def cuCtxSetLimit(limit not None : CUlimit, size_t value): """ Set resource limits. Setting `limit` to `value` is a request by the application to update the current limit maintained by the context. The driver is free to modify the requested value to meet h/w requirements (this could be clamping to minimum or maximum values, rounding up to nearest element size, etc). The application can use :py:obj:`~.cuCtxGetLimit()` to find out exactly what the limit has been set to. Setting each :py:obj:`~.CUlimit` has its own specific restrictions, so each is discussed here. - :py:obj:`~.CU_LIMIT_STACK_SIZE` controls the stack size in bytes of each GPU thread. The driver automatically increases the per-thread stack size for each kernel launch as needed. This size isn't reset back to the original value after each launch. Setting this value will take effect immediately, and if necessary, the device will block until all preceding requested tasks are complete. - :py:obj:`~.CU_LIMIT_PRINTF_FIFO_SIZE` controls the size in bytes of the FIFO used by the :py:obj:`~.printf()` device system call. Setting :py:obj:`~.CU_LIMIT_PRINTF_FIFO_SIZE` must be performed before launching any kernel that uses the :py:obj:`~.printf()` device system call, otherwise :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. - :py:obj:`~.CU_LIMIT_MALLOC_HEAP_SIZE` controls the size in bytes of the heap used by the :py:obj:`~.malloc()` and :py:obj:`~.free()` device system calls. Setting :py:obj:`~.CU_LIMIT_MALLOC_HEAP_SIZE` must be performed before launching any kernel that uses the :py:obj:`~.malloc()` or :py:obj:`~.free()` device system calls, otherwise :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. - :py:obj:`~.CU_LIMIT_DEV_RUNTIME_SYNC_DEPTH` controls the maximum nesting depth of a grid at which a thread can safely call :py:obj:`~.cudaDeviceSynchronize()`. Setting this limit must be performed before any launch of a kernel that uses the device runtime and calls :py:obj:`~.cudaDeviceSynchronize()` above the default sync depth, two levels of grids. Calls to :py:obj:`~.cudaDeviceSynchronize()` will fail with error code :py:obj:`~.cudaErrorSyncDepthExceeded` if the limitation is violated. This limit can be set smaller than the default or up the maximum launch depth of 24. When setting this limit, keep in mind that additional levels of sync depth require the driver to reserve large amounts of device memory which can no longer be used for user allocations. If these reservations of device memory fail, :py:obj:`~.cuCtxSetLimit()` will return :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, and the limit can be reset to a lower value. This limit is only applicable to devices of compute capability < 9.0. Attempting to set this limit on devices of other compute capability versions will result in the error :py:obj:`~.CUDA_ERROR_UNSUPPORTED_LIMIT` being returned. - :py:obj:`~.CU_LIMIT_DEV_RUNTIME_PENDING_LAUNCH_COUNT` controls the maximum number of outstanding device runtime launches that can be made from the current context. A grid is outstanding from the point of launch up until the grid is known to have been completed. Device runtime launches which violate this limitation fail and return :py:obj:`~.cudaErrorLaunchPendingCountExceeded` when :py:obj:`~.cudaGetLastError()` is called after launch. If more pending launches than the default (2048 launches) are needed for a module using the device runtime, this limit can be increased. Keep in mind that being able to sustain additional pending launches will require the driver to reserve larger amounts of device memory upfront which can no longer be used for allocations. If these reservations fail, :py:obj:`~.cuCtxSetLimit()` will return :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, and the limit can be reset to a lower value. This limit is only applicable to devices of compute capability 3.5 and higher. Attempting to set this limit on devices of compute capability less than 3.5 will result in the error :py:obj:`~.CUDA_ERROR_UNSUPPORTED_LIMIT` being returned. - :py:obj:`~.CU_LIMIT_MAX_L2_FETCH_GRANULARITY` controls the L2 cache fetch granularity. Values can range from 0B to 128B. This is purely a performance hint and it can be ignored or clamped depending on the platform. - :py:obj:`~.CU_LIMIT_PERSISTING_L2_CACHE_SIZE` controls size in bytes available for persisting L2 cache. This is purely a performance hint and it can be ignored or clamped depending on the platform. Parameters ---------- limit : :py:obj:`~.CUlimit` Limit to set value : size_t Size of limit Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_LIMIT`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cudaDeviceSetLimit` """ cdef ccuda.CUlimit climit = limit.value err = ccuda.cuCtxSetLimit(climit, value) return (CUresult(err),) {{endif}} {{if 'cuCtxGetLimit' in found_functions}} @cython.embedsignature(True) def cuCtxGetLimit(limit not None : CUlimit): """ Returns resource limits. Returns in `*pvalue` the current size of `limit`. The supported :py:obj:`~.CUlimit` values are: - :py:obj:`~.CU_LIMIT_STACK_SIZE`: stack size in bytes of each GPU thread. - :py:obj:`~.CU_LIMIT_PRINTF_FIFO_SIZE`: size in bytes of the FIFO used by the :py:obj:`~.printf()` device system call. - :py:obj:`~.CU_LIMIT_MALLOC_HEAP_SIZE`: size in bytes of the heap used by the :py:obj:`~.malloc()` and :py:obj:`~.free()` device system calls. - :py:obj:`~.CU_LIMIT_DEV_RUNTIME_SYNC_DEPTH`: maximum grid depth at which a thread can issue the device runtime call :py:obj:`~.cudaDeviceSynchronize()` to wait on child grid launches to complete. - :py:obj:`~.CU_LIMIT_DEV_RUNTIME_PENDING_LAUNCH_COUNT`: maximum number of outstanding device runtime launches that can be made from this context. - :py:obj:`~.CU_LIMIT_MAX_L2_FETCH_GRANULARITY`: L2 cache fetch granularity. - :py:obj:`~.CU_LIMIT_PERSISTING_L2_CACHE_SIZE`: Persisting L2 cache size in bytes Parameters ---------- limit : :py:obj:`~.CUlimit` Limit to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_LIMIT` pvalue : int Returned size of limit See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cudaDeviceGetLimit` """ cdef size_t pvalue = 0 cdef ccuda.CUlimit climit = limit.value err = ccuda.cuCtxGetLimit(&pvalue, climit) return (CUresult(err), pvalue) {{endif}} {{if 'cuCtxGetCacheConfig' in found_functions}} @cython.embedsignature(True) def cuCtxGetCacheConfig(): """ Returns the preferred cache configuration for the current context. On devices where the L1 cache and shared memory use the same hardware resources, this function returns through `pconfig` the preferred cache configuration for the current context. This is only a preference. The driver will use the requested configuration if possible, but it is free to choose a different configuration if required to execute functions. This will return a `pconfig` of :py:obj:`~.CU_FUNC_CACHE_PREFER_NONE` on devices where the size of the L1 cache and shared memory are fixed. The supported cache configurations are: - :py:obj:`~.CU_FUNC_CACHE_PREFER_NONE`: no preference for shared memory or L1 (default) - :py:obj:`~.CU_FUNC_CACHE_PREFER_SHARED`: prefer larger shared memory and smaller L1 cache - :py:obj:`~.CU_FUNC_CACHE_PREFER_L1`: prefer larger L1 cache and smaller shared memory - :py:obj:`~.CU_FUNC_CACHE_PREFER_EQUAL`: prefer equal sized L1 cache and shared memory Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pconfig : :py:obj:`~.CUfunc_cache` Returned cache configuration See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cudaDeviceGetCacheConfig` """ cdef ccuda.CUfunc_cache pconfig err = ccuda.cuCtxGetCacheConfig(&pconfig) return (CUresult(err), CUfunc_cache(pconfig)) {{endif}} {{if 'cuCtxSetCacheConfig' in found_functions}} @cython.embedsignature(True) def cuCtxSetCacheConfig(config not None : CUfunc_cache): """ Sets the preferred cache configuration for the current context. On devices where the L1 cache and shared memory use the same hardware resources, this sets through `config` the preferred cache configuration for the current context. This is only a preference. The driver will use the requested configuration if possible, but it is free to choose a different configuration if required to execute the function. Any function preference set via :py:obj:`~.cuFuncSetCacheConfig()` or :py:obj:`~.cuKernelSetCacheConfig()` will be preferred over this context-wide setting. Setting the context-wide cache configuration to :py:obj:`~.CU_FUNC_CACHE_PREFER_NONE` will cause subsequent kernel launches to prefer to not change the cache configuration unless required to launch the kernel. This setting does nothing on devices where the size of the L1 cache and shared memory are fixed. Launching a kernel with a different preference than the most recent preference setting may insert a device-side synchronization point. The supported cache configurations are: - :py:obj:`~.CU_FUNC_CACHE_PREFER_NONE`: no preference for shared memory or L1 (default) - :py:obj:`~.CU_FUNC_CACHE_PREFER_SHARED`: prefer larger shared memory and smaller L1 cache - :py:obj:`~.CU_FUNC_CACHE_PREFER_L1`: prefer larger L1 cache and smaller shared memory - :py:obj:`~.CU_FUNC_CACHE_PREFER_EQUAL`: prefer equal sized L1 cache and shared memory Parameters ---------- config : :py:obj:`~.CUfunc_cache` Requested cache configuration Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cudaDeviceSetCacheConfig`, :py:obj:`~.cuKernelSetCacheConfig` """ cdef ccuda.CUfunc_cache cconfig = config.value err = ccuda.cuCtxSetCacheConfig(cconfig) return (CUresult(err),) {{endif}} {{if 'cuCtxGetApiVersion' in found_functions}} @cython.embedsignature(True) def cuCtxGetApiVersion(ctx): """ Gets the context's API version. Returns a version number in `version` corresponding to the capabilities of the context (e.g. 3010 or 3020), which library developers can use to direct callers to a specific API version. If `ctx` is NULL, returns the API version used to create the currently bound context. Note that new API versions are only introduced when context capabilities are changed that break binary compatibility, so the API version and driver version may be different. For example, it is valid for the API version to be 3020 while the driver version is 4020. Parameters ---------- ctx : :py:obj:`~.CUcontext` Context to check Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` version : unsigned int Pointer to version See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx cdef unsigned int version = 0 err = ccuda.cuCtxGetApiVersion(cctx, &version) return (CUresult(err), version) {{endif}} {{if 'cuCtxGetStreamPriorityRange' in found_functions}} @cython.embedsignature(True) def cuCtxGetStreamPriorityRange(): """ Returns numerical values that correspond to the least and greatest stream priorities. Returns in `*leastPriority` and `*greatestPriority` the numerical values that correspond to the least and greatest stream priorities respectively. Stream priorities follow a convention where lower numbers imply greater priorities. The range of meaningful stream priorities is given by [`*greatestPriority`, `*leastPriority`]. If the user attempts to create a stream with a priority value that is outside the meaningful range as specified by this API, the priority is automatically clamped down or up to either `*leastPriority` or `*greatestPriority` respectively. See :py:obj:`~.cuStreamCreateWithPriority` for details on creating a priority stream. A NULL may be passed in for `*leastPriority` or `*greatestPriority` if the value is not desired. This function will return '0' in both `*leastPriority` and `*greatestPriority` if the current context's device does not support stream priorities (see :py:obj:`~.cuDeviceGetAttribute`). Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, leastPriority : int Pointer to an int in which the numerical value for least stream priority is returned greatestPriority : int Pointer to an int in which the numerical value for greatest stream priority is returned See Also -------- :py:obj:`~.cuStreamCreateWithPriority`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cudaDeviceGetStreamPriorityRange` """ cdef int leastPriority = 0 cdef int greatestPriority = 0 err = ccuda.cuCtxGetStreamPriorityRange(&leastPriority, &greatestPriority) return (CUresult(err), leastPriority, greatestPriority) {{endif}} {{if 'cuCtxResetPersistingL2Cache' in found_functions}} @cython.embedsignature(True) def cuCtxResetPersistingL2Cache(): """ Resets all persisting lines in cache to normal status. :py:obj:`~.cuCtxResetPersistingL2Cache` Resets all persisting lines in cache to normal status. Takes effect on function return. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.CUaccessPolicyWindow` """ err = ccuda.cuCtxResetPersistingL2Cache() return (CUresult(err),) {{endif}} {{if 'cuCtxGetExecAffinity' in found_functions}} @cython.embedsignature(True) def cuCtxGetExecAffinity(typename not None : CUexecAffinityType): """ Returns the execution affinity setting for the current context. Returns in `*pExecAffinity` the current value of `typename`. The supported :py:obj:`~.CUexecAffinityType` values are: - :py:obj:`~.CU_EXEC_AFFINITY_TYPE_SM_COUNT`: number of SMs the context is limited to use. Parameters ---------- typename : :py:obj:`~.CUexecAffinityType` Execution affinity type to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_EXEC_AFFINITY` pExecAffinity : :py:obj:`~.CUexecAffinityParam` Returned execution affinity See Also -------- :py:obj:`~.CUexecAffinityParam` """ cdef CUexecAffinityParam pExecAffinity = CUexecAffinityParam() cdef ccuda.CUexecAffinityType ctypename = typename.value err = ccuda.cuCtxGetExecAffinity(pExecAffinity._ptr, ctypename) return (CUresult(err), pExecAffinity) {{endif}} {{if 'cuCtxRecordEvent' in found_functions}} @cython.embedsignature(True) def cuCtxRecordEvent(hCtx, hEvent): """ Records an event. Captures in `hEvent` all the activities of the context `hCtx` at the time of this call. `hEvent` and `hCtx` must be from the same CUDA context, otherwise :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` will be returned. Calls such as :py:obj:`~.cuEventQuery()` or :py:obj:`~.cuCtxWaitEvent()` will then examine or wait for completion of the work that was captured. Uses of `hCtx` after this call do not modify `hEvent`. If the context passed to `hCtx` is the primary context, `hEvent` will capture all the activities of the primary context and its green contexts. If the context passed to `hCtx` is a context converted from green context via :py:obj:`~.cuCtxFromGreenCtx()`, `hEvent` will capture only the activities of the green context. Parameters ---------- hCtx : :py:obj:`~.CUcontext` Context to record event for hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to record Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` See Also -------- :py:obj:`~.cuCtxWaitEvent`, :py:obj:`~.cuGreenCtxRecordEvent`, :py:obj:`~.cuGreenCtxWaitEvent`, :py:obj:`~.cuEventRecord` Notes ----- The API will return :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` if the specified context `hCtx` has a stream in the capture mode. In such a case, the call will invalidate all the conflicting captures. """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent cdef ccuda.CUcontext chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUcontext,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUcontext(hCtx)) chCtx = phCtx err = ccuda.cuCtxRecordEvent(chCtx, chEvent) return (CUresult(err),) {{endif}} {{if 'cuCtxWaitEvent' in found_functions}} @cython.embedsignature(True) def cuCtxWaitEvent(hCtx, hEvent): """ Make a context wait on an event. Makes all future work submitted to context `hCtx` wait for all work captured in `hEvent`. The synchronization will be performed on the device and will not block the calling CPU thread. See :py:obj:`~.cuCtxRecordEvent()` for details on what is captured by an event. If the context passed to `hCtx` is the primary context, the primary context and its green contexts will wait for `hEvent`. If the context passed to `hCtx` is a context converted from green context via :py:obj:`~.cuCtxFromGreenCtx()`, the green context will wait for `hEvent`. Parameters ---------- hCtx : :py:obj:`~.CUcontext` Context to wait hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to wait on Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` See Also -------- :py:obj:`~.cuCtxRecordEvent`, :py:obj:`~.cuGreenCtxRecordEvent`, :py:obj:`~.cuGreenCtxWaitEvent`, :py:obj:`~.cuStreamWaitEvent` Notes ----- `hEvent` may be from a different context or device than `hCtx`. The API will return :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` and invalidate the capture if the specified event `hEvent` is part of an ongoing capture sequence or if the specified context `hCtx` has a stream in the capture mode. """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent cdef ccuda.CUcontext chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUcontext,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUcontext(hCtx)) chCtx = phCtx err = ccuda.cuCtxWaitEvent(chCtx, chEvent) return (CUresult(err),) {{endif}} {{if 'cuCtxAttach' in found_functions}} @cython.embedsignature(True) def cuCtxAttach(unsigned int flags): """ Increment a context's usage-count. [Deprecated] Note that this function is deprecated and should not be used. Increments the usage count of the context and passes back a context handle in `*pctx` that must be passed to :py:obj:`~.cuCtxDetach()` when the application is done with the context. :py:obj:`~.cuCtxAttach()` fails if there is no context current to the thread. Currently, the `flags` parameter must be 0. Parameters ---------- flags : unsigned int Context attach flags (must be 0) Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pctx : :py:obj:`~.CUcontext` Returned context handle of the current context See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxDetach`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` """ cdef CUcontext pctx = CUcontext() err = ccuda.cuCtxAttach(pctx._ptr, flags) return (CUresult(err), pctx) {{endif}} {{if 'cuCtxDetach' in found_functions}} @cython.embedsignature(True) def cuCtxDetach(ctx): """ Decrement a context's usage-count. [Deprecated] Note that this function is deprecated and should not be used. Decrements the usage count of the context `ctx`, and destroys the context if the usage count goes to 0. The context must be a handle that was passed back by :py:obj:`~.cuCtxCreate()` or :py:obj:`~.cuCtxAttach()`, and must be current to the calling thread. Parameters ---------- ctx : :py:obj:`~.CUcontext` Context to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx err = ccuda.cuCtxDetach(cctx) return (CUresult(err),) {{endif}} {{if 'cuCtxGetSharedMemConfig' in found_functions}} @cython.embedsignature(True) def cuCtxGetSharedMemConfig(): """ Returns the current shared memory configuration for the current context. [Deprecated] This function will return in `pConfig` the current size of shared memory banks in the current context. On devices with configurable shared memory banks, :py:obj:`~.cuCtxSetSharedMemConfig` can be used to change this setting, so that all subsequent kernel launches will by default use the new bank size. When :py:obj:`~.cuCtxGetSharedMemConfig` is called on devices without configurable shared memory, it will return the fixed bank size of the hardware. The returned bank configurations can be either: - :py:obj:`~.CU_SHARED_MEM_CONFIG_FOUR_BYTE_BANK_SIZE`: shared memory bank width is four bytes. - :py:obj:`~.CU_SHARED_MEM_CONFIG_EIGHT_BYTE_BANK_SIZE`: shared memory bank width will eight bytes. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pConfig : :py:obj:`~.CUsharedconfig` returned shared memory configuration See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cuCtxGetSharedMemConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cudaDeviceGetSharedMemConfig` """ cdef ccuda.CUsharedconfig pConfig err = ccuda.cuCtxGetSharedMemConfig(&pConfig) return (CUresult(err), CUsharedconfig(pConfig)) {{endif}} {{if 'cuCtxSetSharedMemConfig' in found_functions}} @cython.embedsignature(True) def cuCtxSetSharedMemConfig(config not None : CUsharedconfig): """ Sets the shared memory configuration for the current context. [Deprecated] On devices with configurable shared memory banks, this function will set the context's shared memory bank size which is used for subsequent kernel launches. Changed the shared memory configuration between launches may insert a device side synchronization point between those launches. Changing the shared memory bank size will not increase shared memory usage or affect occupancy of kernels, but may have major effects on performance. Larger bank sizes will allow for greater potential bandwidth to shared memory, but will change what kinds of accesses to shared memory will result in bank conflicts. This function will do nothing on devices with fixed shared memory bank size. The supported bank configurations are: - :py:obj:`~.CU_SHARED_MEM_CONFIG_DEFAULT_BANK_SIZE`: set bank width to the default initial setting (currently, four bytes). - :py:obj:`~.CU_SHARED_MEM_CONFIG_FOUR_BYTE_BANK_SIZE`: set shared memory bank width to be natively four bytes. - :py:obj:`~.CU_SHARED_MEM_CONFIG_EIGHT_BYTE_BANK_SIZE`: set shared memory bank width to be natively eight bytes. Parameters ---------- config : :py:obj:`~.CUsharedconfig` requested shared memory configuration Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxDestroy`, :py:obj:`~.cuCtxGetApiVersion`, :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxGetDevice`, :py:obj:`~.cuCtxGetFlags`, :py:obj:`~.cuCtxGetLimit`, :py:obj:`~.cuCtxPopCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuCtxSetLimit`, :py:obj:`~.cuCtxSynchronize`, :py:obj:`~.cuCtxGetSharedMemConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cudaDeviceSetSharedMemConfig` """ cdef ccuda.CUsharedconfig cconfig = config.value err = ccuda.cuCtxSetSharedMemConfig(cconfig) return (CUresult(err),) {{endif}} {{if 'cuModuleLoad' in found_functions}} @cython.embedsignature(True) def cuModuleLoad(char* fname): """ Loads a compute module. Takes a filename `fname` and loads the corresponding module `module` into the current context. The CUDA driver API does not attempt to lazily allocate the resources needed by a module; if the memory for functions and data (constant and global) needed by the module cannot be allocated, :py:obj:`~.cuModuleLoad()` fails. The file should be a `cubin` file as output by nvcc, or a `PTX` file either as output by nvcc or handwritten, or a `fatbin` file as output by nvcc from toolchain 4.0 or later. Parameters ---------- fname : bytes Filename of module to load Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_FILE_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_JIT_COMPILER_NOT_FOUND` module : :py:obj:`~.CUmodule` Returned module See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleLoadFatBinary`, :py:obj:`~.cuModuleUnload` """ cdef CUmodule module = CUmodule() err = ccuda.cuModuleLoad(module._ptr, fname) return (CUresult(err), module) {{endif}} {{if 'cuModuleLoadData' in found_functions}} @cython.embedsignature(True) def cuModuleLoadData(image): """ Load a module's data. Takes a pointer `image` and loads the corresponding module `module` into the current context. The `image` may be a `cubin` or `fatbin` as output by nvcc, or a NULL-terminated `PTX`, either as output by nvcc or hand-written. Parameters ---------- image : Any Module data to load Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_JIT_COMPILER_NOT_FOUND` module : :py:obj:`~.CUmodule` Returned module See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleLoadFatBinary`, :py:obj:`~.cuModuleUnload` """ cdef CUmodule module = CUmodule() cimage = utils.HelperInputVoidPtr(image) cdef void* cimage_ptr = cimage.cptr err = ccuda.cuModuleLoadData(module._ptr, cimage_ptr) return (CUresult(err), module) {{endif}} {{if 'cuModuleLoadDataEx' in found_functions}} @cython.embedsignature(True) def cuModuleLoadDataEx(image, unsigned int numOptions, options : Optional[Tuple[CUjit_option] | List[CUjit_option]], optionValues : Optional[Tuple[Any] | List[Any]]): """ Load a module's data with options. Takes a pointer `image` and loads the corresponding module `module` into the current context. The `image` may be a `cubin` or `fatbin` as output by nvcc, or a NULL-terminated `PTX`, either as output by nvcc or hand-written. Parameters ---------- image : Any Module data to load numOptions : unsigned int Number of options options : List[:py:obj:`~.CUjit_option`] Options for JIT optionValues : List[Any] Option values for JIT Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_JIT_COMPILER_NOT_FOUND` module : :py:obj:`~.CUmodule` Returned module See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadFatBinary`, :py:obj:`~.cuModuleUnload` """ optionValues = [] if optionValues is None else optionValues options = [] if options is None else options if not all(isinstance(_x, (CUjit_option)) for _x in options): raise TypeError("Argument 'options' is not instance of type (expected Tuple[ccuda.CUjit_option] or List[ccuda.CUjit_option]") cdef CUmodule module = CUmodule() cimage = utils.HelperInputVoidPtr(image) cdef void* cimage_ptr = cimage.cptr if numOptions > len(options): raise RuntimeError("List is too small: " + str(len(options)) + " < " + str(numOptions)) if numOptions > len(optionValues): raise RuntimeError("List is too small: " + str(len(optionValues)) + " < " + str(numOptions)) cdef vector[ccuda.CUjit_option] coptions = [pyoptions.value for pyoptions in (options)] pylist = [utils.HelperCUjit_option(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(options, optionValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperoptionValues = utils.InputVoidPtrPtrHelper(pylist) err = ccuda.cuModuleLoadDataEx(module._ptr, cimage_ptr, numOptions, coptions.data(), voidStarHelperoptionValues.cptr) return (CUresult(err), module) {{endif}} {{if 'cuModuleLoadFatBinary' in found_functions}} @cython.embedsignature(True) def cuModuleLoadFatBinary(fatCubin): """ Load a module's data. Takes a pointer `fatCubin` and loads the corresponding module `module` into the current context. The pointer represents a `fat binary` object, which is a collection of different `cubin` and/or `PTX` files, all representing the same device code, but compiled and optimized for different architectures. Prior to CUDA 4.0, there was no documented API for constructing and using fat binary objects by programmers. Starting with CUDA 4.0, fat binary objects can be constructed by providing the `-fatbin option` to nvcc. More information can be found in the nvcc document. Parameters ---------- fatCubin : Any Fat binary to load Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_JIT_COMPILER_NOT_FOUND` module : :py:obj:`~.CUmodule` Returned module See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleUnload` """ cdef CUmodule module = CUmodule() cfatCubin = utils.HelperInputVoidPtr(fatCubin) cdef void* cfatCubin_ptr = cfatCubin.cptr err = ccuda.cuModuleLoadFatBinary(module._ptr, cfatCubin_ptr) return (CUresult(err), module) {{endif}} {{if 'cuModuleUnload' in found_functions}} @cython.embedsignature(True) def cuModuleUnload(hmod): """ Unloads a module. Unloads a module `hmod` from the current context. Attempting to unload a module which was obtained from the Library Management API such as :py:obj:`~.cuLibraryGetModule` will return :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`. Parameters ---------- hmod : :py:obj:`~.CUmodule` Module to unload Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED` See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleLoadFatBinary` """ cdef ccuda.CUmodule chmod if hmod is None: chmod = 0 elif isinstance(hmod, (CUmodule,)): phmod = int(hmod) chmod = phmod else: phmod = int(CUmodule(hmod)) chmod = phmod err = ccuda.cuModuleUnload(chmod) return (CUresult(err),) {{endif}} {{if 'cuModuleGetLoadingMode' in found_functions}} @cython.embedsignature(True) def cuModuleGetLoadingMode(): """ Query lazy loading mode. Returns lazy loading mode Module loading mode is controlled by CUDA_MODULE_LOADING env variable Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, mode : :py:obj:`~.CUmoduleLoadingMode` Returns the lazy loading mode See Also -------- :py:obj:`~.cuModuleLoad`, """ cdef ccuda.CUmoduleLoadingMode mode err = ccuda.cuModuleGetLoadingMode(&mode) return (CUresult(err), CUmoduleLoadingMode(mode)) {{endif}} {{if 'cuModuleGetFunction' in found_functions}} @cython.embedsignature(True) def cuModuleGetFunction(hmod, char* name): """ Returns a function handle. Returns in `*hfunc` the handle of the function of name `name` located in module `hmod`. If no function of that name exists, :py:obj:`~.cuModuleGetFunction()` returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. Parameters ---------- hmod : :py:obj:`~.CUmodule` Module to retrieve function from name : bytes Name of function to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` hfunc : :py:obj:`~.CUfunction` Returned function handle See Also -------- :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleLoadFatBinary`, :py:obj:`~.cuModuleUnload` """ cdef ccuda.CUmodule chmod if hmod is None: chmod = 0 elif isinstance(hmod, (CUmodule,)): phmod = int(hmod) chmod = phmod else: phmod = int(CUmodule(hmod)) chmod = phmod cdef CUfunction hfunc = CUfunction() err = ccuda.cuModuleGetFunction(hfunc._ptr, chmod, name) return (CUresult(err), hfunc) {{endif}} {{if 'cuModuleGetFunctionCount' in found_functions}} @cython.embedsignature(True) def cuModuleGetFunctionCount(mod): """ Returns the number of functions within a module. Returns in `count` the number of functions in `mod`. Parameters ---------- mod : :py:obj:`~.CUmodule` Module to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` count : unsigned int Number of functions found within the module """ cdef ccuda.CUmodule cmod if mod is None: cmod = 0 elif isinstance(mod, (CUmodule,)): pmod = int(mod) cmod = pmod else: pmod = int(CUmodule(mod)) cmod = pmod cdef unsigned int count = 0 err = ccuda.cuModuleGetFunctionCount(&count, cmod) return (CUresult(err), count) {{endif}} {{if 'cuModuleEnumerateFunctions' in found_functions}} @cython.embedsignature(True) def cuModuleEnumerateFunctions(unsigned int numFunctions, mod): """ Returns the function handles within a module. Returns in `functions` a maximum number of `numFunctions` function handles within `mod`. When function loading mode is set to LAZY the function retrieved may be partially loaded. The loading state of a function can be queried using :py:obj:`~.cuFunctionIsLoaded`. CUDA APIs may load the function automatically when called with partially loaded function handle which may incur additional latency. Alternatively, :py:obj:`~.cuFunctionLoad` can be used to explicitly load a function. The returned function handles become invalid when the module is unloaded. Parameters ---------- numFunctions : unsigned int Maximum number of function handles may be returned to the buffer mod : :py:obj:`~.CUmodule` Module to query from Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` functions : List[:py:obj:`~.CUfunction`] Buffer where the function handles are returned to See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetFunctionCount`, :py:obj:`~.cuFuncIsLoaded`, :py:obj:`~.cuFuncLoad` """ cdef ccuda.CUmodule cmod if mod is None: cmod = 0 elif isinstance(mod, (CUmodule,)): pmod = int(mod) cmod = pmod else: pmod = int(CUmodule(mod)) cmod = pmod cdef ccuda.CUfunction* cfunctions = NULL pyfunctions = [] if numFunctions != 0: cfunctions = calloc(numFunctions, sizeof(ccuda.CUfunction)) if cfunctions is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(numFunctions) + 'x' + str(sizeof(ccuda.CUfunction))) err = ccuda.cuModuleEnumerateFunctions(cfunctions, numFunctions, cmod) if CUresult(err) == CUresult(0): pyfunctions = [CUfunction(init_value=cfunctions[idx]) for idx in range(numFunctions)] if cfunctions is not NULL: free(cfunctions) return (CUresult(err), pyfunctions) {{endif}} {{if 'cuModuleGetGlobal_v2' in found_functions}} @cython.embedsignature(True) def cuModuleGetGlobal(hmod, char* name): """ Returns a global pointer from a module. Returns in `*dptr` and `*bytes` the base pointer and size of the global of name `name` located in module `hmod`. If no variable of that name exists, :py:obj:`~.cuModuleGetGlobal()` returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. One of the parameters `dptr` or `numbytes` (not both) can be NULL in which case it is ignored. Parameters ---------- hmod : :py:obj:`~.CUmodule` Module to retrieve global from name : bytes Name of global to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` dptr : :py:obj:`~.CUdeviceptr` Returned global device pointer numbytes : int Returned global size in bytes See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleLoadFatBinary`, :py:obj:`~.cuModuleUnload`, :py:obj:`~.cudaGetSymbolAddress`, :py:obj:`~.cudaGetSymbolSize` """ cdef ccuda.CUmodule chmod if hmod is None: chmod = 0 elif isinstance(hmod, (CUmodule,)): phmod = int(hmod) chmod = phmod else: phmod = int(CUmodule(hmod)) chmod = phmod cdef CUdeviceptr dptr = CUdeviceptr() cdef size_t numbytes = 0 err = ccuda.cuModuleGetGlobal(dptr._ptr, &numbytes, chmod, name) return (CUresult(err), dptr, numbytes) {{endif}} {{if 'cuLinkCreate_v2' in found_functions}} @cython.embedsignature(True) def cuLinkCreate(unsigned int numOptions, options : Optional[Tuple[CUjit_option] | List[CUjit_option]], optionValues : Optional[Tuple[Any] | List[Any]]): """ Creates a pending JIT linker invocation. If the call is successful, the caller owns the returned CUlinkState, which should eventually be destroyed with :py:obj:`~.cuLinkDestroy`. The device code machine size (32 or 64 bit) will match the calling application. Both linker and compiler options may be specified. Compiler options will be applied to inputs to this linker action which must be compiled from PTX. The options :py:obj:`~.CU_JIT_WALL_TIME`, :py:obj:`~.CU_JIT_INFO_LOG_BUFFER_SIZE_BYTES`, and :py:obj:`~.CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES` will accumulate data until the CUlinkState is destroyed. The data passed in via :py:obj:`~.cuLinkAddData` and :py:obj:`~.cuLinkAddFile` will be treated as relocatable (-rdc=true to nvcc) when linking the final cubin during :py:obj:`~.cuLinkComplete` and will have similar consequences as offline relocatable device code linking. `optionValues` must remain valid for the life of the CUlinkState if output options are used. No other references to inputs are maintained after this call returns. Parameters ---------- numOptions : unsigned int Size of options arrays options : List[:py:obj:`~.CUjit_option`] Array of linker and compiler options optionValues : List[Any] Array of option values, each cast to void * Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_JIT_COMPILER_NOT_FOUND` stateOut : :py:obj:`~.CUlinkState` On success, this will contain a CUlinkState to specify and complete this action See Also -------- :py:obj:`~.cuLinkAddData`, :py:obj:`~.cuLinkAddFile`, :py:obj:`~.cuLinkComplete`, :py:obj:`~.cuLinkDestroy` Notes ----- For LTO-IR input, only LTO-IR compiled with toolkits prior to CUDA 12.0 will be accepted """ optionValues = [] if optionValues is None else optionValues options = [] if options is None else options if not all(isinstance(_x, (CUjit_option)) for _x in options): raise TypeError("Argument 'options' is not instance of type (expected Tuple[ccuda.CUjit_option] or List[ccuda.CUjit_option]") if numOptions > len(options): raise RuntimeError("List is too small: " + str(len(options)) + " < " + str(numOptions)) if numOptions > len(optionValues): raise RuntimeError("List is too small: " + str(len(optionValues)) + " < " + str(numOptions)) cdef vector[ccuda.CUjit_option] coptions = [pyoptions.value for pyoptions in (options)] pylist = [utils.HelperCUjit_option(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(options, optionValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperoptionValues = utils.InputVoidPtrPtrHelper(pylist) cdef CUlinkState stateOut = CUlinkState() err = ccuda.cuLinkCreate(numOptions, coptions.data(), voidStarHelperoptionValues.cptr, stateOut._ptr) stateOut._keepalive.append(voidStarHelperoptionValues) for option in pylist: stateOut._keepalive.append(option) return (CUresult(err), stateOut) {{endif}} {{if 'cuLinkAddData_v2' in found_functions}} @cython.embedsignature(True) def cuLinkAddData(state, typename not None : CUjitInputType, data, size_t size, char* name, unsigned int numOptions, options : Optional[Tuple[CUjit_option] | List[CUjit_option]], optionValues : Optional[Tuple[Any] | List[Any]]): """ Add an input to a pending linker invocation. Ownership of `data` is retained by the caller. No reference is retained to any inputs after this call returns. This method accepts only compiler options, which are used if the data must be compiled from PTX, and does not accept any of :py:obj:`~.CU_JIT_WALL_TIME`, :py:obj:`~.CU_JIT_INFO_LOG_BUFFER`, :py:obj:`~.CU_JIT_ERROR_LOG_BUFFER`, :py:obj:`~.CU_JIT_TARGET_FROM_CUCONTEXT`, or :py:obj:`~.CU_JIT_TARGET`. Parameters ---------- state : :py:obj:`~.CUlinkState` A pending linker action. typename : :py:obj:`~.CUjitInputType` The type of the input data. data : Any The input data. PTX must be NULL-terminated. size : size_t The length of the input data. name : bytes An optional name for this input in log messages. numOptions : unsigned int Size of options. options : List[:py:obj:`~.CUjit_option`] Options to be applied only for this input (overrides options from :py:obj:`~.cuLinkCreate`). optionValues : List[Any] Array of option values, each cast to void *. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU` See Also -------- :py:obj:`~.cuLinkCreate`, :py:obj:`~.cuLinkAddFile`, :py:obj:`~.cuLinkComplete`, :py:obj:`~.cuLinkDestroy` Notes ----- For LTO-IR input, only LTO-IR compiled with toolkits prior to CUDA 12.0 will be accepted """ optionValues = [] if optionValues is None else optionValues options = [] if options is None else options if not all(isinstance(_x, (CUjit_option)) for _x in options): raise TypeError("Argument 'options' is not instance of type (expected Tuple[ccuda.CUjit_option] or List[ccuda.CUjit_option]") cdef ccuda.CUlinkState cstate if state is None: cstate = 0 elif isinstance(state, (CUlinkState,)): pstate = int(state) cstate = pstate else: pstate = int(CUlinkState(state)) cstate = pstate cdef ccuda.CUjitInputType ctypename = typename.value cdata = utils.HelperInputVoidPtr(data) cdef void* cdata_ptr = cdata.cptr if numOptions > len(options): raise RuntimeError("List is too small: " + str(len(options)) + " < " + str(numOptions)) if numOptions > len(optionValues): raise RuntimeError("List is too small: " + str(len(optionValues)) + " < " + str(numOptions)) cdef vector[ccuda.CUjit_option] coptions = [pyoptions.value for pyoptions in (options)] pylist = [utils.HelperCUjit_option(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(options, optionValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperoptionValues = utils.InputVoidPtrPtrHelper(pylist) err = ccuda.cuLinkAddData(cstate, ctypename, cdata_ptr, size, name, numOptions, coptions.data(), voidStarHelperoptionValues.cptr) return (CUresult(err),) {{endif}} {{if 'cuLinkAddFile_v2' in found_functions}} @cython.embedsignature(True) def cuLinkAddFile(state, typename not None : CUjitInputType, char* path, unsigned int numOptions, options : Optional[Tuple[CUjit_option] | List[CUjit_option]], optionValues : Optional[Tuple[Any] | List[Any]]): """ Add a file input to a pending linker invocation. No reference is retained to any inputs after this call returns. This method accepts only compiler options, which are used if the input must be compiled from PTX, and does not accept any of :py:obj:`~.CU_JIT_WALL_TIME`, :py:obj:`~.CU_JIT_INFO_LOG_BUFFER`, :py:obj:`~.CU_JIT_ERROR_LOG_BUFFER`, :py:obj:`~.CU_JIT_TARGET_FROM_CUCONTEXT`, or :py:obj:`~.CU_JIT_TARGET`. This method is equivalent to invoking :py:obj:`~.cuLinkAddData` on the contents of the file. Parameters ---------- state : :py:obj:`~.CUlinkState` A pending linker action typename : :py:obj:`~.CUjitInputType` The type of the input data path : bytes Path to the input file numOptions : unsigned int Size of options options : List[:py:obj:`~.CUjit_option`] Options to be applied only for this input (overrides options from :py:obj:`~.cuLinkCreate`) optionValues : List[Any] Array of option values, each cast to void * Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_FILE_NOT_FOUND` :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU` See Also -------- :py:obj:`~.cuLinkCreate`, :py:obj:`~.cuLinkAddData`, :py:obj:`~.cuLinkComplete`, :py:obj:`~.cuLinkDestroy` Notes ----- For LTO-IR input, only LTO-IR compiled with toolkits prior to CUDA 12.0 will be accepted """ optionValues = [] if optionValues is None else optionValues options = [] if options is None else options if not all(isinstance(_x, (CUjit_option)) for _x in options): raise TypeError("Argument 'options' is not instance of type (expected Tuple[ccuda.CUjit_option] or List[ccuda.CUjit_option]") cdef ccuda.CUlinkState cstate if state is None: cstate = 0 elif isinstance(state, (CUlinkState,)): pstate = int(state) cstate = pstate else: pstate = int(CUlinkState(state)) cstate = pstate cdef ccuda.CUjitInputType ctypename = typename.value if numOptions > len(options): raise RuntimeError("List is too small: " + str(len(options)) + " < " + str(numOptions)) if numOptions > len(optionValues): raise RuntimeError("List is too small: " + str(len(optionValues)) + " < " + str(numOptions)) cdef vector[ccuda.CUjit_option] coptions = [pyoptions.value for pyoptions in (options)] pylist = [utils.HelperCUjit_option(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(options, optionValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperoptionValues = utils.InputVoidPtrPtrHelper(pylist) err = ccuda.cuLinkAddFile(cstate, ctypename, path, numOptions, coptions.data(), voidStarHelperoptionValues.cptr) return (CUresult(err),) {{endif}} {{if 'cuLinkComplete' in found_functions}} @cython.embedsignature(True) def cuLinkComplete(state): """ Complete a pending linker invocation. Completes the pending linker action and returns the cubin image for the linked device code, which can be used with :py:obj:`~.cuModuleLoadData`. The cubin is owned by `state`, so it should be loaded before `state` is destroyed via :py:obj:`~.cuLinkDestroy`. This call does not destroy `state`. Parameters ---------- state : :py:obj:`~.CUlinkState` A pending linker invocation Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` cubinOut : Any On success, this will point to the output image sizeOut : int Optional parameter to receive the size of the generated image See Also -------- :py:obj:`~.cuLinkCreate`, :py:obj:`~.cuLinkAddData`, :py:obj:`~.cuLinkAddFile`, :py:obj:`~.cuLinkDestroy`, :py:obj:`~.cuModuleLoadData` """ cdef ccuda.CUlinkState cstate if state is None: cstate = 0 elif isinstance(state, (CUlinkState,)): pstate = int(state) cstate = pstate else: pstate = int(CUlinkState(state)) cstate = pstate cdef void_ptr cubinOut = 0 cdef size_t sizeOut = 0 err = ccuda.cuLinkComplete(cstate, &cubinOut, &sizeOut) return (CUresult(err), cubinOut, sizeOut) {{endif}} {{if 'cuLinkDestroy' in found_functions}} @cython.embedsignature(True) def cuLinkDestroy(state): """ Destroys state for a JIT linker invocation. Parameters ---------- state : :py:obj:`~.CUlinkState` State object for the linker invocation Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuLinkCreate` """ cdef ccuda.CUlinkState cstate if state is None: cstate = 0 elif isinstance(state, (CUlinkState,)): pstate = int(state) cstate = pstate else: pstate = int(CUlinkState(state)) cstate = pstate err = ccuda.cuLinkDestroy(cstate) return (CUresult(err),) {{endif}} {{if 'cuModuleGetTexRef' in found_functions}} @cython.embedsignature(True) def cuModuleGetTexRef(hmod, char* name): """ Returns a handle to a texture reference. [Deprecated] Returns in `*pTexRef` the handle of the texture reference of name `name` in the module `hmod`. If no texture reference of that name exists, :py:obj:`~.cuModuleGetTexRef()` returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. This texture reference handle should not be destroyed, since it will be destroyed when the module is unloaded. Parameters ---------- hmod : :py:obj:`~.CUmodule` Module to retrieve texture reference from name : bytes Name of texture reference to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` pTexRef : :py:obj:`~.CUtexref` Returned texture reference See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetSurfRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleLoadFatBinary`, :py:obj:`~.cuModuleUnload` """ cdef ccuda.CUmodule chmod if hmod is None: chmod = 0 elif isinstance(hmod, (CUmodule,)): phmod = int(hmod) chmod = phmod else: phmod = int(CUmodule(hmod)) chmod = phmod cdef CUtexref pTexRef = CUtexref() err = ccuda.cuModuleGetTexRef(pTexRef._ptr, chmod, name) return (CUresult(err), pTexRef) {{endif}} {{if 'cuModuleGetSurfRef' in found_functions}} @cython.embedsignature(True) def cuModuleGetSurfRef(hmod, char* name): """ Returns a handle to a surface reference. [Deprecated] Returns in `*pSurfRef` the handle of the surface reference of name `name` in the module `hmod`. If no surface reference of that name exists, :py:obj:`~.cuModuleGetSurfRef()` returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. Parameters ---------- hmod : :py:obj:`~.CUmodule` Module to retrieve surface reference from name : bytes Name of surface reference to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` pSurfRef : :py:obj:`~.CUsurfref` Returned surface reference See Also -------- :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuModuleGetGlobal`, :py:obj:`~.cuModuleGetTexRef`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx`, :py:obj:`~.cuModuleLoadFatBinary`, :py:obj:`~.cuModuleUnload` """ cdef ccuda.CUmodule chmod if hmod is None: chmod = 0 elif isinstance(hmod, (CUmodule,)): phmod = int(hmod) chmod = phmod else: phmod = int(CUmodule(hmod)) chmod = phmod cdef CUsurfref pSurfRef = CUsurfref() err = ccuda.cuModuleGetSurfRef(pSurfRef._ptr, chmod, name) return (CUresult(err), pSurfRef) {{endif}} {{if 'cuLibraryLoadData' in found_functions}} @cython.embedsignature(True) def cuLibraryLoadData(code, jitOptions : Optional[Tuple[CUjit_option] | List[CUjit_option]], jitOptionsValues : Optional[Tuple[Any] | List[Any]], unsigned int numJitOptions, libraryOptions : Optional[Tuple[CUlibraryOption] | List[CUlibraryOption]], libraryOptionValues : Optional[Tuple[Any] | List[Any]], unsigned int numLibraryOptions): """ Load a library with specified code and options. Takes a pointer `code` and loads the corresponding library `library` based on the application defined library loading mode: - If module loading is set to EAGER, via the environment variables described in "Module loading", `library` is loaded eagerly into all contexts at the time of the call and future contexts at the time of creation until the library is unloaded with :py:obj:`~.cuLibraryUnload()`. - If the environment variables are set to LAZY, `library` is not immediately loaded onto all existent contexts and will only be loaded when a function is needed for that context, such as a kernel launch. These environment variables are described in the CUDA programming guide under the "CUDA environment variables" section. The `code` may be a `cubin` or `fatbin` as output by nvcc, or a NULL- terminated `PTX`, either as output by nvcc or hand-written. A fatbin should also contain relocatable code when doing separate compilation. Options are passed as an array via `jitOptions` and any corresponding parameters are passed in `jitOptionsValues`. The number of total JIT options is supplied via `numJitOptions`. Any outputs will be returned via `jitOptionsValues`. Library load options are passed as an array via `libraryOptions` and any corresponding parameters are passed in `libraryOptionValues`. The number of total library load options is supplied via `numLibraryOptions`. Parameters ---------- code : Any Code to load jitOptions : List[:py:obj:`~.CUjit_option`] Options for JIT jitOptionsValues : List[Any] Option values for JIT numJitOptions : unsigned int Number of options libraryOptions : List[:py:obj:`~.CUlibraryOption`] Options for loading libraryOptionValues : List[Any] Option values for loading numLibraryOptions : unsigned int Number of options for loading Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_JIT_COMPILER_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` library : :py:obj:`~.CUlibrary` Returned library See Also -------- :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx` Notes ----- If the library contains managed variables and no device in the system supports managed variables this call is expected to return :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` """ libraryOptionValues = [] if libraryOptionValues is None else libraryOptionValues libraryOptions = [] if libraryOptions is None else libraryOptions if not all(isinstance(_x, (CUlibraryOption)) for _x in libraryOptions): raise TypeError("Argument 'libraryOptions' is not instance of type (expected Tuple[ccuda.CUlibraryOption] or List[ccuda.CUlibraryOption]") jitOptionsValues = [] if jitOptionsValues is None else jitOptionsValues jitOptions = [] if jitOptions is None else jitOptions if not all(isinstance(_x, (CUjit_option)) for _x in jitOptions): raise TypeError("Argument 'jitOptions' is not instance of type (expected Tuple[ccuda.CUjit_option] or List[ccuda.CUjit_option]") cdef CUlibrary library = CUlibrary() ccode = utils.HelperInputVoidPtr(code) cdef void* ccode_ptr = ccode.cptr cdef vector[ccuda.CUjit_option] cjitOptions = [pyjitOptions.value for pyjitOptions in (jitOptions)] pylist = [utils.HelperCUjit_option(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(jitOptions, jitOptionsValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperjitOptionsValues = utils.InputVoidPtrPtrHelper(pylist) if numJitOptions > len(jitOptions): raise RuntimeError("List is too small: " + str(len(jitOptions)) + " < " + str(numJitOptions)) if numJitOptions > len(jitOptionsValues): raise RuntimeError("List is too small: " + str(len(jitOptionsValues)) + " < " + str(numJitOptions)) cdef vector[ccuda.CUlibraryOption] clibraryOptions = [pylibraryOptions.value for pylibraryOptions in (libraryOptions)] pylist = [utils.HelperCUlibraryOption(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(libraryOptions, libraryOptionValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperlibraryOptionValues = utils.InputVoidPtrPtrHelper(pylist) if numLibraryOptions > len(libraryOptions): raise RuntimeError("List is too small: " + str(len(libraryOptions)) + " < " + str(numLibraryOptions)) if numLibraryOptions > len(libraryOptionValues): raise RuntimeError("List is too small: " + str(len(libraryOptionValues)) + " < " + str(numLibraryOptions)) err = ccuda.cuLibraryLoadData(library._ptr, ccode_ptr, cjitOptions.data(), voidStarHelperjitOptionsValues.cptr, numJitOptions, clibraryOptions.data(), voidStarHelperlibraryOptionValues.cptr, numLibraryOptions) return (CUresult(err), library) {{endif}} {{if 'cuLibraryLoadFromFile' in found_functions}} @cython.embedsignature(True) def cuLibraryLoadFromFile(char* fileName, jitOptions : Optional[Tuple[CUjit_option] | List[CUjit_option]], jitOptionsValues : Optional[Tuple[Any] | List[Any]], unsigned int numJitOptions, libraryOptions : Optional[Tuple[CUlibraryOption] | List[CUlibraryOption]], libraryOptionValues : Optional[Tuple[Any] | List[Any]], unsigned int numLibraryOptions): """ Load a library with specified file and options. Takes a pointer `code` and loads the corresponding library `library` based on the application defined library loading mode: - If module loading is set to EAGER, via the environment variables described in "Module loading", `library` is loaded eagerly into all contexts at the time of the call and future contexts at the time of creation until the library is unloaded with :py:obj:`~.cuLibraryUnload()`. - If the environment variables are set to LAZY, `library` is not immediately loaded onto all existent contexts and will only be loaded when a function is needed for that context, such as a kernel launch. These environment variables are described in the CUDA programming guide under the "CUDA environment variables" section. The file should be a `cubin` file as output by nvcc, or a `PTX` file either as output by nvcc or handwritten, or a `fatbin` file as output by nvcc. A fatbin should also contain relocatable code when doing separate compilation. Options are passed as an array via `jitOptions` and any corresponding parameters are passed in `jitOptionsValues`. The number of total options is supplied via `numJitOptions`. Any outputs will be returned via `jitOptionsValues`. Library load options are passed as an array via `libraryOptions` and any corresponding parameters are passed in `libraryOptionValues`. The number of total library load options is supplied via `numLibraryOptions`. Parameters ---------- fileName : bytes File to load from jitOptions : List[:py:obj:`~.CUjit_option`] Options for JIT jitOptionsValues : List[Any] Option values for JIT numJitOptions : unsigned int Number of options libraryOptions : List[:py:obj:`~.CUlibraryOption`] Options for loading libraryOptionValues : List[Any] Option values for loading numLibraryOptions : unsigned int Number of options for loading Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_PTX`, :py:obj:`~.CUDA_ERROR_UNSUPPORTED_PTX_VERSION`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NO_BINARY_FOR_GPU`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_JIT_COMPILER_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` library : :py:obj:`~.CUlibrary` Returned library See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuModuleLoad`, :py:obj:`~.cuModuleLoadData`, :py:obj:`~.cuModuleLoadDataEx` Notes ----- If the library contains managed variables and no device in the system supports managed variables this call is expected to return :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` """ libraryOptionValues = [] if libraryOptionValues is None else libraryOptionValues libraryOptions = [] if libraryOptions is None else libraryOptions if not all(isinstance(_x, (CUlibraryOption)) for _x in libraryOptions): raise TypeError("Argument 'libraryOptions' is not instance of type (expected Tuple[ccuda.CUlibraryOption] or List[ccuda.CUlibraryOption]") jitOptionsValues = [] if jitOptionsValues is None else jitOptionsValues jitOptions = [] if jitOptions is None else jitOptions if not all(isinstance(_x, (CUjit_option)) for _x in jitOptions): raise TypeError("Argument 'jitOptions' is not instance of type (expected Tuple[ccuda.CUjit_option] or List[ccuda.CUjit_option]") cdef CUlibrary library = CUlibrary() cdef vector[ccuda.CUjit_option] cjitOptions = [pyjitOptions.value for pyjitOptions in (jitOptions)] pylist = [utils.HelperCUjit_option(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(jitOptions, jitOptionsValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperjitOptionsValues = utils.InputVoidPtrPtrHelper(pylist) if numJitOptions > len(jitOptions): raise RuntimeError("List is too small: " + str(len(jitOptions)) + " < " + str(numJitOptions)) if numJitOptions > len(jitOptionsValues): raise RuntimeError("List is too small: " + str(len(jitOptionsValues)) + " < " + str(numJitOptions)) cdef vector[ccuda.CUlibraryOption] clibraryOptions = [pylibraryOptions.value for pylibraryOptions in (libraryOptions)] pylist = [utils.HelperCUlibraryOption(pyoptions, pyoptionValues) for pyoptions, pyoptionValues in zip(libraryOptions, libraryOptionValues)] cdef utils.InputVoidPtrPtrHelper voidStarHelperlibraryOptionValues = utils.InputVoidPtrPtrHelper(pylist) if numLibraryOptions > len(libraryOptions): raise RuntimeError("List is too small: " + str(len(libraryOptions)) + " < " + str(numLibraryOptions)) if numLibraryOptions > len(libraryOptionValues): raise RuntimeError("List is too small: " + str(len(libraryOptionValues)) + " < " + str(numLibraryOptions)) err = ccuda.cuLibraryLoadFromFile(library._ptr, fileName, cjitOptions.data(), voidStarHelperjitOptionsValues.cptr, numJitOptions, clibraryOptions.data(), voidStarHelperlibraryOptionValues.cptr, numLibraryOptions) return (CUresult(err), library) {{endif}} {{if 'cuLibraryUnload' in found_functions}} @cython.embedsignature(True) def cuLibraryUnload(library): """ Unloads a library. Unloads the library specified with `library` Parameters ---------- library : :py:obj:`~.CUlibrary` Library to unload Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuModuleUnload` """ cdef ccuda.CUlibrary clibrary if library is None: clibrary = 0 elif isinstance(library, (CUlibrary,)): plibrary = int(library) clibrary = plibrary else: plibrary = int(CUlibrary(library)) clibrary = plibrary err = ccuda.cuLibraryUnload(clibrary) return (CUresult(err),) {{endif}} {{if 'cuLibraryGetKernel' in found_functions}} @cython.embedsignature(True) def cuLibraryGetKernel(library, char* name): """ Returns a kernel handle. Returns in `pKernel` the handle of the kernel with name `name` located in library `library`. If kernel handle is not found, the call returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. Parameters ---------- library : :py:obj:`~.CUlibrary` Library to retrieve kernel from name : bytes Name of kernel to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` pKernel : :py:obj:`~.CUkernel` Returned kernel handle See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuKernelGetFunction`, :py:obj:`~.cuLibraryGetModule`, :py:obj:`~.cuModuleGetFunction` """ cdef ccuda.CUlibrary clibrary if library is None: clibrary = 0 elif isinstance(library, (CUlibrary,)): plibrary = int(library) clibrary = plibrary else: plibrary = int(CUlibrary(library)) clibrary = plibrary cdef CUkernel pKernel = CUkernel() err = ccuda.cuLibraryGetKernel(pKernel._ptr, clibrary, name) return (CUresult(err), pKernel) {{endif}} {{if 'cuLibraryGetKernelCount' in found_functions}} @cython.embedsignature(True) def cuLibraryGetKernelCount(lib): """ Returns the number of kernels within a library. Returns in `count` the number of kernels in `lib`. Parameters ---------- lib : :py:obj:`~.CUlibrary` Library to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` count : unsigned int Number of kernels found within the library """ cdef ccuda.CUlibrary clib if lib is None: clib = 0 elif isinstance(lib, (CUlibrary,)): plib = int(lib) clib = plib else: plib = int(CUlibrary(lib)) clib = plib cdef unsigned int count = 0 err = ccuda.cuLibraryGetKernelCount(&count, clib) return (CUresult(err), count) {{endif}} {{if 'cuLibraryEnumerateKernels' in found_functions}} @cython.embedsignature(True) def cuLibraryEnumerateKernels(unsigned int numKernels, lib): """ Retrieve the kernel handles within a library. Returns in `kernels` a maximum number of `numKernels` kernel handles within `lib`. The returned kernel handle becomes invalid when the library is unloaded. Parameters ---------- numKernels : unsigned int Maximum number of kernel handles may be returned to the buffer lib : :py:obj:`~.CUlibrary` Library to query from Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` kernels : List[:py:obj:`~.CUkernel`] Buffer where the kernel handles are returned to See Also -------- :py:obj:`~.cuLibraryGetKernelCount` """ cdef ccuda.CUlibrary clib if lib is None: clib = 0 elif isinstance(lib, (CUlibrary,)): plib = int(lib) clib = plib else: plib = int(CUlibrary(lib)) clib = plib cdef ccuda.CUkernel* ckernels = NULL pykernels = [] if numKernels != 0: ckernels = calloc(numKernels, sizeof(ccuda.CUkernel)) if ckernels is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(numKernels) + 'x' + str(sizeof(ccuda.CUkernel))) err = ccuda.cuLibraryEnumerateKernels(ckernels, numKernels, clib) if CUresult(err) == CUresult(0): pykernels = [CUkernel(init_value=ckernels[idx]) for idx in range(numKernels)] if ckernels is not NULL: free(ckernels) return (CUresult(err), pykernels) {{endif}} {{if 'cuLibraryGetModule' in found_functions}} @cython.embedsignature(True) def cuLibraryGetModule(library): """ Returns a module handle. Returns in `pMod` the module handle associated with the current context located in library `library`. If module handle is not found, the call returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. Parameters ---------- library : :py:obj:`~.CUlibrary` Library to retrieve module from Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` pMod : :py:obj:`~.CUmodule` Returned module handle See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuModuleGetFunction` """ cdef ccuda.CUlibrary clibrary if library is None: clibrary = 0 elif isinstance(library, (CUlibrary,)): plibrary = int(library) clibrary = plibrary else: plibrary = int(CUlibrary(library)) clibrary = plibrary cdef CUmodule pMod = CUmodule() err = ccuda.cuLibraryGetModule(pMod._ptr, clibrary) return (CUresult(err), pMod) {{endif}} {{if 'cuKernelGetFunction' in found_functions}} @cython.embedsignature(True) def cuKernelGetFunction(kernel): """ Returns a function handle. Returns in `pFunc` the handle of the function for the requested kernel `kernel` and the current context. If function handle is not found, the call returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. Parameters ---------- kernel : :py:obj:`~.CUkernel` Kernel to retrieve function for the requested context Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` pFunc : :py:obj:`~.CUfunction` Returned function handle See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuLibraryGetKernel`, :py:obj:`~.cuLibraryGetModule`, :py:obj:`~.cuModuleGetFunction` """ cdef ccuda.CUkernel ckernel if kernel is None: ckernel = 0 elif isinstance(kernel, (CUkernel,)): pkernel = int(kernel) ckernel = pkernel else: pkernel = int(CUkernel(kernel)) ckernel = pkernel cdef CUfunction pFunc = CUfunction() err = ccuda.cuKernelGetFunction(pFunc._ptr, ckernel) return (CUresult(err), pFunc) {{endif}} {{if 'cuKernelGetLibrary' in found_functions}} @cython.embedsignature(True) def cuKernelGetLibrary(kernel): """ Returns a library handle. Returns in `pLib` the handle of the library for the requested kernel `kernel` Parameters ---------- kernel : :py:obj:`~.CUkernel` Kernel to retrieve library handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` pLib : :py:obj:`~.CUlibrary` Returned library handle See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuLibraryGetKernel` """ cdef ccuda.CUkernel ckernel if kernel is None: ckernel = 0 elif isinstance(kernel, (CUkernel,)): pkernel = int(kernel) ckernel = pkernel else: pkernel = int(CUkernel(kernel)) ckernel = pkernel cdef CUlibrary pLib = CUlibrary() err = ccuda.cuKernelGetLibrary(pLib._ptr, ckernel) return (CUresult(err), pLib) {{endif}} {{if 'cuLibraryGetGlobal' in found_functions}} @cython.embedsignature(True) def cuLibraryGetGlobal(library, char* name): """ Returns a global device pointer. Returns in `*dptr` and `*bytes` the base pointer and size of the global with name `name` for the requested library `library` and the current context. If no global for the requested name `name` exists, the call returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. One of the parameters `dptr` or `numbytes` (not both) can be NULL in which case it is ignored. Parameters ---------- library : :py:obj:`~.CUlibrary` Library to retrieve global from name : bytes Name of global to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` dptr : :py:obj:`~.CUdeviceptr` Returned global device pointer for the requested context numbytes : int Returned global size in bytes See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuLibraryGetModule`, :py:obj:`~.cuModuleGetGlobal` """ cdef ccuda.CUlibrary clibrary if library is None: clibrary = 0 elif isinstance(library, (CUlibrary,)): plibrary = int(library) clibrary = plibrary else: plibrary = int(CUlibrary(library)) clibrary = plibrary cdef CUdeviceptr dptr = CUdeviceptr() cdef size_t numbytes = 0 err = ccuda.cuLibraryGetGlobal(dptr._ptr, &numbytes, clibrary, name) return (CUresult(err), dptr, numbytes) {{endif}} {{if 'cuLibraryGetManaged' in found_functions}} @cython.embedsignature(True) def cuLibraryGetManaged(library, char* name): """ Returns a pointer to managed memory. Returns in `*dptr` and `*bytes` the base pointer and size of the managed memory with name `name` for the requested library `library`. If no managed memory with the requested name `name` exists, the call returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. One of the parameters `dptr` or `numbytes` (not both) can be NULL in which case it is ignored. Note that managed memory for library `library` is shared across devices and is registered when the library is loaded into atleast one context. Parameters ---------- library : :py:obj:`~.CUlibrary` Library to retrieve managed memory from name : bytes Name of managed memory to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` dptr : :py:obj:`~.CUdeviceptr` Returned pointer to the managed memory numbytes : int Returned memory size in bytes See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload` """ cdef ccuda.CUlibrary clibrary if library is None: clibrary = 0 elif isinstance(library, (CUlibrary,)): plibrary = int(library) clibrary = plibrary else: plibrary = int(CUlibrary(library)) clibrary = plibrary cdef CUdeviceptr dptr = CUdeviceptr() cdef size_t numbytes = 0 err = ccuda.cuLibraryGetManaged(dptr._ptr, &numbytes, clibrary, name) return (CUresult(err), dptr, numbytes) {{endif}} {{if 'cuLibraryGetUnifiedFunction' in found_functions}} @cython.embedsignature(True) def cuLibraryGetUnifiedFunction(library, char* symbol): """ Returns a pointer to a unified function. Returns in `*fptr` the function pointer to a unified function denoted by `symbol`. If no unified function with name `symbol` exists, the call returns :py:obj:`~.CUDA_ERROR_NOT_FOUND`. If there is no device with attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_UNIFIED_FUNCTION_POINTERS` present in the system, the call may return :py:obj:`~.CUDA_ERROR_NOT_FOUND`. Parameters ---------- library : :py:obj:`~.CUlibrary` Library to retrieve function pointer memory from symbol : bytes Name of function pointer to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` fptr : Any Returned pointer to a unified function See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload` """ cdef ccuda.CUlibrary clibrary if library is None: clibrary = 0 elif isinstance(library, (CUlibrary,)): plibrary = int(library) clibrary = plibrary else: plibrary = int(CUlibrary(library)) clibrary = plibrary cdef void_ptr fptr = 0 err = ccuda.cuLibraryGetUnifiedFunction(&fptr, clibrary, symbol) return (CUresult(err), fptr) {{endif}} {{if 'cuKernelGetAttribute' in found_functions}} @cython.embedsignature(True) def cuKernelGetAttribute(attrib not None : CUfunction_attribute, kernel, dev): """ Returns information about a kernel. Returns in `*pi` the integer value of the attribute `attrib` for the kernel `kernel` for the requested device `dev`. The supported attributes are: - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK`: The maximum number of threads per block, beyond which a launch of the kernel would fail. This number depends on both the kernel and the requested device. - :py:obj:`~.CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES`: The size in bytes of statically-allocated shared memory per block required by this kernel. This does not include dynamically-allocated shared memory requested by the user at runtime. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CONST_SIZE_BYTES`: The size in bytes of user-allocated constant memory required by this kernel. - :py:obj:`~.CU_FUNC_ATTRIBUTE_LOCAL_SIZE_BYTES`: The size in bytes of local memory used by each thread of this kernel. - :py:obj:`~.CU_FUNC_ATTRIBUTE_NUM_REGS`: The number of registers used by each thread of this kernel. - :py:obj:`~.CU_FUNC_ATTRIBUTE_PTX_VERSION`: The PTX virtual architecture version for which the kernel was compiled. This value is the major PTX version * 10 - the minor PTX version, so a PTX version 1.3 function would return the value 13. Note that this may return the undefined value of 0 for cubins compiled prior to CUDA 3.0. - :py:obj:`~.CU_FUNC_ATTRIBUTE_BINARY_VERSION`: The binary architecture version for which the kernel was compiled. This value is the major binary version * 10 + the minor binary version, so a binary version 1.3 function would return the value 13. Note that this will return a value of 10 for legacy cubins that do not have a properly-encoded binary architecture version. - :py:obj:`~.CU_FUNC_CACHE_MODE_CA`: The attribute to indicate whether the kernel has been compiled with user specified option "-Xptxas --dlcm=ca" set. - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES`: The maximum size in bytes of dynamically-allocated shared memory. - :py:obj:`~.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT`: Preferred shared memory-L1 cache split ratio in percent of total shared memory. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CLUSTER_SIZE_MUST_BE_SET`: If this attribute is set, the kernel must launch with a valid cluster size specified. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_WIDTH`: The required cluster width in blocks. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_HEIGHT`: The required cluster height in blocks. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_DEPTH`: The required cluster depth in blocks. - :py:obj:`~.CU_FUNC_ATTRIBUTE_NON_PORTABLE_CLUSTER_SIZE_ALLOWED`: Indicates whether the function can be launched with non-portable cluster size. 1 is allowed, 0 is disallowed. A non-portable cluster size may only function on the specific SKUs the program is tested on. The launch might fail if the program is run on a different hardware platform. CUDA API provides cudaOccupancyMaxActiveClusters to assist with checking whether the desired size can be launched on the current device. A portable cluster size is guaranteed to be functional on all compute capabilities higher than the target compute capability. The portable cluster size for sm_90 is 8 blocks per cluster. This value may increase for future compute capabilities. The specific hardware unit may support higher cluster sizes that’s not guaranteed to be portable. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE`: The block scheduling policy of a function. The value type is CUclusterSchedulingPolicy. Parameters ---------- attrib : :py:obj:`~.CUfunction_attribute` Attribute requested kernel : :py:obj:`~.CUkernel` Kernel to query attribute of dev : :py:obj:`~.CUdevice` Device to query attribute of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` pi : int Returned attribute value See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuKernelSetAttribute`, :py:obj:`~.cuLibraryGetKernel`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cuKernelGetFunction`, :py:obj:`~.cuLibraryGetModule`, :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuFuncGetAttribute` Notes ----- If another thread is trying to set the same attribute on the same device using :py:obj:`~.cuKernelSetAttribute()` simultaneously, the attribute query will give the old or new value depending on the interleavings chosen by the OS scheduler and memory consistency. """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef ccuda.CUkernel ckernel if kernel is None: ckernel = 0 elif isinstance(kernel, (CUkernel,)): pkernel = int(kernel) ckernel = pkernel else: pkernel = int(CUkernel(kernel)) ckernel = pkernel cdef int pi = 0 cdef ccuda.CUfunction_attribute cattrib = attrib.value err = ccuda.cuKernelGetAttribute(&pi, cattrib, ckernel, cdev) return (CUresult(err), pi) {{endif}} {{if 'cuKernelSetAttribute' in found_functions}} @cython.embedsignature(True) def cuKernelSetAttribute(attrib not None : CUfunction_attribute, int val, kernel, dev): """ Sets information about a kernel. This call sets the value of a specified attribute `attrib` on the kernel `kernel` for the requested device `dev` to an integer value specified by `val`. This function returns CUDA_SUCCESS if the new value of the attribute could be successfully set. If the set fails, this call will return an error. Not all attributes can have values set. Attempting to set a value on a read-only attribute will result in an error (CUDA_ERROR_INVALID_VALUE) Note that attributes set using :py:obj:`~.cuFuncSetAttribute()` will override the attribute set by this API irrespective of whether the call to :py:obj:`~.cuFuncSetAttribute()` is made before or after this API call. However, :py:obj:`~.cuKernelGetAttribute()` will always return the attribute value set by this API. Supported attributes are: - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES`: This is the maximum size in bytes of dynamically-allocated shared memory. The value should contain the requested maximum size of dynamically- allocated shared memory. The sum of this value and the function attribute :py:obj:`~.CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES` cannot exceed the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK_OPTIN`. The maximal size of requestable dynamic shared memory may differ by GPU architecture. - :py:obj:`~.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT`: On devices where the L1 cache and shared memory use the same hardware resources, this sets the shared memory carveout preference, in percent of the total shared memory. See :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_MULTIPROCESSOR` This is only a hint, and the driver can choose a different ratio if required to execute the function. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_WIDTH`: The required cluster width in blocks. The width, height, and depth values must either all be 0 or all be positive. The validity of the cluster dimensions is checked at launch time. If the value is set during compile time, it cannot be set at runtime. Setting it at runtime will return CUDA_ERROR_NOT_PERMITTED. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_HEIGHT`: The required cluster height in blocks. The width, height, and depth values must either all be 0 or all be positive. The validity of the cluster dimensions is checked at launch time. If the value is set during compile time, it cannot be set at runtime. Setting it at runtime will return CUDA_ERROR_NOT_PERMITTED. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_DEPTH`: The required cluster depth in blocks. The width, height, and depth values must either all be 0 or all be positive. The validity of the cluster dimensions is checked at launch time. If the value is set during compile time, it cannot be set at runtime. Setting it at runtime will return CUDA_ERROR_NOT_PERMITTED. - :py:obj:`~.CU_FUNC_ATTRIBUTE_NON_PORTABLE_CLUSTER_SIZE_ALLOWED`: Indicates whether the function can be launched with non-portable cluster size. 1 is allowed, 0 is disallowed. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE`: The block scheduling policy of a function. The value type is CUclusterSchedulingPolicy. Parameters ---------- attrib : :py:obj:`~.CUfunction_attribute` Attribute requested val : int Value to set kernel : :py:obj:`~.CUkernel` Kernel to set attribute of dev : :py:obj:`~.CUdevice` Device to set attribute of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuKernelGetAttribute`, :py:obj:`~.cuLibraryGetKernel`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cuKernelGetFunction`, :py:obj:`~.cuLibraryGetModule`, :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuFuncSetAttribute` Notes ----- The API has stricter locking requirements in comparison to its legacy counterpart :py:obj:`~.cuFuncSetAttribute()` due to device-wide semantics. If multiple threads are trying to set the same attribute on the same device simultaneously, the attribute setting will depend on the interleavings chosen by the OS scheduler and memory consistency. """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef ccuda.CUkernel ckernel if kernel is None: ckernel = 0 elif isinstance(kernel, (CUkernel,)): pkernel = int(kernel) ckernel = pkernel else: pkernel = int(CUkernel(kernel)) ckernel = pkernel cdef ccuda.CUfunction_attribute cattrib = attrib.value err = ccuda.cuKernelSetAttribute(cattrib, val, ckernel, cdev) return (CUresult(err),) {{endif}} {{if 'cuKernelSetCacheConfig' in found_functions}} @cython.embedsignature(True) def cuKernelSetCacheConfig(kernel, config not None : CUfunc_cache, dev): """ Sets the preferred cache configuration for a device kernel. On devices where the L1 cache and shared memory use the same hardware resources, this sets through `config` the preferred cache configuration for the device kernel `kernel` on the requested device `dev`. This is only a preference. The driver will use the requested configuration if possible, but it is free to choose a different configuration if required to execute `kernel`. Any context-wide preference set via :py:obj:`~.cuCtxSetCacheConfig()` will be overridden by this per-kernel setting. Note that attributes set using :py:obj:`~.cuFuncSetCacheConfig()` will override the attribute set by this API irrespective of whether the call to :py:obj:`~.cuFuncSetCacheConfig()` is made before or after this API call. This setting does nothing on devices where the size of the L1 cache and shared memory are fixed. Launching a kernel with a different preference than the most recent preference setting may insert a device-side synchronization point. The supported cache configurations are: - :py:obj:`~.CU_FUNC_CACHE_PREFER_NONE`: no preference for shared memory or L1 (default) - :py:obj:`~.CU_FUNC_CACHE_PREFER_SHARED`: prefer larger shared memory and smaller L1 cache - :py:obj:`~.CU_FUNC_CACHE_PREFER_L1`: prefer larger L1 cache and smaller shared memory - :py:obj:`~.CU_FUNC_CACHE_PREFER_EQUAL`: prefer equal sized L1 cache and shared memory Parameters ---------- kernel : :py:obj:`~.CUkernel` Kernel to configure cache for config : :py:obj:`~.CUfunc_cache` Requested cache configuration dev : :py:obj:`~.CUdevice` Device to set attribute of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuLibraryLoadData`, :py:obj:`~.cuLibraryLoadFromFile`, :py:obj:`~.cuLibraryUnload`, :py:obj:`~.cuLibraryGetKernel`, :py:obj:`~.cuKernelGetFunction`, :py:obj:`~.cuLibraryGetModule`, :py:obj:`~.cuModuleGetFunction`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuLaunchKernel` Notes ----- The API has stricter locking requirements in comparison to its legacy counterpart :py:obj:`~.cuFuncSetCacheConfig()` due to device-wide semantics. If multiple threads are trying to set a config on the same device simultaneously, the cache config setting will depend on the interleavings chosen by the OS scheduler and memory consistency. """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef ccuda.CUkernel ckernel if kernel is None: ckernel = 0 elif isinstance(kernel, (CUkernel,)): pkernel = int(kernel) ckernel = pkernel else: pkernel = int(CUkernel(kernel)) ckernel = pkernel cdef ccuda.CUfunc_cache cconfig = config.value err = ccuda.cuKernelSetCacheConfig(ckernel, cconfig, cdev) return (CUresult(err),) {{endif}} {{if 'cuKernelGetName' in found_functions}} @cython.embedsignature(True) def cuKernelGetName(hfunc): """ Returns the function name for a :py:obj:`~.CUkernel` handle. Returns in `**name` the function name associated with the kernel handle `hfunc` . The function name is returned as a null-terminated string. The returned name is only valid when the kernel handle is valid. If the library is unloaded or reloaded, one must call the API again to get the updated name. This API may return a mangled name if the function is not declared as having C linkage. If either `**name` or `hfunc` is NULL, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. Parameters ---------- hfunc : :py:obj:`~.CUkernel` The function handle to retrieve the name for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` name : bytes The returned name of the function """ cdef ccuda.CUkernel chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUkernel,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUkernel(hfunc)) chfunc = phfunc cdef const char* name = NULL err = ccuda.cuKernelGetName(&name, chfunc) return (CUresult(err), name) {{endif}} {{if 'cuKernelGetParamInfo' in found_functions}} @cython.embedsignature(True) def cuKernelGetParamInfo(kernel, size_t paramIndex): """ Returns the offset and size of a kernel parameter in the device-side parameter layout. Queries the kernel parameter at `paramIndex` into `kernel's` list of parameters, and returns in `paramOffset` and `paramSize` the offset and size, respectively, where the parameter will reside in the device-side parameter layout. This information can be used to update kernel node parameters from the device via :py:obj:`~.cudaGraphKernelNodeSetParam()` and :py:obj:`~.cudaGraphKernelNodeUpdatesApply()`. `paramIndex` must be less than the number of parameters that `kernel` takes. `paramSize` can be set to NULL if only the parameter offset is desired. Parameters ---------- kernel : :py:obj:`~.CUkernel` The kernel to query paramIndex : size_t The parameter index to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, paramOffset : int Returns the offset into the device-side parameter layout at which the parameter resides paramSize : int Optionally returns the size of the parameter in the device-side parameter layout See Also -------- :py:obj:`~.cuFuncGetParamInfo` """ cdef ccuda.CUkernel ckernel if kernel is None: ckernel = 0 elif isinstance(kernel, (CUkernel,)): pkernel = int(kernel) ckernel = pkernel else: pkernel = int(CUkernel(kernel)) ckernel = pkernel cdef size_t paramOffset = 0 cdef size_t paramSize = 0 err = ccuda.cuKernelGetParamInfo(ckernel, paramIndex, ¶mOffset, ¶mSize) return (CUresult(err), paramOffset, paramSize) {{endif}} {{if 'cuMemGetInfo_v2' in found_functions}} @cython.embedsignature(True) def cuMemGetInfo(): """ Gets free and total memory. Returns in `*total` the total amount of memory available to the the current context. Returns in `*free` the amount of memory on the device that is free according to the OS. CUDA is not guaranteed to be able to allocate all of the memory that the OS reports as free. In a multi- tenet situation, free estimate returned is prone to race condition where a new allocation/free done by a different process or a different thread in the same process between the time when free memory was estimated and reported, will result in deviation in free value reported and actual free memory. The integrated GPU on Tegra shares memory with CPU and other component of the SoC. The free and total values returned by the API excludes the SWAP memory space maintained by the OS on some platforms. The OS may move some of the memory pages into swap area as the GPU or CPU allocate or access memory. See Tegra app note on how to calculate total and free memory on Tegra. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` free : int Returned free memory in bytes total : int Returned total memory in bytes See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemGetInfo` """ cdef size_t free = 0 cdef size_t total = 0 err = ccuda.cuMemGetInfo(&free, &total) return (CUresult(err), free, total) {{endif}} {{if 'cuMemAlloc_v2' in found_functions}} @cython.embedsignature(True) def cuMemAlloc(size_t bytesize): """ Allocates device memory. Allocates `bytesize` bytes of linear memory on the device and returns in `*dptr` a pointer to the allocated memory. The allocated memory is suitably aligned for any kind of variable. The memory is not cleared. If `bytesize` is 0, :py:obj:`~.cuMemAlloc()` returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. Parameters ---------- bytesize : size_t Requested allocation size in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` dptr : :py:obj:`~.CUdeviceptr` Returned device pointer See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMalloc` """ cdef CUdeviceptr dptr = CUdeviceptr() err = ccuda.cuMemAlloc(dptr._ptr, bytesize) return (CUresult(err), dptr) {{endif}} {{if 'cuMemAllocPitch_v2' in found_functions}} @cython.embedsignature(True) def cuMemAllocPitch(size_t WidthInBytes, size_t Height, unsigned int ElementSizeBytes): """ Allocates pitched device memory. Allocates at least `WidthInBytes` * `Height` bytes of linear memory on the device and returns in `*dptr` a pointer to the allocated memory. The function may pad the allocation to ensure that corresponding pointers in any given row will continue to meet the alignment requirements for coalescing as the address is updated from row to row. `ElementSizeBytes` specifies the size of the largest reads and writes that will be performed on the memory range. `ElementSizeBytes` may be 4, 8 or 16 (since coalesced memory transactions are not possible on other data sizes). If `ElementSizeBytes` is smaller than the actual read/write size of a kernel, the kernel will run correctly, but possibly at reduced speed. The pitch returned in `*pPitch` by :py:obj:`~.cuMemAllocPitch()` is the width in bytes of the allocation. The intended usage of pitch is as a separate parameter of the allocation, used to compute addresses within the 2D array. Given the row and column of an array element of type T, the address is computed as: **View CUDA Toolkit Documentation for a C++ code example** The pitch returned by :py:obj:`~.cuMemAllocPitch()` is guaranteed to work with :py:obj:`~.cuMemcpy2D()` under all circumstances. For allocations of 2D arrays, it is recommended that programmers consider performing pitch allocations using :py:obj:`~.cuMemAllocPitch()`. Due to alignment restrictions in the hardware, this is especially true if the application will be performing 2D memory copies between different regions of device memory (whether linear memory or CUDA arrays). The byte alignment of the pitch returned by :py:obj:`~.cuMemAllocPitch()` is guaranteed to match or exceed the alignment requirement for texture binding with :py:obj:`~.cuTexRefSetAddress2D()`. Parameters ---------- WidthInBytes : size_t Requested allocation width in bytes Height : size_t Requested allocation height in rows ElementSizeBytes : unsigned int Size of largest reads/writes for range Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` dptr : :py:obj:`~.CUdeviceptr` Returned device pointer pPitch : int Returned pitch of allocation in bytes See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMallocPitch` """ cdef CUdeviceptr dptr = CUdeviceptr() cdef size_t pPitch = 0 err = ccuda.cuMemAllocPitch(dptr._ptr, &pPitch, WidthInBytes, Height, ElementSizeBytes) return (CUresult(err), dptr, pPitch) {{endif}} {{if 'cuMemFree_v2' in found_functions}} @cython.embedsignature(True) def cuMemFree(dptr): """ Frees device memory. Frees the memory space pointed to by `dptr`, which must have been returned by a previous call to one of the following memory allocation APIs - :py:obj:`~.cuMemAlloc()`, :py:obj:`~.cuMemAllocPitch()`, :py:obj:`~.cuMemAllocManaged()`, :py:obj:`~.cuMemAllocAsync()`, :py:obj:`~.cuMemAllocFromPoolAsync()` Note - This API will not perform any implict synchronization when the pointer was allocated with :py:obj:`~.cuMemAllocAsync` or :py:obj:`~.cuMemAllocFromPoolAsync`. Callers must ensure that all accesses to these pointer have completed before invoking :py:obj:`~.cuMemFree`. For best performance and memory reuse, users should use :py:obj:`~.cuMemFreeAsync` to free memory allocated via the stream ordered memory allocator. For all other pointers, this API may perform implicit synchronization. Parameters ---------- dptr : :py:obj:`~.CUdeviceptr` Pointer to memory to free Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemAllocManaged`, :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemAllocFromPoolAsync`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaFree` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr err = ccuda.cuMemFree(cdptr) return (CUresult(err),) {{endif}} {{if 'cuMemGetAddressRange_v2' in found_functions}} @cython.embedsignature(True) def cuMemGetAddressRange(dptr): """ Get information on memory allocations. Returns the base address in `*pbase` and size in `*psize` of the allocation by :py:obj:`~.cuMemAlloc()` or :py:obj:`~.cuMemAllocPitch()` that contains the input pointer `dptr`. Both parameters `pbase` and `psize` are optional. If one of them is NULL, it is ignored. Parameters ---------- dptr : :py:obj:`~.CUdeviceptr` Device pointer to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_NOT_FOUND`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pbase : :py:obj:`~.CUdeviceptr` Returned base address psize : int Returned size of device memory allocation See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr cdef CUdeviceptr pbase = CUdeviceptr() cdef size_t psize = 0 err = ccuda.cuMemGetAddressRange(pbase._ptr, &psize, cdptr) return (CUresult(err), pbase, psize) {{endif}} {{if 'cuMemAllocHost_v2' in found_functions}} @cython.embedsignature(True) def cuMemAllocHost(size_t bytesize): """ Allocates page-locked host memory. Allocates `bytesize` bytes of host memory that is page-locked and accessible to the device. The driver tracks the virtual memory ranges allocated with this function and automatically accelerates calls to functions such as :py:obj:`~.cuMemcpy()`. Since the memory can be accessed directly by the device, it can be read or written with much higher bandwidth than pageable memory obtained with functions such as :py:obj:`~.malloc()`. On systems where :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES` is true, :py:obj:`~.cuMemAllocHost` may not page-lock the allocated memory. Page-locking excessive amounts of memory with :py:obj:`~.cuMemAllocHost()` may degrade system performance, since it reduces the amount of memory available to the system for paging. As a result, this function is best used sparingly to allocate staging areas for data exchange between host and device. Note all host memory allocated using :py:obj:`~.cuMemAllocHost()` will automatically be immediately accessible to all contexts on all devices which support unified addressing (as may be queried using :py:obj:`~.CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING`). The device pointer that may be used to access this host memory from those contexts is always equal to the returned host pointer `*pp`. See :py:obj:`~.Unified Addressing` for additional details. Parameters ---------- bytesize : size_t Requested allocation size in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` pp : Any Returned pointer to host memory See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMallocHost` """ cdef void_ptr pp = 0 err = ccuda.cuMemAllocHost(&pp, bytesize) return (CUresult(err), pp) {{endif}} {{if 'cuMemFreeHost' in found_functions}} @cython.embedsignature(True) def cuMemFreeHost(p): """ Frees page-locked host memory. Frees the memory space pointed to by `p`, which must have been returned by a previous call to :py:obj:`~.cuMemAllocHost()`. Parameters ---------- p : Any Pointer to memory to free Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaFreeHost` """ cp = utils.HelperInputVoidPtr(p) cdef void* cp_ptr = cp.cptr err = ccuda.cuMemFreeHost(cp_ptr) return (CUresult(err),) {{endif}} {{if 'cuMemHostAlloc' in found_functions}} @cython.embedsignature(True) def cuMemHostAlloc(size_t bytesize, unsigned int Flags): """ Allocates page-locked host memory. Allocates `bytesize` bytes of host memory that is page-locked and accessible to the device. The driver tracks the virtual memory ranges allocated with this function and automatically accelerates calls to functions such as :py:obj:`~.cuMemcpyHtoD()`. Since the memory can be accessed directly by the device, it can be read or written with much higher bandwidth than pageable memory obtained with functions such as :py:obj:`~.malloc()`. On systems where :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES` is true, :py:obj:`~.cuMemHostAlloc` may not page-lock the allocated memory. Page-locking excessive amounts of memory may degrade system performance, since it reduces the amount of memory available to the system for paging. As a result, this function is best used sparingly to allocate staging areas for data exchange between host and device. The `Flags` parameter enables different options to be specified that affect the allocation, as follows. - :py:obj:`~.CU_MEMHOSTALLOC_PORTABLE`: The memory returned by this call will be considered as pinned memory by all CUDA contexts, not just the one that performed the allocation. - :py:obj:`~.CU_MEMHOSTALLOC_DEVICEMAP`: Maps the allocation into the CUDA address space. The device pointer to the memory may be obtained by calling :py:obj:`~.cuMemHostGetDevicePointer()`. - :py:obj:`~.CU_MEMHOSTALLOC_WRITECOMBINED`: Allocates the memory as write-combined (WC). WC memory can be transferred across the PCI Express bus more quickly on some system configurations, but cannot be read efficiently by most CPUs. WC memory is a good option for buffers that will be written by the CPU and read by the GPU via mapped pinned memory or host->device transfers. All of these flags are orthogonal to one another: a developer may allocate memory that is portable, mapped and/or write-combined with no restrictions. The :py:obj:`~.CU_MEMHOSTALLOC_DEVICEMAP` flag may be specified on CUDA contexts for devices that do not support mapped pinned memory. The failure is deferred to :py:obj:`~.cuMemHostGetDevicePointer()` because the memory may be mapped into other CUDA contexts via the :py:obj:`~.CU_MEMHOSTALLOC_PORTABLE` flag. The memory allocated by this function must be freed with :py:obj:`~.cuMemFreeHost()`. Note all host memory allocated using :py:obj:`~.cuMemHostAlloc()` will automatically be immediately accessible to all contexts on all devices which support unified addressing (as may be queried using :py:obj:`~.CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING`). Unless the flag :py:obj:`~.CU_MEMHOSTALLOC_WRITECOMBINED` is specified, the device pointer that may be used to access this host memory from those contexts is always equal to the returned host pointer `*pp`. If the flag :py:obj:`~.CU_MEMHOSTALLOC_WRITECOMBINED` is specified, then the function :py:obj:`~.cuMemHostGetDevicePointer()` must be used to query the device pointer, even if the context supports unified addressing. See :py:obj:`~.Unified Addressing` for additional details. Parameters ---------- bytesize : size_t Requested allocation size in bytes Flags : unsigned int Flags for allocation request Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` pp : Any Returned pointer to host memory See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaHostAlloc` """ cdef void_ptr pp = 0 err = ccuda.cuMemHostAlloc(&pp, bytesize, Flags) return (CUresult(err), pp) {{endif}} {{if 'cuMemHostGetDevicePointer_v2' in found_functions}} @cython.embedsignature(True) def cuMemHostGetDevicePointer(p, unsigned int Flags): """ Passes back device pointer of mapped pinned memory. Passes back the device pointer `pdptr` corresponding to the mapped, pinned host buffer `p` allocated by :py:obj:`~.cuMemHostAlloc`. :py:obj:`~.cuMemHostGetDevicePointer()` will fail if the :py:obj:`~.CU_MEMHOSTALLOC_DEVICEMAP` flag was not specified at the time the memory was allocated, or if the function is called on a GPU that does not support mapped pinned memory. For devices that have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_USE_HOST_POINTER_FOR_REGISTERED_MEM`, the memory can also be accessed from the device using the host pointer `p`. The device pointer returned by :py:obj:`~.cuMemHostGetDevicePointer()` may or may not match the original host pointer `p` and depends on the devices visible to the application. If all devices visible to the application have a non-zero value for the device attribute, the device pointer returned by :py:obj:`~.cuMemHostGetDevicePointer()` will match the original pointer `p`. If any device visible to the application has a zero value for the device attribute, the device pointer returned by :py:obj:`~.cuMemHostGetDevicePointer()` will not match the original host pointer `p`, but it will be suitable for use on all devices provided Unified Virtual Addressing is enabled. In such systems, it is valid to access the memory using either pointer on devices that have a non-zero value for the device attribute. Note however that such devices should access the memory using only one of the two pointers and not both. `Flags` provides for future releases. For now, it must be set to 0. Parameters ---------- p : Any Host pointer Flags : unsigned int Options (must be 0) Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pdptr : :py:obj:`~.CUdeviceptr` Returned device pointer See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaHostGetDevicePointer` """ cdef CUdeviceptr pdptr = CUdeviceptr() cp = utils.HelperInputVoidPtr(p) cdef void* cp_ptr = cp.cptr err = ccuda.cuMemHostGetDevicePointer(pdptr._ptr, cp_ptr, Flags) return (CUresult(err), pdptr) {{endif}} {{if 'cuMemHostGetFlags' in found_functions}} @cython.embedsignature(True) def cuMemHostGetFlags(p): """ Passes back flags that were used for a pinned allocation. Passes back the flags `pFlags` that were specified when allocating the pinned host buffer `p` allocated by :py:obj:`~.cuMemHostAlloc`. :py:obj:`~.cuMemHostGetFlags()` will fail if the pointer does not reside in an allocation performed by :py:obj:`~.cuMemAllocHost()` or :py:obj:`~.cuMemHostAlloc()`. Parameters ---------- p : Any Host pointer Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pFlags : unsigned int Returned flags word See Also -------- :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cudaHostGetFlags` """ cdef unsigned int pFlags = 0 cp = utils.HelperInputVoidPtr(p) cdef void* cp_ptr = cp.cptr err = ccuda.cuMemHostGetFlags(&pFlags, cp_ptr) return (CUresult(err), pFlags) {{endif}} {{if 'cuMemAllocManaged' in found_functions}} @cython.embedsignature(True) def cuMemAllocManaged(size_t bytesize, unsigned int flags): """ Allocates memory that will be automatically managed by the Unified Memory system. Allocates `bytesize` bytes of managed memory on the device and returns in `*dptr` a pointer to the allocated memory. If the device doesn't support allocating managed memory, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` is returned. Support for managed memory can be queried using the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MANAGED_MEMORY`. The allocated memory is suitably aligned for any kind of variable. The memory is not cleared. If `bytesize` is 0, :py:obj:`~.cuMemAllocManaged` returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. The pointer is valid on the CPU and on all GPUs in the system that support managed memory. All accesses to this pointer must obey the Unified Memory programming model. `flags` specifies the default stream association for this allocation. `flags` must be one of :py:obj:`~.CU_MEM_ATTACH_GLOBAL` or :py:obj:`~.CU_MEM_ATTACH_HOST`. If :py:obj:`~.CU_MEM_ATTACH_GLOBAL` is specified, then this memory is accessible from any stream on any device. If :py:obj:`~.CU_MEM_ATTACH_HOST` is specified, then the allocation should not be accessed from devices that have a zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`; an explicit call to :py:obj:`~.cuStreamAttachMemAsync` will be required to enable access on such devices. If the association is later changed via :py:obj:`~.cuStreamAttachMemAsync` to a single stream, the default association as specified during :py:obj:`~.cuMemAllocManaged` is restored when that stream is destroyed. For managed variables, the default association is always :py:obj:`~.CU_MEM_ATTACH_GLOBAL`. Note that destroying a stream is an asynchronous operation, and as a result, the change to default association won't happen until all work in the stream has completed. Memory allocated with :py:obj:`~.cuMemAllocManaged` should be released with :py:obj:`~.cuMemFree`. Device memory oversubscription is possible for GPUs that have a non- zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. Managed memory on such GPUs may be evicted from device memory to host memory at any time by the Unified Memory driver in order to make room for other allocations. In a system where all GPUs have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`, managed memory may not be populated when this API returns and instead may be populated on access. In such systems, managed memory can migrate to any processor's memory at any time. The Unified Memory driver will employ heuristics to maintain data locality and prevent excessive page faults to the extent possible. The application can also guide the driver about memory usage patterns via :py:obj:`~.cuMemAdvise`. The application can also explicitly migrate memory to a desired processor's memory via :py:obj:`~.cuMemPrefetchAsync`. In a multi-GPU system where all of the GPUs have a zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS` and all the GPUs have peer-to-peer support with each other, the physical storage for managed memory is created on the GPU which is active at the time :py:obj:`~.cuMemAllocManaged` is called. All other GPUs will reference the data at reduced bandwidth via peer mappings over the PCIe bus. The Unified Memory driver does not migrate memory among such GPUs. In a multi-GPU system where not all GPUs have peer-to-peer support with each other and where the value of the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS` is zero for at least one of those GPUs, the location chosen for physical storage of managed memory is system-dependent. - On Linux, the location chosen will be device memory as long as the current set of active contexts are on devices that either have peer- to-peer support with each other or have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. If there is an active context on a GPU that does not have a non-zero value for that device attribute and it does not have peer-to-peer support with the other devices that have active contexts on them, then the location for physical storage will be 'zero-copy' or host memory. Note that this means that managed memory that is located in device memory is migrated to host memory if a new context is created on a GPU that doesn't have a non-zero value for the device attribute and does not support peer-to-peer with at least one of the other devices that has an active context. This in turn implies that context creation may fail if there is insufficient host memory to migrate all managed allocations. - On Windows, the physical storage is always created in 'zero-copy' or host memory. All GPUs will reference the data at reduced bandwidth over the PCIe bus. In these circumstances, use of the environment variable CUDA_VISIBLE_DEVICES is recommended to restrict CUDA to only use those GPUs that have peer-to-peer support. Alternatively, users can also set CUDA_MANAGED_FORCE_DEVICE_ALLOC to a non-zero value to force the driver to always use device memory for physical storage. When this environment variable is set to a non-zero value, all contexts created in that process on devices that support managed memory have to be peer-to-peer compatible with each other. Context creation will fail if a context is created on a device that supports managed memory and is not peer-to-peer compatible with any of the other managed memory supporting devices on which contexts were previously created, even if those contexts have been destroyed. These environment variables are described in the CUDA programming guide under the "CUDA environment variables" section. - On ARM, managed memory is not available on discrete gpu with Drive PX-2. Parameters ---------- bytesize : size_t Requested allocation size in bytes flags : unsigned int Must be one of :py:obj:`~.CU_MEM_ATTACH_GLOBAL` or :py:obj:`~.CU_MEM_ATTACH_HOST` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` dptr : :py:obj:`~.CUdeviceptr` Returned device pointer See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuStreamAttachMemAsync`, :py:obj:`~.cudaMallocManaged` """ cdef CUdeviceptr dptr = CUdeviceptr() err = ccuda.cuMemAllocManaged(dptr._ptr, bytesize, flags) return (CUresult(err), dptr) {{endif}} {{if 'cuDeviceRegisterAsyncNotification' in found_functions}} @cython.embedsignature(True) def cuDeviceRegisterAsyncNotification(device, callbackFunc, userData): """ Registers a callback function to receive async notifications. Registers `callbackFunc` to receive async notifications. The `userData` parameter is passed to the callback function at async notification time. Likewise, `callback` is also passed to the callback function to distinguish between multiple registered callbacks. The callback function being registered should be designed to return quickly (~10ms). Any long running tasks should be queued for execution on an application thread. Callbacks may not call cuDeviceRegisterAsyncNotification or cuDeviceUnregisterAsyncNotification. Doing so will result in :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`. Async notification callbacks execute in an undefined order and may be serialized. Returns in `*callback` a handle representing the registered callback instance. Parameters ---------- device : :py:obj:`~.CUdevice` The device on which to register the callback callbackFunc : :py:obj:`~.CUasyncCallback` The function to register as a callback userData : Any A generic pointer to user data. This is passed into the callback function. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` :py:obj:`~.CUDA_ERROR_NOT_PERMITTED` :py:obj:`~.CUDA_ERROR_UNKNOWN` callback : :py:obj:`~.CUasyncCallbackHandle` A handle representing the registered callback instance See Also -------- :py:obj:`~.cuDeviceUnregisterAsyncNotification` """ cdef ccuda.CUasyncCallback ccallbackFunc if callbackFunc is None: ccallbackFunc = 0 elif isinstance(callbackFunc, (CUasyncCallback,)): pcallbackFunc = int(callbackFunc) ccallbackFunc = pcallbackFunc else: pcallbackFunc = int(CUasyncCallback(callbackFunc)) ccallbackFunc = pcallbackFunc cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cuserData = utils.HelperInputVoidPtr(userData) cdef void* cuserData_ptr = cuserData.cptr cdef CUasyncCallbackHandle callback = CUasyncCallbackHandle() err = ccuda.cuDeviceRegisterAsyncNotification(cdevice, ccallbackFunc, cuserData_ptr, callback._ptr) return (CUresult(err), callback) {{endif}} {{if 'cuDeviceUnregisterAsyncNotification' in found_functions}} @cython.embedsignature(True) def cuDeviceUnregisterAsyncNotification(device, callback): """ Unregisters an async notification callback. Unregisters `callback` so that the corresponding callback function will stop receiving async notifications. Parameters ---------- device : :py:obj:`~.CUdevice` The device from which to remove `callback`. callback : :py:obj:`~.CUasyncCallbackHandle` The callback instance to unregister from receiving async notifications. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` :py:obj:`~.CUDA_ERROR_NOT_PERMITTED` :py:obj:`~.CUDA_ERROR_UNKNOWN` See Also -------- :py:obj:`~.cuDeviceRegisterAsyncNotification` """ cdef ccuda.CUasyncCallbackHandle ccallback if callback is None: ccallback = 0 elif isinstance(callback, (CUasyncCallbackHandle,)): pcallback = int(callback) ccallback = pcallback else: pcallback = int(CUasyncCallbackHandle(callback)) ccallback = pcallback cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice err = ccuda.cuDeviceUnregisterAsyncNotification(cdevice, ccallback) return (CUresult(err),) {{endif}} {{if 'cuDeviceGetByPCIBusId' in found_functions}} @cython.embedsignature(True) def cuDeviceGetByPCIBusId(char* pciBusId): """ Returns a handle to a compute device. Returns in `*device` a device handle given a PCI bus ID string. where `domain`, `bus`, `device`, and `function` are all hexadecimal values Parameters ---------- pciBusId : bytes String in one of the following forms: Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` dev : :py:obj:`~.CUdevice` Returned device handle See Also -------- :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetPCIBusId`, :py:obj:`~.cudaDeviceGetByPCIBusId` """ cdef CUdevice dev = CUdevice() err = ccuda.cuDeviceGetByPCIBusId(dev._ptr, pciBusId) return (CUresult(err), dev) {{endif}} {{if 'cuDeviceGetPCIBusId' in found_functions}} @cython.embedsignature(True) def cuDeviceGetPCIBusId(int length, dev): """ Returns a PCI Bus Id string for the device. Returns an ASCII string identifying the device `dev` in the NULL- terminated string pointed to by `pciBusId`. `length` specifies the maximum length of the string that may be returned. where `domain`, `bus`, `device`, and `function` are all hexadecimal values. pciBusId should be large enough to store 13 characters including the NULL-terminator. Parameters ---------- length : int Maximum length of string to store in `name` dev : :py:obj:`~.CUdevice` Device to get identifier string for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` pciBusId : bytes Returned identifier string for the device in the following format See Also -------- :py:obj:`~.cuDeviceGet`, :py:obj:`~.cuDeviceGetAttribute`, :py:obj:`~.cuDeviceGetByPCIBusId`, :py:obj:`~.cudaDeviceGetPCIBusId` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev pypciBusId = b" " * length cdef char* pciBusId = pypciBusId err = ccuda.cuDeviceGetPCIBusId(pciBusId, length, cdev) return (CUresult(err), pypciBusId) {{endif}} {{if 'cuIpcGetEventHandle' in found_functions}} @cython.embedsignature(True) def cuIpcGetEventHandle(event): """ Gets an interprocess handle for a previously allocated event. Takes as input a previously allocated event. This event must have been created with the :py:obj:`~.CU_EVENT_INTERPROCESS` and :py:obj:`~.CU_EVENT_DISABLE_TIMING` flags set. This opaque handle may be copied into other processes and opened with :py:obj:`~.cuIpcOpenEventHandle` to allow efficient hardware synchronization between GPU work in different processes. After the event has been opened in the importing process, :py:obj:`~.cuEventRecord`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuStreamWaitEvent` and :py:obj:`~.cuEventQuery` may be used in either process. Performing operations on the imported event after the exported event has been freed with :py:obj:`~.cuEventDestroy` will result in undefined behavior. IPC functionality is restricted to devices with support for unified addressing on Linux and Windows operating systems. IPC functionality on Windows is supported for compatibility purposes but not recommended as it comes with performance cost. Users can test their device for IPC functionality by calling :py:obj:`~.cuapiDeviceGetAttribute` with :py:obj:`~.CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED` Parameters ---------- event : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event allocated with :py:obj:`~.CU_EVENT_INTERPROCESS` and :py:obj:`~.CU_EVENT_DISABLE_TIMING` flags. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_MAP_FAILED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pHandle : :py:obj:`~.CUipcEventHandle` Pointer to a user allocated CUipcEventHandle in which to return the opaque event handle See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuIpcOpenEventHandle`, :py:obj:`~.cuIpcGetMemHandle`, :py:obj:`~.cuIpcOpenMemHandle`, :py:obj:`~.cuIpcCloseMemHandle`, :py:obj:`~.cudaIpcGetEventHandle` """ cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent cdef CUipcEventHandle pHandle = CUipcEventHandle() err = ccuda.cuIpcGetEventHandle(pHandle._ptr, cevent) return (CUresult(err), pHandle) {{endif}} {{if 'cuIpcOpenEventHandle' in found_functions}} @cython.embedsignature(True) def cuIpcOpenEventHandle(handle not None : CUipcEventHandle): """ Opens an interprocess event handle for use in the current process. Opens an interprocess event handle exported from another process with :py:obj:`~.cuIpcGetEventHandle`. This function returns a :py:obj:`~.CUevent` that behaves like a locally created event with the :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag specified. This event must be freed with :py:obj:`~.cuEventDestroy`. Performing operations on the imported event after the exported event has been freed with :py:obj:`~.cuEventDestroy` will result in undefined behavior. IPC functionality is restricted to devices with support for unified addressing on Linux and Windows operating systems. IPC functionality on Windows is supported for compatibility purposes but not recommended as it comes with performance cost. Users can test their device for IPC functionality by calling :py:obj:`~.cuapiDeviceGetAttribute` with :py:obj:`~.CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED` Parameters ---------- handle : :py:obj:`~.CUipcEventHandle` Interprocess handle to open Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_MAP_FAILED`, :py:obj:`~.CUDA_ERROR_PEER_ACCESS_UNSUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phEvent : :py:obj:`~.CUevent` Returns the imported event See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuIpcGetEventHandle`, :py:obj:`~.cuIpcGetMemHandle`, :py:obj:`~.cuIpcOpenMemHandle`, :py:obj:`~.cuIpcCloseMemHandle`, :py:obj:`~.cudaIpcOpenEventHandle` """ cdef CUevent phEvent = CUevent() err = ccuda.cuIpcOpenEventHandle(phEvent._ptr, handle._ptr[0]) return (CUresult(err), phEvent) {{endif}} {{if 'cuIpcGetMemHandle' in found_functions}} @cython.embedsignature(True) def cuIpcGetMemHandle(dptr): """ Gets an interprocess memory handle for an existing device memory allocation. Takes a pointer to the base of an existing device memory allocation created with :py:obj:`~.cuMemAlloc` and exports it for use in another process. This is a lightweight operation and may be called multiple times on an allocation without adverse effects. If a region of memory is freed with :py:obj:`~.cuMemFree` and a subsequent call to :py:obj:`~.cuMemAlloc` returns memory with the same device address, :py:obj:`~.cuIpcGetMemHandle` will return a unique handle for the new memory. IPC functionality is restricted to devices with support for unified addressing on Linux and Windows operating systems. IPC functionality on Windows is supported for compatibility purposes but not recommended as it comes with performance cost. Users can test their device for IPC functionality by calling :py:obj:`~.cuapiDeviceGetAttribute` with :py:obj:`~.CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED` Parameters ---------- dptr : :py:obj:`~.CUdeviceptr` Base pointer to previously allocated device memory Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_MAP_FAILED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pHandle : :py:obj:`~.CUipcMemHandle` Pointer to user allocated :py:obj:`~.CUipcMemHandle` to return the handle in. See Also -------- :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuIpcGetEventHandle`, :py:obj:`~.cuIpcOpenEventHandle`, :py:obj:`~.cuIpcOpenMemHandle`, :py:obj:`~.cuIpcCloseMemHandle`, :py:obj:`~.cudaIpcGetMemHandle` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr cdef CUipcMemHandle pHandle = CUipcMemHandle() err = ccuda.cuIpcGetMemHandle(pHandle._ptr, cdptr) return (CUresult(err), pHandle) {{endif}} {{if 'cuIpcOpenMemHandle_v2' in found_functions}} @cython.embedsignature(True) def cuIpcOpenMemHandle(handle not None : CUipcMemHandle, unsigned int Flags): """ Opens an interprocess memory handle exported from another process and returns a device pointer usable in the local process. Maps memory exported from another process with :py:obj:`~.cuIpcGetMemHandle` into the current device address space. For contexts on different devices :py:obj:`~.cuIpcOpenMemHandle` can attempt to enable peer access between the devices as if the user called :py:obj:`~.cuCtxEnablePeerAccess`. This behavior is controlled by the :py:obj:`~.CU_IPC_MEM_LAZY_ENABLE_PEER_ACCESS` flag. :py:obj:`~.cuDeviceCanAccessPeer` can determine if a mapping is possible. Contexts that may open :py:obj:`~.CUipcMemHandles` are restricted in the following way. :py:obj:`~.CUipcMemHandles` from each :py:obj:`~.CUdevice` in a given process may only be opened by one :py:obj:`~.CUcontext` per :py:obj:`~.CUdevice` per other process. If the memory handle has already been opened by the current context, the reference count on the handle is incremented by 1 and the existing device pointer is returned. Memory returned from :py:obj:`~.cuIpcOpenMemHandle` must be freed with :py:obj:`~.cuIpcCloseMemHandle`. Calling :py:obj:`~.cuMemFree` on an exported memory region before calling :py:obj:`~.cuIpcCloseMemHandle` in the importing context will result in undefined behavior. IPC functionality is restricted to devices with support for unified addressing on Linux and Windows operating systems. IPC functionality on Windows is supported for compatibility purposes but not recommended as it comes with performance cost. Users can test their device for IPC functionality by calling :py:obj:`~.cuapiDeviceGetAttribute` with :py:obj:`~.CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED` Parameters ---------- handle : :py:obj:`~.CUipcMemHandle` :py:obj:`~.CUipcMemHandle` to open Flags : unsigned int Flags for this operation. Must be specified as :py:obj:`~.CU_IPC_MEM_LAZY_ENABLE_PEER_ACCESS` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_MAP_FAILED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_TOO_MANY_PEERS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pdptr : :py:obj:`~.CUdeviceptr` Returned device pointer See Also -------- :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuIpcGetEventHandle`, :py:obj:`~.cuIpcOpenEventHandle`, :py:obj:`~.cuIpcGetMemHandle`, :py:obj:`~.cuIpcCloseMemHandle`, :py:obj:`~.cuCtxEnablePeerAccess`, :py:obj:`~.cuDeviceCanAccessPeer`, :py:obj:`~.cudaIpcOpenMemHandle` Notes ----- No guarantees are made about the address returned in `*pdptr`. In particular, multiple processes may not receive the same address for the same `handle`. """ cdef CUdeviceptr pdptr = CUdeviceptr() err = ccuda.cuIpcOpenMemHandle(pdptr._ptr, handle._ptr[0], Flags) return (CUresult(err), pdptr) {{endif}} {{if 'cuIpcCloseMemHandle' in found_functions}} @cython.embedsignature(True) def cuIpcCloseMemHandle(dptr): """ Attempts to close memory mapped with :py:obj:`~.cuIpcOpenMemHandle`. Decrements the reference count of the memory returned by :py:obj:`~.cuIpcOpenMemHandle` by 1. When the reference count reaches 0, this API unmaps the memory. The original allocation in the exporting process as well as imported mappings in other processes will be unaffected. Any resources used to enable peer access will be freed if this is the last mapping using them. IPC functionality is restricted to devices with support for unified addressing on Linux and Windows operating systems. IPC functionality on Windows is supported for compatibility purposes but not recommended as it comes with performance cost. Users can test their device for IPC functionality by calling :py:obj:`~.cuapiDeviceGetAttribute` with :py:obj:`~.CU_DEVICE_ATTRIBUTE_IPC_EVENT_SUPPORTED` Parameters ---------- dptr : :py:obj:`~.CUdeviceptr` Device pointer returned by :py:obj:`~.cuIpcOpenMemHandle` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_MAP_FAILED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuIpcGetEventHandle`, :py:obj:`~.cuIpcOpenEventHandle`, :py:obj:`~.cuIpcGetMemHandle`, :py:obj:`~.cuIpcOpenMemHandle`, :py:obj:`~.cudaIpcCloseMemHandle` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr err = ccuda.cuIpcCloseMemHandle(cdptr) return (CUresult(err),) {{endif}} {{if 'cuMemHostRegister_v2' in found_functions}} @cython.embedsignature(True) def cuMemHostRegister(p, size_t bytesize, unsigned int Flags): """ Registers an existing host memory range for use by CUDA. Page-locks the memory range specified by `p` and `bytesize` and maps it for the device(s) as specified by `Flags`. This memory range also is added to the same tracking mechanism as :py:obj:`~.cuMemHostAlloc` to automatically accelerate calls to functions such as :py:obj:`~.cuMemcpyHtoD()`. Since the memory can be accessed directly by the device, it can be read or written with much higher bandwidth than pageable memory that has not been registered. Page-locking excessive amounts of memory may degrade system performance, since it reduces the amount of memory available to the system for paging. As a result, this function is best used sparingly to register staging areas for data exchange between host and device. On systems where :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES` is true, :py:obj:`~.cuMemHostRegister` will not page-lock the memory range specified by `ptr` but only populate unpopulated pages. The `Flags` parameter enables different options to be specified that affect the allocation, as follows. - :py:obj:`~.CU_MEMHOSTREGISTER_PORTABLE`: The memory returned by this call will be considered as pinned memory by all CUDA contexts, not just the one that performed the allocation. - :py:obj:`~.CU_MEMHOSTREGISTER_DEVICEMAP`: Maps the allocation into the CUDA address space. The device pointer to the memory may be obtained by calling :py:obj:`~.cuMemHostGetDevicePointer()`. - :py:obj:`~.CU_MEMHOSTREGISTER_IOMEMORY`: The pointer is treated as pointing to some I/O memory space, e.g. the PCI Express resource of a 3rd party device. - :py:obj:`~.CU_MEMHOSTREGISTER_READ_ONLY`: The pointer is treated as pointing to memory that is considered read-only by the device. On platforms without :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, this flag is required in order to register memory mapped to the CPU as read-only. Support for the use of this flag can be queried from the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_READ_ONLY_HOST_REGISTER_SUPPORTED`. Using this flag with a current context associated with a device that does not have this attribute set will cause :py:obj:`~.cuMemHostRegister` to error with CUDA_ERROR_NOT_SUPPORTED. All of these flags are orthogonal to one another: a developer may page- lock memory that is portable or mapped with no restrictions. The :py:obj:`~.CU_MEMHOSTREGISTER_DEVICEMAP` flag may be specified on CUDA contexts for devices that do not support mapped pinned memory. The failure is deferred to :py:obj:`~.cuMemHostGetDevicePointer()` because the memory may be mapped into other CUDA contexts via the :py:obj:`~.CU_MEMHOSTREGISTER_PORTABLE` flag. For devices that have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_USE_HOST_POINTER_FOR_REGISTERED_MEM`, the memory can also be accessed from the device using the host pointer `p`. The device pointer returned by :py:obj:`~.cuMemHostGetDevicePointer()` may or may not match the original host pointer `ptr` and depends on the devices visible to the application. If all devices visible to the application have a non-zero value for the device attribute, the device pointer returned by :py:obj:`~.cuMemHostGetDevicePointer()` will match the original pointer `ptr`. If any device visible to the application has a zero value for the device attribute, the device pointer returned by :py:obj:`~.cuMemHostGetDevicePointer()` will not match the original host pointer `ptr`, but it will be suitable for use on all devices provided Unified Virtual Addressing is enabled. In such systems, it is valid to access the memory using either pointer on devices that have a non-zero value for the device attribute. Note however that such devices should access the memory using only of the two pointers and not both. The memory page-locked by this function must be unregistered with :py:obj:`~.cuMemHostUnregister()`. Parameters ---------- p : Any Host pointer to memory to page-lock bytesize : size_t Size in bytes of the address range to page-lock Flags : unsigned int Flags for allocation request Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_HOST_MEMORY_ALREADY_REGISTERED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMemHostUnregister`, :py:obj:`~.cuMemHostGetFlags`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cudaHostRegister` """ cp = utils.HelperInputVoidPtr(p) cdef void* cp_ptr = cp.cptr err = ccuda.cuMemHostRegister(cp_ptr, bytesize, Flags) return (CUresult(err),) {{endif}} {{if 'cuMemHostUnregister' in found_functions}} @cython.embedsignature(True) def cuMemHostUnregister(p): """ Unregisters a memory range that was registered with cuMemHostRegister. Unmaps the memory range whose base address is specified by `p`, and makes it pageable again. The base address must be the same one specified to :py:obj:`~.cuMemHostRegister()`. Parameters ---------- p : Any Host pointer to memory to unregister Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_HOST_MEMORY_NOT_REGISTERED`, See Also -------- :py:obj:`~.cuMemHostRegister`, :py:obj:`~.cudaHostUnregister` """ cp = utils.HelperInputVoidPtr(p) cdef void* cp_ptr = cp.cptr err = ccuda.cuMemHostUnregister(cp_ptr) return (CUresult(err),) {{endif}} {{if 'cuMemcpy' in found_functions}} @cython.embedsignature(True) def cuMemcpy(dst, src, size_t ByteCount): """ Copies memory. Copies data between two pointers. `dst` and `src` are base pointers of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Note that this function infers the type of the transfer (host to host, host to device, device to device, or device to host) from the pointer values. This function is only allowed in contexts which support unified addressing. Parameters ---------- dst : :py:obj:`~.CUdeviceptr` Destination unified virtual address space pointer src : :py:obj:`~.CUdeviceptr` Source unified virtual address space pointer ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpy`, :py:obj:`~.cudaMemcpyToSymbol`, :py:obj:`~.cudaMemcpyFromSymbol` """ cdef ccuda.CUdeviceptr csrc if src is None: csrc = 0 elif isinstance(src, (CUdeviceptr,)): psrc = int(src) csrc = psrc else: psrc = int(CUdeviceptr(src)) csrc = psrc cdef ccuda.CUdeviceptr cdst if dst is None: cdst = 0 elif isinstance(dst, (CUdeviceptr,)): pdst = int(dst) cdst = pdst else: pdst = int(CUdeviceptr(dst)) cdst = pdst err = ccuda.cuMemcpy(cdst, csrc, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyPeer' in found_functions}} @cython.embedsignature(True) def cuMemcpyPeer(dstDevice, dstContext, srcDevice, srcContext, size_t ByteCount): """ Copies device memory between two contexts. Copies from device memory in one context to device memory in another context. `dstDevice` is the base device pointer of the destination memory and `dstContext` is the destination context. `srcDevice` is the base device pointer of the source memory and `srcContext` is the source pointer. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstContext : :py:obj:`~.CUcontext` Destination context srcDevice : :py:obj:`~.CUdeviceptr` Source device pointer srcContext : :py:obj:`~.CUcontext` Source context ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpy3DPeer`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyPeerAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cudaMemcpyPeer` """ cdef ccuda.CUcontext csrcContext if srcContext is None: csrcContext = 0 elif isinstance(srcContext, (CUcontext,)): psrcContext = int(srcContext) csrcContext = psrcContext else: psrcContext = int(CUcontext(srcContext)) csrcContext = psrcContext cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice cdef ccuda.CUcontext cdstContext if dstContext is None: cdstContext = 0 elif isinstance(dstContext, (CUcontext,)): pdstContext = int(dstContext) cdstContext = pdstContext else: pdstContext = int(CUcontext(dstContext)) cdstContext = pdstContext cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemcpyPeer(cdstDevice, cdstContext, csrcDevice, csrcContext, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyHtoD_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyHtoD(dstDevice, srcHost, size_t ByteCount): """ Copies memory from Host to Device. Copies from host memory to device memory. `dstDevice` and `srcHost` are the base addresses of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer srcHost : Any Source host pointer ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpy`, :py:obj:`~.cudaMemcpyToSymbol` """ cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice csrcHost = utils.HelperInputVoidPtr(srcHost) cdef void* csrcHost_ptr = csrcHost.cptr err = ccuda.cuMemcpyHtoD(cdstDevice, csrcHost_ptr, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyDtoH_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyDtoH(dstHost, srcDevice, size_t ByteCount): """ Copies memory from Device to Host. Copies from device to host memory. `dstHost` and `srcDevice` specify the base pointers of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstHost : Any Destination host pointer srcDevice : :py:obj:`~.CUdeviceptr` Source device pointer ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpy`, :py:obj:`~.cudaMemcpyFromSymbol` """ cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice cdstHost = utils.HelperInputVoidPtr(dstHost) cdef void* cdstHost_ptr = cdstHost.cptr err = ccuda.cuMemcpyDtoH(cdstHost_ptr, csrcDevice, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyDtoD_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyDtoD(dstDevice, srcDevice, size_t ByteCount): """ Copies memory from Device to Device. Copies from device memory to device memory. `dstDevice` and `srcDevice` are the base pointers of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer srcDevice : :py:obj:`~.CUdeviceptr` Source device pointer ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpy`, :py:obj:`~.cudaMemcpyToSymbol`, :py:obj:`~.cudaMemcpyFromSymbol` """ cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemcpyDtoD(cdstDevice, csrcDevice, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyDtoA_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyDtoA(dstArray, size_t dstOffset, srcDevice, size_t ByteCount): """ Copies memory from Device to Array. Copies from device memory to a 1D CUDA array. `dstArray` and `dstOffset` specify the CUDA array handle and starting index of the destination data. `srcDevice` specifies the base pointer of the source. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstArray : :py:obj:`~.CUarray` Destination array dstOffset : size_t Offset in bytes of destination array srcDevice : :py:obj:`~.CUdeviceptr` Source device pointer ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpyToArray` """ cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice cdef ccuda.CUarray cdstArray if dstArray is None: cdstArray = 0 elif isinstance(dstArray, (CUarray,)): pdstArray = int(dstArray) cdstArray = pdstArray else: pdstArray = int(CUarray(dstArray)) cdstArray = pdstArray err = ccuda.cuMemcpyDtoA(cdstArray, dstOffset, csrcDevice, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyAtoD_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyAtoD(dstDevice, srcArray, size_t srcOffset, size_t ByteCount): """ Copies memory from Array to Device. Copies from one 1D CUDA array to device memory. `dstDevice` specifies the base pointer of the destination and must be naturally aligned with the CUDA array elements. `srcArray` and `srcOffset` specify the CUDA array handle and the offset in bytes into the array where the copy is to begin. `ByteCount` specifies the number of bytes to copy and must be evenly divisible by the array element size. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer srcArray : :py:obj:`~.CUarray` Source array srcOffset : size_t Offset in bytes of source array ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpyFromArray` """ cdef ccuda.CUarray csrcArray if srcArray is None: csrcArray = 0 elif isinstance(srcArray, (CUarray,)): psrcArray = int(srcArray) csrcArray = psrcArray else: psrcArray = int(CUarray(srcArray)) csrcArray = psrcArray cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemcpyAtoD(cdstDevice, csrcArray, srcOffset, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyHtoA_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyHtoA(dstArray, size_t dstOffset, srcHost, size_t ByteCount): """ Copies memory from Host to Array. Copies from host memory to a 1D CUDA array. `dstArray` and `dstOffset` specify the CUDA array handle and starting offset in bytes of the destination data. `pSrc` specifies the base address of the source. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstArray : :py:obj:`~.CUarray` Destination array dstOffset : size_t Offset in bytes of destination array srcHost : Any Source host pointer ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpyToArray` """ cdef ccuda.CUarray cdstArray if dstArray is None: cdstArray = 0 elif isinstance(dstArray, (CUarray,)): pdstArray = int(dstArray) cdstArray = pdstArray else: pdstArray = int(CUarray(dstArray)) cdstArray = pdstArray csrcHost = utils.HelperInputVoidPtr(srcHost) cdef void* csrcHost_ptr = csrcHost.cptr err = ccuda.cuMemcpyHtoA(cdstArray, dstOffset, csrcHost_ptr, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyAtoH_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyAtoH(dstHost, srcArray, size_t srcOffset, size_t ByteCount): """ Copies memory from Array to Host. Copies from one 1D CUDA array to host memory. `dstHost` specifies the base pointer of the destination. `srcArray` and `srcOffset` specify the CUDA array handle and starting offset in bytes of the source data. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstHost : Any Destination device pointer srcArray : :py:obj:`~.CUarray` Source array srcOffset : size_t Offset in bytes of source array ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpyFromArray` """ cdef ccuda.CUarray csrcArray if srcArray is None: csrcArray = 0 elif isinstance(srcArray, (CUarray,)): psrcArray = int(srcArray) csrcArray = psrcArray else: psrcArray = int(CUarray(srcArray)) csrcArray = psrcArray cdstHost = utils.HelperInputVoidPtr(dstHost) cdef void* cdstHost_ptr = cdstHost.cptr err = ccuda.cuMemcpyAtoH(cdstHost_ptr, csrcArray, srcOffset, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpyAtoA_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyAtoA(dstArray, size_t dstOffset, srcArray, size_t srcOffset, size_t ByteCount): """ Copies memory from Array to Array. Copies from one 1D CUDA array to another. `dstArray` and `srcArray` specify the handles of the destination and source CUDA arrays for the copy, respectively. `dstOffset` and `srcOffset` specify the destination and source offsets in bytes into the CUDA arrays. `ByteCount` is the number of bytes to be copied. The size of the elements in the CUDA arrays need not be the same format, but the elements must be the same size; and count must be evenly divisible by that size. Parameters ---------- dstArray : :py:obj:`~.CUarray` Destination array dstOffset : size_t Offset in bytes of destination array srcArray : :py:obj:`~.CUarray` Source array srcOffset : size_t Offset in bytes of source array ByteCount : size_t Size of memory copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpyArrayToArray` """ cdef ccuda.CUarray csrcArray if srcArray is None: csrcArray = 0 elif isinstance(srcArray, (CUarray,)): psrcArray = int(srcArray) csrcArray = psrcArray else: psrcArray = int(CUarray(srcArray)) csrcArray = psrcArray cdef ccuda.CUarray cdstArray if dstArray is None: cdstArray = 0 elif isinstance(dstArray, (CUarray,)): pdstArray = int(dstArray) cdstArray = pdstArray else: pdstArray = int(CUarray(dstArray)) cdstArray = pdstArray err = ccuda.cuMemcpyAtoA(cdstArray, dstOffset, csrcArray, srcOffset, ByteCount) return (CUresult(err),) {{endif}} {{if 'cuMemcpy2D_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpy2D(pCopy : Optional[CUDA_MEMCPY2D]): """ Copies memory for 2D arrays. Perform a 2D memory copy according to the parameters specified in `pCopy`. The :py:obj:`~.CUDA_MEMCPY2D` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.srcMemoryType` and :py:obj:`~.dstMemoryType` specify the type of memory of the source and destination, respectively; :py:obj:`~.CUmemorytype_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.srcHost` and :py:obj:`~.srcPitch` specify the (host) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (device) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.srcArray` specifies the handle of the source data. :py:obj:`~.srcHost`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` are ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.dstHost` and :py:obj:`~.dstPitch` specify the (host) base address of the destination data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (device) base address of the destination data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.dstArray` specifies the handle of the destination data. :py:obj:`~.dstHost`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` are ignored. - :py:obj:`~.srcXInBytes` and :py:obj:`~.srcY` specify the base address of the source data for the copy. For host pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.srcXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.dstXInBytes` and :py:obj:`~.dstY` specify the base address of the destination data for the copy. For host pointers, the base address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.dstXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.WidthInBytes` and :py:obj:`~.Height` specify the width (in bytes) and height of the 2D copy being performed. - If specified, :py:obj:`~.srcPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + :py:obj:`~.srcXInBytes`, and :py:obj:`~.dstPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + dstXInBytes. :py:obj:`~.cuMemcpy2D()` returns an error if any pitch is greater than the maximum allowed (:py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_PITCH`). :py:obj:`~.cuMemAllocPitch()` passes back pitches that always work with :py:obj:`~.cuMemcpy2D()`. On intra-device memory copies (device to device, CUDA array to device, CUDA array to CUDA array), :py:obj:`~.cuMemcpy2D()` may fail for pitches not computed by :py:obj:`~.cuMemAllocPitch()`. :py:obj:`~.cuMemcpy2DUnaligned()` does not have this restriction, but may run significantly slower in the cases where :py:obj:`~.cuMemcpy2D()` would have returned an error code. Parameters ---------- pCopy : :py:obj:`~.CUDA_MEMCPY2D` Parameters for the memory copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpy2D`, :py:obj:`~.cudaMemcpy2DToArray`, :py:obj:`~.cudaMemcpy2DFromArray` """ cdef ccuda.CUDA_MEMCPY2D* cpCopy_ptr = pCopy._ptr if pCopy != None else NULL err = ccuda.cuMemcpy2D(cpCopy_ptr) return (CUresult(err),) {{endif}} {{if 'cuMemcpy2DUnaligned_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpy2DUnaligned(pCopy : Optional[CUDA_MEMCPY2D]): """ Copies memory for 2D arrays. Perform a 2D memory copy according to the parameters specified in `pCopy`. The :py:obj:`~.CUDA_MEMCPY2D` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.srcMemoryType` and :py:obj:`~.dstMemoryType` specify the type of memory of the source and destination, respectively; :py:obj:`~.CUmemorytype_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.srcHost` and :py:obj:`~.srcPitch` specify the (host) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (device) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.srcArray` specifies the handle of the source data. :py:obj:`~.srcHost`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` are ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.dstHost` and :py:obj:`~.dstPitch` specify the (host) base address of the destination data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (device) base address of the destination data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.dstArray` specifies the handle of the destination data. :py:obj:`~.dstHost`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` are ignored. - :py:obj:`~.srcXInBytes` and :py:obj:`~.srcY` specify the base address of the source data for the copy. For host pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.srcXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.dstXInBytes` and :py:obj:`~.dstY` specify the base address of the destination data for the copy. For host pointers, the base address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.dstXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.WidthInBytes` and :py:obj:`~.Height` specify the width (in bytes) and height of the 2D copy being performed. - If specified, :py:obj:`~.srcPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + :py:obj:`~.srcXInBytes`, and :py:obj:`~.dstPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + dstXInBytes. :py:obj:`~.cuMemcpy2D()` returns an error if any pitch is greater than the maximum allowed (:py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_PITCH`). :py:obj:`~.cuMemAllocPitch()` passes back pitches that always work with :py:obj:`~.cuMemcpy2D()`. On intra-device memory copies (device to device, CUDA array to device, CUDA array to CUDA array), :py:obj:`~.cuMemcpy2D()` may fail for pitches not computed by :py:obj:`~.cuMemAllocPitch()`. :py:obj:`~.cuMemcpy2DUnaligned()` does not have this restriction, but may run significantly slower in the cases where :py:obj:`~.cuMemcpy2D()` would have returned an error code. Parameters ---------- pCopy : :py:obj:`~.CUDA_MEMCPY2D` Parameters for the memory copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpy2D`, :py:obj:`~.cudaMemcpy2DToArray`, :py:obj:`~.cudaMemcpy2DFromArray` """ cdef ccuda.CUDA_MEMCPY2D* cpCopy_ptr = pCopy._ptr if pCopy != None else NULL err = ccuda.cuMemcpy2DUnaligned(cpCopy_ptr) return (CUresult(err),) {{endif}} {{if 'cuMemcpy3D_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpy3D(pCopy : Optional[CUDA_MEMCPY3D]): """ Copies memory for 3D arrays. Perform a 3D memory copy according to the parameters specified in `pCopy`. The :py:obj:`~.CUDA_MEMCPY3D` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.srcMemoryType` and :py:obj:`~.dstMemoryType` specify the type of memory of the source and destination, respectively; :py:obj:`~.CUmemorytype_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.srcHost`, :py:obj:`~.srcPitch` and :py:obj:`~.srcHeight` specify the (host) base address of the source data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.srcDevice`, :py:obj:`~.srcPitch` and :py:obj:`~.srcHeight` specify the (device) base address of the source data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.srcArray` specifies the handle of the source data. :py:obj:`~.srcHost`, :py:obj:`~.srcDevice`, :py:obj:`~.srcPitch` and :py:obj:`~.srcHeight` are ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.dstHost` and :py:obj:`~.dstPitch` specify the (host) base address of the destination data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (device) base address of the destination data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.dstArray` specifies the handle of the destination data. :py:obj:`~.dstHost`, :py:obj:`~.dstDevice`, :py:obj:`~.dstPitch` and :py:obj:`~.dstHeight` are ignored. - :py:obj:`~.srcXInBytes`, :py:obj:`~.srcY` and :py:obj:`~.srcZ` specify the base address of the source data for the copy. For host pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.srcXInBytes` must be evenly divisible by the array element size. - dstXInBytes, :py:obj:`~.dstY` and :py:obj:`~.dstZ` specify the base address of the destination data for the copy. For host pointers, the base address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.dstXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.WidthInBytes`, :py:obj:`~.Height` and :py:obj:`~.Depth` specify the width (in bytes), height and depth of the 3D copy being performed. - If specified, :py:obj:`~.srcPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + :py:obj:`~.srcXInBytes`, and :py:obj:`~.dstPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + dstXInBytes. - If specified, :py:obj:`~.srcHeight` must be greater than or equal to :py:obj:`~.Height` + :py:obj:`~.srcY`, and :py:obj:`~.dstHeight` must be greater than or equal to :py:obj:`~.Height` + :py:obj:`~.dstY`. :py:obj:`~.cuMemcpy3D()` returns an error if any pitch is greater than the maximum allowed (:py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_PITCH`). The :py:obj:`~.srcLOD` and :py:obj:`~.dstLOD` members of the :py:obj:`~.CUDA_MEMCPY3D` structure must be set to 0. Parameters ---------- pCopy : :py:obj:`~.CUDA_MEMCPY3D` Parameters for the memory copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemcpy3D` """ cdef ccuda.CUDA_MEMCPY3D* cpCopy_ptr = pCopy._ptr if pCopy != None else NULL err = ccuda.cuMemcpy3D(cpCopy_ptr) return (CUresult(err),) {{endif}} {{if 'cuMemcpy3DPeer' in found_functions}} @cython.embedsignature(True) def cuMemcpy3DPeer(pCopy : Optional[CUDA_MEMCPY3D_PEER]): """ Copies memory between contexts. Perform a 3D memory copy according to the parameters specified in `pCopy`. See the definition of the :py:obj:`~.CUDA_MEMCPY3D_PEER` structure for documentation of its parameters. Parameters ---------- pCopy : :py:obj:`~.CUDA_MEMCPY3D_PEER` Parameters for the memory copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyPeer`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyPeerAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cudaMemcpy3DPeer` """ cdef ccuda.CUDA_MEMCPY3D_PEER* cpCopy_ptr = pCopy._ptr if pCopy != None else NULL err = ccuda.cuMemcpy3DPeer(cpCopy_ptr) return (CUresult(err),) {{endif}} {{if 'cuMemcpyAsync' in found_functions}} @cython.embedsignature(True) def cuMemcpyAsync(dst, src, size_t ByteCount, hStream): """ Copies memory asynchronously. Copies data between two pointers. `dst` and `src` are base pointers of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Note that this function infers the type of the transfer (host to host, host to device, device to device, or device to host) from the pointer values. This function is only allowed in contexts which support unified addressing. Parameters ---------- dst : :py:obj:`~.CUdeviceptr` Destination unified virtual address space pointer src : :py:obj:`~.CUdeviceptr` Source unified virtual address space pointer ByteCount : size_t Size of memory copy in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpyAsync`, :py:obj:`~.cudaMemcpyToSymbolAsync`, :py:obj:`~.cudaMemcpyFromSymbolAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr csrc if src is None: csrc = 0 elif isinstance(src, (CUdeviceptr,)): psrc = int(src) csrc = psrc else: psrc = int(CUdeviceptr(src)) csrc = psrc cdef ccuda.CUdeviceptr cdst if dst is None: cdst = 0 elif isinstance(dst, (CUdeviceptr,)): pdst = int(dst) cdst = pdst else: pdst = int(CUdeviceptr(dst)) cdst = pdst err = ccuda.cuMemcpyAsync(cdst, csrc, ByteCount, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpyPeerAsync' in found_functions}} @cython.embedsignature(True) def cuMemcpyPeerAsync(dstDevice, dstContext, srcDevice, srcContext, size_t ByteCount, hStream): """ Copies device memory between two contexts asynchronously. Copies from device memory in one context to device memory in another context. `dstDevice` is the base device pointer of the destination memory and `dstContext` is the destination context. `srcDevice` is the base device pointer of the source memory and `srcContext` is the source pointer. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstContext : :py:obj:`~.CUcontext` Destination context srcDevice : :py:obj:`~.CUdeviceptr` Source device pointer srcContext : :py:obj:`~.CUcontext` Source context ByteCount : size_t Size of memory copy in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyPeer`, :py:obj:`~.cuMemcpy3DPeer`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cudaMemcpyPeerAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUcontext csrcContext if srcContext is None: csrcContext = 0 elif isinstance(srcContext, (CUcontext,)): psrcContext = int(srcContext) csrcContext = psrcContext else: psrcContext = int(CUcontext(srcContext)) csrcContext = psrcContext cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice cdef ccuda.CUcontext cdstContext if dstContext is None: cdstContext = 0 elif isinstance(dstContext, (CUcontext,)): pdstContext = int(dstContext) cdstContext = pdstContext else: pdstContext = int(CUcontext(dstContext)) cdstContext = pdstContext cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemcpyPeerAsync(cdstDevice, cdstContext, csrcDevice, csrcContext, ByteCount, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpyHtoDAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyHtoDAsync(dstDevice, srcHost, size_t ByteCount, hStream): """ Copies memory from Host to Device. Copies from host memory to device memory. `dstDevice` and `srcHost` are the base addresses of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer srcHost : Any Source host pointer ByteCount : size_t Size of memory copy in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpyAsync`, :py:obj:`~.cudaMemcpyToSymbolAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice csrcHost = utils.HelperInputVoidPtr(srcHost) cdef void* csrcHost_ptr = csrcHost.cptr err = ccuda.cuMemcpyHtoDAsync(cdstDevice, csrcHost_ptr, ByteCount, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpyDtoHAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyDtoHAsync(dstHost, srcDevice, size_t ByteCount, hStream): """ Copies memory from Device to Host. Copies from device to host memory. `dstHost` and `srcDevice` specify the base pointers of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstHost : Any Destination host pointer srcDevice : :py:obj:`~.CUdeviceptr` Source device pointer ByteCount : size_t Size of memory copy in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpyAsync`, :py:obj:`~.cudaMemcpyFromSymbolAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice cdstHost = utils.HelperInputVoidPtr(dstHost) cdef void* cdstHost_ptr = cdstHost.cptr err = ccuda.cuMemcpyDtoHAsync(cdstHost_ptr, csrcDevice, ByteCount, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpyDtoDAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyDtoDAsync(dstDevice, srcDevice, size_t ByteCount, hStream): """ Copies memory from Device to Device. Copies from device memory to device memory. `dstDevice` and `srcDevice` are the base pointers of the destination and source, respectively. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer srcDevice : :py:obj:`~.CUdeviceptr` Source device pointer ByteCount : size_t Size of memory copy in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpyAsync`, :py:obj:`~.cudaMemcpyToSymbolAsync`, :py:obj:`~.cudaMemcpyFromSymbolAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdeviceptr,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdeviceptr(srcDevice)) csrcDevice = psrcDevice cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemcpyDtoDAsync(cdstDevice, csrcDevice, ByteCount, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpyHtoAAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyHtoAAsync(dstArray, size_t dstOffset, srcHost, size_t ByteCount, hStream): """ Copies memory from Host to Array. Copies from host memory to a 1D CUDA array. `dstArray` and `dstOffset` specify the CUDA array handle and starting offset in bytes of the destination data. `srcHost` specifies the base address of the source. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstArray : :py:obj:`~.CUarray` Destination array dstOffset : size_t Offset in bytes of destination array srcHost : Any Source host pointer ByteCount : size_t Size of memory copy in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpyToArrayAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUarray cdstArray if dstArray is None: cdstArray = 0 elif isinstance(dstArray, (CUarray,)): pdstArray = int(dstArray) cdstArray = pdstArray else: pdstArray = int(CUarray(dstArray)) cdstArray = pdstArray csrcHost = utils.HelperInputVoidPtr(srcHost) cdef void* csrcHost_ptr = csrcHost.cptr err = ccuda.cuMemcpyHtoAAsync(cdstArray, dstOffset, csrcHost_ptr, ByteCount, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpyAtoHAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpyAtoHAsync(dstHost, srcArray, size_t srcOffset, size_t ByteCount, hStream): """ Copies memory from Array to Host. Copies from one 1D CUDA array to host memory. `dstHost` specifies the base pointer of the destination. `srcArray` and `srcOffset` specify the CUDA array handle and starting offset in bytes of the source data. `ByteCount` specifies the number of bytes to copy. Parameters ---------- dstHost : Any Destination pointer srcArray : :py:obj:`~.CUarray` Source array srcOffset : size_t Offset in bytes of source array ByteCount : size_t Size of memory copy in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpyFromArrayAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUarray csrcArray if srcArray is None: csrcArray = 0 elif isinstance(srcArray, (CUarray,)): psrcArray = int(srcArray) csrcArray = psrcArray else: psrcArray = int(CUarray(srcArray)) csrcArray = psrcArray cdstHost = utils.HelperInputVoidPtr(dstHost) cdef void* cdstHost_ptr = cdstHost.cptr err = ccuda.cuMemcpyAtoHAsync(cdstHost_ptr, csrcArray, srcOffset, ByteCount, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpy2DAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpy2DAsync(pCopy : Optional[CUDA_MEMCPY2D], hStream): """ Copies memory for 2D arrays. Perform a 2D memory copy according to the parameters specified in `pCopy`. The :py:obj:`~.CUDA_MEMCPY2D` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.srcMemoryType` and :py:obj:`~.dstMemoryType` specify the type of memory of the source and destination, respectively; :py:obj:`~.CUmemorytype_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.srcHost` and :py:obj:`~.srcPitch` specify the (host) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (device) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.srcArray` specifies the handle of the source data. :py:obj:`~.srcHost`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` are ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.dstHost` and :py:obj:`~.dstPitch` specify the (host) base address of the destination data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (device) base address of the destination data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.dstArray` specifies the handle of the destination data. :py:obj:`~.dstHost`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` are ignored. - :py:obj:`~.srcXInBytes` and :py:obj:`~.srcY` specify the base address of the source data for the copy. For host pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.srcXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.dstXInBytes` and :py:obj:`~.dstY` specify the base address of the destination data for the copy. For host pointers, the base address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.dstXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.WidthInBytes` and :py:obj:`~.Height` specify the width (in bytes) and height of the 2D copy being performed. - If specified, :py:obj:`~.srcPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + :py:obj:`~.srcXInBytes`, and :py:obj:`~.dstPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + dstXInBytes. - If specified, :py:obj:`~.srcPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + :py:obj:`~.srcXInBytes`, and :py:obj:`~.dstPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + dstXInBytes. - If specified, :py:obj:`~.srcHeight` must be greater than or equal to :py:obj:`~.Height` + :py:obj:`~.srcY`, and :py:obj:`~.dstHeight` must be greater than or equal to :py:obj:`~.Height` + :py:obj:`~.dstY`. :py:obj:`~.cuMemcpy2DAsync()` returns an error if any pitch is greater than the maximum allowed (:py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_PITCH`). :py:obj:`~.cuMemAllocPitch()` passes back pitches that always work with :py:obj:`~.cuMemcpy2D()`. On intra-device memory copies (device to device, CUDA array to device, CUDA array to CUDA array), :py:obj:`~.cuMemcpy2DAsync()` may fail for pitches not computed by :py:obj:`~.cuMemAllocPitch()`. Parameters ---------- pCopy : :py:obj:`~.CUDA_MEMCPY2D` Parameters for the memory copy hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpy2DAsync`, :py:obj:`~.cudaMemcpy2DToArrayAsync`, :py:obj:`~.cudaMemcpy2DFromArrayAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUDA_MEMCPY2D* cpCopy_ptr = pCopy._ptr if pCopy != None else NULL err = ccuda.cuMemcpy2DAsync(cpCopy_ptr, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpy3DAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemcpy3DAsync(pCopy : Optional[CUDA_MEMCPY3D], hStream): """ Copies memory for 3D arrays. Perform a 3D memory copy according to the parameters specified in `pCopy`. The :py:obj:`~.CUDA_MEMCPY3D` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.srcMemoryType` and :py:obj:`~.dstMemoryType` specify the type of memory of the source and destination, respectively; :py:obj:`~.CUmemorytype_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.srcDevice` and :py:obj:`~.srcPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.srcArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.srcHost`, :py:obj:`~.srcPitch` and :py:obj:`~.srcHeight` specify the (host) base address of the source data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.srcDevice`, :py:obj:`~.srcPitch` and :py:obj:`~.srcHeight` specify the (device) base address of the source data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.srcArray` is ignored. If :py:obj:`~.srcMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.srcArray` specifies the handle of the source data. :py:obj:`~.srcHost`, :py:obj:`~.srcDevice`, :py:obj:`~.srcPitch` and :py:obj:`~.srcHeight` are ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_UNIFIED`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (unified virtual address space) base address of the source data and the bytes per row to apply. :py:obj:`~.dstArray` is ignored. This value may be used only if unified addressing is supported in the calling context. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_HOST`, :py:obj:`~.dstHost` and :py:obj:`~.dstPitch` specify the (host) base address of the destination data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_DEVICE`, :py:obj:`~.dstDevice` and :py:obj:`~.dstPitch` specify the (device) base address of the destination data, the bytes per row, and the height of each 2D slice of the 3D array. :py:obj:`~.dstArray` is ignored. If :py:obj:`~.dstMemoryType` is :py:obj:`~.CU_MEMORYTYPE_ARRAY`, :py:obj:`~.dstArray` specifies the handle of the destination data. :py:obj:`~.dstHost`, :py:obj:`~.dstDevice`, :py:obj:`~.dstPitch` and :py:obj:`~.dstHeight` are ignored. - :py:obj:`~.srcXInBytes`, :py:obj:`~.srcY` and :py:obj:`~.srcZ` specify the base address of the source data for the copy. For host pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.srcXInBytes` must be evenly divisible by the array element size. - dstXInBytes, :py:obj:`~.dstY` and :py:obj:`~.dstZ` specify the base address of the destination data for the copy. For host pointers, the base address is **View CUDA Toolkit Documentation for a C++ code example** For device pointers, the starting address is **View CUDA Toolkit Documentation for a C++ code example** For CUDA arrays, :py:obj:`~.dstXInBytes` must be evenly divisible by the array element size. - :py:obj:`~.WidthInBytes`, :py:obj:`~.Height` and :py:obj:`~.Depth` specify the width (in bytes), height and depth of the 3D copy being performed. - If specified, :py:obj:`~.srcPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + :py:obj:`~.srcXInBytes`, and :py:obj:`~.dstPitch` must be greater than or equal to :py:obj:`~.WidthInBytes` + dstXInBytes. - If specified, :py:obj:`~.srcHeight` must be greater than or equal to :py:obj:`~.Height` + :py:obj:`~.srcY`, and :py:obj:`~.dstHeight` must be greater than or equal to :py:obj:`~.Height` + :py:obj:`~.dstY`. :py:obj:`~.cuMemcpy3DAsync()` returns an error if any pitch is greater than the maximum allowed (:py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_PITCH`). The :py:obj:`~.srcLOD` and :py:obj:`~.dstLOD` members of the :py:obj:`~.CUDA_MEMCPY3D` structure must be set to 0. Parameters ---------- pCopy : :py:obj:`~.CUDA_MEMCPY3D` Parameters for the memory copy hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemcpy3DAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUDA_MEMCPY3D* cpCopy_ptr = pCopy._ptr if pCopy != None else NULL err = ccuda.cuMemcpy3DAsync(cpCopy_ptr, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemcpy3DPeerAsync' in found_functions}} @cython.embedsignature(True) def cuMemcpy3DPeerAsync(pCopy : Optional[CUDA_MEMCPY3D_PEER], hStream): """ Copies memory between contexts asynchronously. Perform a 3D memory copy according to the parameters specified in `pCopy`. See the definition of the :py:obj:`~.CUDA_MEMCPY3D_PEER` structure for documentation of its parameters. Parameters ---------- pCopy : :py:obj:`~.CUDA_MEMCPY3D_PEER` Parameters for the memory copy hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyPeer`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyPeerAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cudaMemcpy3DPeerAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUDA_MEMCPY3D_PEER* cpCopy_ptr = pCopy._ptr if pCopy != None else NULL err = ccuda.cuMemcpy3DPeerAsync(cpCopy_ptr, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemsetD8_v2' in found_functions}} @cython.embedsignature(True) def cuMemsetD8(dstDevice, unsigned char uc, size_t N): """ Initializes device memory. Sets the memory range of `N` 8-bit values to the specified value `uc`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer uc : unsigned char Value to set N : size_t Number of elements Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset` """ cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD8(cdstDevice, uc, N) return (CUresult(err),) {{endif}} {{if 'cuMemsetD16_v2' in found_functions}} @cython.embedsignature(True) def cuMemsetD16(dstDevice, unsigned short us, size_t N): """ Initializes device memory. Sets the memory range of `N` 16-bit values to the specified value `us`. The `dstDevice` pointer must be two byte aligned. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer us : unsigned short Value to set N : size_t Number of elements Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset` """ cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD16(cdstDevice, us, N) return (CUresult(err),) {{endif}} {{if 'cuMemsetD32_v2' in found_functions}} @cython.embedsignature(True) def cuMemsetD32(dstDevice, unsigned int ui, size_t N): """ Initializes device memory. Sets the memory range of `N` 32-bit values to the specified value `ui`. The `dstDevice` pointer must be four byte aligned. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer ui : unsigned int Value to set N : size_t Number of elements Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset` """ cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD32(cdstDevice, ui, N) return (CUresult(err),) {{endif}} {{if 'cuMemsetD2D8_v2' in found_functions}} @cython.embedsignature(True) def cuMemsetD2D8(dstDevice, size_t dstPitch, unsigned char uc, size_t Width, size_t Height): """ Initializes device memory. Sets the 2D memory range of `Width` 8-bit values to the specified value `uc`. `Height` specifies the number of rows to set, and `dstPitch` specifies the number of bytes between each row. This function performs fastest when the pitch is one that has been passed back by :py:obj:`~.cuMemAllocPitch()`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstPitch : size_t Pitch of destination device pointer(Unused if `Height` is 1) uc : unsigned char Value to set Width : size_t Width of row Height : size_t Number of rows Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset2D` """ cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD2D8(cdstDevice, dstPitch, uc, Width, Height) return (CUresult(err),) {{endif}} {{if 'cuMemsetD2D16_v2' in found_functions}} @cython.embedsignature(True) def cuMemsetD2D16(dstDevice, size_t dstPitch, unsigned short us, size_t Width, size_t Height): """ Initializes device memory. Sets the 2D memory range of `Width` 16-bit values to the specified value `us`. `Height` specifies the number of rows to set, and `dstPitch` specifies the number of bytes between each row. The `dstDevice` pointer and `dstPitch` offset must be two byte aligned. This function performs fastest when the pitch is one that has been passed back by :py:obj:`~.cuMemAllocPitch()`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstPitch : size_t Pitch of destination device pointer(Unused if `Height` is 1) us : unsigned short Value to set Width : size_t Width of row Height : size_t Number of rows Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset2D` """ cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD2D16(cdstDevice, dstPitch, us, Width, Height) return (CUresult(err),) {{endif}} {{if 'cuMemsetD2D32_v2' in found_functions}} @cython.embedsignature(True) def cuMemsetD2D32(dstDevice, size_t dstPitch, unsigned int ui, size_t Width, size_t Height): """ Initializes device memory. Sets the 2D memory range of `Width` 32-bit values to the specified value `ui`. `Height` specifies the number of rows to set, and `dstPitch` specifies the number of bytes between each row. The `dstDevice` pointer and `dstPitch` offset must be four byte aligned. This function performs fastest when the pitch is one that has been passed back by :py:obj:`~.cuMemAllocPitch()`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstPitch : size_t Pitch of destination device pointer(Unused if `Height` is 1) ui : unsigned int Value to set Width : size_t Width of row Height : size_t Number of rows Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset2D` """ cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD2D32(cdstDevice, dstPitch, ui, Width, Height) return (CUresult(err),) {{endif}} {{if 'cuMemsetD8Async' in found_functions}} @cython.embedsignature(True) def cuMemsetD8Async(dstDevice, unsigned char uc, size_t N, hStream): """ Sets device memory. Sets the memory range of `N` 8-bit values to the specified value `uc`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer uc : unsigned char Value to set N : size_t Number of elements hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemsetAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD8Async(cdstDevice, uc, N, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemsetD16Async' in found_functions}} @cython.embedsignature(True) def cuMemsetD16Async(dstDevice, unsigned short us, size_t N, hStream): """ Sets device memory. Sets the memory range of `N` 16-bit values to the specified value `us`. The `dstDevice` pointer must be two byte aligned. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer us : unsigned short Value to set N : size_t Number of elements hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemsetAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD16Async(cdstDevice, us, N, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemsetD32Async' in found_functions}} @cython.embedsignature(True) def cuMemsetD32Async(dstDevice, unsigned int ui, size_t N, hStream): """ Sets device memory. Sets the memory range of `N` 32-bit values to the specified value `ui`. The `dstDevice` pointer must be four byte aligned. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer ui : unsigned int Value to set N : size_t Number of elements hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMemsetAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD32Async(cdstDevice, ui, N, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemsetD2D8Async' in found_functions}} @cython.embedsignature(True) def cuMemsetD2D8Async(dstDevice, size_t dstPitch, unsigned char uc, size_t Width, size_t Height, hStream): """ Sets device memory. Sets the 2D memory range of `Width` 8-bit values to the specified value `uc`. `Height` specifies the number of rows to set, and `dstPitch` specifies the number of bytes between each row. This function performs fastest when the pitch is one that has been passed back by :py:obj:`~.cuMemAllocPitch()`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstPitch : size_t Pitch of destination device pointer(Unused if `Height` is 1) uc : unsigned char Value to set Width : size_t Width of row Height : size_t Number of rows hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset2DAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD2D8Async(cdstDevice, dstPitch, uc, Width, Height, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemsetD2D16Async' in found_functions}} @cython.embedsignature(True) def cuMemsetD2D16Async(dstDevice, size_t dstPitch, unsigned short us, size_t Width, size_t Height, hStream): """ Sets device memory. Sets the 2D memory range of `Width` 16-bit values to the specified value `us`. `Height` specifies the number of rows to set, and `dstPitch` specifies the number of bytes between each row. The `dstDevice` pointer and `dstPitch` offset must be two byte aligned. This function performs fastest when the pitch is one that has been passed back by :py:obj:`~.cuMemAllocPitch()`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstPitch : size_t Pitch of destination device pointer(Unused if `Height` is 1) us : unsigned short Value to set Width : size_t Width of row Height : size_t Number of rows hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD2D32Async`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset2DAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD2D16Async(cdstDevice, dstPitch, us, Width, Height, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemsetD2D32Async' in found_functions}} @cython.embedsignature(True) def cuMemsetD2D32Async(dstDevice, size_t dstPitch, unsigned int ui, size_t Width, size_t Height, hStream): """ Sets device memory. Sets the 2D memory range of `Width` 32-bit values to the specified value `ui`. `Height` specifies the number of rows to set, and `dstPitch` specifies the number of bytes between each row. The `dstDevice` pointer and `dstPitch` offset must be four byte aligned. This function performs fastest when the pitch is one that has been passed back by :py:obj:`~.cuMemAllocPitch()`. Parameters ---------- dstDevice : :py:obj:`~.CUdeviceptr` Destination device pointer dstPitch : size_t Pitch of destination device pointer(Unused if `Height` is 1) ui : unsigned int Value to set Width : size_t Width of row Height : size_t Number of rows hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D8Async`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D16Async`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD8Async`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD16Async`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cuMemsetD32Async`, :py:obj:`~.cudaMemset2DAsync` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdeviceptr,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdeviceptr(dstDevice)) cdstDevice = pdstDevice err = ccuda.cuMemsetD2D32Async(cdstDevice, dstPitch, ui, Width, Height, chStream) return (CUresult(err),) {{endif}} {{if 'cuArrayCreate_v2' in found_functions}} @cython.embedsignature(True) def cuArrayCreate(pAllocateArray : Optional[CUDA_ARRAY_DESCRIPTOR]): """ Creates a 1D or 2D CUDA array. Creates a CUDA array according to the :py:obj:`~.CUDA_ARRAY_DESCRIPTOR` structure `pAllocateArray` and returns a handle to the new CUDA array in `*pHandle`. The :py:obj:`~.CUDA_ARRAY_DESCRIPTOR` is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - `Width`, and `Height` are the width, and height of the CUDA array (in elements); the CUDA array is one-dimensional if height is 0, two- dimensional otherwise; - :py:obj:`~.Format` specifies the format of the elements; :py:obj:`~.CUarray_format` is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - `NumChannels` specifies the number of packed components per CUDA array element; it may be 1, 2, or 4; Here are examples of CUDA array descriptions: Description for a CUDA array of 2048 floats: **View CUDA Toolkit Documentation for a C++ code example** Description for a 64 x 64 CUDA array of floats: **View CUDA Toolkit Documentation for a C++ code example** Description for a `width` x `height` CUDA array of 64-bit, 4x16-bit float16's: **View CUDA Toolkit Documentation for a C++ code example** Description for a `width` x `height` CUDA array of 16-bit elements, each of which is two 8-bit unsigned chars: **View CUDA Toolkit Documentation for a C++ code example** Parameters ---------- pAllocateArray : :py:obj:`~.CUDA_ARRAY_DESCRIPTOR` Array descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pHandle : :py:obj:`~.CUarray` Returned array See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMallocArray` """ cdef CUarray pHandle = CUarray() cdef ccuda.CUDA_ARRAY_DESCRIPTOR* cpAllocateArray_ptr = pAllocateArray._ptr if pAllocateArray != None else NULL err = ccuda.cuArrayCreate(pHandle._ptr, cpAllocateArray_ptr) return (CUresult(err), pHandle) {{endif}} {{if 'cuArrayGetDescriptor_v2' in found_functions}} @cython.embedsignature(True) def cuArrayGetDescriptor(hArray): """ Get a 1D or 2D CUDA array descriptor. Returns in `*pArrayDescriptor` a descriptor containing information on the format and dimensions of the CUDA array `hArray`. It is useful for subroutines that have been passed a CUDA array, but need to know the CUDA array parameters for validation or other purposes. Parameters ---------- hArray : :py:obj:`~.CUarray` Array to get descriptor of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` pArrayDescriptor : :py:obj:`~.CUDA_ARRAY_DESCRIPTOR` Returned array descriptor See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaArrayGetInfo` """ cdef ccuda.CUarray chArray if hArray is None: chArray = 0 elif isinstance(hArray, (CUarray,)): phArray = int(hArray) chArray = phArray else: phArray = int(CUarray(hArray)) chArray = phArray cdef CUDA_ARRAY_DESCRIPTOR pArrayDescriptor = CUDA_ARRAY_DESCRIPTOR() err = ccuda.cuArrayGetDescriptor(pArrayDescriptor._ptr, chArray) return (CUresult(err), pArrayDescriptor) {{endif}} {{if 'cuArrayGetSparseProperties' in found_functions}} @cython.embedsignature(True) def cuArrayGetSparseProperties(array): """ Returns the layout properties of a sparse CUDA array. Returns the layout properties of a sparse CUDA array in `sparseProperties` If the CUDA array is not allocated with flag :py:obj:`~.CUDA_ARRAY3D_SPARSE` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. If the returned value in :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.flags` contains :py:obj:`~.CU_ARRAY_SPARSE_PROPERTIES_SINGLE_MIPTAIL`, then :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.miptailSize` represents the total size of the array. Otherwise, it will be zero. Also, the returned value in :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.miptailFirstLevel` is always zero. Note that the `array` must have been allocated using :py:obj:`~.cuArrayCreate` or :py:obj:`~.cuArray3DCreate`. For CUDA arrays obtained using :py:obj:`~.cuMipmappedArrayGetLevel`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. Instead, :py:obj:`~.cuMipmappedArrayGetSparseProperties` must be used to obtain the sparse properties of the entire CUDA mipmapped array to which `array` belongs to. Parameters ---------- array : :py:obj:`~.CUarray` CUDA array to get the sparse properties of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` sparseProperties : :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES` Pointer to :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES` See Also -------- :py:obj:`~.cuMipmappedArrayGetSparseProperties`, :py:obj:`~.cuMemMapArrayAsync` """ cdef ccuda.CUarray carray if array is None: carray = 0 elif isinstance(array, (CUarray,)): parray = int(array) carray = parray else: parray = int(CUarray(array)) carray = parray cdef CUDA_ARRAY_SPARSE_PROPERTIES sparseProperties = CUDA_ARRAY_SPARSE_PROPERTIES() err = ccuda.cuArrayGetSparseProperties(sparseProperties._ptr, carray) return (CUresult(err), sparseProperties) {{endif}} {{if 'cuMipmappedArrayGetSparseProperties' in found_functions}} @cython.embedsignature(True) def cuMipmappedArrayGetSparseProperties(mipmap): """ Returns the layout properties of a sparse CUDA mipmapped array. Returns the sparse array layout properties in `sparseProperties` If the CUDA mipmapped array is not allocated with flag :py:obj:`~.CUDA_ARRAY3D_SPARSE` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. For non-layered CUDA mipmapped arrays, :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.miptailSize` returns the size of the mip tail region. The mip tail region includes all mip levels whose width, height or depth is less than that of the tile. For layered CUDA mipmapped arrays, if :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.flags` contains :py:obj:`~.CU_ARRAY_SPARSE_PROPERTIES_SINGLE_MIPTAIL`, then :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.miptailSize` specifies the size of the mip tail of all layers combined. Otherwise, :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.miptailSize` specifies mip tail size per layer. The returned value of :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.miptailFirstLevel` is valid only if :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.miptailSize` is non- zero. Parameters ---------- mipmap : :py:obj:`~.CUmipmappedArray` CUDA mipmapped array to get the sparse properties of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` sparseProperties : :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES` Pointer to :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES` See Also -------- :py:obj:`~.cuArrayGetSparseProperties`, :py:obj:`~.cuMemMapArrayAsync` """ cdef ccuda.CUmipmappedArray cmipmap if mipmap is None: cmipmap = 0 elif isinstance(mipmap, (CUmipmappedArray,)): pmipmap = int(mipmap) cmipmap = pmipmap else: pmipmap = int(CUmipmappedArray(mipmap)) cmipmap = pmipmap cdef CUDA_ARRAY_SPARSE_PROPERTIES sparseProperties = CUDA_ARRAY_SPARSE_PROPERTIES() err = ccuda.cuMipmappedArrayGetSparseProperties(sparseProperties._ptr, cmipmap) return (CUresult(err), sparseProperties) {{endif}} {{if 'cuArrayGetMemoryRequirements' in found_functions}} @cython.embedsignature(True) def cuArrayGetMemoryRequirements(array, device): """ Returns the memory requirements of a CUDA array. Returns the memory requirements of a CUDA array in `memoryRequirements` If the CUDA array is not allocated with flag :py:obj:`~.CUDA_ARRAY3D_DEFERRED_MAPPING` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. The returned value in :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS.size` represents the total size of the CUDA array. The returned value in :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS.alignment` represents the alignment necessary for mapping the CUDA array. Parameters ---------- array : :py:obj:`~.CUarray` CUDA array to get the memory requirements of device : :py:obj:`~.CUdevice` Device to get the memory requirements for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` memoryRequirements : :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS` Pointer to :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS` See Also -------- :py:obj:`~.cuMipmappedArrayGetMemoryRequirements`, :py:obj:`~.cuMemMapArrayAsync` """ cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cdef ccuda.CUarray carray if array is None: carray = 0 elif isinstance(array, (CUarray,)): parray = int(array) carray = parray else: parray = int(CUarray(array)) carray = parray cdef CUDA_ARRAY_MEMORY_REQUIREMENTS memoryRequirements = CUDA_ARRAY_MEMORY_REQUIREMENTS() err = ccuda.cuArrayGetMemoryRequirements(memoryRequirements._ptr, carray, cdevice) return (CUresult(err), memoryRequirements) {{endif}} {{if 'cuMipmappedArrayGetMemoryRequirements' in found_functions}} @cython.embedsignature(True) def cuMipmappedArrayGetMemoryRequirements(mipmap, device): """ Returns the memory requirements of a CUDA mipmapped array. Returns the memory requirements of a CUDA mipmapped array in `memoryRequirements` If the CUDA mipmapped array is not allocated with flag :py:obj:`~.CUDA_ARRAY3D_DEFERRED_MAPPING` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. The returned value in :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS.size` represents the total size of the CUDA mipmapped array. The returned value in :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS.alignment` represents the alignment necessary for mapping the CUDA mipmapped array. Parameters ---------- mipmap : :py:obj:`~.CUmipmappedArray` CUDA mipmapped array to get the memory requirements of device : :py:obj:`~.CUdevice` Device to get the memory requirements for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_INVALID_VALUE` memoryRequirements : :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS` Pointer to :py:obj:`~.CUDA_ARRAY_MEMORY_REQUIREMENTS` See Also -------- :py:obj:`~.cuArrayGetMemoryRequirements`, :py:obj:`~.cuMemMapArrayAsync` """ cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cdef ccuda.CUmipmappedArray cmipmap if mipmap is None: cmipmap = 0 elif isinstance(mipmap, (CUmipmappedArray,)): pmipmap = int(mipmap) cmipmap = pmipmap else: pmipmap = int(CUmipmappedArray(mipmap)) cmipmap = pmipmap cdef CUDA_ARRAY_MEMORY_REQUIREMENTS memoryRequirements = CUDA_ARRAY_MEMORY_REQUIREMENTS() err = ccuda.cuMipmappedArrayGetMemoryRequirements(memoryRequirements._ptr, cmipmap, cdevice) return (CUresult(err), memoryRequirements) {{endif}} {{if 'cuArrayGetPlane' in found_functions}} @cython.embedsignature(True) def cuArrayGetPlane(hArray, unsigned int planeIdx): """ Gets a CUDA array plane from a CUDA array. Returns in `pPlaneArray` a CUDA array that represents a single format plane of the CUDA array `hArray`. If `planeIdx` is greater than the maximum number of planes in this array or if the array does not have a multi-planar format e.g: :py:obj:`~.CU_AD_FORMAT_NV12`, then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. Note that if the `hArray` has format :py:obj:`~.CU_AD_FORMAT_NV12`, then passing in 0 for `planeIdx` returns a CUDA array of the same size as `hArray` but with one channel and :py:obj:`~.CU_AD_FORMAT_UNSIGNED_INT8` as its format. If 1 is passed for `planeIdx`, then the returned CUDA array has half the height and width of `hArray` with two channels and :py:obj:`~.CU_AD_FORMAT_UNSIGNED_INT8` as its format. Parameters ---------- hArray : :py:obj:`~.CUarray` Multiplanar CUDA array planeIdx : unsigned int Plane index Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` pPlaneArray : :py:obj:`~.CUarray` Returned CUDA array referenced by the `planeIdx` See Also -------- :py:obj:`~.cuArrayCreate`, :py:obj:`~.cudaArrayGetPlane` """ cdef ccuda.CUarray chArray if hArray is None: chArray = 0 elif isinstance(hArray, (CUarray,)): phArray = int(hArray) chArray = phArray else: phArray = int(CUarray(hArray)) chArray = phArray cdef CUarray pPlaneArray = CUarray() err = ccuda.cuArrayGetPlane(pPlaneArray._ptr, chArray, planeIdx) return (CUresult(err), pPlaneArray) {{endif}} {{if 'cuArrayDestroy' in found_functions}} @cython.embedsignature(True) def cuArrayDestroy(hArray): """ Destroys a CUDA array. Destroys the CUDA array `hArray`. Parameters ---------- hArray : :py:obj:`~.CUarray` Array to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ARRAY_IS_MAPPED`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaFreeArray` """ cdef ccuda.CUarray chArray if hArray is None: chArray = 0 elif isinstance(hArray, (CUarray,)): phArray = int(hArray) chArray = phArray else: phArray = int(CUarray(hArray)) chArray = phArray err = ccuda.cuArrayDestroy(chArray) return (CUresult(err),) {{endif}} {{if 'cuArray3DCreate_v2' in found_functions}} @cython.embedsignature(True) def cuArray3DCreate(pAllocateArray : Optional[CUDA_ARRAY3D_DESCRIPTOR]): """ Creates a 3D CUDA array. Creates a CUDA array according to the :py:obj:`~.CUDA_ARRAY3D_DESCRIPTOR` structure `pAllocateArray` and returns a handle to the new CUDA array in `*pHandle`. The :py:obj:`~.CUDA_ARRAY3D_DESCRIPTOR` is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - `Width`, `Height`, and `Depth` are the width, height, and depth of the CUDA array (in elements); the following types of CUDA arrays can be allocated: - A 1D array is allocated if `Height` and `Depth` extents are both zero. - A 2D array is allocated if only `Depth` extent is zero. - A 3D array is allocated if all three extents are non-zero. - A 1D layered CUDA array is allocated if only `Height` is zero and the :py:obj:`~.CUDA_ARRAY3D_LAYERED` flag is set. Each layer is a 1D array. The number of layers is determined by the depth extent. - A 2D layered CUDA array is allocated if all three extents are non- zero and the :py:obj:`~.CUDA_ARRAY3D_LAYERED` flag is set. Each layer is a 2D array. The number of layers is determined by the depth extent. - A cubemap CUDA array is allocated if all three extents are non-zero and the :py:obj:`~.CUDA_ARRAY3D_CUBEMAP` flag is set. `Width` must be equal to `Height`, and `Depth` must be six. A cubemap is a special type of 2D layered CUDA array, where the six layers represent the six faces of a cube. The order of the six layers in memory is the same as that listed in :py:obj:`~.CUarray_cubemap_face`. - A cubemap layered CUDA array is allocated if all three extents are non-zero, and both, :py:obj:`~.CUDA_ARRAY3D_CUBEMAP` and :py:obj:`~.CUDA_ARRAY3D_LAYERED` flags are set. `Width` must be equal to `Height`, and `Depth` must be a multiple of six. A cubemap layered CUDA array is a special type of 2D layered CUDA array that consists of a collection of cubemaps. The first six layers represent the first cubemap, the next six layers form the second cubemap, and so on. - :py:obj:`~.Format` specifies the format of the elements; :py:obj:`~.CUarray_format` is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - `NumChannels` specifies the number of packed components per CUDA array element; it may be 1, 2, or 4; - :py:obj:`~.Flags` may be set to - :py:obj:`~.CUDA_ARRAY3D_LAYERED` to enable creation of layered CUDA arrays. If this flag is set, `Depth` specifies the number of layers, not the depth of a 3D array. - :py:obj:`~.CUDA_ARRAY3D_SURFACE_LDST` to enable surface references to be bound to the CUDA array. If this flag is not set, :py:obj:`~.cuSurfRefSetArray` will fail when attempting to bind the CUDA array to a surface reference. - :py:obj:`~.CUDA_ARRAY3D_CUBEMAP` to enable creation of cubemaps. If this flag is set, `Width` must be equal to `Height`, and `Depth` must be six. If the :py:obj:`~.CUDA_ARRAY3D_LAYERED` flag is also set, then `Depth` must be a multiple of six. - :py:obj:`~.CUDA_ARRAY3D_TEXTURE_GATHER` to indicate that the CUDA array will be used for texture gather. Texture gather can only be performed on 2D CUDA arrays. `Width`, `Height` and `Depth` must meet certain size requirements as listed in the following table. All values are specified in elements. Note that for brevity's sake, the full name of the device attribute is not specified. For ex., TEXTURE1D_WIDTH refers to the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_WIDTH`. Note that 2D CUDA arrays have different size requirements if the :py:obj:`~.CUDA_ARRAY3D_TEXTURE_GATHER` flag is set. `Width` and `Height` must not be greater than :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_WIDTH` and :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_GATHER_HEIGHT` respectively, in that case. **View CUDA Toolkit Documentation for a table example** Here are examples of CUDA array descriptions: Description for a CUDA array of 2048 floats: **View CUDA Toolkit Documentation for a C++ code example** Description for a 64 x 64 CUDA array of floats: **View CUDA Toolkit Documentation for a C++ code example** Description for a `width` x `height` x `depth` CUDA array of 64-bit, 4x16-bit float16's: **View CUDA Toolkit Documentation for a C++ code example** Parameters ---------- pAllocateArray : :py:obj:`~.CUDA_ARRAY3D_DESCRIPTOR` 3D array descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pHandle : :py:obj:`~.CUarray` Returned array See Also -------- :py:obj:`~.cuArray3DGetDescriptor`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaMalloc3DArray` """ cdef CUarray pHandle = CUarray() cdef ccuda.CUDA_ARRAY3D_DESCRIPTOR* cpAllocateArray_ptr = pAllocateArray._ptr if pAllocateArray != None else NULL err = ccuda.cuArray3DCreate(pHandle._ptr, cpAllocateArray_ptr) return (CUresult(err), pHandle) {{endif}} {{if 'cuArray3DGetDescriptor_v2' in found_functions}} @cython.embedsignature(True) def cuArray3DGetDescriptor(hArray): """ Get a 3D CUDA array descriptor. Returns in `*pArrayDescriptor` a descriptor containing information on the format and dimensions of the CUDA array `hArray`. It is useful for subroutines that have been passed a CUDA array, but need to know the CUDA array parameters for validation or other purposes. This function may be called on 1D and 2D arrays, in which case the `Height` and/or `Depth` members of the descriptor struct will be set to 0. Parameters ---------- hArray : :py:obj:`~.CUarray` 3D array to get descriptor of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` pArrayDescriptor : :py:obj:`~.CUDA_ARRAY3D_DESCRIPTOR` Returned 3D array descriptor See Also -------- :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArrayDestroy`, :py:obj:`~.cuArrayGetDescriptor`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemAllocPitch`, :py:obj:`~.cuMemcpy2D`, :py:obj:`~.cuMemcpy2DAsync`, :py:obj:`~.cuMemcpy2DUnaligned`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuMemcpy3DAsync`, :py:obj:`~.cuMemcpyAtoA`, :py:obj:`~.cuMemcpyAtoD`, :py:obj:`~.cuMemcpyAtoH`, :py:obj:`~.cuMemcpyAtoHAsync`, :py:obj:`~.cuMemcpyDtoA`, :py:obj:`~.cuMemcpyDtoD`, :py:obj:`~.cuMemcpyDtoDAsync`, :py:obj:`~.cuMemcpyDtoH`, :py:obj:`~.cuMemcpyDtoHAsync`, :py:obj:`~.cuMemcpyHtoA`, :py:obj:`~.cuMemcpyHtoAAsync`, :py:obj:`~.cuMemcpyHtoD`, :py:obj:`~.cuMemcpyHtoDAsync`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemGetAddressRange`, :py:obj:`~.cuMemGetInfo`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostGetDevicePointer`, :py:obj:`~.cuMemsetD2D8`, :py:obj:`~.cuMemsetD2D16`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuMemsetD8`, :py:obj:`~.cuMemsetD16`, :py:obj:`~.cuMemsetD32`, :py:obj:`~.cudaArrayGetInfo` """ cdef ccuda.CUarray chArray if hArray is None: chArray = 0 elif isinstance(hArray, (CUarray,)): phArray = int(hArray) chArray = phArray else: phArray = int(CUarray(hArray)) chArray = phArray cdef CUDA_ARRAY3D_DESCRIPTOR pArrayDescriptor = CUDA_ARRAY3D_DESCRIPTOR() err = ccuda.cuArray3DGetDescriptor(pArrayDescriptor._ptr, chArray) return (CUresult(err), pArrayDescriptor) {{endif}} {{if 'cuMipmappedArrayCreate' in found_functions}} @cython.embedsignature(True) def cuMipmappedArrayCreate(pMipmappedArrayDesc : Optional[CUDA_ARRAY3D_DESCRIPTOR], unsigned int numMipmapLevels): """ Creates a CUDA mipmapped array. Creates a CUDA mipmapped array according to the :py:obj:`~.CUDA_ARRAY3D_DESCRIPTOR` structure `pMipmappedArrayDesc` and returns a handle to the new CUDA mipmapped array in `*pHandle`. `numMipmapLevels` specifies the number of mipmap levels to be allocated. This value is clamped to the range [1, 1 + floor(log2(max(width, height, depth)))]. The :py:obj:`~.CUDA_ARRAY3D_DESCRIPTOR` is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - `Width`, `Height`, and `Depth` are the width, height, and depth of the CUDA array (in elements); the following types of CUDA arrays can be allocated: - A 1D mipmapped array is allocated if `Height` and `Depth` extents are both zero. - A 2D mipmapped array is allocated if only `Depth` extent is zero. - A 3D mipmapped array is allocated if all three extents are non- zero. - A 1D layered CUDA mipmapped array is allocated if only `Height` is zero and the :py:obj:`~.CUDA_ARRAY3D_LAYERED` flag is set. Each layer is a 1D array. The number of layers is determined by the depth extent. - A 2D layered CUDA mipmapped array is allocated if all three extents are non-zero and the :py:obj:`~.CUDA_ARRAY3D_LAYERED` flag is set. Each layer is a 2D array. The number of layers is determined by the depth extent. - A cubemap CUDA mipmapped array is allocated if all three extents are non-zero and the :py:obj:`~.CUDA_ARRAY3D_CUBEMAP` flag is set. `Width` must be equal to `Height`, and `Depth` must be six. A cubemap is a special type of 2D layered CUDA array, where the six layers represent the six faces of a cube. The order of the six layers in memory is the same as that listed in :py:obj:`~.CUarray_cubemap_face`. - A cubemap layered CUDA mipmapped array is allocated if all three extents are non-zero, and both, :py:obj:`~.CUDA_ARRAY3D_CUBEMAP` and :py:obj:`~.CUDA_ARRAY3D_LAYERED` flags are set. `Width` must be equal to `Height`, and `Depth` must be a multiple of six. A cubemap layered CUDA array is a special type of 2D layered CUDA array that consists of a collection of cubemaps. The first six layers represent the first cubemap, the next six layers form the second cubemap, and so on. - :py:obj:`~.Format` specifies the format of the elements; :py:obj:`~.CUarray_format` is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - `NumChannels` specifies the number of packed components per CUDA array element; it may be 1, 2, or 4; - :py:obj:`~.Flags` may be set to - :py:obj:`~.CUDA_ARRAY3D_LAYERED` to enable creation of layered CUDA mipmapped arrays. If this flag is set, `Depth` specifies the number of layers, not the depth of a 3D array. - :py:obj:`~.CUDA_ARRAY3D_SURFACE_LDST` to enable surface references to be bound to individual mipmap levels of the CUDA mipmapped array. If this flag is not set, :py:obj:`~.cuSurfRefSetArray` will fail when attempting to bind a mipmap level of the CUDA mipmapped array to a surface reference. - :py:obj:`~.CUDA_ARRAY3D_CUBEMAP` to enable creation of mipmapped cubemaps. If this flag is set, `Width` must be equal to `Height`, and `Depth` must be six. If the :py:obj:`~.CUDA_ARRAY3D_LAYERED` flag is also set, then `Depth` must be a multiple of six. - :py:obj:`~.CUDA_ARRAY3D_TEXTURE_GATHER` to indicate that the CUDA mipmapped array will be used for texture gather. Texture gather can only be performed on 2D CUDA mipmapped arrays. `Width`, `Height` and `Depth` must meet certain size requirements as listed in the following table. All values are specified in elements. Note that for brevity's sake, the full name of the device attribute is not specified. For ex., TEXTURE1D_MIPMAPPED_WIDTH refers to the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_MIPMAPPED_WIDTH`. **View CUDA Toolkit Documentation for a table example** Parameters ---------- pMipmappedArrayDesc : :py:obj:`~.CUDA_ARRAY3D_DESCRIPTOR` mipmapped array descriptor numMipmapLevels : unsigned int Number of mipmap levels Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pHandle : :py:obj:`~.CUmipmappedArray` Returned mipmapped array See Also -------- :py:obj:`~.cuMipmappedArrayDestroy`, :py:obj:`~.cuMipmappedArrayGetLevel`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cudaMallocMipmappedArray` """ cdef CUmipmappedArray pHandle = CUmipmappedArray() cdef ccuda.CUDA_ARRAY3D_DESCRIPTOR* cpMipmappedArrayDesc_ptr = pMipmappedArrayDesc._ptr if pMipmappedArrayDesc != None else NULL err = ccuda.cuMipmappedArrayCreate(pHandle._ptr, cpMipmappedArrayDesc_ptr, numMipmapLevels) return (CUresult(err), pHandle) {{endif}} {{if 'cuMipmappedArrayGetLevel' in found_functions}} @cython.embedsignature(True) def cuMipmappedArrayGetLevel(hMipmappedArray, unsigned int level): """ Gets a mipmap level of a CUDA mipmapped array. Returns in `*pLevelArray` a CUDA array that represents a single mipmap level of the CUDA mipmapped array `hMipmappedArray`. If `level` is greater than the maximum number of levels in this mipmapped array, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. Parameters ---------- hMipmappedArray : :py:obj:`~.CUmipmappedArray` CUDA mipmapped array level : unsigned int Mipmap level Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` pLevelArray : :py:obj:`~.CUarray` Returned mipmap level CUDA array See Also -------- :py:obj:`~.cuMipmappedArrayCreate`, :py:obj:`~.cuMipmappedArrayDestroy`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cudaGetMipmappedArrayLevel` """ cdef ccuda.CUmipmappedArray chMipmappedArray if hMipmappedArray is None: chMipmappedArray = 0 elif isinstance(hMipmappedArray, (CUmipmappedArray,)): phMipmappedArray = int(hMipmappedArray) chMipmappedArray = phMipmappedArray else: phMipmappedArray = int(CUmipmappedArray(hMipmappedArray)) chMipmappedArray = phMipmappedArray cdef CUarray pLevelArray = CUarray() err = ccuda.cuMipmappedArrayGetLevel(pLevelArray._ptr, chMipmappedArray, level) return (CUresult(err), pLevelArray) {{endif}} {{if 'cuMipmappedArrayDestroy' in found_functions}} @cython.embedsignature(True) def cuMipmappedArrayDestroy(hMipmappedArray): """ Destroys a CUDA mipmapped array. Destroys the CUDA mipmapped array `hMipmappedArray`. Parameters ---------- hMipmappedArray : :py:obj:`~.CUmipmappedArray` Mipmapped array to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ARRAY_IS_MAPPED`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` See Also -------- :py:obj:`~.cuMipmappedArrayCreate`, :py:obj:`~.cuMipmappedArrayGetLevel`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cudaFreeMipmappedArray` """ cdef ccuda.CUmipmappedArray chMipmappedArray if hMipmappedArray is None: chMipmappedArray = 0 elif isinstance(hMipmappedArray, (CUmipmappedArray,)): phMipmappedArray = int(hMipmappedArray) chMipmappedArray = phMipmappedArray else: phMipmappedArray = int(CUmipmappedArray(hMipmappedArray)) chMipmappedArray = phMipmappedArray err = ccuda.cuMipmappedArrayDestroy(chMipmappedArray) return (CUresult(err),) {{endif}} {{if 'cuMemGetHandleForAddressRange' in found_functions}} @cython.embedsignature(True) def cuMemGetHandleForAddressRange(dptr, size_t size, handleType not None : CUmemRangeHandleType, unsigned long long flags): """ Retrieve handle for an address range. Get a handle of the specified type to an address range. The address range must have been obtained by a prior call to either :py:obj:`~.cuMemAlloc` or :py:obj:`~.cuMemAddressReserve`. If the address range was obtained via :py:obj:`~.cuMemAddressReserve`, it must also be fully mapped via :py:obj:`~.cuMemMap`. The address range must have been obtained by a prior call to either :py:obj:`~.cuMemAllocHost` or :py:obj:`~.cuMemHostAlloc` on Tegra. Users must ensure the `dptr` and `size` are aligned to the host page size. When requesting CUmemRangeHandleType::CU_MEM_RANGE_HANDLE_TYPE_DMA_BUF_FD, users are expected to query for dma_buf support for the platform by using :py:obj:`~.CU_DEVICE_ATTRIBUTE_DMA_BUF_SUPPORTED` device attribute before calling this API. The `handle` will be interpreted as a pointer to an integer to store the dma_buf file descriptor. Users must ensure the entire address range is backed and mapped when the address range is allocated by :py:obj:`~.cuMemAddressReserve`. All the physical allocations backing the address range must be resident on the same device and have identical allocation properties. Users are also expected to retrieve a new handle every time the underlying physical allocation(s) corresponding to a previously queried VA range are changed. Parameters ---------- dptr : :py:obj:`~.CUdeviceptr` Pointer to a valid CUDA device allocation. Must be aligned to host page size. size : size_t Length of the address range. Must be aligned to host page size. handleType : :py:obj:`~.CUmemRangeHandleType` Type of handle requested (defines type and size of the `handle` output parameter) flags : unsigned long long Reserved, must be zero Returns ------- CUresult CUDA_SUCCESS CUDA_ERROR_INVALID_VALUE CUDA_ERROR_NOT_SUPPORTED handle : Any Pointer to the location where the returned handle will be stored. """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr cdef int handle = 0 cdef void* chandle_ptr = &handle cdef ccuda.CUmemRangeHandleType chandleType = handleType.value err = ccuda.cuMemGetHandleForAddressRange(chandle_ptr, cdptr, size, chandleType, flags) return (CUresult(err), handle) {{endif}} {{if 'cuMemAddressReserve' in found_functions}} @cython.embedsignature(True) def cuMemAddressReserve(size_t size, size_t alignment, addr, unsigned long long flags): """ Allocate an address range reservation. Reserves a virtual address range based on the given parameters, giving the starting address of the range in `ptr`. This API requires a system that supports UVA. The size and address parameters must be a multiple of the host page size and the alignment must be a power of two or zero for default alignment. Parameters ---------- size : size_t Size of the reserved virtual address range requested alignment : size_t Alignment of the reserved virtual address range requested addr : :py:obj:`~.CUdeviceptr` Fixed starting address range requested flags : unsigned long long Currently unused, must be zero Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` ptr : :py:obj:`~.CUdeviceptr` Resulting pointer to start of virtual address range allocated See Also -------- :py:obj:`~.cuMemAddressFree` """ cdef ccuda.CUdeviceptr caddr if addr is None: caddr = 0 elif isinstance(addr, (CUdeviceptr,)): paddr = int(addr) caddr = paddr else: paddr = int(CUdeviceptr(addr)) caddr = paddr cdef CUdeviceptr ptr = CUdeviceptr() err = ccuda.cuMemAddressReserve(ptr._ptr, size, alignment, caddr, flags) return (CUresult(err), ptr) {{endif}} {{if 'cuMemAddressFree' in found_functions}} @cython.embedsignature(True) def cuMemAddressFree(ptr, size_t size): """ Free an address range reservation. Frees a virtual address range reserved by cuMemAddressReserve. The size must match what was given to memAddressReserve and the ptr given must match what was returned from memAddressReserve. Parameters ---------- ptr : :py:obj:`~.CUdeviceptr` Starting address of the virtual address range to free size : size_t Size of the virtual address region to free Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMemAddressReserve` """ cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr err = ccuda.cuMemAddressFree(cptr, size) return (CUresult(err),) {{endif}} {{if 'cuMemCreate' in found_functions}} @cython.embedsignature(True) def cuMemCreate(size_t size, prop : Optional[CUmemAllocationProp], unsigned long long flags): """ Create a CUDA memory handle representing a memory allocation of a given size described by the given properties. This creates a memory allocation on the target device specified through the `prop` structure. The created allocation will not have any device or host mappings. The generic memory `handle` for the allocation can be mapped to the address space of calling process via :py:obj:`~.cuMemMap`. This handle cannot be transmitted directly to other processes (see :py:obj:`~.cuMemExportToShareableHandle`). On Windows, the caller must also pass an LPSECURITYATTRIBUTE in `prop` to be associated with this handle which limits or allows access to this handle for a recipient process (see :py:obj:`~.CUmemAllocationProp.win32HandleMetaData` for more). The `size` of this allocation must be a multiple of the the value given via :py:obj:`~.cuMemGetAllocationGranularity` with the :py:obj:`~.CU_MEM_ALLOC_GRANULARITY_MINIMUM` flag. To create a CPU allocation targeting a specific host NUMA node, applications must set :py:obj:`~.CUmemAllocationProp`::CUmemLocation::type to :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA` and :py:obj:`~.CUmemAllocationProp`::CUmemLocation::id must specify the NUMA ID of the CPU. On systems where NUMA is not available :py:obj:`~.CUmemAllocationProp`::CUmemLocation::id must be set to 0. Specifying :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT` or :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST` as the :py:obj:`~.CUmemLocation.type` will result in :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. Applications can set :py:obj:`~.CUmemAllocationProp.requestedHandleTypes` to :py:obj:`~.CU_MEM_HANDLE_TYPE_FABRIC` in order to create allocations suitable for sharing within an IMEX domain. An IMEX domain is either an OS instance or a group of securely connected OS instances using the NVIDIA IMEX daemon. An IMEX channel is a global resource within the IMEX domain that represents a logical entity that aims to provide fine grained accessibility control for the participating processes. When exporter and importer CUDA processes have been granted access to the same IMEX channel, they can securely share memory. If the allocating process does not have access setup for an IMEX channel, attempting to create a :py:obj:`~.CUmemGenericAllocationHandle` with :py:obj:`~.CU_MEM_HANDLE_TYPE_FABRIC` will result in :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`. The nvidia-modprobe CLI provides more information regarding setting up of IMEX channels. If :py:obj:`~.CUmemAllocationProp`::allocFlags::usage contains :py:obj:`~.CU_MEM_CREATE_USAGE_TILE_POOL` flag then the memory allocation is intended only to be used as backing tile pool for sparse CUDA arrays and sparse CUDA mipmapped arrays. (see :py:obj:`~.cuMemMapArrayAsync`). Parameters ---------- size : size_t Size of the allocation requested prop : :py:obj:`~.CUmemAllocationProp` Properties of the allocation to create. flags : unsigned long long flags for future use, must be zero now. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` handle : :py:obj:`~.CUmemGenericAllocationHandle` Value of handle returned. All operations on this allocation are to be performed using this handle. See Also -------- :py:obj:`~.cuMemRelease`, :py:obj:`~.cuMemExportToShareableHandle`, :py:obj:`~.cuMemImportFromShareableHandle` """ cdef CUmemGenericAllocationHandle handle = CUmemGenericAllocationHandle() cdef ccuda.CUmemAllocationProp* cprop_ptr = prop._ptr if prop != None else NULL err = ccuda.cuMemCreate(handle._ptr, size, cprop_ptr, flags) return (CUresult(err), handle) {{endif}} {{if 'cuMemRelease' in found_functions}} @cython.embedsignature(True) def cuMemRelease(handle): """ Release a memory handle representing a memory allocation which was previously allocated through cuMemCreate. Frees the memory that was allocated on a device through cuMemCreate. The memory allocation will be freed when all outstanding mappings to the memory are unmapped and when all outstanding references to the handle (including it's shareable counterparts) are also released. The generic memory handle can be freed when there are still outstanding mappings made with this handle. Each time a recipient process imports a shareable handle, it needs to pair it with :py:obj:`~.cuMemRelease` for the handle to be freed. If `handle` is not a valid handle the behavior is undefined. Parameters ---------- handle : :py:obj:`~.CUmemGenericAllocationHandle` Value of handle which was returned previously by cuMemCreate. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMemCreate` """ cdef ccuda.CUmemGenericAllocationHandle chandle if handle is None: chandle = 0 elif isinstance(handle, (CUmemGenericAllocationHandle,)): phandle = int(handle) chandle = phandle else: phandle = int(CUmemGenericAllocationHandle(handle)) chandle = phandle err = ccuda.cuMemRelease(chandle) return (CUresult(err),) {{endif}} {{if 'cuMemMap' in found_functions}} @cython.embedsignature(True) def cuMemMap(ptr, size_t size, size_t offset, handle, unsigned long long flags): """ Maps an allocation handle to a reserved virtual address range. Maps bytes of memory represented by `handle` starting from byte `offset` to `size` to address range [`addr`, `addr` + `size`]. This range must be an address reservation previously reserved with :py:obj:`~.cuMemAddressReserve`, and `offset` + `size` must be less than the size of the memory allocation. Both `ptr`, `size`, and `offset` must be a multiple of the value given via :py:obj:`~.cuMemGetAllocationGranularity` with the :py:obj:`~.CU_MEM_ALLOC_GRANULARITY_MINIMUM` flag. If `handle` represents a multicast object, `ptr`, `size` and `offset` must be aligned to the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_MINIMUM_GRANULARITY`. For best performance however, it is recommended that `ptr`, `size` and `offset` be aligned to the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_RECOMMENDED_GRANULARITY`. Please note calling :py:obj:`~.cuMemMap` does not make the address accessible, the caller needs to update accessibility of a contiguous mapped VA range by calling :py:obj:`~.cuMemSetAccess`. Once a recipient process obtains a shareable memory handle from :py:obj:`~.cuMemImportFromShareableHandle`, the process must use :py:obj:`~.cuMemMap` to map the memory into its address ranges before setting accessibility with :py:obj:`~.cuMemSetAccess`. :py:obj:`~.cuMemMap` can only create mappings on VA range reservations that are not currently mapped. Parameters ---------- ptr : :py:obj:`~.CUdeviceptr` Address where memory will be mapped. size : size_t Size of the memory mapping. offset : size_t Offset into the memory represented by handle : :py:obj:`~.CUmemGenericAllocationHandle` Handle to a shareable memory flags : unsigned long long flags for future use, must be zero now. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMemUnmap`, :py:obj:`~.cuMemSetAccess`, :py:obj:`~.cuMemCreate`, :py:obj:`~.cuMemAddressReserve`, :py:obj:`~.cuMemImportFromShareableHandle` """ cdef ccuda.CUmemGenericAllocationHandle chandle if handle is None: chandle = 0 elif isinstance(handle, (CUmemGenericAllocationHandle,)): phandle = int(handle) chandle = phandle else: phandle = int(CUmemGenericAllocationHandle(handle)) chandle = phandle cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr err = ccuda.cuMemMap(cptr, size, offset, chandle, flags) return (CUresult(err),) {{endif}} {{if 'cuMemMapArrayAsync' in found_functions}} @cython.embedsignature(True) def cuMemMapArrayAsync(mapInfoList : Optional[Tuple[CUarrayMapInfo] | List[CUarrayMapInfo]], unsigned int count, hStream): """ Maps or unmaps subregions of sparse CUDA arrays and sparse CUDA mipmapped arrays. Performs map or unmap operations on subregions of sparse CUDA arrays and sparse CUDA mipmapped arrays. Each operation is specified by a :py:obj:`~.CUarrayMapInfo` entry in the `mapInfoList` array of size `count`. The structure :py:obj:`~.CUarrayMapInfo` is defined as follow: **View CUDA Toolkit Documentation for a C++ code example** where :py:obj:`~.CUarrayMapInfo.resourceType` specifies the type of resource to be operated on. If :py:obj:`~.CUarrayMapInfo.resourceType` is set to :py:obj:`~.CUresourcetype`::CU_RESOURCE_TYPE_ARRAY then :py:obj:`~.CUarrayMapInfo`::resource::array must be set to a valid sparse CUDA array handle. The CUDA array must be either a 2D, 2D layered or 3D CUDA array and must have been allocated using :py:obj:`~.cuArrayCreate` or :py:obj:`~.cuArray3DCreate` with the flag :py:obj:`~.CUDA_ARRAY3D_SPARSE` or :py:obj:`~.CUDA_ARRAY3D_DEFERRED_MAPPING`. For CUDA arrays obtained using :py:obj:`~.cuMipmappedArrayGetLevel`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned. If :py:obj:`~.CUarrayMapInfo.resourceType` is set to :py:obj:`~.CUresourcetype`::CU_RESOURCE_TYPE_MIPMAPPED_ARRAY then :py:obj:`~.CUarrayMapInfo`::resource::mipmap must be set to a valid sparse CUDA mipmapped array handle. The CUDA mipmapped array must be either a 2D, 2D layered or 3D CUDA mipmapped array and must have been allocated using :py:obj:`~.cuMipmappedArrayCreate` with the flag :py:obj:`~.CUDA_ARRAY3D_SPARSE` or :py:obj:`~.CUDA_ARRAY3D_DEFERRED_MAPPING`. :py:obj:`~.CUarrayMapInfo.subresourceType` specifies the type of subresource within the resource. :py:obj:`~.CUarraySparseSubresourceType_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** where :py:obj:`~.CUarraySparseSubresourceType`::CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_SPARSE_LEVEL indicates a sparse-miplevel which spans at least one tile in every dimension. The remaining miplevels which are too small to span at least one tile in any dimension constitute the mip tail region as indicated by :py:obj:`~.CUarraySparseSubresourceType`::CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_MIPTAIL subresource type. If :py:obj:`~.CUarrayMapInfo.subresourceType` is set to :py:obj:`~.CUarraySparseSubresourceType`::CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_SPARSE_LEVEL then :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel struct must contain valid array subregion offsets and extents. The :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::offsetX, :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::offsetY and :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::offsetZ must specify valid X, Y and Z offsets respectively. The :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::extentWidth, :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::extentHeight and :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::extentDepth must specify valid width, height and depth extents respectively. These offsets and extents must be aligned to the corresponding tile dimension. For CUDA mipmapped arrays :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::level must specify a valid mip level index. Otherwise, must be zero. For layered CUDA arrays and layered CUDA mipmapped arrays :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::layer must specify a valid layer index. Otherwise, must be zero. :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::offsetZ must be zero and :py:obj:`~.CUarrayMapInfo`::subresource::sparseLevel::extentDepth must be set to 1 for 2D and 2D layered CUDA arrays and CUDA mipmapped arrays. Tile extents can be obtained by calling :py:obj:`~.cuArrayGetSparseProperties` and :py:obj:`~.cuMipmappedArrayGetSparseProperties` If :py:obj:`~.CUarrayMapInfo.subresourceType` is set to :py:obj:`~.CUarraySparseSubresourceType`::CU_ARRAY_SPARSE_SUBRESOURCE_TYPE_MIPTAIL then :py:obj:`~.CUarrayMapInfo`::subresource::miptail struct must contain valid mip tail offset in :py:obj:`~.CUarrayMapInfo`::subresource::miptail::offset and size in :py:obj:`~.CUarrayMapInfo`::subresource::miptail::size. Both, mip tail offset and mip tail size must be aligned to the tile size. For layered CUDA mipmapped arrays which don't have the flag :py:obj:`~.CU_ARRAY_SPARSE_PROPERTIES_SINGLE_MIPTAIL` set in :py:obj:`~.CUDA_ARRAY_SPARSE_PROPERTIES.flags` as returned by :py:obj:`~.cuMipmappedArrayGetSparseProperties`, :py:obj:`~.CUarrayMapInfo`::subresource::miptail::layer must specify a valid layer index. Otherwise, must be zero. If :py:obj:`~.CUarrayMapInfo`::resource::array or :py:obj:`~.CUarrayMapInfo`::resource::mipmap was created with :py:obj:`~.CUDA_ARRAY3D_DEFERRED_MAPPING` flag set the :py:obj:`~.CUarrayMapInfo.subresourceType` and the contents of :py:obj:`~.CUarrayMapInfo`::subresource will be ignored. :py:obj:`~.CUarrayMapInfo.memOperationType` specifies the type of operation. :py:obj:`~.CUmemOperationType` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.CUarrayMapInfo.memOperationType` is set to :py:obj:`~.CUmemOperationType`::CU_MEM_OPERATION_TYPE_MAP then the subresource will be mapped onto the tile pool memory specified by :py:obj:`~.CUarrayMapInfo`::memHandle at offset :py:obj:`~.CUarrayMapInfo.offset`. The tile pool allocation has to be created by specifying the :py:obj:`~.CU_MEM_CREATE_USAGE_TILE_POOL` flag when calling :py:obj:`~.cuMemCreate`. Also, :py:obj:`~.CUarrayMapInfo.memHandleType` must be set to :py:obj:`~.CUmemHandleType`::CU_MEM_HANDLE_TYPE_GENERIC. If :py:obj:`~.CUarrayMapInfo.memOperationType` is set to :py:obj:`~.CUmemOperationType`::CU_MEM_OPERATION_TYPE_UNMAP then an unmapping operation is performed. :py:obj:`~.CUarrayMapInfo`::memHandle must be NULL. :py:obj:`~.CUarrayMapInfo.deviceBitMask` specifies the list of devices that must map or unmap physical memory. Currently, this mask must have exactly one bit set, and the corresponding device must match the device associated with the stream. If :py:obj:`~.CUarrayMapInfo.memOperationType` is set to :py:obj:`~.CUmemOperationType`::CU_MEM_OPERATION_TYPE_MAP, the device must also match the device associated with the tile pool memory allocation as specified by :py:obj:`~.CUarrayMapInfo`::memHandle. :py:obj:`~.CUarrayMapInfo.flags` and :py:obj:`~.CUarrayMapInfo.reserved`[] are unused and must be set to zero. Parameters ---------- mapInfoList : List[:py:obj:`~.CUarrayMapInfo`] List of :py:obj:`~.CUarrayMapInfo` count : unsigned int Count of :py:obj:`~.CUarrayMapInfo` in `mapInfoList` hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier for the stream to use for map or unmap operations Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuMipmappedArrayCreate`, :py:obj:`~.cuArrayCreate`, :py:obj:`~.cuArray3DCreate`, :py:obj:`~.cuMemCreate`, :py:obj:`~.cuArrayGetSparseProperties`, :py:obj:`~.cuMipmappedArrayGetSparseProperties` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream mapInfoList = [] if mapInfoList is None else mapInfoList if not all(isinstance(_x, (CUarrayMapInfo,)) for _x in mapInfoList): raise TypeError("Argument 'mapInfoList' is not instance of type (expected Tuple[ccuda.CUarrayMapInfo,] or List[ccuda.CUarrayMapInfo,]") cdef ccuda.CUarrayMapInfo* cmapInfoList = NULL if len(mapInfoList) > 0: cmapInfoList = calloc(len(mapInfoList), sizeof(ccuda.CUarrayMapInfo)) if cmapInfoList is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(mapInfoList)) + 'x' + str(sizeof(ccuda.CUarrayMapInfo))) for idx in range(len(mapInfoList)): string.memcpy(&cmapInfoList[idx], (mapInfoList[idx])._ptr, sizeof(ccuda.CUarrayMapInfo)) if count > len(mapInfoList): raise RuntimeError("List is too small: " + str(len(mapInfoList)) + " < " + str(count)) err = ccuda.cuMemMapArrayAsync((mapInfoList[0])._ptr if len(mapInfoList) == 1 else cmapInfoList, count, chStream) if cmapInfoList is not NULL: free(cmapInfoList) return (CUresult(err),) {{endif}} {{if 'cuMemUnmap' in found_functions}} @cython.embedsignature(True) def cuMemUnmap(ptr, size_t size): """ Unmap the backing memory of a given address range. The range must be the entire contiguous address range that was mapped to. In other words, :py:obj:`~.cuMemUnmap` cannot unmap a sub-range of an address range mapped by :py:obj:`~.cuMemCreate` / :py:obj:`~.cuMemMap`. Any backing memory allocations will be freed if there are no existing mappings and there are no unreleased memory handles. When :py:obj:`~.cuMemUnmap` returns successfully the address range is converted to an address reservation and can be used for a future calls to :py:obj:`~.cuMemMap`. Any new mapping to this virtual address will need to have access granted through :py:obj:`~.cuMemSetAccess`, as all mappings start with no accessibility setup. Parameters ---------- ptr : :py:obj:`~.CUdeviceptr` Starting address for the virtual address range to unmap size : size_t Size of the virtual address range to unmap Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMemCreate`, :py:obj:`~.cuMemAddressReserve` """ cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr err = ccuda.cuMemUnmap(cptr, size) return (CUresult(err),) {{endif}} {{if 'cuMemSetAccess' in found_functions}} @cython.embedsignature(True) def cuMemSetAccess(ptr, size_t size, desc : Optional[Tuple[CUmemAccessDesc] | List[CUmemAccessDesc]], size_t count): """ Set the access flags for each location specified in `desc` for the given virtual address range. Given the virtual address range via `ptr` and `size`, and the locations in the array given by `desc` and `count`, set the access flags for the target locations. The range must be a fully mapped address range containing all allocations created by :py:obj:`~.cuMemMap` / :py:obj:`~.cuMemCreate`. Users cannot specify :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA` accessibility for allocations created on with other location types. Note: When :py:obj:`~.CUmemAccessDesc`::CUmemLocation::type is :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA`, :py:obj:`~.CUmemAccessDesc`::CUmemLocation::id is ignored. When setting the access flags for a virtual address range mapping a multicast object, `ptr` and `size` must be aligned to the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_MINIMUM_GRANULARITY`. For best performance however, it is recommended that `ptr` and `size` be aligned to the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_RECOMMENDED_GRANULARITY`. Parameters ---------- ptr : :py:obj:`~.CUdeviceptr` Starting address for the virtual address range size : size_t Length of the virtual address range desc : List[:py:obj:`~.CUmemAccessDesc`] Array of :py:obj:`~.CUmemAccessDesc` that describe how to change the count : size_t Number of :py:obj:`~.CUmemAccessDesc` in `desc` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMemSetAccess`, :py:obj:`~.cuMemCreate`, :py:obj:`~.py`:obj:`~.cuMemMap` """ desc = [] if desc is None else desc if not all(isinstance(_x, (CUmemAccessDesc,)) for _x in desc): raise TypeError("Argument 'desc' is not instance of type (expected Tuple[ccuda.CUmemAccessDesc,] or List[ccuda.CUmemAccessDesc,]") cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr cdef ccuda.CUmemAccessDesc* cdesc = NULL if len(desc) > 0: cdesc = calloc(len(desc), sizeof(ccuda.CUmemAccessDesc)) if cdesc is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(desc)) + 'x' + str(sizeof(ccuda.CUmemAccessDesc))) for idx in range(len(desc)): string.memcpy(&cdesc[idx], (desc[idx])._ptr, sizeof(ccuda.CUmemAccessDesc)) if count > len(desc): raise RuntimeError("List is too small: " + str(len(desc)) + " < " + str(count)) err = ccuda.cuMemSetAccess(cptr, size, (desc[0])._ptr if len(desc) == 1 else cdesc, count) if cdesc is not NULL: free(cdesc) return (CUresult(err),) {{endif}} {{if 'cuMemGetAccess' in found_functions}} @cython.embedsignature(True) def cuMemGetAccess(location : Optional[CUmemLocation], ptr): """ Get the access `flags` set for the given `location` and `ptr`. Parameters ---------- location : :py:obj:`~.CUmemLocation` Location in which to check the flags for ptr : :py:obj:`~.CUdeviceptr` Address in which to check the access flags for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` flags : unsigned long long Flags set for this location See Also -------- :py:obj:`~.cuMemSetAccess` """ cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr cdef unsigned long long flags = 0 cdef ccuda.CUmemLocation* clocation_ptr = location._ptr if location != None else NULL err = ccuda.cuMemGetAccess(&flags, clocation_ptr, cptr) return (CUresult(err), flags) {{endif}} {{if 'cuMemExportToShareableHandle' in found_functions}} @cython.embedsignature(True) def cuMemExportToShareableHandle(handle, handleType not None : CUmemAllocationHandleType, unsigned long long flags): """ Exports an allocation to a requested shareable handle type. Given a CUDA memory handle, create a shareable memory allocation handle that can be used to share the memory with other processes. The recipient process can convert the shareable handle back into a CUDA memory handle using :py:obj:`~.cuMemImportFromShareableHandle` and map it with :py:obj:`~.cuMemMap`. The implementation of what this handle is and how it can be transferred is defined by the requested handle type in `handleType` Once all shareable handles are closed and the allocation is released, the allocated memory referenced will be released back to the OS and uses of the CUDA handle afterward will lead to undefined behavior. This API can also be used in conjunction with other APIs (e.g. Vulkan, OpenGL) that support importing memory from the shareable type Parameters ---------- handle : :py:obj:`~.CUmemGenericAllocationHandle` CUDA handle for the memory allocation handleType : :py:obj:`~.CUmemAllocationHandleType` Type of shareable handle requested (defines type and size of the `shareableHandle` output parameter) flags : unsigned long long Reserved, must be zero Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` shareableHandle : Any Pointer to the location in which to store the requested handle type See Also -------- :py:obj:`~.cuMemImportFromShareableHandle` """ cdef ccuda.CUmemGenericAllocationHandle chandle if handle is None: chandle = 0 elif isinstance(handle, (CUmemGenericAllocationHandle,)): phandle = int(handle) chandle = phandle else: phandle = int(CUmemGenericAllocationHandle(handle)) chandle = phandle cdef utils.HelperCUmemAllocationHandleType cshareableHandle = utils.HelperCUmemAllocationHandleType(handleType) cdef void* cshareableHandle_ptr = cshareableHandle.cptr cdef ccuda.CUmemAllocationHandleType chandleType = handleType.value err = ccuda.cuMemExportToShareableHandle(cshareableHandle_ptr, chandle, chandleType, flags) return (CUresult(err), cshareableHandle.pyObj()) {{endif}} {{if 'cuMemImportFromShareableHandle' in found_functions}} @cython.embedsignature(True) def cuMemImportFromShareableHandle(osHandle, shHandleType not None : CUmemAllocationHandleType): """ Imports an allocation from a requested shareable handle type. If the current process cannot support the memory described by this shareable handle, this API will error as :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`. If `shHandleType` is :py:obj:`~.CU_MEM_HANDLE_TYPE_FABRIC` and the importer process has not been granted access to the same IMEX channel as the exporter process, this API will error as :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`. Parameters ---------- osHandle : Any Shareable Handle representing the memory allocation that is to be imported. shHandleType : :py:obj:`~.CUmemAllocationHandleType` handle type of the exported handle :py:obj:`~.CUmemAllocationHandleType`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` handle : :py:obj:`~.CUmemGenericAllocationHandle` CUDA Memory handle for the memory allocation. See Also -------- :py:obj:`~.cuMemExportToShareableHandle`, :py:obj:`~.cuMemMap`, :py:obj:`~.cuMemRelease` Notes ----- Importing shareable handles exported from some graphics APIs(VUlkan, OpenGL, etc) created on devices under an SLI group may not be supported, and thus this API will return CUDA_ERROR_NOT_SUPPORTED. There is no guarantee that the contents of `handle` will be the same CUDA memory handle for the same given OS shareable handle, or the same underlying allocation. """ cdef CUmemGenericAllocationHandle handle = CUmemGenericAllocationHandle() cosHandle = utils.HelperInputVoidPtr(osHandle) cdef void* cosHandle_ptr = cosHandle.cptr cdef ccuda.CUmemAllocationHandleType cshHandleType = shHandleType.value err = ccuda.cuMemImportFromShareableHandle(handle._ptr, cosHandle_ptr, cshHandleType) return (CUresult(err), handle) {{endif}} {{if 'cuMemGetAllocationGranularity' in found_functions}} @cython.embedsignature(True) def cuMemGetAllocationGranularity(prop : Optional[CUmemAllocationProp], option not None : CUmemAllocationGranularity_flags): """ Calculates either the minimal or recommended granularity. Calculates either the minimal or recommended granularity for a given allocation specification and returns it in granularity. This granularity can be used as a multiple for alignment, size, or address mapping. Parameters ---------- prop : :py:obj:`~.CUmemAllocationProp` Property for which to determine the granularity for option : :py:obj:`~.CUmemAllocationGranularity_flags` Determines which granularity to return Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` granularity : int Returned granularity. See Also -------- :py:obj:`~.cuMemCreate`, :py:obj:`~.cuMemMap` """ cdef size_t granularity = 0 cdef ccuda.CUmemAllocationProp* cprop_ptr = prop._ptr if prop != None else NULL cdef ccuda.CUmemAllocationGranularity_flags coption = option.value err = ccuda.cuMemGetAllocationGranularity(&granularity, cprop_ptr, coption) return (CUresult(err), granularity) {{endif}} {{if 'cuMemGetAllocationPropertiesFromHandle' in found_functions}} @cython.embedsignature(True) def cuMemGetAllocationPropertiesFromHandle(handle): """ Retrieve the contents of the property structure defining properties for this handle. Parameters ---------- handle : :py:obj:`~.CUmemGenericAllocationHandle` Handle which to perform the query on Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` prop : :py:obj:`~.CUmemAllocationProp` Pointer to a properties structure which will hold the information about this handle See Also -------- :py:obj:`~.cuMemCreate`, :py:obj:`~.cuMemImportFromShareableHandle` """ cdef ccuda.CUmemGenericAllocationHandle chandle if handle is None: chandle = 0 elif isinstance(handle, (CUmemGenericAllocationHandle,)): phandle = int(handle) chandle = phandle else: phandle = int(CUmemGenericAllocationHandle(handle)) chandle = phandle cdef CUmemAllocationProp prop = CUmemAllocationProp() err = ccuda.cuMemGetAllocationPropertiesFromHandle(prop._ptr, chandle) return (CUresult(err), prop) {{endif}} {{if 'cuMemRetainAllocationHandle' in found_functions}} @cython.embedsignature(True) def cuMemRetainAllocationHandle(addr): """ Given an address `addr`, returns the allocation handle of the backing memory allocation. The handle is guaranteed to be the same handle value used to map the memory. If the address requested is not mapped, the function will fail. The returned handle must be released with corresponding number of calls to :py:obj:`~.cuMemRelease`. Parameters ---------- addr : Any Memory address to query, that has been mapped previously. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` handle : :py:obj:`~.CUmemGenericAllocationHandle` CUDA Memory handle for the backing memory allocation. See Also -------- :py:obj:`~.cuMemCreate`, :py:obj:`~.cuMemRelease`, :py:obj:`~.cuMemMap` Notes ----- The address `addr`, can be any address in a range previously mapped by :py:obj:`~.cuMemMap`, and not necessarily the start address. """ cdef CUmemGenericAllocationHandle handle = CUmemGenericAllocationHandle() caddr = utils.HelperInputVoidPtr(addr) cdef void* caddr_ptr = caddr.cptr err = ccuda.cuMemRetainAllocationHandle(handle._ptr, caddr_ptr) return (CUresult(err), handle) {{endif}} {{if 'cuMemFreeAsync' in found_functions}} @cython.embedsignature(True) def cuMemFreeAsync(dptr, hStream): """ Frees memory with stream ordered semantics. Inserts a free operation into `hStream`. The allocation must not be accessed after stream execution reaches the free. After this API returns, accessing the memory from any subsequent work launched on the GPU or querying its pointer attributes results in undefined behavior. Parameters ---------- dptr : :py:obj:`~.CUdeviceptr` memory to free hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream establishing the stream ordering contract. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` (default stream specified with no current context), :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` Notes ----- During stream capture, this function results in the creation of a free node and must therefore be passed the address of a graph allocation. """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr err = ccuda.cuMemFreeAsync(cdptr, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemAllocAsync' in found_functions}} @cython.embedsignature(True) def cuMemAllocAsync(size_t bytesize, hStream): """ Allocates memory with stream ordered semantics. Inserts an allocation operation into `hStream`. A pointer to the allocated memory is returned immediately in *dptr. The allocation must not be accessed until the the allocation operation completes. The allocation comes from the memory pool current to the stream's device. Parameters ---------- bytesize : size_t Number of bytes to allocate hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream establishing the stream ordering contract and the memory pool to allocate from Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` (default stream specified with no current context), :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` dptr : :py:obj:`~.CUdeviceptr` Returned device pointer See Also -------- :py:obj:`~.cuMemAllocFromPoolAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceSetMemPool`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate`, :py:obj:`~.cuMemPoolSetAccess`, :py:obj:`~.cuMemPoolSetAttribute` Notes ----- The default memory pool of a device contains device memory from that device. Basic stream ordering allows future work submitted into the same stream to use the allocation. Stream query, stream synchronize, and CUDA events can be used to guarantee that the allocation operation completes before work submitted in a separate stream runs. During stream capture, this function results in the creation of an allocation node. In this case, the allocation is owned by the graph instead of the memory pool. The memory pool's properties are used to set the node's creation parameters. """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef CUdeviceptr dptr = CUdeviceptr() err = ccuda.cuMemAllocAsync(dptr._ptr, bytesize, chStream) return (CUresult(err), dptr) {{endif}} {{if 'cuMemPoolTrimTo' in found_functions}} @cython.embedsignature(True) def cuMemPoolTrimTo(pool, size_t minBytesToKeep): """ Tries to release memory back to the OS. Releases memory back to the OS until the pool contains fewer than minBytesToKeep reserved bytes, or there is no more memory that the allocator can safely release. The allocator cannot release OS allocations that back outstanding asynchronous allocations. The OS allocations may happen at different granularity from the user allocations. Parameters ---------- pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` The memory pool to trim minBytesToKeep : size_t If the pool has less than minBytesToKeep reserved, the TrimTo operation is a no-op. Otherwise the pool will be guaranteed to have at least minBytesToKeep bytes reserved after the operation. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate` Notes ----- : Allocations that have not been freed count as outstanding. : Allocations that have been asynchronously freed but whose completion has not been observed on the host (eg. by a synchronize) can count as outstanding. """ cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool err = ccuda.cuMemPoolTrimTo(cpool, minBytesToKeep) return (CUresult(err),) {{endif}} {{if 'cuMemPoolSetAttribute' in found_functions}} @cython.embedsignature(True) def cuMemPoolSetAttribute(pool, attr not None : CUmemPool_attribute, value): """ Sets attributes of a memory pool. Supported attributes are: - :py:obj:`~.CU_MEMPOOL_ATTR_RELEASE_THRESHOLD`: (value type = cuuint64_t) Amount of reserved memory in bytes to hold onto before trying to release memory back to the OS. When more than the release threshold bytes of memory are held by the memory pool, the allocator will try to release memory back to the OS on the next call to stream, event or context synchronize. (default 0) - :py:obj:`~.CU_MEMPOOL_ATTR_REUSE_FOLLOW_EVENT_DEPENDENCIES`: (value type = int) Allow :py:obj:`~.cuMemAllocAsync` to use memory asynchronously freed in another stream as long as a stream ordering dependency of the allocating stream on the free action exists. Cuda events and null stream interactions can create the required stream ordered dependencies. (default enabled) - :py:obj:`~.CU_MEMPOOL_ATTR_REUSE_ALLOW_OPPORTUNISTIC`: (value type = int) Allow reuse of already completed frees when there is no dependency between the free and allocation. (default enabled) - :py:obj:`~.CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES`: (value type = int) Allow :py:obj:`~.cuMemAllocAsync` to insert new stream dependencies in order to establish the stream ordering required to reuse a piece of memory released by :py:obj:`~.cuMemFreeAsync` (default enabled). - :py:obj:`~.CU_MEMPOOL_ATTR_RESERVED_MEM_HIGH`: (value type = cuuint64_t) Reset the high watermark that tracks the amount of backing memory that was allocated for the memory pool. It is illegal to set this attribute to a non-zero value. - :py:obj:`~.CU_MEMPOOL_ATTR_USED_MEM_HIGH`: (value type = cuuint64_t) Reset the high watermark that tracks the amount of used memory that was allocated for the memory pool. Parameters ---------- pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` The memory pool to modify attr : :py:obj:`~.CUmemPool_attribute` The attribute to modify value : Any Pointer to the value to assign Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate` """ cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool cdef ccuda.CUmemPool_attribute cattr = attr.value cdef utils.HelperCUmemPool_attribute cvalue = utils.HelperCUmemPool_attribute(attr, value, is_getter=False) cdef void* cvalue_ptr = cvalue.cptr err = ccuda.cuMemPoolSetAttribute(cpool, cattr, cvalue_ptr) return (CUresult(err),) {{endif}} {{if 'cuMemPoolGetAttribute' in found_functions}} @cython.embedsignature(True) def cuMemPoolGetAttribute(pool, attr not None : CUmemPool_attribute): """ Gets attributes of a memory pool. Supported attributes are: - :py:obj:`~.CU_MEMPOOL_ATTR_RELEASE_THRESHOLD`: (value type = cuuint64_t) Amount of reserved memory in bytes to hold onto before trying to release memory back to the OS. When more than the release threshold bytes of memory are held by the memory pool, the allocator will try to release memory back to the OS on the next call to stream, event or context synchronize. (default 0) - :py:obj:`~.CU_MEMPOOL_ATTR_REUSE_FOLLOW_EVENT_DEPENDENCIES`: (value type = int) Allow :py:obj:`~.cuMemAllocAsync` to use memory asynchronously freed in another stream as long as a stream ordering dependency of the allocating stream on the free action exists. Cuda events and null stream interactions can create the required stream ordered dependencies. (default enabled) - :py:obj:`~.CU_MEMPOOL_ATTR_REUSE_ALLOW_OPPORTUNISTIC`: (value type = int) Allow reuse of already completed frees when there is no dependency between the free and allocation. (default enabled) - :py:obj:`~.CU_MEMPOOL_ATTR_REUSE_ALLOW_INTERNAL_DEPENDENCIES`: (value type = int) Allow :py:obj:`~.cuMemAllocAsync` to insert new stream dependencies in order to establish the stream ordering required to reuse a piece of memory released by :py:obj:`~.cuMemFreeAsync` (default enabled). - :py:obj:`~.CU_MEMPOOL_ATTR_RESERVED_MEM_CURRENT`: (value type = cuuint64_t) Amount of backing memory currently allocated for the mempool - :py:obj:`~.CU_MEMPOOL_ATTR_RESERVED_MEM_HIGH`: (value type = cuuint64_t) High watermark of backing memory allocated for the mempool since the last time it was reset. - :py:obj:`~.CU_MEMPOOL_ATTR_USED_MEM_CURRENT`: (value type = cuuint64_t) Amount of memory from the pool that is currently in use by the application. - :py:obj:`~.CU_MEMPOOL_ATTR_USED_MEM_HIGH`: (value type = cuuint64_t) High watermark of the amount of memory from the pool that was in use by the application. Parameters ---------- pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` The memory pool to get attributes of attr : :py:obj:`~.CUmemPool_attribute` The attribute to get Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` value : Any Retrieved value See Also -------- :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate` """ cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool cdef ccuda.CUmemPool_attribute cattr = attr.value cdef utils.HelperCUmemPool_attribute cvalue = utils.HelperCUmemPool_attribute(attr, 0, is_getter=True) cdef void* cvalue_ptr = cvalue.cptr err = ccuda.cuMemPoolGetAttribute(cpool, cattr, cvalue_ptr) return (CUresult(err), cvalue.pyObj()) {{endif}} {{if 'cuMemPoolSetAccess' in found_functions}} @cython.embedsignature(True) def cuMemPoolSetAccess(pool, map : Optional[Tuple[CUmemAccessDesc] | List[CUmemAccessDesc]], size_t count): """ Controls visibility of pools between devices. Parameters ---------- pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` The pool being modified map : List[:py:obj:`~.CUmemAccessDesc`] Array of access descriptors. Each descriptor instructs the access to enable for a single gpu. count : size_t Number of descriptors in the map array. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate` """ map = [] if map is None else map if not all(isinstance(_x, (CUmemAccessDesc,)) for _x in map): raise TypeError("Argument 'map' is not instance of type (expected Tuple[ccuda.CUmemAccessDesc,] or List[ccuda.CUmemAccessDesc,]") cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool cdef ccuda.CUmemAccessDesc* cmap = NULL if len(map) > 0: cmap = calloc(len(map), sizeof(ccuda.CUmemAccessDesc)) if cmap is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(map)) + 'x' + str(sizeof(ccuda.CUmemAccessDesc))) for idx in range(len(map)): string.memcpy(&cmap[idx], (map[idx])._ptr, sizeof(ccuda.CUmemAccessDesc)) if count > len(map): raise RuntimeError("List is too small: " + str(len(map)) + " < " + str(count)) err = ccuda.cuMemPoolSetAccess(cpool, (map[0])._ptr if len(map) == 1 else cmap, count) if cmap is not NULL: free(cmap) return (CUresult(err),) {{endif}} {{if 'cuMemPoolGetAccess' in found_functions}} @cython.embedsignature(True) def cuMemPoolGetAccess(memPool, location : Optional[CUmemLocation]): """ Returns the accessibility of a pool from a device. Returns the accessibility of the pool's memory from the specified location. Parameters ---------- memPool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` the pool being queried location : :py:obj:`~.CUmemLocation` the location accessing the pool Returns ------- CUresult flags : :py:obj:`~.CUmemAccess_flags` the accessibility of the pool from the specified location See Also -------- :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate` """ cdef ccuda.CUmemoryPool cmemPool if memPool is None: cmemPool = 0 elif isinstance(memPool, (CUmemoryPool,)): pmemPool = int(memPool) cmemPool = pmemPool else: pmemPool = int(CUmemoryPool(memPool)) cmemPool = pmemPool cdef ccuda.CUmemAccess_flags flags cdef ccuda.CUmemLocation* clocation_ptr = location._ptr if location != None else NULL err = ccuda.cuMemPoolGetAccess(&flags, cmemPool, clocation_ptr) return (CUresult(err), CUmemAccess_flags(flags)) {{endif}} {{if 'cuMemPoolCreate' in found_functions}} @cython.embedsignature(True) def cuMemPoolCreate(poolProps : Optional[CUmemPoolProps]): """ Creates a memory pool. Creates a CUDA memory pool and returns the handle in `pool`. The `poolProps` determines the properties of the pool such as the backing device and IPC capabilities. To create a memory pool targeting a specific host NUMA node, applications must set :py:obj:`~.CUmemPoolProps`::CUmemLocation::type to :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA` and :py:obj:`~.CUmemPoolProps`::CUmemLocation::id must specify the NUMA ID of the host memory node. Specifying :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT` or :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST` as the :py:obj:`~.CUmemPoolProps`::CUmemLocation::type will result in :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. By default, the pool's memory will be accessible from the device it is allocated on. In the case of pools created with :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA`, their default accessibility will be from the host CPU. Applications can control the maximum size of the pool by specifying a non-zero value for :py:obj:`~.CUmemPoolProps.maxSize`. If set to 0, the maximum size of the pool will default to a system dependent value. Applications can set :py:obj:`~.CUmemPoolProps.handleTypes` to :py:obj:`~.CU_MEM_HANDLE_TYPE_FABRIC` in order to create :py:obj:`~.CUmemoryPool` suitable for sharing within an IMEX domain. An IMEX domain is either an OS instance or a group of securely connected OS instances using the NVIDIA IMEX daemon. An IMEX channel is a global resource within the IMEX domain that represents a logical entity that aims to provide fine grained accessibility control for the participating processes. When exporter and importer CUDA processes have been granted access to the same IMEX channel, they can securely share memory. If the allocating process does not have access setup for an IMEX channel, attempting to export a :py:obj:`~.CUmemoryPool` with :py:obj:`~.CU_MEM_HANDLE_TYPE_FABRIC` will result in :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`. The nvidia-modprobe CLI provides more information regarding setting up of IMEX channels. Parameters ---------- poolProps : :py:obj:`~.CUmemPoolProps` None Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED` :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` pool : :py:obj:`~.CUmemoryPool` None See Also -------- :py:obj:`~.cuDeviceSetMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuMemAllocFromPoolAsync`, :py:obj:`~.cuMemPoolExportToShareableHandle` Notes ----- Specifying CU_MEM_HANDLE_TYPE_NONE creates a memory pool that will not support IPC. """ cdef CUmemoryPool pool = CUmemoryPool() cdef ccuda.CUmemPoolProps* cpoolProps_ptr = poolProps._ptr if poolProps != None else NULL err = ccuda.cuMemPoolCreate(pool._ptr, cpoolProps_ptr) return (CUresult(err), pool) {{endif}} {{if 'cuMemPoolDestroy' in found_functions}} @cython.embedsignature(True) def cuMemPoolDestroy(pool): """ Destroys the specified memory pool. If any pointers obtained from this pool haven't been freed or the pool has free operations that haven't completed when :py:obj:`~.cuMemPoolDestroy` is invoked, the function will return immediately and the resources associated with the pool will be released automatically once there are no more outstanding allocations. Destroying the current mempool of a device sets the default mempool of that device as the current mempool for that device. Parameters ---------- pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` None Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceSetMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuMemPoolCreate` Notes ----- A device's default memory pool cannot be destroyed. """ cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool err = ccuda.cuMemPoolDestroy(cpool) return (CUresult(err),) {{endif}} {{if 'cuMemAllocFromPoolAsync' in found_functions}} @cython.embedsignature(True) def cuMemAllocFromPoolAsync(size_t bytesize, pool, hStream): """ Allocates memory from a specified pool with stream ordered semantics. Inserts an allocation operation into `hStream`. A pointer to the allocated memory is returned immediately in *dptr. The allocation must not be accessed until the the allocation operation completes. The allocation comes from the specified memory pool. Parameters ---------- bytesize : size_t Number of bytes to allocate pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` The pool to allocate from hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream establishing the stream ordering semantic Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` (default stream specified with no current context), :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` dptr : :py:obj:`~.CUdeviceptr` Returned device pointer See Also -------- :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate`, :py:obj:`~.cuMemPoolSetAccess`, :py:obj:`~.cuMemPoolSetAttribute` Notes ----- During stream capture, this function results in the creation of an allocation node. In this case, the allocation is owned by the graph instead of the memory pool. The memory pool's properties are used to set the node's creation parameters. """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool cdef CUdeviceptr dptr = CUdeviceptr() err = ccuda.cuMemAllocFromPoolAsync(dptr._ptr, bytesize, cpool, chStream) return (CUresult(err), dptr) {{endif}} {{if 'cuMemPoolExportToShareableHandle' in found_functions}} @cython.embedsignature(True) def cuMemPoolExportToShareableHandle(pool, handleType not None : CUmemAllocationHandleType, unsigned long long flags): """ Exports a memory pool to the requested handle type. Given an IPC capable mempool, create an OS handle to share the pool with another process. A recipient process can convert the shareable handle into a mempool with :py:obj:`~.cuMemPoolImportFromShareableHandle`. Individual pointers can then be shared with the :py:obj:`~.cuMemPoolExportPointer` and :py:obj:`~.cuMemPoolImportPointer` APIs. The implementation of what the shareable handle is and how it can be transferred is defined by the requested handle type. Parameters ---------- pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` pool to export handleType : :py:obj:`~.CUmemAllocationHandleType` the type of handle to create flags : unsigned long long must be 0 Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` handle_out : Any Returned OS handle See Also -------- :py:obj:`~.cuMemPoolImportFromShareableHandle`, :py:obj:`~.cuMemPoolExportPointer`, :py:obj:`~.cuMemPoolImportPointer`, :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuDeviceGetDefaultMemPool`, :py:obj:`~.cuDeviceGetMemPool`, :py:obj:`~.cuMemPoolCreate`, :py:obj:`~.cuMemPoolSetAccess`, :py:obj:`~.cuMemPoolSetAttribute` Notes ----- : To create an IPC capable mempool, create a mempool with a CUmemAllocationHandleType other than CU_MEM_HANDLE_TYPE_NONE. """ cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool cdef utils.HelperCUmemAllocationHandleType chandle_out = utils.HelperCUmemAllocationHandleType(handleType) cdef void* chandle_out_ptr = chandle_out.cptr cdef ccuda.CUmemAllocationHandleType chandleType = handleType.value err = ccuda.cuMemPoolExportToShareableHandle(chandle_out_ptr, cpool, chandleType, flags) return (CUresult(err), chandle_out.pyObj()) {{endif}} {{if 'cuMemPoolImportFromShareableHandle' in found_functions}} @cython.embedsignature(True) def cuMemPoolImportFromShareableHandle(handle, handleType not None : CUmemAllocationHandleType, unsigned long long flags): """ imports a memory pool from a shared handle. Specific allocations can be imported from the imported pool with cuMemPoolImportPointer. If `handleType` is :py:obj:`~.CU_MEM_HANDLE_TYPE_FABRIC` and the importer process has not been granted access to the same IMEX channel as the exporter process, this API will error as :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`. Parameters ---------- handle : Any OS handle of the pool to open handleType : :py:obj:`~.CUmemAllocationHandleType` The type of handle being imported flags : unsigned long long must be 0 Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` pool_out : :py:obj:`~.CUmemoryPool` Returned memory pool See Also -------- :py:obj:`~.cuMemPoolExportToShareableHandle`, :py:obj:`~.cuMemPoolExportPointer`, :py:obj:`~.cuMemPoolImportPointer` Notes ----- Imported memory pools do not support creating new allocations. As such imported memory pools may not be used in cuDeviceSetMemPool or :py:obj:`~.cuMemAllocFromPoolAsync` calls. """ cdef CUmemoryPool pool_out = CUmemoryPool() chandle = utils.HelperInputVoidPtr(handle) cdef void* chandle_ptr = chandle.cptr cdef ccuda.CUmemAllocationHandleType chandleType = handleType.value err = ccuda.cuMemPoolImportFromShareableHandle(pool_out._ptr, chandle_ptr, chandleType, flags) return (CUresult(err), pool_out) {{endif}} {{if 'cuMemPoolExportPointer' in found_functions}} @cython.embedsignature(True) def cuMemPoolExportPointer(ptr): """ Export data to share a memory pool allocation between processes. Constructs `shareData_out` for sharing a specific allocation from an already shared memory pool. The recipient process can import the allocation with the :py:obj:`~.cuMemPoolImportPointer` api. The data is not a handle and may be shared through any IPC mechanism. Parameters ---------- ptr : :py:obj:`~.CUdeviceptr` pointer to memory being exported Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` shareData_out : :py:obj:`~.CUmemPoolPtrExportData` Returned export data See Also -------- :py:obj:`~.cuMemPoolExportToShareableHandle`, :py:obj:`~.cuMemPoolImportFromShareableHandle`, :py:obj:`~.cuMemPoolImportPointer` """ cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr cdef CUmemPoolPtrExportData shareData_out = CUmemPoolPtrExportData() err = ccuda.cuMemPoolExportPointer(shareData_out._ptr, cptr) return (CUresult(err), shareData_out) {{endif}} {{if 'cuMemPoolImportPointer' in found_functions}} @cython.embedsignature(True) def cuMemPoolImportPointer(pool, shareData : Optional[CUmemPoolPtrExportData]): """ Import a memory pool allocation from another process. Returns in `ptr_out` a pointer to the imported memory. The imported memory must not be accessed before the allocation operation completes in the exporting process. The imported memory must be freed from all importing processes before being freed in the exporting process. The pointer may be freed with cuMemFree or cuMemFreeAsync. If cuMemFreeAsync is used, the free must be completed on the importing process before the free operation on the exporting process. Parameters ---------- pool : :py:obj:`~.CUmemoryPool` or :py:obj:`~.cudaMemPool_t` pool from which to import shareData : :py:obj:`~.CUmemPoolPtrExportData` data specifying the memory to import Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` ptr_out : :py:obj:`~.CUdeviceptr` pointer to imported memory See Also -------- :py:obj:`~.cuMemPoolExportToShareableHandle`, :py:obj:`~.cuMemPoolImportFromShareableHandle`, :py:obj:`~.cuMemPoolExportPointer` Notes ----- The cuMemFreeAsync api may be used in the exporting process before the cuMemFreeAsync operation completes in its stream as long as the cuMemFreeAsync in the exporting process specifies a stream with a stream dependency on the importing process's cuMemFreeAsync. """ cdef ccuda.CUmemoryPool cpool if pool is None: cpool = 0 elif isinstance(pool, (CUmemoryPool,)): ppool = int(pool) cpool = ppool else: ppool = int(CUmemoryPool(pool)) cpool = ppool cdef CUdeviceptr ptr_out = CUdeviceptr() cdef ccuda.CUmemPoolPtrExportData* cshareData_ptr = shareData._ptr if shareData != None else NULL err = ccuda.cuMemPoolImportPointer(ptr_out._ptr, cpool, cshareData_ptr) return (CUresult(err), ptr_out) {{endif}} {{if 'cuMulticastCreate' in found_functions}} @cython.embedsignature(True) def cuMulticastCreate(prop : Optional[CUmulticastObjectProp]): """ Create a generic allocation handle representing a multicast object described by the given properties. This creates a multicast object as described by `prop`. The number of participating devices is specified by :py:obj:`~.CUmulticastObjectProp.numDevices`. Devices can be added to the multicast object via :py:obj:`~.cuMulticastAddDevice`. All participating devices must be added to the multicast object before memory can be bound to it. Memory is bound to the multicast object via either :py:obj:`~.cuMulticastBindMem` or :py:obj:`~.cuMulticastBindAddr`, and can be unbound via :py:obj:`~.cuMulticastUnbind`. The total amount of memory that can be bound per device is specified by :py:obj:`~.py`:obj:`~.CUmulticastObjectProp.size`. This size must be a multiple of the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_GRANULARITY_MINIMUM`. For best performance however, the size should be aligned to the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_GRANULARITY_RECOMMENDED`. After all participating devices have been added, multicast objects can also be mapped to a device's virtual address space using the virtual memory management APIs (see :py:obj:`~.cuMemMap` and :py:obj:`~.cuMemSetAccess`). Multicast objects can also be shared with other processes by requesting a shareable handle via :py:obj:`~.cuMemExportToShareableHandle`. Note that the desired types of shareable handles must be specified in the bitmask :py:obj:`~.CUmulticastObjectProp.handleTypes`. Multicast objects can be released using the virtual memory management API :py:obj:`~.cuMemRelease`. Parameters ---------- prop : :py:obj:`~.CUmulticastObjectProp` Properties of the multicast object to create. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` mcHandle : :py:obj:`~.CUmemGenericAllocationHandle` Value of handle returned. See Also -------- :py:obj:`~.cuMulticastAddDevice`, :py:obj:`~.cuMulticastBindMem`, :py:obj:`~.cuMulticastBindAddr`, :py:obj:`~.cuMulticastUnbind` :py:obj:`~.cuMemCreate`, :py:obj:`~.cuMemRelease`, :py:obj:`~.cuMemExportToShareableHandle`, :py:obj:`~.cuMemImportFromShareableHandle` """ cdef CUmemGenericAllocationHandle mcHandle = CUmemGenericAllocationHandle() cdef ccuda.CUmulticastObjectProp* cprop_ptr = prop._ptr if prop != None else NULL err = ccuda.cuMulticastCreate(mcHandle._ptr, cprop_ptr) return (CUresult(err), mcHandle) {{endif}} {{if 'cuMulticastAddDevice' in found_functions}} @cython.embedsignature(True) def cuMulticastAddDevice(mcHandle, dev): """ Associate a device to a multicast object. Associates a device to a multicast object. The added device will be a part of the multicast team of size specified by :py:obj:`~.CUmulticastObjectProp.numDevices` during :py:obj:`~.cuMulticastCreate`. The association of the device to the multicast object is permanent during the life time of the multicast object. All devices must be added to the multicast team before any memory can be bound to any device in the team. Any calls to :py:obj:`~.cuMulticastBindMem` or :py:obj:`~.cuMulticastBindAddr` will block until all devices have been added. Similarly all devices must be added to the multicast team before a virtual address range can be mapped to the multicast object. A call to :py:obj:`~.cuMemMap` will block until all devices have been added. Parameters ---------- mcHandle : :py:obj:`~.CUmemGenericAllocationHandle` Handle representing a multicast object. dev : :py:obj:`~.CUdevice` Device that will be associated to the multicast object. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMulticastCreate`, :py:obj:`~.cuMulticastBindMem`, :py:obj:`~.cuMulticastBindAddr` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef ccuda.CUmemGenericAllocationHandle cmcHandle if mcHandle is None: cmcHandle = 0 elif isinstance(mcHandle, (CUmemGenericAllocationHandle,)): pmcHandle = int(mcHandle) cmcHandle = pmcHandle else: pmcHandle = int(CUmemGenericAllocationHandle(mcHandle)) cmcHandle = pmcHandle err = ccuda.cuMulticastAddDevice(cmcHandle, cdev) return (CUresult(err),) {{endif}} {{if 'cuMulticastBindMem' in found_functions}} @cython.embedsignature(True) def cuMulticastBindMem(mcHandle, size_t mcOffset, memHandle, size_t memOffset, size_t size, unsigned long long flags): """ Bind a memory allocation represented by a handle to a multicast object. Binds a memory allocation specified by `memHandle` and created via :py:obj:`~.cuMemCreate` to a multicast object represented by `mcHandle` and created via :py:obj:`~.cuMulticastCreate`. The intended `size` of the bind, the offset in the multicast range `mcOffset` as well as the offset in the memory `memOffset` must be a multiple of the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_GRANULARITY_MINIMUM`. For best performance however, `size`, `mcOffset` and `memOffset` should be aligned to the granularity of the memory allocation(see :py:obj:`~.cuMemGetAllocationGranularity`) or to the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_GRANULARITY_RECOMMENDED`. The `size` + `memOffset` cannot be larger than the size of the allocated memory. Similarly the `size` + `mcOffset` cannot be larger than the size of the multicast object. The memory allocation must have beeen created on one of the devices that was added to the multicast team via :py:obj:`~.cuMulticastAddDevice`. Externally shareable as well as imported multicast objects can be bound only to externally shareable memory. Note that this call will return CUDA_ERROR_OUT_OF_MEMORY if there are insufficient resources required to perform the bind. This call may also return CUDA_ERROR_SYSTEM_NOT_READY if the necessary system software is not initialized or running. Parameters ---------- mcHandle : :py:obj:`~.CUmemGenericAllocationHandle` Handle representing a multicast object. mcOffset : size_t Offset into the multicast object for attachment. memHandle : :py:obj:`~.CUmemGenericAllocationHandle` Handle representing a memory allocation. memOffset : size_t Offset into the memory for attachment. size : size_t Size of the memory that will be bound to the multicast object. flags : unsigned long long Flags for future use, must be zero for now. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_SYSTEM_NOT_READY` See Also -------- :py:obj:`~.cuMulticastCreate`, :py:obj:`~.cuMulticastAddDevice`, :py:obj:`~.cuMemCreate` """ cdef ccuda.CUmemGenericAllocationHandle cmemHandle if memHandle is None: cmemHandle = 0 elif isinstance(memHandle, (CUmemGenericAllocationHandle,)): pmemHandle = int(memHandle) cmemHandle = pmemHandle else: pmemHandle = int(CUmemGenericAllocationHandle(memHandle)) cmemHandle = pmemHandle cdef ccuda.CUmemGenericAllocationHandle cmcHandle if mcHandle is None: cmcHandle = 0 elif isinstance(mcHandle, (CUmemGenericAllocationHandle,)): pmcHandle = int(mcHandle) cmcHandle = pmcHandle else: pmcHandle = int(CUmemGenericAllocationHandle(mcHandle)) cmcHandle = pmcHandle err = ccuda.cuMulticastBindMem(cmcHandle, mcOffset, cmemHandle, memOffset, size, flags) return (CUresult(err),) {{endif}} {{if 'cuMulticastBindAddr' in found_functions}} @cython.embedsignature(True) def cuMulticastBindAddr(mcHandle, size_t mcOffset, memptr, size_t size, unsigned long long flags): """ Bind a memory allocation represented by a virtual address to a multicast object. Binds a memory allocation specified by its mapped address `memptr` to a multicast object represented by `mcHandle`. The memory must have been allocated via :py:obj:`~.cuMemCreate` or :py:obj:`~.cudaMallocAsync`. The intended `size` of the bind, the offset in the multicast range `mcOffset` and `memptr` must be a multiple of the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_GRANULARITY_MINIMUM`. For best performance however, `size`, `mcOffset` and `memptr` should be aligned to the value returned by :py:obj:`~.cuMulticastGetGranularity` with the flag :py:obj:`~.CU_MULTICAST_GRANULARITY_RECOMMENDED`. The `size` cannot be larger than the size of the allocated memory. Similarly the `size` + `mcOffset` cannot be larger than the total size of the multicast object. The memory allocation must have beeen created on one of the devices that was added to the multicast team via :py:obj:`~.cuMulticastAddDevice`. Externally shareable as well as imported multicast objects can be bound only to externally shareable memory. Note that this call will return CUDA_ERROR_OUT_OF_MEMORY if there are insufficient resources required to perform the bind. This call may also return CUDA_ERROR_SYSTEM_NOT_READY if the necessary system software is not initialized or running. Parameters ---------- mcHandle : :py:obj:`~.CUmemGenericAllocationHandle` Handle representing a multicast object. mcOffset : size_t Offset into multicast va range for attachment. memptr : :py:obj:`~.CUdeviceptr` Virtual address of the memory allocation. size : size_t Size of memory that will be bound to the multicast object. flags : unsigned long long Flags for future use, must be zero now. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY`, :py:obj:`~.CUDA_ERROR_SYSTEM_NOT_READY` See Also -------- :py:obj:`~.cuMulticastCreate`, :py:obj:`~.cuMulticastAddDevice`, :py:obj:`~.cuMemCreate` """ cdef ccuda.CUdeviceptr cmemptr if memptr is None: cmemptr = 0 elif isinstance(memptr, (CUdeviceptr,)): pmemptr = int(memptr) cmemptr = pmemptr else: pmemptr = int(CUdeviceptr(memptr)) cmemptr = pmemptr cdef ccuda.CUmemGenericAllocationHandle cmcHandle if mcHandle is None: cmcHandle = 0 elif isinstance(mcHandle, (CUmemGenericAllocationHandle,)): pmcHandle = int(mcHandle) cmcHandle = pmcHandle else: pmcHandle = int(CUmemGenericAllocationHandle(mcHandle)) cmcHandle = pmcHandle err = ccuda.cuMulticastBindAddr(cmcHandle, mcOffset, cmemptr, size, flags) return (CUresult(err),) {{endif}} {{if 'cuMulticastUnbind' in found_functions}} @cython.embedsignature(True) def cuMulticastUnbind(mcHandle, dev, size_t mcOffset, size_t size): """ Unbind any memory allocations bound to a multicast object at a given offset and upto a given size. Unbinds any memory allocations hosted on `dev` and bound to a multicast object at `mcOffset` and upto a given `size`. The intended `size` of the unbind and the offset in the multicast range ( `mcOffset` ) must be a multiple of the value returned by :py:obj:`~.cuMulticastGetGranularity` flag :py:obj:`~.CU_MULTICAST_GRANULARITY_MINIMUM`. The `size` + `mcOffset` cannot be larger than the total size of the multicast object. Parameters ---------- mcHandle : :py:obj:`~.CUmemGenericAllocationHandle` Handle representing a multicast object. dev : :py:obj:`~.CUdevice` Device that hosts the memory allocation. mcOffset : size_t Offset into the multicast object. size : size_t Desired size to unbind. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuMulticastBindMem`, :py:obj:`~.cuMulticastBindAddr` Notes ----- Warning: The `mcOffset` and the `size` must match the corresponding values specified during the bind call. Any other values may result in undefined behavior. """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef ccuda.CUmemGenericAllocationHandle cmcHandle if mcHandle is None: cmcHandle = 0 elif isinstance(mcHandle, (CUmemGenericAllocationHandle,)): pmcHandle = int(mcHandle) cmcHandle = pmcHandle else: pmcHandle = int(CUmemGenericAllocationHandle(mcHandle)) cmcHandle = pmcHandle err = ccuda.cuMulticastUnbind(cmcHandle, cdev, mcOffset, size) return (CUresult(err),) {{endif}} {{if 'cuMulticastGetGranularity' in found_functions}} @cython.embedsignature(True) def cuMulticastGetGranularity(prop : Optional[CUmulticastObjectProp], option not None : CUmulticastGranularity_flags): """ Calculates either the minimal or recommended granularity for multicast object. Calculates either the minimal or recommended granularity for a given set of multicast object properties and returns it in granularity. This granularity can be used as a multiple for size, bind offsets and address mappings of the multicast object. Parameters ---------- prop : :py:obj:`~.CUmulticastObjectProp` Properties of the multicast object. option : :py:obj:`~.CUmulticastGranularity_flags` Determines which granularity to return. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` granularity : int Returned granularity. See Also -------- :py:obj:`~.cuMulticastCreate`, :py:obj:`~.cuMulticastBindMem`, :py:obj:`~.cuMulticastBindAddr`, :py:obj:`~.cuMulticastUnbind` """ cdef size_t granularity = 0 cdef ccuda.CUmulticastObjectProp* cprop_ptr = prop._ptr if prop != None else NULL cdef ccuda.CUmulticastGranularity_flags coption = option.value err = ccuda.cuMulticastGetGranularity(&granularity, cprop_ptr, coption) return (CUresult(err), granularity) {{endif}} {{if 'cuPointerGetAttribute' in found_functions}} @cython.embedsignature(True) def cuPointerGetAttribute(attribute not None : CUpointer_attribute, ptr): """ Returns information about a pointer. The supported attributes are: - :py:obj:`~.CU_POINTER_ATTRIBUTE_CONTEXT`: - Returns in `*data` the :py:obj:`~.CUcontext` in which `ptr` was allocated or registered. The type of `data` must be :py:obj:`~.CUcontext` *. - If `ptr` was not allocated by, mapped by, or registered with a :py:obj:`~.CUcontext` which uses unified virtual addressing then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. - :py:obj:`~.CU_POINTER_ATTRIBUTE_MEMORY_TYPE`: - Returns in `*data` the physical memory type of the memory that `ptr` addresses as a :py:obj:`~.CUmemorytype` enumerated value. The type of `data` must be unsigned int. - If `ptr` addresses device memory then `*data` is set to :py:obj:`~.CU_MEMORYTYPE_DEVICE`. The particular :py:obj:`~.CUdevice` on which the memory resides is the :py:obj:`~.CUdevice` of the :py:obj:`~.CUcontext` returned by the :py:obj:`~.CU_POINTER_ATTRIBUTE_CONTEXT` attribute of `ptr`. - If `ptr` addresses host memory then `*data` is set to :py:obj:`~.CU_MEMORYTYPE_HOST`. - If `ptr` was not allocated by, mapped by, or registered with a :py:obj:`~.CUcontext` which uses unified virtual addressing then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. - If the current :py:obj:`~.CUcontext` does not support unified virtual addressing then :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` is returned. - :py:obj:`~.CU_POINTER_ATTRIBUTE_DEVICE_POINTER`: - Returns in `*data` the device pointer value through which `ptr` may be accessed by kernels running in the current :py:obj:`~.CUcontext`. The type of `data` must be CUdeviceptr *. - If there exists no device pointer value through which kernels running in the current :py:obj:`~.CUcontext` may access `ptr` then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. - If there is no current :py:obj:`~.CUcontext` then :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` is returned. - Except in the exceptional disjoint addressing cases discussed below, the value returned in `*data` will equal the input value `ptr`. - :py:obj:`~.CU_POINTER_ATTRIBUTE_HOST_POINTER`: - Returns in `*data` the host pointer value through which `ptr` may be accessed by by the host program. The type of `data` must be void **. If there exists no host pointer value through which the host program may directly access `ptr` then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. - Except in the exceptional disjoint addressing cases discussed below, the value returned in `*data` will equal the input value `ptr`. - :py:obj:`~.CU_POINTER_ATTRIBUTE_P2P_TOKENS`: - Returns in `*data` two tokens for use with the nv-p2p.h Linux kernel interface. `data` must be a struct of type CUDA_POINTER_ATTRIBUTE_P2P_TOKENS. - `ptr` must be a pointer to memory obtained from :py:obj:`~.py`:obj:`~.cuMemAlloc()`. Note that p2pToken and vaSpaceToken are only valid for the lifetime of the source allocation. A subsequent allocation at the same address may return completely different tokens. Querying this attribute has a side effect of setting the attribute :py:obj:`~.CU_POINTER_ATTRIBUTE_SYNC_MEMOPS` for the region of memory that `ptr` points to. - :py:obj:`~.CU_POINTER_ATTRIBUTE_SYNC_MEMOPS`: - A boolean attribute which when set, ensures that synchronous memory operations initiated on the region of memory that `ptr` points to will always synchronize. See further documentation in the section titled "API synchronization behavior" to learn more about cases when synchronous memory operations can exhibit asynchronous behavior. - :py:obj:`~.CU_POINTER_ATTRIBUTE_BUFFER_ID`: - Returns in `*data` a buffer ID which is guaranteed to be unique within the process. `data` must point to an unsigned long long. - `ptr` must be a pointer to memory obtained from a CUDA memory allocation API. Every memory allocation from any of the CUDA memory allocation APIs will have a unique ID over a process lifetime. Subsequent allocations do not reuse IDs from previous freed allocations. IDs are only unique within a single process. - :py:obj:`~.CU_POINTER_ATTRIBUTE_IS_MANAGED`: - Returns in `*data` a boolean that indicates whether the pointer points to managed memory or not. - If `ptr` is not a valid CUDA pointer then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. - :py:obj:`~.CU_POINTER_ATTRIBUTE_DEVICE_ORDINAL`: - Returns in `*data` an integer representing a device ordinal of a device against which the memory was allocated or registered. - :py:obj:`~.CU_POINTER_ATTRIBUTE_IS_LEGACY_CUDA_IPC_CAPABLE`: - Returns in `*data` a boolean that indicates if this pointer maps to an allocation that is suitable for :py:obj:`~.cudaIpcGetMemHandle`. - :py:obj:`~.CU_POINTER_ATTRIBUTE_RANGE_START_ADDR`: - Returns in `*data` the starting address for the allocation referenced by the device pointer `ptr`. Note that this is not necessarily the address of the mapped region, but the address of the mappable address range `ptr` references (e.g. from :py:obj:`~.cuMemAddressReserve`). - :py:obj:`~.CU_POINTER_ATTRIBUTE_RANGE_SIZE`: - Returns in `*data` the size for the allocation referenced by the device pointer `ptr`. Note that this is not necessarily the size of the mapped region, but the size of the mappable address range `ptr` references (e.g. from :py:obj:`~.cuMemAddressReserve`). To retrieve the size of the mapped region, see :py:obj:`~.cuMemGetAddressRange` - :py:obj:`~.CU_POINTER_ATTRIBUTE_MAPPED`: - Returns in `*data` a boolean that indicates if this pointer is in a valid address range that is mapped to a backing allocation. - :py:obj:`~.CU_POINTER_ATTRIBUTE_ALLOWED_HANDLE_TYPES`: - Returns a bitmask of the allowed handle types for an allocation that may be passed to :py:obj:`~.cuMemExportToShareableHandle`. - :py:obj:`~.CU_POINTER_ATTRIBUTE_MEMPOOL_HANDLE`: - Returns in `*data` the handle to the mempool that the allocation was obtained from. Note that for most allocations in the unified virtual address space the host and device pointer for accessing the allocation will be the same. The exceptions to this are - user memory registered using :py:obj:`~.cuMemHostRegister` - host memory allocated using :py:obj:`~.cuMemHostAlloc` with the :py:obj:`~.CU_MEMHOSTALLOC_WRITECOMBINED` flag For these types of allocation there will exist separate, disjoint host and device addresses for accessing the allocation. In particular - The host address will correspond to an invalid unmapped device address (which will result in an exception if accessed from the device) - The device address will correspond to an invalid unmapped host address (which will result in an exception if accessed from the host). For these types of allocations, querying :py:obj:`~.CU_POINTER_ATTRIBUTE_HOST_POINTER` and :py:obj:`~.CU_POINTER_ATTRIBUTE_DEVICE_POINTER` may be used to retrieve the host and device addresses from either address. Parameters ---------- attribute : :py:obj:`~.CUpointer_attribute` Pointer attribute to query ptr : :py:obj:`~.CUdeviceptr` Pointer Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` data : Any Returned pointer attribute value See Also -------- :py:obj:`~.cuPointerSetAttribute`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostRegister`, :py:obj:`~.cuMemHostUnregister`, :py:obj:`~.cudaPointerGetAttributes` """ cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr cdef utils.HelperCUpointer_attribute cdata = utils.HelperCUpointer_attribute(attribute, 0, is_getter=True) cdef void* cdata_ptr = cdata.cptr cdef ccuda.CUpointer_attribute cattribute = attribute.value err = ccuda.cuPointerGetAttribute(cdata_ptr, cattribute, cptr) return (CUresult(err), cdata.pyObj()) {{endif}} {{if 'cuMemPrefetchAsync' in found_functions}} @cython.embedsignature(True) def cuMemPrefetchAsync(devPtr, size_t count, dstDevice, hStream): """ Prefetches memory to the specified destination device. Note there is a later version of this API, :py:obj:`~.cuMemPrefetchAsync_v2`. It will supplant this version in 13.0, which is retained for minor version compatibility. Prefetches memory to the specified destination device. `devPtr` is the base device pointer of the memory to be prefetched and `dstDevice` is the destination device. `count` specifies the number of bytes to copy. `hStream` is the stream in which the operation is enqueued. The memory range must refer to managed memory allocated via :py:obj:`~.cuMemAllocManaged` or declared via managed variables or it may also refer to system-allocated memory on systems with non-zero CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS. Passing in CU_DEVICE_CPU for `dstDevice` will prefetch the data to host memory. If `dstDevice` is a GPU, then the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS` must be non- zero. Additionally, `hStream` must be associated with a device that has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. The start address and end address of the memory range will be rounded down and rounded up respectively to be aligned to CPU page size before the prefetch operation is enqueued in the stream. If no physical memory has been allocated for this region, then this memory region will be populated and mapped on the destination device. If there's insufficient memory to prefetch the desired region, the Unified Memory driver may evict pages from other :py:obj:`~.cuMemAllocManaged` allocations to host memory in order to make room. Device memory allocated using :py:obj:`~.cuMemAlloc` or :py:obj:`~.cuArrayCreate` will not be evicted. By default, any mappings to the previous location of the migrated pages are removed and mappings for the new location are only setup on `dstDevice`. The exact behavior however also depends on the settings applied to this memory range via :py:obj:`~.cuMemAdvise` as described below: If :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` was set on any subset of this memory range, then that subset will create a read-only copy of the pages on `dstDevice`. If :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION` was called on any subset of this memory range, then the pages will be migrated to `dstDevice` even if `dstDevice` is not the preferred location of any pages in the memory range. If :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY` was called on any subset of this memory range, then mappings to those pages from all the appropriate processors are updated to refer to the new location if establishing such a mapping is possible. Otherwise, those mappings are cleared. Note that this API is not required for functionality and only serves to improve performance by allowing the application to migrate data to a suitable location before it is accessed. Memory accesses to this range are always coherent and are allowed even when the data is actively being migrated. Note that this function is asynchronous with respect to the host and all work on other devices. Parameters ---------- devPtr : :py:obj:`~.CUdeviceptr` Pointer to be prefetched count : size_t Size in bytes dstDevice : :py:obj:`~.CUdevice` Destination device to prefetch to hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to enqueue prefetch operation Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` See Also -------- :py:obj:`~.cuMemcpy`, :py:obj:`~.cuMemcpyPeer`, :py:obj:`~.cuMemcpyAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cuMemAdvise`, :py:obj:`~.cuMemPrefetchAsync` :py:obj:`~.cudaMemPrefetchAsync_v2` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdevice cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdevice,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdevice(dstDevice)) cdstDevice = pdstDevice cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr,)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr err = ccuda.cuMemPrefetchAsync(cdevPtr, count, cdstDevice, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemPrefetchAsync_v2' in found_functions}} @cython.embedsignature(True) def cuMemPrefetchAsync_v2(devPtr, size_t count, location not None : CUmemLocation, unsigned int flags, hStream): """ Prefetches memory to the specified destination location. Prefetches memory to the specified destination location. `devPtr` is the base device pointer of the memory to be prefetched and `location` specifies the destination location. `count` specifies the number of bytes to copy. `hStream` is the stream in which the operation is enqueued. The memory range must refer to managed memory allocated via :py:obj:`~.cuMemAllocManaged` or declared via managed variables. Specifying :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE` for :py:obj:`~.CUmemLocation.type` will prefetch memory to GPU specified by device ordinal :py:obj:`~.CUmemLocation.id` which must have non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. Additionally, `hStream` must be associated with a device that has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. Specifying :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST` as :py:obj:`~.CUmemLocation.type` will prefetch data to host memory. Applications can request prefetching memory to a specific host NUMA node by specifying :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA` for :py:obj:`~.CUmemLocation.type` and a valid host NUMA node id in :py:obj:`~.CUmemLocation.id` Users can also request prefetching memory to the host NUMA node closest to the current thread's CPU by specifying :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT` for :py:obj:`~.CUmemLocation.type`. Note when :py:obj:`~.CUmemLocation.type` is etiher :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST` OR :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT`, :py:obj:`~.CUmemLocation.id` will be ignored. The start address and end address of the memory range will be rounded down and rounded up respectively to be aligned to CPU page size before the prefetch operation is enqueued in the stream. If no physical memory has been allocated for this region, then this memory region will be populated and mapped on the destination device. If there's insufficient memory to prefetch the desired region, the Unified Memory driver may evict pages from other :py:obj:`~.cuMemAllocManaged` allocations to host memory in order to make room. Device memory allocated using :py:obj:`~.cuMemAlloc` or :py:obj:`~.cuArrayCreate` will not be evicted. By default, any mappings to the previous location of the migrated pages are removed and mappings for the new location are only setup on the destination location. The exact behavior however also depends on the settings applied to this memory range via :py:obj:`~.cuMemAdvise` as described below: If :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` was set on any subset of this memory range, then that subset will create a read-only copy of the pages on destination location. If however the destination location is a host NUMA node, then any pages of that subset that are already in another host NUMA node will be transferred to the destination. If :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION` was called on any subset of this memory range, then the pages will be migrated to `location` even if `location` is not the preferred location of any pages in the memory range. If :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY` was called on any subset of this memory range, then mappings to those pages from all the appropriate processors are updated to refer to the new location if establishing such a mapping is possible. Otherwise, those mappings are cleared. Note that this API is not required for functionality and only serves to improve performance by allowing the application to migrate data to a suitable location before it is accessed. Memory accesses to this range are always coherent and are allowed even when the data is actively being migrated. Note that this function is asynchronous with respect to the host and all work on other devices. Parameters ---------- devPtr : :py:obj:`~.CUdeviceptr` Pointer to be prefetched count : size_t Size in bytes dstDevice : :py:obj:`~.CUmemLocation` Destination device to prefetch to flags : unsigned int flags for future use, must be zero now. hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to enqueue prefetch operation Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` See Also -------- :py:obj:`~.cuMemcpy`, :py:obj:`~.cuMemcpyPeer`, :py:obj:`~.cuMemcpyAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cuMemAdvise`, :py:obj:`~.cuMemPrefetchAsync` :py:obj:`~.cudaMemPrefetchAsync_v2` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr,)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr err = ccuda.cuMemPrefetchAsync_v2(cdevPtr, count, location._ptr[0], flags, chStream) return (CUresult(err),) {{endif}} {{if 'cuMemAdvise' in found_functions}} @cython.embedsignature(True) def cuMemAdvise(devPtr, size_t count, advice not None : CUmem_advise, device): """ Advise about the usage of a given memory range. Note there is a later version of this API, :py:obj:`~.cuMemAdvise_v2`. It will supplant this version in 13.0, which is retained for minor version compatibility. Advise the Unified Memory subsystem about the usage pattern for the memory range starting at `devPtr` with a size of `count` bytes. The start address and end address of the memory range will be rounded down and rounded up respectively to be aligned to CPU page size before the advice is applied. The memory range must refer to managed memory allocated via :py:obj:`~.cuMemAllocManaged` or declared via managed variables. The memory range could also refer to system-allocated pageable memory provided it represents a valid, host-accessible region of memory and all additional constraints imposed by `advice` as outlined below are also satisfied. Specifying an invalid system- allocated pageable memory range results in an error being returned. The `advice` parameter can take the following values: - :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY`: This implies that the data is mostly going to be read from and only occasionally written to. Any read accesses from any processor to this region will create a read- only copy of at least the accessed pages in that processor's memory. Additionally, if :py:obj:`~.cuMemPrefetchAsync` is called on this region, it will create a read-only copy of the data on the destination processor. If any processor writes to this region, all copies of the corresponding page will be invalidated except for the one where the write occurred. The `device` argument is ignored for this advice. Note that for a page to be read-duplicated, the accessing processor must either be the CPU or a GPU that has a non- zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. Also, if a context is created on a device that does not have the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS` set, then read-duplication will not occur until all such contexts are destroyed. If the memory region refers to valid system-allocated pageable memory, then the accessing device must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS` for a read- only copy to be created on that device. Note however that if the accessing device also has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, then setting this advice will not create a read-only copy when that device accesses this memory region. - :py:obj:`~.CU_MEM_ADVISE_UNSET_READ_MOSTLY`: Undoes the effect of :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` and also prevents the Unified Memory driver from attempting heuristic read-duplication on the memory range. Any read-duplicated copies of the data will be collapsed into a single copy. The location for the collapsed copy will be the preferred location if the page has a preferred location and one of the read-duplicated copies was resident at that location. Otherwise, the location chosen is arbitrary. - :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION`: This advice sets the preferred location for the data to be the memory belonging to `device`. Passing in CU_DEVICE_CPU for `device` sets the preferred location as host memory. If `device` is a GPU, then it must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. Setting the preferred location does not cause data to migrate to that location immediately. Instead, it guides the migration policy when a fault occurs on that memory region. If the data is already in its preferred location and the faulting processor can establish a mapping without requiring the data to be migrated, then data migration will be avoided. On the other hand, if the data is not in its preferred location or if a direct mapping cannot be established, then it will be migrated to the processor accessing it. It is important to note that setting the preferred location does not prevent data prefetching done using :py:obj:`~.cuMemPrefetchAsync`. Having a preferred location can override the page thrash detection and resolution logic in the Unified Memory driver. Normally, if a page is detected to be constantly thrashing between for example host and device memory, the page may eventually be pinned to host memory by the Unified Memory driver. But if the preferred location is set as device memory, then the page will continue to thrash indefinitely. If :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` is also set on this memory region or any subset of it, then the policies associated with that advice will override the policies of this advice, unless read accesses from `device` will not result in a read-only copy being created on that device as outlined in description for the advice :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY`. If the memory region refers to valid system-allocated pageable memory, then `device` must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`. - :py:obj:`~.CU_MEM_ADVISE_UNSET_PREFERRED_LOCATION`: Undoes the effect of :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION` and changes the preferred location to none. - :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY`: This advice implies that the data will be accessed by `device`. Passing in :py:obj:`~.CU_DEVICE_CPU` for `device` will set the advice for the CPU. If `device` is a GPU, then the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS` must be non-zero. This advice does not cause data migration and has no impact on the location of the data per se. Instead, it causes the data to always be mapped in the specified processor's page tables, as long as the location of the data permits a mapping to be established. If the data gets migrated for any reason, the mappings are updated accordingly. This advice is recommended in scenarios where data locality is not important, but avoiding faults is. Consider for example a system containing multiple GPUs with peer-to-peer access enabled, where the data located on one GPU is occasionally accessed by peer GPUs. In such scenarios, migrating data over to the other GPUs is not as important because the accesses are infrequent and the overhead of migration may be too high. But preventing faults can still help improve performance, and so having a mapping set up in advance is useful. Note that on CPU access of this data, the data may be migrated to host memory because the CPU typically cannot access device memory directly. Any GPU that had the :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY` flag set for this data will now have its mapping updated to point to the page in host memory. If :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` is also set on this memory region or any subset of it, then the policies associated with that advice will override the policies of this advice. Additionally, if the preferred location of this memory region or any subset of it is also `device`, then the policies associated with :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION` will override the policies of this advice. If the memory region refers to valid system- allocated pageable memory, then `device` must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`. Additionally, if `device` has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, then this call has no effect. - :py:obj:`~.CU_MEM_ADVISE_UNSET_ACCESSED_BY`: Undoes the effect of :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY`. Any mappings to the data from `device` may be removed at any time causing accesses to result in non-fatal page faults. If the memory region refers to valid system-allocated pageable memory, then `device` must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`. Additionally, if `device` has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, then this call has no effect. Parameters ---------- devPtr : :py:obj:`~.CUdeviceptr` Pointer to memory to set the advice for count : size_t Size in bytes of the memory range advice : :py:obj:`~.CUmem_advise` Advice to be applied for the specified memory range device : :py:obj:`~.CUdevice` Device to apply the advice for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` See Also -------- :py:obj:`~.cuMemcpy`, :py:obj:`~.cuMemcpyPeer`, :py:obj:`~.cuMemcpyAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cuMemPrefetchAsync`, :py:obj:`~.cuMemAdvise_v2` :py:obj:`~.cudaMemAdvise` """ cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr,)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr cdef ccuda.CUmem_advise cadvice = advice.value err = ccuda.cuMemAdvise(cdevPtr, count, cadvice, cdevice) return (CUresult(err),) {{endif}} {{if 'cuMemAdvise_v2' in found_functions}} @cython.embedsignature(True) def cuMemAdvise_v2(devPtr, size_t count, advice not None : CUmem_advise, location not None : CUmemLocation): """ Advise about the usage of a given memory range. Advise the Unified Memory subsystem about the usage pattern for the memory range starting at `devPtr` with a size of `count` bytes. The start address and end address of the memory range will be rounded down and rounded up respectively to be aligned to CPU page size before the advice is applied. The memory range must refer to managed memory allocated via :py:obj:`~.cuMemAllocManaged` or declared via managed variables. The memory range could also refer to system-allocated pageable memory provided it represents a valid, host-accessible region of memory and all additional constraints imposed by `advice` as outlined below are also satisfied. Specifying an invalid system- allocated pageable memory range results in an error being returned. The `advice` parameter can take the following values: - :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY`: This implies that the data is mostly going to be read from and only occasionally written to. Any read accesses from any processor to this region will create a read- only copy of at least the accessed pages in that processor's memory. Additionally, if :py:obj:`~.cuMemPrefetchAsync` or :py:obj:`~.cuMemPrefetchAsync_v2` is called on this region, it will create a read-only copy of the data on the destination processor. If the target location for :py:obj:`~.cuMemPrefetchAsync_v2` is a host NUMA node and a read-only copy already exists on another host NUMA node, that copy will be migrated to the targeted host NUMA node. If any processor writes to this region, all copies of the corresponding page will be invalidated except for the one where the write occurred. If the writing processor is the CPU and the preferred location of the page is a host NUMA node, then the page will also be migrated to that host NUMA node. The `location` argument is ignored for this advice. Note that for a page to be read-duplicated, the accessing processor must either be the CPU or a GPU that has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. Also, if a context is created on a device that does not have the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS` set, then read-duplication will not occur until all such contexts are destroyed. If the memory region refers to valid system-allocated pageable memory, then the accessing device must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS` for a read- only copy to be created on that device. Note however that if the accessing device also has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, then setting this advice will not create a read-only copy when that device accesses this memory region. - :py:obj:`~.CU_MEM_ADVISE_UNSET_READ_MOSTLY`: Undoes the effect of :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` and also prevents the Unified Memory driver from attempting heuristic read-duplication on the memory range. Any read-duplicated copies of the data will be collapsed into a single copy. The location for the collapsed copy will be the preferred location if the page has a preferred location and one of the read-duplicated copies was resident at that location. Otherwise, the location chosen is arbitrary. Note: The `location` argument is ignored for this advice. - :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION`: This advice sets the preferred location for the data to be the memory belonging to `location`. When :py:obj:`~.CUmemLocation.type` is :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST`, :py:obj:`~.CUmemLocation.id` is ignored and the preferred location is set to be host memory. To set the preferred location to a specific host NUMA node, applications must set :py:obj:`~.CUmemLocation.type` to :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA` and :py:obj:`~.CUmemLocation.id` must specify the NUMA ID of the host NUMA node. If :py:obj:`~.CUmemLocation.type` is set to :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA_CURRENT`, :py:obj:`~.CUmemLocation.id` will be ignored and the the host NUMA node closest to the calling thread's CPU will be used as the preferred location. If :py:obj:`~.CUmemLocation.type` is a :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE`, then :py:obj:`~.CUmemLocation.id` must be a valid device ordinal and the device must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. Setting the preferred location does not cause data to migrate to that location immediately. Instead, it guides the migration policy when a fault occurs on that memory region. If the data is already in its preferred location and the faulting processor can establish a mapping without requiring the data to be migrated, then data migration will be avoided. On the other hand, if the data is not in its preferred location or if a direct mapping cannot be established, then it will be migrated to the processor accessing it. It is important to note that setting the preferred location does not prevent data prefetching done using :py:obj:`~.cuMemPrefetchAsync`. Having a preferred location can override the page thrash detection and resolution logic in the Unified Memory driver. Normally, if a page is detected to be constantly thrashing between for example host and device memory, the page may eventually be pinned to host memory by the Unified Memory driver. But if the preferred location is set as device memory, then the page will continue to thrash indefinitely. If :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` is also set on this memory region or any subset of it, then the policies associated with that advice will override the policies of this advice, unless read accesses from `location` will not result in a read-only copy being created on that procesor as outlined in description for the advice :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY`. If the memory region refers to valid system-allocated pageable memory, and :py:obj:`~.CUmemLocation.type` is CU_MEM_LOCATION_TYPE_DEVICE then :py:obj:`~.CUmemLocation.id` must be a valid device that has a non- zero alue for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`. - :py:obj:`~.CU_MEM_ADVISE_UNSET_PREFERRED_LOCATION`: Undoes the effect of :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION` and changes the preferred location to none. The `location` argument is ignored for this advice. - :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY`: This advice implies that the data will be accessed by processor `location`. The :py:obj:`~.CUmemLocation.type` must be either :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE` with :py:obj:`~.CUmemLocation.id` representing a valid device ordinal or :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST` and :py:obj:`~.CUmemLocation.id` will be ignored. All other location types are invalid. If :py:obj:`~.CUmemLocation.id` is a GPU, then the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS` must be non-zero. This advice does not cause data migration and has no impact on the location of the data per se. Instead, it causes the data to always be mapped in the specified processor's page tables, as long as the location of the data permits a mapping to be established. If the data gets migrated for any reason, the mappings are updated accordingly. This advice is recommended in scenarios where data locality is not important, but avoiding faults is. Consider for example a system containing multiple GPUs with peer-to-peer access enabled, where the data located on one GPU is occasionally accessed by peer GPUs. In such scenarios, migrating data over to the other GPUs is not as important because the accesses are infrequent and the overhead of migration may be too high. But preventing faults can still help improve performance, and so having a mapping set up in advance is useful. Note that on CPU access of this data, the data may be migrated to host memory because the CPU typically cannot access device memory directly. Any GPU that had the :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY` flag set for this data will now have its mapping updated to point to the page in host memory. If :py:obj:`~.CU_MEM_ADVISE_SET_READ_MOSTLY` is also set on this memory region or any subset of it, then the policies associated with that advice will override the policies of this advice. Additionally, if the preferred location of this memory region or any subset of it is also `location`, then the policies associated with :py:obj:`~.CU_MEM_ADVISE_SET_PREFERRED_LOCATION` will override the policies of this advice. If the memory region refers to valid system- allocated pageable memory, and :py:obj:`~.CUmemLocation.type` is :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE` then device in :py:obj:`~.CUmemLocation.id` must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`. Additionally, if :py:obj:`~.CUmemLocation.id` has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, then this call has no effect. - :py:obj:`~.CU_MEM_ADVISE_UNSET_ACCESSED_BY`: Undoes the effect of :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY`. Any mappings to the data from `location` may be removed at any time causing accesses to result in non-fatal page faults. If the memory region refers to valid system-allocated pageable memory, and :py:obj:`~.CUmemLocation.type` is :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE` then device in :py:obj:`~.CUmemLocation.id` must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`. Additionally, if :py:obj:`~.CUmemLocation.id` has a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS_USES_HOST_PAGE_TABLES`, then this call has no effect. Parameters ---------- devPtr : :py:obj:`~.CUdeviceptr` Pointer to memory to set the advice for count : size_t Size in bytes of the memory range advice : :py:obj:`~.CUmem_advise` Advice to be applied for the specified memory range location : :py:obj:`~.CUmemLocation` location to apply the advice for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` See Also -------- :py:obj:`~.cuMemcpy`, :py:obj:`~.cuMemcpyPeer`, :py:obj:`~.cuMemcpyAsync`, :py:obj:`~.cuMemcpy3DPeerAsync`, :py:obj:`~.cuMemPrefetchAsync`, :py:obj:`~.cuMemAdvise` :py:obj:`~.cudaMemAdvise` """ cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr,)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr cdef ccuda.CUmem_advise cadvice = advice.value err = ccuda.cuMemAdvise_v2(cdevPtr, count, cadvice, location._ptr[0]) return (CUresult(err),) {{endif}} {{if 'cuMemRangeGetAttribute' in found_functions}} @cython.embedsignature(True) def cuMemRangeGetAttribute(size_t dataSize, attribute not None : CUmem_range_attribute, devPtr, size_t count): """ Query an attribute of a given memory range. Query an attribute about the memory range starting at `devPtr` with a size of `count` bytes. The memory range must refer to managed memory allocated via :py:obj:`~.cuMemAllocManaged` or declared via managed variables. The `attribute` parameter can take the following values: - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_READ_MOSTLY`: If this attribute is specified, `data` will be interpreted as a 32-bit integer, and `dataSize` must be 4. The result returned will be 1 if all pages in the given memory range have read-duplication enabled, or 0 otherwise. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION`: If this attribute is specified, `data` will be interpreted as a 32-bit integer, and `dataSize` must be 4. The result returned will be a GPU device id if all pages in the memory range have that GPU as their preferred location, or it will be CU_DEVICE_CPU if all pages in the memory range have the CPU as their preferred location, or it will be CU_DEVICE_INVALID if either all the pages don't have the same preferred location or some of the pages don't have a preferred location at all. Note that the actual location of the pages in the memory range at the time of the query may be different from the preferred location. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_ACCESSED_BY`: If this attribute is specified, `data` will be interpreted as an array of 32-bit integers, and `dataSize` must be a non-zero multiple of 4. The result returned will be a list of device ids that had :py:obj:`~.CU_MEM_ADVISE_SET_ACCESSED_BY` set for that entire memory range. If any device does not have that advice set for the entire memory range, that device will not be included. If `data` is larger than the number of devices that have that advice set for that memory range, CU_DEVICE_INVALID will be returned in all the extra space provided. For ex., if `dataSize` is 12 (i.e. `data` has 3 elements) and only device 0 has the advice set, then the result returned will be { 0, CU_DEVICE_INVALID, CU_DEVICE_INVALID }. If `data` is smaller than the number of devices that have that advice set, then only as many devices will be returned as can fit in the array. There is no guarantee on which specific devices will be returned, however. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION`: If this attribute is specified, `data` will be interpreted as a 32-bit integer, and `dataSize` must be 4. The result returned will be the last location to which all pages in the memory range were prefetched explicitly via :py:obj:`~.cuMemPrefetchAsync`. This will either be a GPU id or CU_DEVICE_CPU depending on whether the last location for prefetch was a GPU or the CPU respectively. If any page in the memory range was never explicitly prefetched or if all pages were not prefetched to the same location, CU_DEVICE_INVALID will be returned. Note that this simply returns the last location that the application requested to prefetch the memory range to. It gives no indication as to whether the prefetch operation to that location has completed or even begun. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_TYPE`: If this attribute is specified, `data` will be interpreted as a :py:obj:`~.CUmemLocationType`, and `dataSize` must be sizeof(CUmemLocationType). The :py:obj:`~.CUmemLocationType` returned will be :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE` if all pages in the memory range have the same GPU as their preferred location, or :py:obj:`~.CUmemLocationType` will be :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST` if all pages in the memory range have the CPU as their preferred location, or it will be :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA` if all the pages in the memory range have the same host NUMA node ID as their preferred location or it will be :py:obj:`~.CU_MEM_LOCATION_TYPE_INVALID` if either all the pages don't have the same preferred location or some of the pages don't have a preferred location at all. Note that the actual location type of the pages in the memory range at the time of the query may be different from the preferred location type. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_ID`: If this attribute is specified, `data` will be interpreted as a 32-bit integer, and `dataSize` must be 4. If the :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_TYPE` query for the same address range returns :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE`, it will be a valid device ordinal or if it returns :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA`, it will be a valid host NUMA node ID or if it returns any other location type, the id should be ignored. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_TYPE`: If this attribute is specified, `data` will be interpreted as a :py:obj:`~.CUmemLocationType`, and `dataSize` must be sizeof(CUmemLocationType). The result returned will be the last location to which all pages in the memory range were prefetched explicitly via :py:obj:`~.cuMemPrefetchAsync`. The :py:obj:`~.CUmemLocationType` returned will be :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE` if the last prefetch location was a GPU or :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST` if it was the CPU or :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA` if the last prefetch location was a specific host NUMA node. If any page in the memory range was never explicitly prefetched or if all pages were not prefetched to the same location, :py:obj:`~.CUmemLocationType` will be :py:obj:`~.CU_MEM_LOCATION_TYPE_INVALID`. Note that this simply returns the last location type that the application requested to prefetch the memory range to. It gives no indication as to whether the prefetch operation to that location has completed or even begun. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_ID`: If this attribute is specified, `data` will be interpreted as a 32-bit integer, and `dataSize` must be 4. If the :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_TYPE` query for the same address range returns :py:obj:`~.CU_MEM_LOCATION_TYPE_DEVICE`, it will be a valid device ordinal or if it returns :py:obj:`~.CU_MEM_LOCATION_TYPE_HOST_NUMA`, it will be a valid host NUMA node ID or if it returns any other location type, the id should be ignored. Parameters ---------- dataSize : size_t Array containing the size of data attribute : :py:obj:`~.CUmem_range_attribute` The attribute to query devPtr : :py:obj:`~.CUdeviceptr` Start of the range to query count : size_t Size of the range to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` data : Any A pointers to a memory location where the result of each attribute query will be written to. See Also -------- :py:obj:`~.cuMemRangeGetAttributes`, :py:obj:`~.cuMemPrefetchAsync`, :py:obj:`~.cuMemAdvise`, :py:obj:`~.cudaMemRangeGetAttribute` """ cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr,)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr cdef utils.HelperCUmem_range_attribute cdata = utils.HelperCUmem_range_attribute(attribute, dataSize) cdef void* cdata_ptr = cdata.cptr cdef ccuda.CUmem_range_attribute cattribute = attribute.value err = ccuda.cuMemRangeGetAttribute(cdata_ptr, dataSize, cattribute, cdevPtr, count) return (CUresult(err), cdata.pyObj()) {{endif}} {{if 'cuMemRangeGetAttributes' in found_functions}} @cython.embedsignature(True) def cuMemRangeGetAttributes(dataSizes : Tuple[int] | List[int], attributes : Optional[Tuple[CUmem_range_attribute] | List[CUmem_range_attribute]], size_t numAttributes, devPtr, size_t count): """ Query attributes of a given memory range. Query attributes of the memory range starting at `devPtr` with a size of `count` bytes. The memory range must refer to managed memory allocated via :py:obj:`~.cuMemAllocManaged` or declared via managed variables. The `attributes` array will be interpreted to have `numAttributes` entries. The `dataSizes` array will also be interpreted to have `numAttributes` entries. The results of the query will be stored in `data`. The list of supported attributes are given below. Please refer to :py:obj:`~.cuMemRangeGetAttribute` for attribute descriptions and restrictions. - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_READ_MOSTLY` - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION` - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_ACCESSED_BY` - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION` - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_TYPE` - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_PREFERRED_LOCATION_ID` - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_TYPE` - :py:obj:`~.CU_MEM_RANGE_ATTRIBUTE_LAST_PREFETCH_LOCATION_ID` Parameters ---------- dataSizes : List[int] Array containing the sizes of each result attributes : List[:py:obj:`~.CUmem_range_attribute`] An array of attributes to query (numAttributes and the number of attributes in this array should match) numAttributes : size_t Number of attributes to query devPtr : :py:obj:`~.CUdeviceptr` Start of the range to query count : size_t Size of the range to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` data : List[Any] A two-dimensional array containing pointers to memory locations where the result of each attribute query will be written to. See Also -------- :py:obj:`~.cuMemRangeGetAttribute`, :py:obj:`~.cuMemAdvise`, :py:obj:`~.cuMemPrefetchAsync`, :py:obj:`~.cudaMemRangeGetAttributes` """ cdef ccuda.CUdeviceptr cdevPtr if devPtr is None: cdevPtr = 0 elif isinstance(devPtr, (CUdeviceptr,)): pdevPtr = int(devPtr) cdevPtr = pdevPtr else: pdevPtr = int(CUdeviceptr(devPtr)) cdevPtr = pdevPtr attributes = [] if attributes is None else attributes if not all(isinstance(_x, (CUmem_range_attribute)) for _x in attributes): raise TypeError("Argument 'attributes' is not instance of type (expected Tuple[ccuda.CUmem_range_attribute] or List[ccuda.CUmem_range_attribute]") if not all(isinstance(_x, (int)) for _x in dataSizes): raise TypeError("Argument 'dataSizes' is not instance of type (expected Tuple[int] or List[int]") pylist = [utils.HelperCUmem_range_attribute(pyattributes, pydataSizes) for (pyattributes, pydataSizes) in zip(attributes, dataSizes)] cdef utils.InputVoidPtrPtrHelper voidStarHelperdata = utils.InputVoidPtrPtrHelper(pylist) cdef void** cvoidStarHelper_ptr = voidStarHelperdata.cptr cdef vector[size_t] cdataSizes = dataSizes cdef vector[ccuda.CUmem_range_attribute] cattributes = [pyattributes.value for pyattributes in (attributes)] if numAttributes > len(dataSizes): raise RuntimeError("List is too small: " + str(len(dataSizes)) + " < " + str(numAttributes)) if numAttributes > len(attributes): raise RuntimeError("List is too small: " + str(len(attributes)) + " < " + str(numAttributes)) err = ccuda.cuMemRangeGetAttributes(cvoidStarHelper_ptr, cdataSizes.data(), cattributes.data(), numAttributes, cdevPtr, count) return (CUresult(err), [obj.pyObj() for obj in pylist]) {{endif}} {{if 'cuPointerSetAttribute' in found_functions}} @cython.embedsignature(True) def cuPointerSetAttribute(value, attribute not None : CUpointer_attribute, ptr): """ Set attributes on a previously allocated memory region. The supported attributes are: - :py:obj:`~.CU_POINTER_ATTRIBUTE_SYNC_MEMOPS`: - A boolean attribute that can either be set (1) or unset (0). When set, the region of memory that `ptr` points to is guaranteed to always synchronize memory operations that are synchronous. If there are some previously initiated synchronous memory operations that are pending when this attribute is set, the function does not return until those memory operations are complete. See further documentation in the section titled "API synchronization behavior" to learn more about cases when synchronous memory operations can exhibit asynchronous behavior. `value` will be considered as a pointer to an unsigned integer to which this attribute is to be set. Parameters ---------- value : Any Pointer to memory containing the value to be set attribute : :py:obj:`~.CUpointer_attribute` Pointer attribute to set ptr : :py:obj:`~.CUdeviceptr` Pointer to a memory region allocated using CUDA memory allocation APIs Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` See Also -------- :py:obj:`~.cuPointerGetAttribute`, :py:obj:`~.cuPointerGetAttributes`, :py:obj:`~.cuMemAlloc`, :py:obj:`~.cuMemFree`, :py:obj:`~.cuMemAllocHost`, :py:obj:`~.cuMemFreeHost`, :py:obj:`~.cuMemHostAlloc`, :py:obj:`~.cuMemHostRegister`, :py:obj:`~.cuMemHostUnregister` """ cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr cdef utils.HelperCUpointer_attribute cvalue = utils.HelperCUpointer_attribute(attribute, value, is_getter=False) cdef void* cvalue_ptr = cvalue.cptr cdef ccuda.CUpointer_attribute cattribute = attribute.value err = ccuda.cuPointerSetAttribute(cvalue_ptr, cattribute, cptr) return (CUresult(err),) {{endif}} {{if 'cuPointerGetAttributes' in found_functions}} @cython.embedsignature(True) def cuPointerGetAttributes(unsigned int numAttributes, attributes : Optional[Tuple[CUpointer_attribute] | List[CUpointer_attribute]], ptr): """ Returns information about a pointer. The supported attributes are (refer to :py:obj:`~.cuPointerGetAttribute` for attribute descriptions and restrictions): - :py:obj:`~.CU_POINTER_ATTRIBUTE_CONTEXT` - :py:obj:`~.CU_POINTER_ATTRIBUTE_MEMORY_TYPE` - :py:obj:`~.CU_POINTER_ATTRIBUTE_DEVICE_POINTER` - :py:obj:`~.CU_POINTER_ATTRIBUTE_HOST_POINTER` - :py:obj:`~.CU_POINTER_ATTRIBUTE_SYNC_MEMOPS` - :py:obj:`~.CU_POINTER_ATTRIBUTE_BUFFER_ID` - :py:obj:`~.CU_POINTER_ATTRIBUTE_IS_MANAGED` - :py:obj:`~.CU_POINTER_ATTRIBUTE_DEVICE_ORDINAL` - :py:obj:`~.CU_POINTER_ATTRIBUTE_RANGE_START_ADDR` - :py:obj:`~.CU_POINTER_ATTRIBUTE_RANGE_SIZE` - :py:obj:`~.CU_POINTER_ATTRIBUTE_MAPPED` - :py:obj:`~.CU_POINTER_ATTRIBUTE_IS_LEGACY_CUDA_IPC_CAPABLE` - :py:obj:`~.CU_POINTER_ATTRIBUTE_ALLOWED_HANDLE_TYPES` - :py:obj:`~.CU_POINTER_ATTRIBUTE_MEMPOOL_HANDLE` Unlike :py:obj:`~.cuPointerGetAttribute`, this function will not return an error when the `ptr` encountered is not a valid CUDA pointer. Instead, the attributes are assigned default NULL values and CUDA_SUCCESS is returned. If `ptr` was not allocated by, mapped by, or registered with a :py:obj:`~.CUcontext` which uses UVA (Unified Virtual Addressing), :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` is returned. Parameters ---------- numAttributes : unsigned int Number of attributes to query attributes : List[:py:obj:`~.CUpointer_attribute`] An array of attributes to query (numAttributes and the number of attributes in this array should match) ptr : :py:obj:`~.CUdeviceptr` Pointer to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` data : List[Any] A two-dimensional array containing pointers to memory locations where the result of each attribute query will be written to. See Also -------- :py:obj:`~.cuPointerGetAttribute`, :py:obj:`~.cuPointerSetAttribute`, :py:obj:`~.cudaPointerGetAttributes` """ cdef ccuda.CUdeviceptr cptr if ptr is None: cptr = 0 elif isinstance(ptr, (CUdeviceptr,)): pptr = int(ptr) cptr = pptr else: pptr = int(CUdeviceptr(ptr)) cptr = pptr attributes = [] if attributes is None else attributes if not all(isinstance(_x, (CUpointer_attribute)) for _x in attributes): raise TypeError("Argument 'attributes' is not instance of type (expected Tuple[ccuda.CUpointer_attribute] or List[ccuda.CUpointer_attribute]") if numAttributes > len(attributes): raise RuntimeError("List is too small: " + str(len(attributes)) + " < " + str(numAttributes)) cdef vector[ccuda.CUpointer_attribute] cattributes = [pyattributes.value for pyattributes in (attributes)] pylist = [utils.HelperCUpointer_attribute(pyattributes, 0, is_getter=True) for pyattributes in attributes] cdef utils.InputVoidPtrPtrHelper voidStarHelperdata = utils.InputVoidPtrPtrHelper(pylist) cdef void** cvoidStarHelper_ptr = voidStarHelperdata.cptr err = ccuda.cuPointerGetAttributes(numAttributes, cattributes.data(), cvoidStarHelper_ptr, cptr) return (CUresult(err), [obj.pyObj() for obj in pylist]) {{endif}} {{if 'cuStreamCreate' in found_functions}} @cython.embedsignature(True) def cuStreamCreate(unsigned int Flags): """ Create a stream. Creates a stream and returns a handle in `phStream`. The `Flags` argument determines behaviors of the stream. Valid values for `Flags` are: - :py:obj:`~.CU_STREAM_DEFAULT`: Default stream creation flag. - :py:obj:`~.CU_STREAM_NON_BLOCKING`: Specifies that work running in the created stream may run concurrently with work in stream 0 (the NULL stream), and that the created stream should perform no implicit synchronization with stream 0. Parameters ---------- Flags : unsigned int Parameters for stream creation Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phStream : :py:obj:`~.CUstream` Returned newly created stream See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreateWithPriority`, :py:obj:`~.cuGreenCtxStreamCreate`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cuStreamGetFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cudaStreamCreateWithFlags` """ cdef CUstream phStream = CUstream() err = ccuda.cuStreamCreate(phStream._ptr, Flags) return (CUresult(err), phStream) {{endif}} {{if 'cuStreamCreateWithPriority' in found_functions}} @cython.embedsignature(True) def cuStreamCreateWithPriority(unsigned int flags, int priority): """ Create a stream with the given priority. Creates a stream with the specified priority and returns a handle in `phStream`. This affects the scheduling priority of work in the stream. Priorities provide a hint to preferentially run work with higher priority when possible, but do not preempt already-running work or provide any other functional guarantee on execution order. `priority` follows a convention where lower numbers represent higher priorities. '0' represents default priority. The range of meaningful numerical priorities can be queried using :py:obj:`~.cuCtxGetStreamPriorityRange`. If the specified priority is outside the numerical range returned by :py:obj:`~.cuCtxGetStreamPriorityRange`, it will automatically be clamped to the lowest or the highest number in the range. Parameters ---------- flags : unsigned int Flags for stream creation. See :py:obj:`~.cuStreamCreate` for a list of valid flags priority : int Stream priority. Lower numbers represent higher priorities. See :py:obj:`~.cuCtxGetStreamPriorityRange` for more information about meaningful stream priorities that can be passed. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phStream : :py:obj:`~.CUstream` Returned newly created stream See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuGreenCtxStreamCreate`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cuCtxGetStreamPriorityRange`, :py:obj:`~.cuStreamGetFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamCreateWithPriority` Notes ----- Stream priorities are supported only on GPUs with compute capability 3.5 or higher. In the current implementation, only compute kernels launched in priority streams are affected by the stream's priority. Stream priorities have no effect on host-to-device and device-to-host memory operations. """ cdef CUstream phStream = CUstream() err = ccuda.cuStreamCreateWithPriority(phStream._ptr, flags, priority) return (CUresult(err), phStream) {{endif}} {{if 'cuStreamGetPriority' in found_functions}} @cython.embedsignature(True) def cuStreamGetPriority(hStream): """ Query the priority of a given stream. Query the priority of a stream created using :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamCreateWithPriority` or :py:obj:`~.cuGreenCtxStreamCreate` and return the priority in `priority`. Note that if the stream was created with a priority outside the numerical range returned by :py:obj:`~.cuCtxGetStreamPriorityRange`, this function returns the clamped priority. See :py:obj:`~.cuStreamCreateWithPriority` for details about priority clamping. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Handle to the stream to be queried Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` priority : int Pointer to a signed integer in which the stream's priority is returned See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamCreateWithPriority`, :py:obj:`~.cuGreenCtxStreamCreate`, :py:obj:`~.cuCtxGetStreamPriorityRange`, :py:obj:`~.cuStreamGetFlags`, :py:obj:`~.cudaStreamGetPriority` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef int priority = 0 err = ccuda.cuStreamGetPriority(chStream, &priority) return (CUresult(err), priority) {{endif}} {{if 'cuStreamGetFlags' in found_functions}} @cython.embedsignature(True) def cuStreamGetFlags(hStream): """ Query the flags of a given stream. Query the flags of a stream created using :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamCreateWithPriority` or :py:obj:`~.cuGreenCtxStreamCreate` and return the flags in `flags`. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Handle to the stream to be queried Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` flags : unsigned int Pointer to an unsigned integer in which the stream's flags are returned The value returned in `flags` is a logical 'OR' of all flags that were used while creating this stream. See :py:obj:`~.cuStreamCreate` for the list of valid flags See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuGreenCtxStreamCreate`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cudaStreamGetFlags` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef unsigned int flags = 0 err = ccuda.cuStreamGetFlags(chStream, &flags) return (CUresult(err), flags) {{endif}} {{if 'cuStreamGetId' in found_functions}} @cython.embedsignature(True) def cuStreamGetId(hStream): """ Returns the unique Id associated with the stream handle supplied. Returns in `streamId` the unique Id which is associated with the given stream handle. The Id is unique for the life of the program. The stream handle `hStream` can refer to any of the following: - a stream created via any of the CUDA driver APIs such as :py:obj:`~.cuStreamCreate` and :py:obj:`~.cuStreamCreateWithPriority`, or their runtime API equivalents such as :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cudaStreamCreateWithFlags` and :py:obj:`~.cudaStreamCreateWithPriority`. Passing an invalid handle will result in undefined behavior. - any of the special streams such as the NULL stream, :py:obj:`~.CU_STREAM_LEGACY` and :py:obj:`~.CU_STREAM_PER_THREAD`. The runtime API equivalents of these are also accepted, which are NULL, :py:obj:`~.cudaStreamLegacy` and :py:obj:`~.cudaStreamPerThread` respectively. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Handle to the stream to be queried Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` streamId : unsigned long long Pointer to store the Id of the stream See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cudaStreamGetId` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef unsigned long long streamId = 0 err = ccuda.cuStreamGetId(chStream, &streamId) return (CUresult(err), streamId) {{endif}} {{if 'cuStreamGetCtx' in found_functions}} @cython.embedsignature(True) def cuStreamGetCtx(hStream): """ Query the context associated with a stream. Returns the CUDA context that the stream is associated with. Note there is a later version of this API, :py:obj:`~.cuStreamGetCtx_v2`. It will supplant this version in CUDA 13.0. It is recommended to use :py:obj:`~.cuStreamGetCtx_v2` till then as this version will return :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` for streams created via the API :py:obj:`~.cuGreenCtxStreamCreate`. The stream handle `hStream` can refer to any of the following: - a stream created via any of the CUDA driver APIs such as :py:obj:`~.cuStreamCreate` and :py:obj:`~.cuStreamCreateWithPriority`, or their runtime API equivalents such as :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cudaStreamCreateWithFlags` and :py:obj:`~.cudaStreamCreateWithPriority`. The returned context is the context that was active in the calling thread when the stream was created. Passing an invalid handle will result in undefined behavior. - any of the special streams such as the NULL stream, :py:obj:`~.CU_STREAM_LEGACY` and :py:obj:`~.CU_STREAM_PER_THREAD`. The runtime API equivalents of these are also accepted, which are NULL, :py:obj:`~.cudaStreamLegacy` and :py:obj:`~.cudaStreamPerThread` respectively. Specifying any of the special handles will return the context current to the calling thread. If no context is current to the calling thread, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` is returned. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Handle to the stream to be queried Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` pctx : :py:obj:`~.CUcontext` Returned context associated with the stream See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreateWithPriority`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cuStreamGetFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cuStreamGetCtx_v2`, :py:obj:`~.cudaStreamCreateWithFlags` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef CUcontext pctx = CUcontext() err = ccuda.cuStreamGetCtx(chStream, pctx._ptr) return (CUresult(err), pctx) {{endif}} {{if 'cuStreamGetCtx_v2' in found_functions}} @cython.embedsignature(True) def cuStreamGetCtx_v2(hStream): """ Query the contexts associated with a stream. Returns the contexts that the stream is associated with. If the stream is associated with a green context, the API returns the green context in `pGreenCtx` and the primary context of the associated device in `pCtx`. If the stream is associated with a regular context, the API returns the regular context in `pCtx` and NULL in `pGreenCtx`. The stream handle `hStream` can refer to any of the following: - a stream created via any of the CUDA driver APIs such as :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamCreateWithPriority` and :py:obj:`~.cuGreenCtxStreamCreate`, or their runtime API equivalents such as :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cudaStreamCreateWithFlags` and :py:obj:`~.cudaStreamCreateWithPriority`. Passing an invalid handle will result in undefined behavior. - any of the special streams such as the NULL stream, :py:obj:`~.CU_STREAM_LEGACY` and :py:obj:`~.CU_STREAM_PER_THREAD`. The runtime API equivalents of these are also accepted, which are NULL, :py:obj:`~.cudaStreamLegacy` and :py:obj:`~.cudaStreamPerThread` respectively. If any of the special handles are specified, the API will operate on the context current to the calling thread. If a green context (that was converted via :py:obj:`~.cuCtxFromGreenCtx()` before setting it current) is current to the calling thread, the API will return the green context in `pGreenCtx` and the primary context of the associated device in `pCtx`. If a regular context is current, the API returns the regular context in `pCtx` and NULL in `pGreenCtx`. Note that specifying :py:obj:`~.CU_STREAM_PER_THREAD` or :py:obj:`~.cudaStreamPerThread` will return :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` if a green context is current to the calling thread. If no context is current to the calling thread, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` is returned. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Handle to the stream to be queried Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` pCtx : :py:obj:`~.CUcontext` Returned regular context associated with the stream pGreenCtx : :py:obj:`~.CUgreenCtx` Returned green context if the stream is associated with a green context or NULL if not See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreate` :py:obj:`~.cuStreamCreateWithPriority`, :py:obj:`~.cuGreenCtxStreamCreate`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cuStreamGetFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cudaStreamCreateWithFlags`, """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef CUcontext pCtx = CUcontext() cdef CUgreenCtx pGreenCtx = CUgreenCtx() err = ccuda.cuStreamGetCtx_v2(chStream, pCtx._ptr, pGreenCtx._ptr) return (CUresult(err), pCtx, pGreenCtx) {{endif}} {{if 'cuStreamWaitEvent' in found_functions}} @cython.embedsignature(True) def cuStreamWaitEvent(hStream, hEvent, unsigned int Flags): """ Make a compute stream wait on an event. Makes all future work submitted to `hStream` wait for all work captured in `hEvent`. See :py:obj:`~.cuEventRecord()` for details on what is captured by an event. The synchronization will be performed efficiently on the device when applicable. `hEvent` may be from a different context or device than `hStream`. flags include: - :py:obj:`~.CU_EVENT_WAIT_DEFAULT`: Default event creation flag. - :py:obj:`~.CU_EVENT_WAIT_EXTERNAL`: Event is captured in the graph as an external event node when performing stream capture. This flag is invalid outside of stream capture. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to wait hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to wait on (may not be NULL) Flags : unsigned int See :py:obj:`~.CUevent_capture_flags` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuEventRecord`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cudaStreamWaitEvent` """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream err = ccuda.cuStreamWaitEvent(chStream, chEvent, Flags) return (CUresult(err),) {{endif}} {{if 'cuStreamAddCallback' in found_functions}} @cython.embedsignature(True) def cuStreamAddCallback(hStream, callback, userData, unsigned int flags): """ Add a callback to a compute stream. Adds a callback to be called on the host after all currently enqueued items in the stream have completed. For each cuStreamAddCallback call, the callback will be executed exactly once. The callback will block later work in the stream until it is finished. The callback may be passed :py:obj:`~.CUDA_SUCCESS` or an error code. In the event of a device error, all subsequently executed callbacks will receive an appropriate :py:obj:`~.CUresult`. Callbacks must not make any CUDA API calls. Attempting to use a CUDA API will result in :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`. Callbacks must not perform any synchronization that may depend on outstanding device work or other callbacks that are not mandated to run earlier. Callbacks without a mandated order (in independent streams) execute in undefined order and may be serialized. For the purposes of Unified Memory, callback execution makes a number of guarantees: - The callback stream is considered idle for the duration of the callback. Thus, for example, a callback may always use memory attached to the callback stream. - The start of execution of a callback has the same effect as synchronizing an event recorded in the same stream immediately prior to the callback. It thus synchronizes streams which have been "joined" prior to the callback. - Adding device work to any stream does not have the effect of making the stream active until all preceding host functions and stream callbacks have executed. Thus, for example, a callback might use global attached memory even if work has been added to another stream, if the work has been ordered behind the callback with an event. - Completion of a callback does not cause a stream to become active except as described above. The callback stream will remain idle if no device work follows the callback, and will remain idle across consecutive callbacks without device work in between. Thus, for example, stream synchronization can be done by signaling from a callback at the end of the stream. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to add callback to callback : :py:obj:`~.CUstreamCallback` The function to call once preceding stream operations are complete userData : Any User specified data to be passed to the callback function flags : unsigned int Reserved for future use, must be 0 Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuMemAllocManaged`, :py:obj:`~.cuStreamAttachMemAsync`, :py:obj:`~.cuLaunchHostFunc`, :py:obj:`~.cudaStreamAddCallback` Notes ----- This function is slated for eventual deprecation and removal. If you do not require the callback to execute in case of a device error, consider using :py:obj:`~.cuLaunchHostFunc`. Additionally, this function is not supported with :py:obj:`~.cuStreamBeginCapture` and :py:obj:`~.cuStreamEndCapture`, unlike :py:obj:`~.cuLaunchHostFunc`. """ cdef ccuda.CUstreamCallback ccallback if callback is None: ccallback = 0 elif isinstance(callback, (CUstreamCallback,)): pcallback = int(callback) ccallback = pcallback else: pcallback = int(CUstreamCallback(callback)) ccallback = pcallback cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cuserData = utils.HelperInputVoidPtr(userData) cdef void* cuserData_ptr = cuserData.cptr err = ccuda.cuStreamAddCallback(chStream, ccallback, cuserData_ptr, flags) return (CUresult(err),) {{endif}} {{if 'cuStreamBeginCapture_v2' in found_functions}} @cython.embedsignature(True) def cuStreamBeginCapture(hStream, mode not None : CUstreamCaptureMode): """ Begins graph capture on a stream. Begin graph capture on `hStream`. When a stream is in capture mode, all operations pushed into the stream will not be executed, but will instead be captured into a graph, which will be returned via :py:obj:`~.cuStreamEndCapture`. Capture may not be initiated if `stream` is CU_STREAM_LEGACY. Capture must be ended on the same stream in which it was initiated, and it may only be initiated if the stream is not already in capture mode. The capture mode may be queried via :py:obj:`~.cuStreamIsCapturing`. A unique id representing the capture sequence may be queried via :py:obj:`~.cuStreamGetCaptureInfo`. If `mode` is not :py:obj:`~.CU_STREAM_CAPTURE_MODE_RELAXED`, :py:obj:`~.cuStreamEndCapture` must be called on this stream from the same thread. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream in which to initiate capture mode : :py:obj:`~.CUstreamCaptureMode` Controls the interaction of this capture sequence with other API calls that are potentially unsafe. For more details see :py:obj:`~.cuThreadExchangeStreamCaptureMode`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamIsCapturing`, :py:obj:`~.cuStreamEndCapture`, :py:obj:`~.cuThreadExchangeStreamCaptureMode` Notes ----- Kernels captured using this API must not use texture and surface references. Reading or writing through any texture or surface reference is undefined behavior. This restriction does not apply to texture and surface objects. """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUstreamCaptureMode cmode = mode.value err = ccuda.cuStreamBeginCapture(chStream, cmode) return (CUresult(err),) {{endif}} {{if 'cuStreamBeginCaptureToGraph' in found_functions}} @cython.embedsignature(True) def cuStreamBeginCaptureToGraph(hStream, hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], dependencyData : Optional[Tuple[CUgraphEdgeData] | List[CUgraphEdgeData]], size_t numDependencies, mode not None : CUstreamCaptureMode): """ Begins graph capture on a stream to an existing graph. Begin graph capture on `hStream`, placing new nodes into an existing graph. When a stream is in capture mode, all operations pushed into the stream will not be executed, but will instead be captured into `hGraph`. The graph will not be instantiable until the user calls :py:obj:`~.cuStreamEndCapture`. Capture may not be initiated if `stream` is CU_STREAM_LEGACY. Capture must be ended on the same stream in which it was initiated, and it may only be initiated if the stream is not already in capture mode. The capture mode may be queried via :py:obj:`~.cuStreamIsCapturing`. A unique id representing the capture sequence may be queried via :py:obj:`~.cuStreamGetCaptureInfo`. If `mode` is not :py:obj:`~.CU_STREAM_CAPTURE_MODE_RELAXED`, :py:obj:`~.cuStreamEndCapture` must be called on this stream from the same thread. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream in which to initiate capture. hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to capture into. dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the first node captured in the stream. Can be NULL if numDependencies is 0. dependencyData : List[:py:obj:`~.CUgraphEdgeData`] Optional array of data associated with each dependency. numDependencies : size_t Number of dependencies. mode : :py:obj:`~.CUstreamCaptureMode` Controls the interaction of this capture sequence with other API calls that are potentially unsafe. For more details see :py:obj:`~.cuThreadExchangeStreamCaptureMode`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuStreamBeginCapture`, :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamIsCapturing`, :py:obj:`~.cuStreamEndCapture`, :py:obj:`~.cuThreadExchangeStreamCaptureMode`, :py:obj:`~.cuGraphAddNode`, Notes ----- Kernels captured using this API must not use texture and surface references. Reading or writing through any texture or surface reference is undefined behavior. This restriction does not apply to texture and surface objects. """ dependencyData = [] if dependencyData is None else dependencyData if not all(isinstance(_x, (CUgraphEdgeData,)) for _x in dependencyData): raise TypeError("Argument 'dependencyData' is not instance of type (expected Tuple[ccuda.CUgraphEdgeData,] or List[ccuda.CUgraphEdgeData,]") dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] cdef ccuda.CUgraphEdgeData* cdependencyData = NULL if len(dependencyData) > 0: cdependencyData = calloc(len(dependencyData), sizeof(ccuda.CUgraphEdgeData)) if cdependencyData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencyData)) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) for idx in range(len(dependencyData)): string.memcpy(&cdependencyData[idx], (dependencyData[idx])._ptr, sizeof(ccuda.CUgraphEdgeData)) if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) if numDependencies > len(dependencyData): raise RuntimeError("List is too small: " + str(len(dependencyData)) + " < " + str(numDependencies)) cdef ccuda.CUstreamCaptureMode cmode = mode.value err = ccuda.cuStreamBeginCaptureToGraph(chStream, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, (dependencyData[0])._ptr if len(dependencyData) == 1 else cdependencyData, numDependencies, cmode) if cdependencies is not NULL: free(cdependencies) if cdependencyData is not NULL: free(cdependencyData) return (CUresult(err),) {{endif}} {{if 'cuThreadExchangeStreamCaptureMode' in found_functions}} @cython.embedsignature(True) def cuThreadExchangeStreamCaptureMode(mode not None : CUstreamCaptureMode): """ Swaps the stream capture interaction mode for a thread. Sets the calling thread's stream capture interaction mode to the value contained in `*mode`, and overwrites `*mode` with the previous mode for the thread. To facilitate deterministic behavior across function or module boundaries, callers are encouraged to use this API in a push-pop fashion: **View CUDA Toolkit Documentation for a C++ code example** During stream capture (see :py:obj:`~.cuStreamBeginCapture`), some actions, such as a call to :py:obj:`~.cudaMalloc`, may be unsafe. In the case of :py:obj:`~.cudaMalloc`, the operation is not enqueued asynchronously to a stream, and is not observed by stream capture. Therefore, if the sequence of operations captured via :py:obj:`~.cuStreamBeginCapture` depended on the allocation being replayed whenever the graph is launched, the captured graph would be invalid. Therefore, stream capture places restrictions on API calls that can be made within or concurrently to a :py:obj:`~.cuStreamBeginCapture`-:py:obj:`~.cuStreamEndCapture` sequence. This behavior can be controlled via this API and flags to :py:obj:`~.cuStreamBeginCapture`. A thread's mode is one of the following: - `CU_STREAM_CAPTURE_MODE_GLOBAL:` This is the default mode. If the local thread has an ongoing capture sequence that was not initiated with `CU_STREAM_CAPTURE_MODE_RELAXED` at `cuStreamBeginCapture`, or if any other thread has a concurrent capture sequence initiated with `CU_STREAM_CAPTURE_MODE_GLOBAL`, this thread is prohibited from potentially unsafe API calls. - `CU_STREAM_CAPTURE_MODE_THREAD_LOCAL:` If the local thread has an ongoing capture sequence not initiated with `CU_STREAM_CAPTURE_MODE_RELAXED`, it is prohibited from potentially unsafe API calls. Concurrent capture sequences in other threads are ignored. - `CU_STREAM_CAPTURE_MODE_RELAXED:` The local thread is not prohibited from potentially unsafe API calls. Note that the thread is still prohibited from API calls which necessarily conflict with stream capture, for example, attempting :py:obj:`~.cuEventQuery` on an event that was last recorded inside a capture sequence. Parameters ---------- mode : :py:obj:`~.CUstreamCaptureMode` Pointer to mode value to swap with the current mode Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` mode : :py:obj:`~.CUstreamCaptureMode` Pointer to mode value to swap with the current mode See Also -------- :py:obj:`~.cuStreamBeginCapture` """ cdef ccuda.CUstreamCaptureMode cmode = mode.value err = ccuda.cuThreadExchangeStreamCaptureMode(&cmode) return (CUresult(err), CUstreamCaptureMode(cmode)) {{endif}} {{if 'cuStreamEndCapture' in found_functions}} @cython.embedsignature(True) def cuStreamEndCapture(hStream): """ Ends capture on a stream, returning the captured graph. End capture on `hStream`, returning the captured graph via `phGraph`. Capture must have been initiated on `hStream` via a call to :py:obj:`~.cuStreamBeginCapture`. If capture was invalidated, due to a violation of the rules of stream capture, then a NULL graph will be returned. If the `mode` argument to :py:obj:`~.cuStreamBeginCapture` was not :py:obj:`~.CU_STREAM_CAPTURE_MODE_RELAXED`, this call must be from the same thread as :py:obj:`~.cuStreamBeginCapture`. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_WRONG_THREAD` phGraph : :py:obj:`~.CUgraph` The captured graph See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamBeginCapture`, :py:obj:`~.cuStreamIsCapturing`, :py:obj:`~.cuGraphDestroy` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef CUgraph phGraph = CUgraph() err = ccuda.cuStreamEndCapture(chStream, phGraph._ptr) return (CUresult(err), phGraph) {{endif}} {{if 'cuStreamIsCapturing' in found_functions}} @cython.embedsignature(True) def cuStreamIsCapturing(hStream): """ Returns a stream's capture status. Return the capture status of `hStream` via `captureStatus`. After a successful call, `*captureStatus` will contain one of the following: - :py:obj:`~.CU_STREAM_CAPTURE_STATUS_NONE`: The stream is not capturing. - :py:obj:`~.CU_STREAM_CAPTURE_STATUS_ACTIVE`: The stream is capturing. - :py:obj:`~.CU_STREAM_CAPTURE_STATUS_INVALIDATED`: The stream was capturing but an error has invalidated the capture sequence. The capture sequence must be terminated with :py:obj:`~.cuStreamEndCapture` on the stream where it was initiated in order to continue using `hStream`. Note that, if this is called on :py:obj:`~.CU_STREAM_LEGACY` (the "null stream") while a blocking stream in the same context is capturing, it will return :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_IMPLICIT` and `*captureStatus` is unspecified after the call. The blocking stream capture is not invalidated. When a blocking stream is capturing, the legacy stream is in an unusable state until the blocking stream capture is terminated. The legacy stream is not supported for stream capture, but attempted use would have an implicit dependency on the capturing stream(s). Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_IMPLICIT` captureStatus : :py:obj:`~.CUstreamCaptureStatus` Returns the stream's capture status See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamBeginCapture`, :py:obj:`~.cuStreamEndCapture` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUstreamCaptureStatus captureStatus err = ccuda.cuStreamIsCapturing(chStream, &captureStatus) return (CUresult(err), CUstreamCaptureStatus(captureStatus)) {{endif}} {{if 'cuStreamGetCaptureInfo_v2' in found_functions}} @cython.embedsignature(True) def cuStreamGetCaptureInfo(hStream): """ Query a stream's capture state. Query stream state related to stream capture. If called on :py:obj:`~.CU_STREAM_LEGACY` (the "null stream") while a stream not created with :py:obj:`~.CU_STREAM_NON_BLOCKING` is capturing, returns :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_IMPLICIT`. Valid data (other than capture status) is returned only if both of the following are true: - the call returns CUDA_SUCCESS - the returned capture status is :py:obj:`~.CU_STREAM_CAPTURE_STATUS_ACTIVE` Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_IMPLICIT` captureStatus_out : :py:obj:`~.CUstreamCaptureStatus` Location to return the capture status of the stream; required id_out : :py:obj:`~.cuuint64_t` Optional location to return an id for the capture sequence, which is unique over the lifetime of the process graph_out : :py:obj:`~.CUgraph` Optional location to return the graph being captured into. All operations other than destroy and node removal are permitted on the graph while the capture sequence is in progress. This API does not transfer ownership of the graph, which is transferred or destroyed at :py:obj:`~.cuStreamEndCapture`. Note that the graph handle may be invalidated before end of capture for certain errors. Nodes that are or become unreachable from the original stream at :py:obj:`~.cuStreamEndCapture` due to direct actions on the graph do not trigger :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNJOINED`. dependencies_out : List[:py:obj:`~.CUgraphNode`] Optional location to store a pointer to an array of nodes. The next node to be captured in the stream will depend on this set of nodes, absent operations such as event wait which modify this set. The array pointer is valid until the next API call which operates on the stream or until the capture is terminated. The node handles may be copied out and are valid until they or the graph is destroyed. The driver-owned array may also be passed directly to APIs that operate on the graph (not the stream) without copying. numDependencies_out : int Optional location to store the size of the array returned in dependencies_out. See Also -------- :py:obj:`~.cuStreamGetCaptureInfo_v3` :py:obj:`~.cuStreamBeginCapture`, :py:obj:`~.cuStreamIsCapturing`, :py:obj:`~.cuStreamUpdateCaptureDependencies` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUstreamCaptureStatus captureStatus_out cdef cuuint64_t id_out = cuuint64_t() cdef CUgraph graph_out = CUgraph() cdef const ccuda.CUgraphNode* cdependencies_out = NULL pydependencies_out = [] cdef size_t numDependencies_out = 0 err = ccuda.cuStreamGetCaptureInfo(chStream, &captureStatus_out, id_out._ptr, graph_out._ptr, &cdependencies_out, &numDependencies_out) if CUresult(err) == CUresult(0): pydependencies_out = [CUgraphNode(init_value=cdependencies_out[idx]) for idx in range(numDependencies_out)] return (CUresult(err), CUstreamCaptureStatus(captureStatus_out), id_out, graph_out, pydependencies_out, numDependencies_out) {{endif}} {{if 'cuStreamGetCaptureInfo_v3' in found_functions}} @cython.embedsignature(True) def cuStreamGetCaptureInfo_v3(hStream): """ Query a stream's capture state (12.3+) Query stream state related to stream capture. If called on :py:obj:`~.CU_STREAM_LEGACY` (the "null stream") while a stream not created with :py:obj:`~.CU_STREAM_NON_BLOCKING` is capturing, returns :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_IMPLICIT`. Valid data (other than capture status) is returned only if both of the following are true: - the call returns CUDA_SUCCESS - the returned capture status is :py:obj:`~.CU_STREAM_CAPTURE_STATUS_ACTIVE` If `edgeData_out` is non-NULL then `dependencies_out` must be as well. If `dependencies_out` is non-NULL and `edgeData_out` is NULL, but there is non-zero edge data for one or more of the current stream dependencies, the call will return :py:obj:`~.CUDA_ERROR_LOSSY_QUERY`. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_IMPLICIT`, :py:obj:`~.CUDA_ERROR_LOSSY_QUERY` captureStatus_out : :py:obj:`~.CUstreamCaptureStatus` Location to return the capture status of the stream; required id_out : :py:obj:`~.cuuint64_t` Optional location to return an id for the capture sequence, which is unique over the lifetime of the process graph_out : :py:obj:`~.CUgraph` Optional location to return the graph being captured into. All operations other than destroy and node removal are permitted on the graph while the capture sequence is in progress. This API does not transfer ownership of the graph, which is transferred or destroyed at :py:obj:`~.cuStreamEndCapture`. Note that the graph handle may be invalidated before end of capture for certain errors. Nodes that are or become unreachable from the original stream at :py:obj:`~.cuStreamEndCapture` due to direct actions on the graph do not trigger :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNJOINED`. dependencies_out : List[:py:obj:`~.CUgraphNode`] Optional location to store a pointer to an array of nodes. The next node to be captured in the stream will depend on this set of nodes, absent operations such as event wait which modify this set. The array pointer is valid until the next API call which operates on the stream or until the capture is terminated. The node handles may be copied out and are valid until they or the graph is destroyed. The driver-owned array may also be passed directly to APIs that operate on the graph (not the stream) without copying. edgeData_out : List[:py:obj:`~.CUgraphEdgeData`] Optional location to store a pointer to an array of graph edge data. This array parallels `dependencies_out`; the next node to be added has an edge to `dependencies_out`[i] with annotation `edgeData_out`[i] for each `i`. The array pointer is valid until the next API call which operates on the stream or until the capture is terminated. numDependencies_out : int Optional location to store the size of the array returned in dependencies_out. See Also -------- :py:obj:`~.cuStreamGetCaptureInfo` :py:obj:`~.cuStreamBeginCapture`, :py:obj:`~.cuStreamIsCapturing`, :py:obj:`~.cuStreamUpdateCaptureDependencies` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUstreamCaptureStatus captureStatus_out cdef cuuint64_t id_out = cuuint64_t() cdef CUgraph graph_out = CUgraph() cdef const ccuda.CUgraphNode* cdependencies_out = NULL pydependencies_out = [] cdef const ccuda.CUgraphEdgeData* cedgeData_out = NULL pyedgeData_out = [] cdef size_t numDependencies_out = 0 err = ccuda.cuStreamGetCaptureInfo_v3(chStream, &captureStatus_out, id_out._ptr, graph_out._ptr, &cdependencies_out, &cedgeData_out, &numDependencies_out) if CUresult(err) == CUresult(0): pydependencies_out = [CUgraphNode(init_value=cdependencies_out[idx]) for idx in range(numDependencies_out)] if CUresult(err) == CUresult(0): pyedgeData_out = [CUgraphEdgeData(_ptr=&cedgeData_out[idx]) for idx in range(numDependencies_out)] return (CUresult(err), CUstreamCaptureStatus(captureStatus_out), id_out, graph_out, pydependencies_out, pyedgeData_out, numDependencies_out) {{endif}} {{if 'cuStreamUpdateCaptureDependencies' in found_functions}} @cython.embedsignature(True) def cuStreamUpdateCaptureDependencies(hStream, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, unsigned int flags): """ Update the set of dependencies in a capturing stream (11.3+) Modifies the dependency set of a capturing stream. The dependency set is the set of nodes that the next captured node in the stream will depend on. Valid flags are :py:obj:`~.CU_STREAM_ADD_CAPTURE_DEPENDENCIES` and :py:obj:`~.CU_STREAM_SET_CAPTURE_DEPENDENCIES`. These control whether the set passed to the API is added to the existing set or replaces it. A flags value of 0 defaults to :py:obj:`~.CU_STREAM_ADD_CAPTURE_DEPENDENCIES`. Nodes that are removed from the dependency set via this API do not result in :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNJOINED` if they are unreachable from the stream at :py:obj:`~.cuStreamEndCapture`. Returns :py:obj:`~.CUDA_ERROR_ILLEGAL_STATE` if the stream is not capturing. This API is new in CUDA 11.3. Developers requiring compatibility across minor versions to CUDA 11.0 should not use this API or provide a fallback. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to update dependencies : List[:py:obj:`~.CUgraphNode`] The set of dependencies to add numDependencies : size_t The size of the dependencies array flags : unsigned int See above Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_ILLEGAL_STATE` See Also -------- :py:obj:`~.cuStreamBeginCapture`, :py:obj:`~.cuStreamGetCaptureInfo`, """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) err = ccuda.cuStreamUpdateCaptureDependencies(chStream, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, flags) if cdependencies is not NULL: free(cdependencies) return (CUresult(err),) {{endif}} {{if 'cuStreamUpdateCaptureDependencies_v2' in found_functions}} @cython.embedsignature(True) def cuStreamUpdateCaptureDependencies_v2(hStream, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], dependencyData : Optional[Tuple[CUgraphEdgeData] | List[CUgraphEdgeData]], size_t numDependencies, unsigned int flags): """ Update the set of dependencies in a capturing stream (12.3+) Modifies the dependency set of a capturing stream. The dependency set is the set of nodes that the next captured node in the stream will depend on along with the edge data for those dependencies. Valid flags are :py:obj:`~.CU_STREAM_ADD_CAPTURE_DEPENDENCIES` and :py:obj:`~.CU_STREAM_SET_CAPTURE_DEPENDENCIES`. These control whether the set passed to the API is added to the existing set or replaces it. A flags value of 0 defaults to :py:obj:`~.CU_STREAM_ADD_CAPTURE_DEPENDENCIES`. Nodes that are removed from the dependency set via this API do not result in :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNJOINED` if they are unreachable from the stream at :py:obj:`~.cuStreamEndCapture`. Returns :py:obj:`~.CUDA_ERROR_ILLEGAL_STATE` if the stream is not capturing. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to update dependencies : List[:py:obj:`~.CUgraphNode`] The set of dependencies to add dependencyData : List[:py:obj:`~.CUgraphEdgeData`] Optional array of data associated with each dependency. numDependencies : size_t The size of the dependencies array flags : unsigned int See above Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_ILLEGAL_STATE` See Also -------- :py:obj:`~.cuStreamBeginCapture`, :py:obj:`~.cuStreamGetCaptureInfo`, """ dependencyData = [] if dependencyData is None else dependencyData if not all(isinstance(_x, (CUgraphEdgeData,)) for _x in dependencyData): raise TypeError("Argument 'dependencyData' is not instance of type (expected Tuple[ccuda.CUgraphEdgeData,] or List[ccuda.CUgraphEdgeData,]") dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] cdef ccuda.CUgraphEdgeData* cdependencyData = NULL if len(dependencyData) > 0: cdependencyData = calloc(len(dependencyData), sizeof(ccuda.CUgraphEdgeData)) if cdependencyData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencyData)) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) for idx in range(len(dependencyData)): string.memcpy(&cdependencyData[idx], (dependencyData[idx])._ptr, sizeof(ccuda.CUgraphEdgeData)) err = ccuda.cuStreamUpdateCaptureDependencies_v2(chStream, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, (dependencyData[0])._ptr if len(dependencyData) == 1 else cdependencyData, numDependencies, flags) if cdependencies is not NULL: free(cdependencies) if cdependencyData is not NULL: free(cdependencyData) return (CUresult(err),) {{endif}} {{if 'cuStreamAttachMemAsync' in found_functions}} @cython.embedsignature(True) def cuStreamAttachMemAsync(hStream, dptr, size_t length, unsigned int flags): """ Attach memory to a stream asynchronously. Enqueues an operation in `hStream` to specify stream association of `length` bytes of memory starting from `dptr`. This function is a stream-ordered operation, meaning that it is dependent on, and will only take effect when, previous work in stream has completed. Any previous association is automatically replaced. `dptr` must point to one of the following types of memories: - managed memory declared using the managed keyword or allocated with :py:obj:`~.cuMemAllocManaged`. - a valid host-accessible region of system-allocated pageable memory. This type of memory may only be specified if the device associated with the stream reports a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_PAGEABLE_MEMORY_ACCESS`. For managed allocations, `length` must be either zero or the entire allocation's size. Both indicate that the entire allocation's stream association is being changed. Currently, it is not possible to change stream association for a portion of a managed allocation. For pageable host allocations, `length` must be non-zero. The stream association is specified using `flags` which must be one of :py:obj:`~.CUmemAttach_flags`. If the :py:obj:`~.CU_MEM_ATTACH_GLOBAL` flag is specified, the memory can be accessed by any stream on any device. If the :py:obj:`~.CU_MEM_ATTACH_HOST` flag is specified, the program makes a guarantee that it won't access the memory on the device from any stream on a device that has a zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. If the :py:obj:`~.CU_MEM_ATTACH_SINGLE` flag is specified and `hStream` is associated with a device that has a zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`, the program makes a guarantee that it will only access the memory on the device from `hStream`. It is illegal to attach singly to the NULL stream, because the NULL stream is a virtual global stream and not a specific stream. An error will be returned in this case. When memory is associated with a single stream, the Unified Memory system will allow CPU access to this memory region so long as all operations in `hStream` have completed, regardless of whether other streams are active. In effect, this constrains exclusive ownership of the managed memory region by an active GPU to per-stream activity instead of whole-GPU activity. Accessing memory on the device from streams that are not associated with it will produce undefined results. No error checking is performed by the Unified Memory system to ensure that kernels launched into other streams do not access this region. It is a program's responsibility to order calls to :py:obj:`~.cuStreamAttachMemAsync` via events, synchronization or other means to ensure legal access to memory at all times. Data visibility and coherency will be changed appropriately for all kernels which follow a stream-association change. If `hStream` is destroyed while data is associated with it, the association is removed and the association reverts to the default visibility of the allocation as specified at :py:obj:`~.cuMemAllocManaged`. For managed variables, the default association is always :py:obj:`~.CU_MEM_ATTACH_GLOBAL`. Note that destroying a stream is an asynchronous operation, and as a result, the change to default association won't happen until all work in the stream has completed. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream in which to enqueue the attach operation dptr : :py:obj:`~.CUdeviceptr` Pointer to memory (must be a pointer to managed memory or to a valid host-accessible region of system-allocated pageable memory) length : size_t Length of memory flags : unsigned int Must be one of :py:obj:`~.CUmemAttach_flags` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuMemAllocManaged`, :py:obj:`~.cudaStreamAttachMemAsync` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream err = ccuda.cuStreamAttachMemAsync(chStream, cdptr, length, flags) return (CUresult(err),) {{endif}} {{if 'cuStreamQuery' in found_functions}} @cython.embedsignature(True) def cuStreamQuery(hStream): """ Determine status of a compute stream. Returns :py:obj:`~.CUDA_SUCCESS` if all operations in the stream specified by `hStream` have completed, or :py:obj:`~.CUDA_ERROR_NOT_READY` if not. For the purposes of Unified Memory, a return value of :py:obj:`~.CUDA_SUCCESS` is equivalent to having called :py:obj:`~.cuStreamSynchronize()`. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to query status of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_READY` See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamQuery` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream err = ccuda.cuStreamQuery(chStream) return (CUresult(err),) {{endif}} {{if 'cuStreamSynchronize' in found_functions}} @cython.embedsignature(True) def cuStreamSynchronize(hStream): """ Wait until a stream's tasks are completed. Waits until the device has completed all operations in the stream specified by `hStream`. If the context was created with the :py:obj:`~.CU_CTX_SCHED_BLOCKING_SYNC` flag, the CPU thread will block until the stream is finished with all of its tasks. \note_null_stream Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to wait for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamSynchronize` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream err = ccuda.cuStreamSynchronize(chStream) return (CUresult(err),) {{endif}} {{if 'cuStreamDestroy_v2' in found_functions}} @cython.embedsignature(True) def cuStreamDestroy(hStream): """ Destroys a stream. Destroys the stream specified by `hStream`. In case the device is still doing work in the stream `hStream` when :py:obj:`~.cuStreamDestroy()` is called, the function will return immediately and the resources associated with `hStream` will be released automatically once the device has completed all work in `hStream`. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamDestroy` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream err = ccuda.cuStreamDestroy(chStream) return (CUresult(err),) {{endif}} {{if 'cuStreamCopyAttributes' in found_functions}} @cython.embedsignature(True) def cuStreamCopyAttributes(dst, src): """ Copies attributes from source stream to destination stream. Copies attributes from source stream `src` to destination stream `dst`. Both streams must have the same context. Parameters ---------- dst : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Destination stream src : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Source stream For list of attributes see :py:obj:`~.CUstreamAttrID` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.CUaccessPolicyWindow` """ cdef ccuda.CUstream csrc if src is None: csrc = 0 elif isinstance(src, (CUstream,)): psrc = int(src) csrc = psrc else: psrc = int(CUstream(src)) csrc = psrc cdef ccuda.CUstream cdst if dst is None: cdst = 0 elif isinstance(dst, (CUstream,)): pdst = int(dst) cdst = pdst else: pdst = int(CUstream(dst)) cdst = pdst err = ccuda.cuStreamCopyAttributes(cdst, csrc) return (CUresult(err),) {{endif}} {{if 'cuStreamGetAttribute' in found_functions}} @cython.embedsignature(True) def cuStreamGetAttribute(hStream, attr not None : CUstreamAttrID): """ Queries stream attribute. Queries attribute `attr` from `hStream` and stores it in corresponding member of `value_out`. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` attr : :py:obj:`~.CUstreamAttrID` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` value_out : :py:obj:`~.CUstreamAttrValue` See Also -------- :py:obj:`~.CUaccessPolicyWindow` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUstreamAttrID cattr = attr.value cdef CUstreamAttrValue value_out = CUstreamAttrValue() err = ccuda.cuStreamGetAttribute(chStream, cattr, value_out._ptr) return (CUresult(err), value_out) {{endif}} {{if 'cuStreamSetAttribute' in found_functions}} @cython.embedsignature(True) def cuStreamSetAttribute(hStream, attr not None : CUstreamAttrID, value : Optional[CUstreamAttrValue]): """ Sets stream attribute. Sets attribute `attr` on `hStream` from corresponding attribute of `value`. The updated attribute will be applied to subsequent work submitted to the stream. It will not affect previously submitted work. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` attr : :py:obj:`~.CUstreamAttrID` value : :py:obj:`~.CUstreamAttrValue` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.CUaccessPolicyWindow` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUstreamAttrID cattr = attr.value cdef ccuda.CUstreamAttrValue* cvalue_ptr = value._ptr if value != None else NULL err = ccuda.cuStreamSetAttribute(chStream, cattr, cvalue_ptr) return (CUresult(err),) {{endif}} {{if 'cuEventCreate' in found_functions}} @cython.embedsignature(True) def cuEventCreate(unsigned int Flags): """ Creates an event. Creates an event *phEvent for the current context with the flags specified via `Flags`. Valid flags include: - :py:obj:`~.CU_EVENT_DEFAULT`: Default event creation flag. - :py:obj:`~.CU_EVENT_BLOCKING_SYNC`: Specifies that the created event should use blocking synchronization. A CPU thread that uses :py:obj:`~.cuEventSynchronize()` to wait on an event created with this flag will block until the event has actually been recorded. - :py:obj:`~.CU_EVENT_DISABLE_TIMING`: Specifies that the created event does not need to record timing data. Events created with this flag specified and the :py:obj:`~.CU_EVENT_BLOCKING_SYNC` flag not specified will provide the best performance when used with :py:obj:`~.cuStreamWaitEvent()` and :py:obj:`~.cuEventQuery()`. - :py:obj:`~.CU_EVENT_INTERPROCESS`: Specifies that the created event may be used as an interprocess event by :py:obj:`~.cuIpcGetEventHandle()`. :py:obj:`~.CU_EVENT_INTERPROCESS` must be specified along with :py:obj:`~.CU_EVENT_DISABLE_TIMING`. Parameters ---------- Flags : unsigned int Event creation flags Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phEvent : :py:obj:`~.CUevent` Returns newly created event See Also -------- :py:obj:`~.cuEventRecord`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cuEventElapsedTime`, :py:obj:`~.cudaEventCreate`, :py:obj:`~.cudaEventCreateWithFlags` """ cdef CUevent phEvent = CUevent() err = ccuda.cuEventCreate(phEvent._ptr, Flags) return (CUresult(err), phEvent) {{endif}} {{if 'cuEventRecord' in found_functions}} @cython.embedsignature(True) def cuEventRecord(hEvent, hStream): """ Records an event. Captures in `hEvent` the contents of `hStream` at the time of this call. `hEvent` and `hStream` must be from the same context otherwise :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` is returned. Calls such as :py:obj:`~.cuEventQuery()` or :py:obj:`~.cuStreamWaitEvent()` will then examine or wait for completion of the work that was captured. Uses of `hStream` after this call do not modify `hEvent`. See note on default stream behavior for what is captured in the default case. :py:obj:`~.cuEventRecord()` can be called multiple times on the same event and will overwrite the previously captured state. Other APIs such as :py:obj:`~.cuStreamWaitEvent()` use the most recently captured state at the time of the API call, and are not affected by later calls to :py:obj:`~.cuEventRecord()`. Before the first call to :py:obj:`~.cuEventRecord()`, an event represents an empty set of work, so for example :py:obj:`~.cuEventQuery()` would return :py:obj:`~.CUDA_SUCCESS`. Parameters ---------- hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to record hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to record event for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cuEventElapsedTime`, :py:obj:`~.cudaEventRecord`, :py:obj:`~.cuEventRecordWithFlags` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent err = ccuda.cuEventRecord(chEvent, chStream) return (CUresult(err),) {{endif}} {{if 'cuEventRecordWithFlags' in found_functions}} @cython.embedsignature(True) def cuEventRecordWithFlags(hEvent, hStream, unsigned int flags): """ Records an event. Captures in `hEvent` the contents of `hStream` at the time of this call. `hEvent` and `hStream` must be from the same context otherwise :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` is returned. Calls such as :py:obj:`~.cuEventQuery()` or :py:obj:`~.cuStreamWaitEvent()` will then examine or wait for completion of the work that was captured. Uses of `hStream` after this call do not modify `hEvent`. See note on default stream behavior for what is captured in the default case. :py:obj:`~.cuEventRecordWithFlags()` can be called multiple times on the same event and will overwrite the previously captured state. Other APIs such as :py:obj:`~.cuStreamWaitEvent()` use the most recently captured state at the time of the API call, and are not affected by later calls to :py:obj:`~.cuEventRecordWithFlags()`. Before the first call to :py:obj:`~.cuEventRecordWithFlags()`, an event represents an empty set of work, so for example :py:obj:`~.cuEventQuery()` would return :py:obj:`~.CUDA_SUCCESS`. flags include: - :py:obj:`~.CU_EVENT_RECORD_DEFAULT`: Default event creation flag. - :py:obj:`~.CU_EVENT_RECORD_EXTERNAL`: Event is captured in the graph as an external event node when performing stream capture. This flag is invalid outside of stream capture. Parameters ---------- hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to record hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to record event for flags : unsigned int See :py:obj:`~.CUevent_capture_flags` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cuEventElapsedTime`, :py:obj:`~.cuEventRecord`, :py:obj:`~.cudaEventRecord` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent err = ccuda.cuEventRecordWithFlags(chEvent, chStream, flags) return (CUresult(err),) {{endif}} {{if 'cuEventQuery' in found_functions}} @cython.embedsignature(True) def cuEventQuery(hEvent): """ Queries an event's status. Queries the status of all work currently captured by `hEvent`. See :py:obj:`~.cuEventRecord()` for details on what is captured by an event. Returns :py:obj:`~.CUDA_SUCCESS` if all captured work has been completed, or :py:obj:`~.CUDA_ERROR_NOT_READY` if any captured work is incomplete. For the purposes of Unified Memory, a return value of :py:obj:`~.CUDA_SUCCESS` is equivalent to having called :py:obj:`~.cuEventSynchronize()`. Parameters ---------- hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_READY` See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventRecord`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cuEventElapsedTime`, :py:obj:`~.cudaEventQuery` """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent err = ccuda.cuEventQuery(chEvent) return (CUresult(err),) {{endif}} {{if 'cuEventSynchronize' in found_functions}} @cython.embedsignature(True) def cuEventSynchronize(hEvent): """ Waits for an event to complete. Waits until the completion of all work currently captured in `hEvent`. See :py:obj:`~.cuEventRecord()` for details on what is captured by an event. Waiting for an event that was created with the :py:obj:`~.CU_EVENT_BLOCKING_SYNC` flag will cause the calling CPU thread to block until the event has been completed by the device. If the :py:obj:`~.CU_EVENT_BLOCKING_SYNC` flag has not been set, then the CPU thread will busy-wait until the event has been completed by the device. Parameters ---------- hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to wait for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventRecord`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cuEventElapsedTime`, :py:obj:`~.cudaEventSynchronize` """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent err = ccuda.cuEventSynchronize(chEvent) return (CUresult(err),) {{endif}} {{if 'cuEventDestroy_v2' in found_functions}} @cython.embedsignature(True) def cuEventDestroy(hEvent): """ Destroys an event. Destroys the event specified by `hEvent`. An event may be destroyed before it is complete (i.e., while :py:obj:`~.cuEventQuery()` would return :py:obj:`~.CUDA_ERROR_NOT_READY`). In this case, the call does not block on completion of the event, and any associated resources will automatically be released asynchronously at completion. Parameters ---------- hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventRecord`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuEventElapsedTime`, :py:obj:`~.cudaEventDestroy` """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent err = ccuda.cuEventDestroy(chEvent) return (CUresult(err),) {{endif}} {{if 'cuEventElapsedTime' in found_functions}} @cython.embedsignature(True) def cuEventElapsedTime(hStart, hEnd): """ Computes the elapsed time between two events. Computes the elapsed time between two events (in milliseconds with a resolution of around 0.5 microseconds). If either event was last recorded in a non-NULL stream, the resulting time may be greater than expected (even if both used the same stream handle). This happens because the :py:obj:`~.cuEventRecord()` operation takes place asynchronously and there is no guarantee that the measured latency is actually just between the two events. Any number of other different stream operations could execute in between the two measured events, thus altering the timing in a significant way. If :py:obj:`~.cuEventRecord()` has not been called on either event then :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` is returned. If :py:obj:`~.cuEventRecord()` has been called on both events but one or both of them has not yet been completed (that is, :py:obj:`~.cuEventQuery()` would return :py:obj:`~.CUDA_ERROR_NOT_READY` on at least one of the events), :py:obj:`~.CUDA_ERROR_NOT_READY` is returned. If either event was created with the :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag, then this function will return :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`. Parameters ---------- hStart : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Starting event hEnd : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Ending event Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_READY`, :py:obj:`~.CUDA_ERROR_UNKNOWN` pMilliseconds : float Time between `hStart` and `hEnd` in ms See Also -------- :py:obj:`~.cuEventCreate`, :py:obj:`~.cuEventRecord`, :py:obj:`~.cuEventQuery`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuEventDestroy`, :py:obj:`~.cudaEventElapsedTime` """ cdef ccuda.CUevent chEnd if hEnd is None: chEnd = 0 elif isinstance(hEnd, (CUevent,)): phEnd = int(hEnd) chEnd = phEnd else: phEnd = int(CUevent(hEnd)) chEnd = phEnd cdef ccuda.CUevent chStart if hStart is None: chStart = 0 elif isinstance(hStart, (CUevent,)): phStart = int(hStart) chStart = phStart else: phStart = int(CUevent(hStart)) chStart = phStart cdef float pMilliseconds = 0 err = ccuda.cuEventElapsedTime(&pMilliseconds, chStart, chEnd) return (CUresult(err), pMilliseconds) {{endif}} {{if 'cuImportExternalMemory' in found_functions}} @cython.embedsignature(True) def cuImportExternalMemory(memHandleDesc : Optional[CUDA_EXTERNAL_MEMORY_HANDLE_DESC]): """ Imports an external memory object. Imports an externally allocated memory object and returns a handle to that in `extMem_out`. The properties of the handle being imported must be described in `memHandleDesc`. The :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC` structure is defined as follows: **View CUDA Toolkit Documentation for a C++ code example** where :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` specifies the type of handle being imported. :py:obj:`~.CUexternalMemoryHandleType` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_FD`, then :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::fd must be a valid file descriptor referencing a memory object. Ownership of the file descriptor is transferred to the CUDA driver when the handle is imported successfully. Performing any operations on the file descriptor after it is imported results in undefined behavior. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32`, then exactly one of :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle and :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name must not be NULL. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle is not NULL, then it must represent a valid shared NT handle that references a memory object. Ownership of this handle is not transferred to CUDA after the import operation, so the application must release the handle using the appropriate system call. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name is not NULL, then it must point to a NULL-terminated array of UTF-16 characters that refers to a memory object. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_OPAQUE_WIN32_KMT`, then :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle must be non-NULL and :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name must be NULL. The handle specified must be a globally shared KMT handle. This handle does not hold a reference to the underlying object, and thus will be invalid when all references to the memory object are destroyed. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_HEAP`, then exactly one of :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle and :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name must not be NULL. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle is not NULL, then it must represent a valid shared NT handle that is returned by ID3D12Device::CreateSharedHandle when referring to a ID3D12Heap object. This handle holds a reference to the underlying object. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name is not NULL, then it must point to a NULL-terminated array of UTF-16 characters that refers to a ID3D12Heap object. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_RESOURCE`, then exactly one of :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle and :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name must not be NULL. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle is not NULL, then it must represent a valid shared NT handle that is returned by ID3D12Device::CreateSharedHandle when referring to a ID3D12Resource object. This handle holds a reference to the underlying object. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name is not NULL, then it must point to a NULL-terminated array of UTF-16 characters that refers to a ID3D12Resource object. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE`, then :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle must represent a valid shared NT handle that is returned by IDXGIResource1::CreateSharedHandle when referring to a ID3D11Resource object. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name is not NULL, then it must point to a NULL-terminated array of UTF-16 characters that refers to a ID3D11Resource object. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE_KMT`, then :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::handle must represent a valid shared KMT handle that is returned by IDXGIResource::GetSharedHandle when referring to a ID3D11Resource object and :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::win32::name must be NULL. If :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF`, then :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC`::handle::nvSciBufObject must be non-NULL and reference a valid NvSciBuf object. If the NvSciBuf object imported into CUDA is also mapped by other drivers, then the application must use :py:obj:`~.cuWaitExternalSemaphoresAsync` or :py:obj:`~.cuSignalExternalSemaphoresAsync` as appropriate barriers to maintain coherence between CUDA and the other drivers. See :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_SKIP_NVSCIBUF_MEMSYNC` and :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_WAIT_SKIP_NVSCIBUF_MEMSYNC` for memory synchronization. The size of the memory object must be specified in :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.size`. Specifying the flag :py:obj:`~.CUDA_EXTERNAL_MEMORY_DEDICATED` in :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.flags` indicates that the resource is a dedicated resource. The definition of what a dedicated resource is outside the scope of this extension. This flag must be set if :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC.type` is one of the following: :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D12_RESOURCE` :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE` :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_D3D11_RESOURCE_KMT` Parameters ---------- memHandleDesc : :py:obj:`~.CUDA_EXTERNAL_MEMORY_HANDLE_DESC` Memory import handle descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OPERATING_SYSTEM` extMem_out : :py:obj:`~.CUexternalMemory` Returned handle to an external memory object See Also -------- :py:obj:`~.cuDestroyExternalMemory`, :py:obj:`~.cuExternalMemoryGetMappedBuffer`, :py:obj:`~.cuExternalMemoryGetMappedMipmappedArray` Notes ----- If the Vulkan memory imported into CUDA is mapped on the CPU then the application must use vkInvalidateMappedMemoryRanges/vkFlushMappedMemoryRanges as well as appropriate Vulkan pipeline barriers to maintain coherence between CPU and GPU. For more information on these APIs, please refer to "Synchronization and Cache Control" chapter from Vulkan specification. """ cdef CUexternalMemory extMem_out = CUexternalMemory() cdef ccuda.CUDA_EXTERNAL_MEMORY_HANDLE_DESC* cmemHandleDesc_ptr = memHandleDesc._ptr if memHandleDesc != None else NULL err = ccuda.cuImportExternalMemory(extMem_out._ptr, cmemHandleDesc_ptr) return (CUresult(err), extMem_out) {{endif}} {{if 'cuExternalMemoryGetMappedBuffer' in found_functions}} @cython.embedsignature(True) def cuExternalMemoryGetMappedBuffer(extMem, bufferDesc : Optional[CUDA_EXTERNAL_MEMORY_BUFFER_DESC]): """ Maps a buffer onto an imported memory object. Maps a buffer onto an imported memory object and returns a device pointer in `devPtr`. The properties of the buffer being mapped must be described in `bufferDesc`. The :py:obj:`~.CUDA_EXTERNAL_MEMORY_BUFFER_DESC` structure is defined as follows: **View CUDA Toolkit Documentation for a C++ code example** where :py:obj:`~.CUDA_EXTERNAL_MEMORY_BUFFER_DESC.offset` is the offset in the memory object where the buffer's base address is. :py:obj:`~.CUDA_EXTERNAL_MEMORY_BUFFER_DESC.size` is the size of the buffer. :py:obj:`~.CUDA_EXTERNAL_MEMORY_BUFFER_DESC.flags` must be zero. The offset and size have to be suitably aligned to match the requirements of the external API. Mapping two buffers whose ranges overlap may or may not result in the same virtual address being returned for the overlapped portion. In such cases, the application must ensure that all accesses to that region from the GPU are volatile. Otherwise writes made via one address are not guaranteed to be visible via the other address, even if they're issued by the same thread. It is recommended that applications map the combined range instead of mapping separate buffers and then apply the appropriate offsets to the returned pointer to derive the individual buffers. The returned pointer `devPtr` must be freed using :py:obj:`~.cuMemFree`. Parameters ---------- extMem : :py:obj:`~.CUexternalMemory` Handle to external memory object bufferDesc : :py:obj:`~.CUDA_EXTERNAL_MEMORY_BUFFER_DESC` Buffer descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` devPtr : :py:obj:`~.CUdeviceptr` Returned device pointer to buffer See Also -------- :py:obj:`~.cuImportExternalMemory`, :py:obj:`~.cuDestroyExternalMemory`, :py:obj:`~.cuExternalMemoryGetMappedMipmappedArray` """ cdef ccuda.CUexternalMemory cextMem if extMem is None: cextMem = 0 elif isinstance(extMem, (CUexternalMemory,)): pextMem = int(extMem) cextMem = pextMem else: pextMem = int(CUexternalMemory(extMem)) cextMem = pextMem cdef CUdeviceptr devPtr = CUdeviceptr() cdef ccuda.CUDA_EXTERNAL_MEMORY_BUFFER_DESC* cbufferDesc_ptr = bufferDesc._ptr if bufferDesc != None else NULL err = ccuda.cuExternalMemoryGetMappedBuffer(devPtr._ptr, cextMem, cbufferDesc_ptr) return (CUresult(err), devPtr) {{endif}} {{if 'cuExternalMemoryGetMappedMipmappedArray' in found_functions}} @cython.embedsignature(True) def cuExternalMemoryGetMappedMipmappedArray(extMem, mipmapDesc : Optional[CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC]): """ Maps a CUDA mipmapped array onto an external memory object. Maps a CUDA mipmapped array onto an external object and returns a handle to it in `mipmap`. The properties of the CUDA mipmapped array being mapped must be described in `mipmapDesc`. The structure :py:obj:`~.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC` is defined as follows: **View CUDA Toolkit Documentation for a C++ code example** where :py:obj:`~.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC.offset` is the offset in the memory object where the base level of the mipmap chain is. :py:obj:`~.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC.arrayDesc` describes the format, dimensions and type of the base level of the mipmap chain. For further details on these parameters, please refer to the documentation for :py:obj:`~.cuMipmappedArrayCreate`. Note that if the mipmapped array is bound as a color target in the graphics API, then the flag :py:obj:`~.CUDA_ARRAY3D_COLOR_ATTACHMENT` must be specified in :py:obj:`~.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC`::arrayDesc::Flags. :py:obj:`~.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC.numLevels` specifies the total number of levels in the mipmap chain. If `extMem` was imported from a handle of type :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF`, then :py:obj:`~.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC.numLevels` must be equal to 1. The returned CUDA mipmapped array must be freed using :py:obj:`~.cuMipmappedArrayDestroy`. Parameters ---------- extMem : :py:obj:`~.CUexternalMemory` Handle to external memory object mipmapDesc : :py:obj:`~.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC` CUDA array descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` mipmap : :py:obj:`~.CUmipmappedArray` Returned CUDA mipmapped array See Also -------- :py:obj:`~.cuImportExternalMemory`, :py:obj:`~.cuDestroyExternalMemory`, :py:obj:`~.cuExternalMemoryGetMappedBuffer` """ cdef ccuda.CUexternalMemory cextMem if extMem is None: cextMem = 0 elif isinstance(extMem, (CUexternalMemory,)): pextMem = int(extMem) cextMem = pextMem else: pextMem = int(CUexternalMemory(extMem)) cextMem = pextMem cdef CUmipmappedArray mipmap = CUmipmappedArray() cdef ccuda.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC* cmipmapDesc_ptr = mipmapDesc._ptr if mipmapDesc != None else NULL err = ccuda.cuExternalMemoryGetMappedMipmappedArray(mipmap._ptr, cextMem, cmipmapDesc_ptr) return (CUresult(err), mipmap) {{endif}} {{if 'cuDestroyExternalMemory' in found_functions}} @cython.embedsignature(True) def cuDestroyExternalMemory(extMem): """ Destroys an external memory object. Destroys the specified external memory object. Any existing buffers and CUDA mipmapped arrays mapped onto this object must no longer be used and must be explicitly freed using :py:obj:`~.cuMemFree` and :py:obj:`~.cuMipmappedArrayDestroy` respectively. Parameters ---------- extMem : :py:obj:`~.CUexternalMemory` External memory object to be destroyed Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuImportExternalMemory`, :py:obj:`~.cuExternalMemoryGetMappedBuffer`, :py:obj:`~.cuExternalMemoryGetMappedMipmappedArray` """ cdef ccuda.CUexternalMemory cextMem if extMem is None: cextMem = 0 elif isinstance(extMem, (CUexternalMemory,)): pextMem = int(extMem) cextMem = pextMem else: pextMem = int(CUexternalMemory(extMem)) cextMem = pextMem err = ccuda.cuDestroyExternalMemory(cextMem) return (CUresult(err),) {{endif}} {{if 'cuImportExternalSemaphore' in found_functions}} @cython.embedsignature(True) def cuImportExternalSemaphore(semHandleDesc : Optional[CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC]): """ Imports an external semaphore. Imports an externally allocated synchronization object and returns a handle to that in `extSem_out`. The properties of the handle being imported must be described in `semHandleDesc`. The :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC` is defined as follows: **View CUDA Toolkit Documentation for a C++ code example** where :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` specifies the type of handle being imported. :py:obj:`~.CUexternalSemaphoreHandleType` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_FD`, then :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::fd must be a valid file descriptor referencing a synchronization object. Ownership of the file descriptor is transferred to the CUDA driver when the handle is imported successfully. Performing any operations on the file descriptor after it is imported results in undefined behavior. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32`, then exactly one of :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle and :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name must not be NULL. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle is not NULL, then it must represent a valid shared NT handle that references a synchronization object. Ownership of this handle is not transferred to CUDA after the import operation, so the application must release the handle using the appropriate system call. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name is not NULL, then it must name a valid synchronization object. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32_KMT`, then :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle must be non-NULL and :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name must be NULL. The handle specified must be a globally shared KMT handle. This handle does not hold a reference to the underlying object, and thus will be invalid when all references to the synchronization object are destroyed. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D12_FENCE`, then exactly one of :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle and :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name must not be NULL. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle is not NULL, then it must represent a valid shared NT handle that is returned by ID3D12Device::CreateSharedHandle when referring to a ID3D12Fence object. This handle holds a reference to the underlying object. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name is not NULL, then it must name a valid synchronization object that refers to a valid ID3D12Fence object. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_FENCE`, then :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle represents a valid shared NT handle that is returned by ID3D11Fence::CreateSharedHandle. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name is not NULL, then it must name a valid synchronization object that refers to a valid ID3D11Fence object. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC`, then :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::nvSciSyncObj represents a valid NvSciSyncObj. :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX`, then :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle represents a valid shared NT handle that is returned by IDXGIResource1::CreateSharedHandle when referring to a IDXGIKeyedMutex object. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name is not NULL, then it must name a valid synchronization object that refers to a valid IDXGIKeyedMutex object. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX_KMT`, then :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle represents a valid shared KMT handle that is returned by IDXGIResource::GetSharedHandle when referring to a IDXGIKeyedMutex object and :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name must be NULL. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_FD`, then :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::fd must be a valid file descriptor referencing a synchronization object. Ownership of the file descriptor is transferred to the CUDA driver when the handle is imported successfully. Performing any operations on the file descriptor after it is imported results in undefined behavior. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC.type` is :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_WIN32`, then exactly one of :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle and :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name must not be NULL. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::handle is not NULL, then it must represent a valid shared NT handle that references a synchronization object. Ownership of this handle is not transferred to CUDA after the import operation, so the application must release the handle using the appropriate system call. If :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC`::handle::win32::name is not NULL, then it must name a valid synchronization object. Parameters ---------- semHandleDesc : :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC` Semaphore import handle descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OPERATING_SYSTEM` extSem_out : :py:obj:`~.CUexternalSemaphore` Returned handle to an external semaphore See Also -------- :py:obj:`~.cuDestroyExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef CUexternalSemaphore extSem_out = CUexternalSemaphore() cdef ccuda.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC* csemHandleDesc_ptr = semHandleDesc._ptr if semHandleDesc != None else NULL err = ccuda.cuImportExternalSemaphore(extSem_out._ptr, csemHandleDesc_ptr) return (CUresult(err), extSem_out) {{endif}} {{if 'cuSignalExternalSemaphoresAsync' in found_functions}} @cython.embedsignature(True) def cuSignalExternalSemaphoresAsync(extSemArray : Optional[Tuple[CUexternalSemaphore] | List[CUexternalSemaphore]], paramsArray : Optional[Tuple[CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS] | List[CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS]], unsigned int numExtSems, stream): """ Signals a set of external semaphore objects. Enqueues a signal operation on a set of externally allocated semaphore object in the specified stream. The operations will be executed when all prior operations in the stream complete. The exact semantics of signaling a semaphore depends on the type of the object. If the semaphore object is any one of the following types: :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_FD`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32_KMT` then signaling the semaphore will set it to the signaled state. If the semaphore object is any one of the following types: :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D12_FENCE`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_FENCE`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_FD`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_WIN32` then the semaphore will be set to the value specified in :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS`::params::fence::value. If the semaphore object is of the type :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC` this API sets :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS`::params::nvSciSync::fence to a value that can be used by subsequent waiters of the same NvSciSync object to order operations with those currently submitted in `stream`. Such an update will overwrite previous contents of :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS`::params::nvSciSync::fence. By default, signaling such an external semaphore object causes appropriate memory synchronization operations to be performed over all external memory objects that are imported as :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF`. This ensures that any subsequent accesses made by other importers of the same set of NvSciBuf memory object(s) are coherent. These operations can be skipped by specifying the flag :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_SKIP_NVSCIBUF_MEMSYNC`, which can be used as a performance optimization when data coherency is not required. But specifying this flag in scenarios where data coherency is required results in undefined behavior. Also, for semaphore object of the type :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC`, if the NvSciSyncAttrList used to create the NvSciSyncObj had not set the flags in :py:obj:`~.cuDeviceGetNvSciSyncAttributes` to CUDA_NVSCISYNC_ATTR_SIGNAL, this API will return CUDA_ERROR_NOT_SUPPORTED. NvSciSyncFence associated with semaphore object of the type :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC` can be deterministic. For this the NvSciSyncAttrList used to create the semaphore object must have value of NvSciSyncAttrKey_RequireDeterministicFences key set to true. Deterministic fences allow users to enqueue a wait over the semaphore object even before corresponding signal is enqueued. For such a semaphore object, CUDA guarantees that each signal operation will increment the fence value by '1'. Users are expected to track count of signals enqueued on the semaphore object and insert waits accordingly. When such a semaphore object is signaled from multiple streams, due to concurrent stream execution, it is possible that the order in which the semaphore gets signaled is indeterministic. This could lead to waiters of the semaphore getting unblocked incorrectly. Users are expected to handle such situations, either by not using the same semaphore object with deterministic fence support enabled in different streams or by adding explicit dependency amongst such streams so that the semaphore is signaled in order. If the semaphore object is any one of the following types: :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX_KMT` then the keyed mutex will be released with the key specified in :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_PARAMS`::params::keyedmutex::key. Parameters ---------- extSemArray : List[:py:obj:`~.CUexternalSemaphore`] Set of external semaphores to be signaled paramsArray : List[:py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS`] Array of semaphore parameters numExtSems : unsigned int Number of semaphores to signal stream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to enqueue the signal operations in Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuDestroyExternalSemaphore`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef ccuda.CUstream cstream if stream is None: cstream = 0 elif isinstance(stream, (CUstream,)): pstream = int(stream) cstream = pstream else: pstream = int(CUstream(stream)) cstream = pstream paramsArray = [] if paramsArray is None else paramsArray if not all(isinstance(_x, (CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS,)) for _x in paramsArray): raise TypeError("Argument 'paramsArray' is not instance of type (expected Tuple[ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS,] or List[ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS,]") extSemArray = [] if extSemArray is None else extSemArray if not all(isinstance(_x, (CUexternalSemaphore,)) for _x in extSemArray): raise TypeError("Argument 'extSemArray' is not instance of type (expected Tuple[ccuda.CUexternalSemaphore,] or List[ccuda.CUexternalSemaphore,]") cdef ccuda.CUexternalSemaphore* cextSemArray = NULL if len(extSemArray) > 0: cextSemArray = calloc(len(extSemArray), sizeof(ccuda.CUexternalSemaphore)) if cextSemArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(extSemArray)) + 'x' + str(sizeof(ccuda.CUexternalSemaphore))) else: for idx in range(len(extSemArray)): cextSemArray[idx] = (extSemArray[idx])._ptr[0] cdef ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS* cparamsArray = NULL if len(paramsArray) > 0: cparamsArray = calloc(len(paramsArray), sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS)) if cparamsArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(paramsArray)) + 'x' + str(sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS))) for idx in range(len(paramsArray)): string.memcpy(&cparamsArray[idx], (paramsArray[idx])._ptr, sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS)) if numExtSems > len(extSemArray): raise RuntimeError("List is too small: " + str(len(extSemArray)) + " < " + str(numExtSems)) if numExtSems > len(paramsArray): raise RuntimeError("List is too small: " + str(len(paramsArray)) + " < " + str(numExtSems)) err = ccuda.cuSignalExternalSemaphoresAsync((extSemArray[0])._ptr if len(extSemArray) == 1 else cextSemArray, (paramsArray[0])._ptr if len(paramsArray) == 1 else cparamsArray, numExtSems, cstream) if cextSemArray is not NULL: free(cextSemArray) if cparamsArray is not NULL: free(cparamsArray) return (CUresult(err),) {{endif}} {{if 'cuWaitExternalSemaphoresAsync' in found_functions}} @cython.embedsignature(True) def cuWaitExternalSemaphoresAsync(extSemArray : Optional[Tuple[CUexternalSemaphore] | List[CUexternalSemaphore]], paramsArray : Optional[Tuple[CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS] | List[CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS]], unsigned int numExtSems, stream): """ Waits on a set of external semaphore objects. Enqueues a wait operation on a set of externally allocated semaphore object in the specified stream. The operations will be executed when all prior operations in the stream complete. The exact semantics of waiting on a semaphore depends on the type of the object. If the semaphore object is any one of the following types: :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_FD`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_OPAQUE_WIN32_KMT` then waiting on the semaphore will wait until the semaphore reaches the signaled state. The semaphore will then be reset to the unsignaled state. Therefore for every signal operation, there can only be one wait operation. If the semaphore object is any one of the following types: :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D12_FENCE`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_FENCE`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_FD`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_TIMELINE_SEMAPHORE_WIN32` then waiting on the semaphore will wait until the value of the semaphore is greater than or equal to :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS`::params::fence::value. If the semaphore object is of the type :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC` then, waiting on the semaphore will wait until the :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS`::params::nvSciSync::fence is signaled by the signaler of the NvSciSyncObj that was associated with this semaphore object. By default, waiting on such an external semaphore object causes appropriate memory synchronization operations to be performed over all external memory objects that are imported as :py:obj:`~.CU_EXTERNAL_MEMORY_HANDLE_TYPE_NVSCIBUF`. This ensures that any subsequent accesses made by other importers of the same set of NvSciBuf memory object(s) are coherent. These operations can be skipped by specifying the flag :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_WAIT_SKIP_NVSCIBUF_MEMSYNC`, which can be used as a performance optimization when data coherency is not required. But specifying this flag in scenarios where data coherency is required results in undefined behavior. Also, for semaphore object of the type :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_NVSCISYNC`, if the NvSciSyncAttrList used to create the NvSciSyncObj had not set the flags in :py:obj:`~.cuDeviceGetNvSciSyncAttributes` to CUDA_NVSCISYNC_ATTR_WAIT, this API will return CUDA_ERROR_NOT_SUPPORTED. If the semaphore object is any one of the following types: :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX`, :py:obj:`~.CU_EXTERNAL_SEMAPHORE_HANDLE_TYPE_D3D11_KEYED_MUTEX_KMT` then the keyed mutex will be acquired when it is released with the key specified in :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS`::params::keyedmutex::key or until the timeout specified by :py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS`::params::keyedmutex::timeoutMs has lapsed. The timeout interval can either be a finite value specified in milliseconds or an infinite value. In case an infinite value is specified the timeout never elapses. The windows INFINITE macro must be used to specify infinite timeout. Parameters ---------- extSemArray : List[:py:obj:`~.CUexternalSemaphore`] External semaphores to be waited on paramsArray : List[:py:obj:`~.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS`] Array of semaphore parameters numExtSems : unsigned int Number of semaphores to wait on stream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to enqueue the wait operations in Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_TIMEOUT` See Also -------- :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuDestroyExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync` """ cdef ccuda.CUstream cstream if stream is None: cstream = 0 elif isinstance(stream, (CUstream,)): pstream = int(stream) cstream = pstream else: pstream = int(CUstream(stream)) cstream = pstream paramsArray = [] if paramsArray is None else paramsArray if not all(isinstance(_x, (CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS,)) for _x in paramsArray): raise TypeError("Argument 'paramsArray' is not instance of type (expected Tuple[ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS,] or List[ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS,]") extSemArray = [] if extSemArray is None else extSemArray if not all(isinstance(_x, (CUexternalSemaphore,)) for _x in extSemArray): raise TypeError("Argument 'extSemArray' is not instance of type (expected Tuple[ccuda.CUexternalSemaphore,] or List[ccuda.CUexternalSemaphore,]") cdef ccuda.CUexternalSemaphore* cextSemArray = NULL if len(extSemArray) > 0: cextSemArray = calloc(len(extSemArray), sizeof(ccuda.CUexternalSemaphore)) if cextSemArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(extSemArray)) + 'x' + str(sizeof(ccuda.CUexternalSemaphore))) else: for idx in range(len(extSemArray)): cextSemArray[idx] = (extSemArray[idx])._ptr[0] cdef ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS* cparamsArray = NULL if len(paramsArray) > 0: cparamsArray = calloc(len(paramsArray), sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS)) if cparamsArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(paramsArray)) + 'x' + str(sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS))) for idx in range(len(paramsArray)): string.memcpy(&cparamsArray[idx], (paramsArray[idx])._ptr, sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS)) if numExtSems > len(extSemArray): raise RuntimeError("List is too small: " + str(len(extSemArray)) + " < " + str(numExtSems)) if numExtSems > len(paramsArray): raise RuntimeError("List is too small: " + str(len(paramsArray)) + " < " + str(numExtSems)) err = ccuda.cuWaitExternalSemaphoresAsync((extSemArray[0])._ptr if len(extSemArray) == 1 else cextSemArray, (paramsArray[0])._ptr if len(paramsArray) == 1 else cparamsArray, numExtSems, cstream) if cextSemArray is not NULL: free(cextSemArray) if cparamsArray is not NULL: free(cparamsArray) return (CUresult(err),) {{endif}} {{if 'cuDestroyExternalSemaphore' in found_functions}} @cython.embedsignature(True) def cuDestroyExternalSemaphore(extSem): """ Destroys an external semaphore. Destroys an external semaphore object and releases any references to the underlying resource. Any outstanding signals or waits must have completed before the semaphore is destroyed. Parameters ---------- extSem : :py:obj:`~.CUexternalSemaphore` External semaphore to be destroyed Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef ccuda.CUexternalSemaphore cextSem if extSem is None: cextSem = 0 elif isinstance(extSem, (CUexternalSemaphore,)): pextSem = int(extSem) cextSem = pextSem else: pextSem = int(CUexternalSemaphore(extSem)) cextSem = pextSem err = ccuda.cuDestroyExternalSemaphore(cextSem) return (CUresult(err),) {{endif}} {{if 'cuStreamWaitValue32_v2' in found_functions}} @cython.embedsignature(True) def cuStreamWaitValue32(stream, addr, value, unsigned int flags): """ Wait on a memory location. Enqueues a synchronization of the stream on the given memory location. Work ordered after the operation will block until the given condition on the memory is satisfied. By default, the condition is to wait for (int32_t)(*addr - value) >= 0, a cyclic greater-or-equal. Other condition types can be specified via `flags`. If the memory was registered via :py:obj:`~.cuMemHostRegister()`, the device pointer should be obtained with :py:obj:`~.cuMemHostGetDevicePointer()`. This function cannot be used with managed memory (:py:obj:`~.cuMemAllocManaged`). Support for CU_STREAM_WAIT_VALUE_NOR can be queried with :py:obj:`~.cuDeviceGetAttribute()` and :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_USE_STREAM_WAIT_VALUE_NOR_V2`. Parameters ---------- stream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to synchronize on the memory location. addr : :py:obj:`~.CUdeviceptr` The memory location to wait on. value : Any The value to compare with the memory location. flags : unsigned int See :py:obj:`~.CUstreamWaitValue_flags`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamWaitValue64`, :py:obj:`~.cuStreamWriteValue32`, :py:obj:`~.cuStreamWriteValue64`, :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuMemHostRegister`, :py:obj:`~.cuStreamWaitEvent` Notes ----- Warning: Improper use of this API may deadlock the application. Synchronization ordering established through this API is not visible to CUDA. CUDA tasks that are (even indirectly) ordered by this API should also have that order expressed with CUDA-visible dependencies such as events. This ensures that the scheduler does not serialize them in an improper order. """ cdef ccuda.cuuint32_t cvalue if value is None: cvalue = 0 elif isinstance(value, (cuuint32_t,)): pvalue = int(value) cvalue = pvalue else: pvalue = int(cuuint32_t(value)) cvalue = pvalue cdef ccuda.CUdeviceptr caddr if addr is None: caddr = 0 elif isinstance(addr, (CUdeviceptr,)): paddr = int(addr) caddr = paddr else: paddr = int(CUdeviceptr(addr)) caddr = paddr cdef ccuda.CUstream cstream if stream is None: cstream = 0 elif isinstance(stream, (CUstream,)): pstream = int(stream) cstream = pstream else: pstream = int(CUstream(stream)) cstream = pstream err = ccuda.cuStreamWaitValue32(cstream, caddr, cvalue, flags) return (CUresult(err),) {{endif}} {{if 'cuStreamWaitValue64_v2' in found_functions}} @cython.embedsignature(True) def cuStreamWaitValue64(stream, addr, value, unsigned int flags): """ Wait on a memory location. Enqueues a synchronization of the stream on the given memory location. Work ordered after the operation will block until the given condition on the memory is satisfied. By default, the condition is to wait for (int64_t)(*addr - value) >= 0, a cyclic greater-or-equal. Other condition types can be specified via `flags`. If the memory was registered via :py:obj:`~.cuMemHostRegister()`, the device pointer should be obtained with :py:obj:`~.cuMemHostGetDevicePointer()`. Support for this can be queried with :py:obj:`~.cuDeviceGetAttribute()` and :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS`. Parameters ---------- stream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to synchronize on the memory location. addr : :py:obj:`~.CUdeviceptr` The memory location to wait on. value : Any The value to compare with the memory location. flags : unsigned int See :py:obj:`~.CUstreamWaitValue_flags`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamWaitValue32`, :py:obj:`~.cuStreamWriteValue32`, :py:obj:`~.cuStreamWriteValue64`, :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuMemHostRegister`, :py:obj:`~.cuStreamWaitEvent` Notes ----- Warning: Improper use of this API may deadlock the application. Synchronization ordering established through this API is not visible to CUDA. CUDA tasks that are (even indirectly) ordered by this API should also have that order expressed with CUDA-visible dependencies such as events. This ensures that the scheduler does not serialize them in an improper order. """ cdef ccuda.cuuint64_t cvalue if value is None: cvalue = 0 elif isinstance(value, (cuuint64_t,)): pvalue = int(value) cvalue = pvalue else: pvalue = int(cuuint64_t(value)) cvalue = pvalue cdef ccuda.CUdeviceptr caddr if addr is None: caddr = 0 elif isinstance(addr, (CUdeviceptr,)): paddr = int(addr) caddr = paddr else: paddr = int(CUdeviceptr(addr)) caddr = paddr cdef ccuda.CUstream cstream if stream is None: cstream = 0 elif isinstance(stream, (CUstream,)): pstream = int(stream) cstream = pstream else: pstream = int(CUstream(stream)) cstream = pstream err = ccuda.cuStreamWaitValue64(cstream, caddr, cvalue, flags) return (CUresult(err),) {{endif}} {{if 'cuStreamWriteValue32_v2' in found_functions}} @cython.embedsignature(True) def cuStreamWriteValue32(stream, addr, value, unsigned int flags): """ Write a value to memory. Write a value to memory. If the memory was registered via :py:obj:`~.cuMemHostRegister()`, the device pointer should be obtained with :py:obj:`~.cuMemHostGetDevicePointer()`. This function cannot be used with managed memory (:py:obj:`~.cuMemAllocManaged`). Parameters ---------- stream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to do the write in. addr : :py:obj:`~.CUdeviceptr` The device address to write to. value : Any The value to write. flags : unsigned int See :py:obj:`~.CUstreamWriteValue_flags`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamWriteValue64`, :py:obj:`~.cuStreamWaitValue32`, :py:obj:`~.cuStreamWaitValue64`, :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuMemHostRegister`, :py:obj:`~.cuEventRecord` """ cdef ccuda.cuuint32_t cvalue if value is None: cvalue = 0 elif isinstance(value, (cuuint32_t,)): pvalue = int(value) cvalue = pvalue else: pvalue = int(cuuint32_t(value)) cvalue = pvalue cdef ccuda.CUdeviceptr caddr if addr is None: caddr = 0 elif isinstance(addr, (CUdeviceptr,)): paddr = int(addr) caddr = paddr else: paddr = int(CUdeviceptr(addr)) caddr = paddr cdef ccuda.CUstream cstream if stream is None: cstream = 0 elif isinstance(stream, (CUstream,)): pstream = int(stream) cstream = pstream else: pstream = int(CUstream(stream)) cstream = pstream err = ccuda.cuStreamWriteValue32(cstream, caddr, cvalue, flags) return (CUresult(err),) {{endif}} {{if 'cuStreamWriteValue64_v2' in found_functions}} @cython.embedsignature(True) def cuStreamWriteValue64(stream, addr, value, unsigned int flags): """ Write a value to memory. Write a value to memory. If the memory was registered via :py:obj:`~.cuMemHostRegister()`, the device pointer should be obtained with :py:obj:`~.cuMemHostGetDevicePointer()`. Support for this can be queried with :py:obj:`~.cuDeviceGetAttribute()` and :py:obj:`~.CU_DEVICE_ATTRIBUTE_CAN_USE_64_BIT_STREAM_MEM_OPS`. Parameters ---------- stream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to do the write in. addr : :py:obj:`~.CUdeviceptr` The device address to write to. value : Any The value to write. flags : unsigned int See :py:obj:`~.CUstreamWriteValue_flags`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamWriteValue32`, :py:obj:`~.cuStreamWaitValue32`, :py:obj:`~.cuStreamWaitValue64`, :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuMemHostRegister`, :py:obj:`~.cuEventRecord` """ cdef ccuda.cuuint64_t cvalue if value is None: cvalue = 0 elif isinstance(value, (cuuint64_t,)): pvalue = int(value) cvalue = pvalue else: pvalue = int(cuuint64_t(value)) cvalue = pvalue cdef ccuda.CUdeviceptr caddr if addr is None: caddr = 0 elif isinstance(addr, (CUdeviceptr,)): paddr = int(addr) caddr = paddr else: paddr = int(CUdeviceptr(addr)) caddr = paddr cdef ccuda.CUstream cstream if stream is None: cstream = 0 elif isinstance(stream, (CUstream,)): pstream = int(stream) cstream = pstream else: pstream = int(CUstream(stream)) cstream = pstream err = ccuda.cuStreamWriteValue64(cstream, caddr, cvalue, flags) return (CUresult(err),) {{endif}} {{if 'cuStreamBatchMemOp_v2' in found_functions}} @cython.embedsignature(True) def cuStreamBatchMemOp(stream, unsigned int count, paramArray : Optional[Tuple[CUstreamBatchMemOpParams] | List[CUstreamBatchMemOpParams]], unsigned int flags): """ Batch operations to synchronize the stream via memory operations. This is a batch version of :py:obj:`~.cuStreamWaitValue32()` and :py:obj:`~.cuStreamWriteValue32()`. Batching operations may avoid some performance overhead in both the API call and the device execution versus adding them to the stream in separate API calls. The operations are enqueued in the order they appear in the array. See :py:obj:`~.CUstreamBatchMemOpType` for the full set of supported operations, and :py:obj:`~.cuStreamWaitValue32()`, :py:obj:`~.cuStreamWaitValue64()`, :py:obj:`~.cuStreamWriteValue32()`, and :py:obj:`~.cuStreamWriteValue64()` for details of specific operations. See related APIs for details on querying support for specific operations. Parameters ---------- stream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` The stream to enqueue the operations in. count : unsigned int The number of operations in the array. Must be less than 256. paramArray : List[:py:obj:`~.CUstreamBatchMemOpParams`] The types and parameters of the individual operations. flags : unsigned int Reserved for future expansion; must be 0. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamWaitValue32`, :py:obj:`~.cuStreamWaitValue64`, :py:obj:`~.cuStreamWriteValue32`, :py:obj:`~.cuStreamWriteValue64`, :py:obj:`~.cuMemHostRegister` Notes ----- Warning: Improper use of this API may deadlock the application. Synchronization ordering established through this API is not visible to CUDA. CUDA tasks that are (even indirectly) ordered by this API should also have that order expressed with CUDA-visible dependencies such as events. This ensures that the scheduler does not serialize them in an improper order. For more information, see the Stream Memory Operations section in the programming guide(https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html). """ paramArray = [] if paramArray is None else paramArray if not all(isinstance(_x, (CUstreamBatchMemOpParams,)) for _x in paramArray): raise TypeError("Argument 'paramArray' is not instance of type (expected Tuple[ccuda.CUstreamBatchMemOpParams,] or List[ccuda.CUstreamBatchMemOpParams,]") cdef ccuda.CUstream cstream if stream is None: cstream = 0 elif isinstance(stream, (CUstream,)): pstream = int(stream) cstream = pstream else: pstream = int(CUstream(stream)) cstream = pstream if count > len(paramArray): raise RuntimeError("List is too small: " + str(len(paramArray)) + " < " + str(count)) cdef ccuda.CUstreamBatchMemOpParams* cparamArray = NULL if len(paramArray) > 0: cparamArray = calloc(len(paramArray), sizeof(ccuda.CUstreamBatchMemOpParams)) if cparamArray is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(paramArray)) + 'x' + str(sizeof(ccuda.CUstreamBatchMemOpParams))) for idx in range(len(paramArray)): string.memcpy(&cparamArray[idx], (paramArray[idx])._ptr, sizeof(ccuda.CUstreamBatchMemOpParams)) err = ccuda.cuStreamBatchMemOp(cstream, count, (paramArray[0])._ptr if len(paramArray) == 1 else cparamArray, flags) if cparamArray is not NULL: free(cparamArray) return (CUresult(err),) {{endif}} {{if 'cuFuncGetAttribute' in found_functions}} @cython.embedsignature(True) def cuFuncGetAttribute(attrib not None : CUfunction_attribute, hfunc): """ Returns information about a function. Returns in `*pi` the integer value of the attribute `attrib` on the kernel given by `hfunc`. The supported attributes are: - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK`: The maximum number of threads per block, beyond which a launch of the function would fail. This number depends on both the function and the device on which the function is currently loaded. - :py:obj:`~.CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES`: The size in bytes of statically-allocated shared memory per block required by this function. This does not include dynamically-allocated shared memory requested by the user at runtime. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CONST_SIZE_BYTES`: The size in bytes of user-allocated constant memory required by this function. - :py:obj:`~.CU_FUNC_ATTRIBUTE_LOCAL_SIZE_BYTES`: The size in bytes of local memory used by each thread of this function. - :py:obj:`~.CU_FUNC_ATTRIBUTE_NUM_REGS`: The number of registers used by each thread of this function. - :py:obj:`~.CU_FUNC_ATTRIBUTE_PTX_VERSION`: The PTX virtual architecture version for which the function was compiled. This value is the major PTX version * 10 - the minor PTX version, so a PTX version 1.3 function would return the value 13. Note that this may return the undefined value of 0 for cubins compiled prior to CUDA 3.0. - :py:obj:`~.CU_FUNC_ATTRIBUTE_BINARY_VERSION`: The binary architecture version for which the function was compiled. This value is the major binary version * 10 + the minor binary version, so a binary version 1.3 function would return the value 13. Note that this will return a value of 10 for legacy cubins that do not have a properly-encoded binary architecture version. - :py:obj:`~.CU_FUNC_CACHE_MODE_CA`: The attribute to indicate whether the function has been compiled with user specified option "-Xptxas --dlcm=ca" set . - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES`: The maximum size in bytes of dynamically-allocated shared memory. - :py:obj:`~.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT`: Preferred shared memory-L1 cache split ratio in percent of total shared memory. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CLUSTER_SIZE_MUST_BE_SET`: If this attribute is set, the kernel must launch with a valid cluster size specified. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_WIDTH`: The required cluster width in blocks. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_HEIGHT`: The required cluster height in blocks. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_DEPTH`: The required cluster depth in blocks. - :py:obj:`~.CU_FUNC_ATTRIBUTE_NON_PORTABLE_CLUSTER_SIZE_ALLOWED`: Indicates whether the function can be launched with non-portable cluster size. 1 is allowed, 0 is disallowed. A non-portable cluster size may only function on the specific SKUs the program is tested on. The launch might fail if the program is run on a different hardware platform. CUDA API provides cudaOccupancyMaxActiveClusters to assist with checking whether the desired size can be launched on the current device. A portable cluster size is guaranteed to be functional on all compute capabilities higher than the target compute capability. The portable cluster size for sm_90 is 8 blocks per cluster. This value may increase for future compute capabilities. The specific hardware unit may support higher cluster sizes that’s not guaranteed to be portable. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE`: The block scheduling policy of a function. The value type is CUclusterSchedulingPolicy. With a few execeptions, function attributes may also be queried on unloaded function handles returned from :py:obj:`~.cuModuleEnumerateFunctions`. :py:obj:`~.CUDA_ERROR_FUNCTION_NOT_LOADED` is returned if the attribute requires a fully loaded function but the function is not loaded. The loading state of a function may be queried using :py:obj:`~.cuFuncIsloaded`. :py:obj:`~.cuFuncLoad` may be called to explicitly load a function before querying the following attributes that require the function to be loaded: - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_THREADS_PER_BLOCK` - :py:obj:`~.CU_FUNC_ATTRIBUTE_CONST_SIZE_BYTES` - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES` Parameters ---------- attrib : :py:obj:`~.CUfunction_attribute` Attribute requested hfunc : :py:obj:`~.CUfunction` Function to query attribute of Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_FUNCTION_NOT_LOADED` pi : int Returned attribute value See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cudaFuncGetAttributes`, :py:obj:`~.cudaFuncSetAttribute`, :py:obj:`~.cuFuncIsLoaded`, :py:obj:`~.cuFuncLoad`, :py:obj:`~.cuKernelGetAttribute` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc cdef int pi = 0 cdef ccuda.CUfunction_attribute cattrib = attrib.value err = ccuda.cuFuncGetAttribute(&pi, cattrib, chfunc) return (CUresult(err), pi) {{endif}} {{if 'cuFuncSetAttribute' in found_functions}} @cython.embedsignature(True) def cuFuncSetAttribute(hfunc, attrib not None : CUfunction_attribute, int value): """ Sets information about a function. This call sets the value of a specified attribute `attrib` on the kernel given by `hfunc` to an integer value specified by `val` This function returns CUDA_SUCCESS if the new value of the attribute could be successfully set. If the set fails, this call will return an error. Not all attributes can have values set. Attempting to set a value on a read-only attribute will result in an error (CUDA_ERROR_INVALID_VALUE) Supported attributes for the cuFuncSetAttribute call are: - :py:obj:`~.CU_FUNC_ATTRIBUTE_MAX_DYNAMIC_SHARED_SIZE_BYTES`: This maximum size in bytes of dynamically-allocated shared memory. The value should contain the requested maximum size of dynamically- allocated shared memory. The sum of this value and the function attribute :py:obj:`~.CU_FUNC_ATTRIBUTE_SHARED_SIZE_BYTES` cannot exceed the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_BLOCK_OPTIN`. The maximal size of requestable dynamic shared memory may differ by GPU architecture. - :py:obj:`~.CU_FUNC_ATTRIBUTE_PREFERRED_SHARED_MEMORY_CARVEOUT`: On devices where the L1 cache and shared memory use the same hardware resources, this sets the shared memory carveout preference, in percent of the total shared memory. See :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAX_SHARED_MEMORY_PER_MULTIPROCESSOR` This is only a hint, and the driver can choose a different ratio if required to execute the function. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_WIDTH`: The required cluster width in blocks. The width, height, and depth values must either all be 0 or all be positive. The validity of the cluster dimensions is checked at launch time. If the value is set during compile time, it cannot be set at runtime. Setting it at runtime will return CUDA_ERROR_NOT_PERMITTED. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_HEIGHT`: The required cluster height in blocks. The width, height, and depth values must either all be 0 or all be positive. The validity of the cluster dimensions is checked at launch time. If the value is set during compile time, it cannot be set at runtime. Setting it at runtime will return CUDA_ERROR_NOT_PERMITTED. - :py:obj:`~.CU_FUNC_ATTRIBUTE_REQUIRED_CLUSTER_DEPTH`: The required cluster depth in blocks. The width, height, and depth values must either all be 0 or all be positive. The validity of the cluster dimensions is checked at launch time. If the value is set during compile time, it cannot be set at runtime. Setting it at runtime will return CUDA_ERROR_NOT_PERMITTED. - :py:obj:`~.CU_FUNC_ATTRIBUTE_NON_PORTABLE_CLUSTER_SIZE_ALLOWED`: Indicates whether the function can be launched with non-portable cluster size. 1 is allowed, 0 is disallowed. - :py:obj:`~.CU_FUNC_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE`: The block scheduling policy of a function. The value type is CUclusterSchedulingPolicy. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Function to query attribute of attrib : :py:obj:`~.CUfunction_attribute` Attribute requested value : int The value to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cudaFuncGetAttributes`, :py:obj:`~.cudaFuncSetAttribute`, :py:obj:`~.cuKernelSetAttribute` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc cdef ccuda.CUfunction_attribute cattrib = attrib.value err = ccuda.cuFuncSetAttribute(chfunc, cattrib, value) return (CUresult(err),) {{endif}} {{if 'cuFuncSetCacheConfig' in found_functions}} @cython.embedsignature(True) def cuFuncSetCacheConfig(hfunc, config not None : CUfunc_cache): """ Sets the preferred cache configuration for a device function. On devices where the L1 cache and shared memory use the same hardware resources, this sets through `config` the preferred cache configuration for the device function `hfunc`. This is only a preference. The driver will use the requested configuration if possible, but it is free to choose a different configuration if required to execute `hfunc`. Any context-wide preference set via :py:obj:`~.cuCtxSetCacheConfig()` will be overridden by this per-function setting unless the per-function setting is :py:obj:`~.CU_FUNC_CACHE_PREFER_NONE`. In that case, the current context-wide setting will be used. This setting does nothing on devices where the size of the L1 cache and shared memory are fixed. Launching a kernel with a different preference than the most recent preference setting may insert a device-side synchronization point. The supported cache configurations are: - :py:obj:`~.CU_FUNC_CACHE_PREFER_NONE`: no preference for shared memory or L1 (default) - :py:obj:`~.CU_FUNC_CACHE_PREFER_SHARED`: prefer larger shared memory and smaller L1 cache - :py:obj:`~.CU_FUNC_CACHE_PREFER_L1`: prefer larger L1 cache and smaller shared memory - :py:obj:`~.CU_FUNC_CACHE_PREFER_EQUAL`: prefer equal sized L1 cache and shared memory Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to configure cache for config : :py:obj:`~.CUfunc_cache` Requested cache configuration Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cudaFuncSetCacheConfig`, :py:obj:`~.cuKernelSetCacheConfig` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc cdef ccuda.CUfunc_cache cconfig = config.value err = ccuda.cuFuncSetCacheConfig(chfunc, cconfig) return (CUresult(err),) {{endif}} {{if 'cuFuncGetModule' in found_functions}} @cython.embedsignature(True) def cuFuncGetModule(hfunc): """ Returns a module handle. Returns in `*hmod` the handle of the module that function `hfunc` is located in. The lifetime of the module corresponds to the lifetime of the context it was loaded in or until the module is explicitly unloaded. The CUDA runtime manages its own modules loaded into the primary context. If the handle returned by this API refers to a module loaded by the CUDA runtime, calling :py:obj:`~.cuModuleUnload()` on that module will result in undefined behavior. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Function to retrieve module for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` hmod : :py:obj:`~.CUmodule` Returned module handle """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc cdef CUmodule hmod = CUmodule() err = ccuda.cuFuncGetModule(hmod._ptr, chfunc) return (CUresult(err), hmod) {{endif}} {{if 'cuFuncGetName' in found_functions}} @cython.embedsignature(True) def cuFuncGetName(hfunc): """ Returns the function name for a :py:obj:`~.CUfunction` handle. Returns in `**name` the function name associated with the function handle `hfunc` . The function name is returned as a null-terminated string. The returned name is only valid when the function handle is valid. If the module is unloaded or reloaded, one must call the API again to get the updated name. This API may return a mangled name if the function is not declared as having C linkage. If either `**name` or `hfunc` is NULL, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. Parameters ---------- hfunc : :py:obj:`~.CUfunction` The function handle to retrieve the name for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, name : bytes The returned name of the function """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc cdef const char* name = NULL err = ccuda.cuFuncGetName(&name, chfunc) return (CUresult(err), name) {{endif}} {{if 'cuFuncGetParamInfo' in found_functions}} @cython.embedsignature(True) def cuFuncGetParamInfo(func, size_t paramIndex): """ Returns the offset and size of a kernel parameter in the device-side parameter layout. Queries the kernel parameter at `paramIndex` into `func's` list of parameters, and returns in `paramOffset` and `paramSize` the offset and size, respectively, where the parameter will reside in the device-side parameter layout. This information can be used to update kernel node parameters from the device via :py:obj:`~.cudaGraphKernelNodeSetParam()` and :py:obj:`~.cudaGraphKernelNodeUpdatesApply()`. `paramIndex` must be less than the number of parameters that `func` takes. `paramSize` can be set to NULL if only the parameter offset is desired. Parameters ---------- func : :py:obj:`~.CUfunction` The function to query paramIndex : size_t The parameter index to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, paramOffset : int Returns the offset into the device-side parameter layout at which the parameter resides paramSize : int Optionally returns the size of the parameter in the device-side parameter layout See Also -------- :py:obj:`~.cuKernelGetParamInfo` """ cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef size_t paramOffset = 0 cdef size_t paramSize = 0 err = ccuda.cuFuncGetParamInfo(cfunc, paramIndex, ¶mOffset, ¶mSize) return (CUresult(err), paramOffset, paramSize) {{endif}} {{if 'cuFuncIsLoaded' in found_functions}} @cython.embedsignature(True) def cuFuncIsLoaded(function): """ Returns if the function is loaded. Returns in `state` the loading state of `function`. Parameters ---------- function : :py:obj:`~.CUfunction` the function to check Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` state : :py:obj:`~.CUfunctionLoadingState` returned loading state See Also -------- :py:obj:`~.cuFuncLoad`, :py:obj:`~.cuModuleEnumerateFunctions` """ cdef ccuda.CUfunction cfunction if function is None: cfunction = 0 elif isinstance(function, (CUfunction,)): pfunction = int(function) cfunction = pfunction else: pfunction = int(CUfunction(function)) cfunction = pfunction cdef ccuda.CUfunctionLoadingState state err = ccuda.cuFuncIsLoaded(&state, cfunction) return (CUresult(err), CUfunctionLoadingState(state)) {{endif}} {{if 'cuFuncLoad' in found_functions}} @cython.embedsignature(True) def cuFuncLoad(function): """ Loads a function. Finalizes function loading for `function`. Calling this API with a fully loaded function has no effect. Parameters ---------- function : :py:obj:`~.CUfunction` the function to load Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuModuleEnumerateFunctions`, :py:obj:`~.cuFuncIsLoaded` """ cdef ccuda.CUfunction cfunction if function is None: cfunction = 0 elif isinstance(function, (CUfunction,)): pfunction = int(function) cfunction = pfunction else: pfunction = int(CUfunction(function)) cfunction = pfunction err = ccuda.cuFuncLoad(cfunction) return (CUresult(err),) {{endif}} {{if 'cuLaunchKernel' in found_functions}} @cython.embedsignature(True) def cuLaunchKernel(f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gridDimZ, unsigned int blockDimX, unsigned int blockDimY, unsigned int blockDimZ, unsigned int sharedMemBytes, hStream, kernelParams, void_ptr extra): """ Launches a CUDA function :py:obj:`~.CUfunction` or a CUDA kernel :py:obj:`~.CUkernel`. Invokes the function :py:obj:`~.CUfunction` or the kernel :py:obj:`~.CUkernel` `f` on a `gridDimX` x `gridDimY` x `gridDimZ` grid of blocks. Each block contains `blockDimX` x `blockDimY` x `blockDimZ` threads. `sharedMemBytes` sets the amount of dynamic shared memory that will be available to each thread block. Kernel parameters to `f` can be specified in one of two ways: 1) Kernel parameters can be specified via `kernelParams`. If `f` has N parameters, then `kernelParams` needs to be an array of N pointers. Each of `kernelParams`[0] through `kernelParams`[N-1] must point to a region of memory from which the actual kernel parameter will be copied. The number of kernel parameters and their offsets and sizes do not need to be specified as that information is retrieved directly from the kernel's image. 2) Kernel parameters can also be packaged by the application into a single buffer that is passed in via the `extra` parameter. This places the burden on the application of knowing each kernel parameter's size and alignment/padding within the buffer. Here is an example of using the `extra` parameter in this manner: **View CUDA Toolkit Documentation for a C++ code example** The `extra` parameter exists to allow :py:obj:`~.cuLaunchKernel` to take additional less commonly used arguments. `extra` specifies a list of names of extra settings and their corresponding values. Each extra setting name is immediately followed by the corresponding value. The list must be terminated with either NULL or :py:obj:`~.CU_LAUNCH_PARAM_END`. - :py:obj:`~.CU_LAUNCH_PARAM_END`, which indicates the end of the `extra` array; - :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_POINTER`, which specifies that the next value in `extra` will be a pointer to a buffer containing all the kernel parameters for launching kernel `f`; - :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_SIZE`, which specifies that the next value in `extra` will be a pointer to a size_t containing the size of the buffer specified with :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_POINTER`; The error :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned if kernel parameters are specified with both `kernelParams` and `extra` (i.e. both `kernelParams` and `extra` are non-NULL). Calling :py:obj:`~.cuLaunchKernel()` invalidates the persistent function state set through the following deprecated APIs: :py:obj:`~.cuFuncSetBlockShape()`, :py:obj:`~.cuFuncSetSharedSize()`, :py:obj:`~.cuParamSetSize()`, :py:obj:`~.cuParamSeti()`, :py:obj:`~.cuParamSetf()`, :py:obj:`~.cuParamSetv()`. Note that to use :py:obj:`~.cuLaunchKernel()`, the kernel `f` must either have been compiled with toolchain version 3.2 or later so that it will contain kernel parameter information, or have no kernel parameters. If either of these conditions is not met, then :py:obj:`~.cuLaunchKernel()` will return :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`. Note that the API can also be used to launch context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to launch the kernel on will either be taken from the specified stream `hStream` or the current context in case of NULL stream. Parameters ---------- f : :py:obj:`~.CUfunction` Function :py:obj:`~.CUfunction` or Kernel :py:obj:`~.CUkernel` to launch gridDimX : unsigned int Width of grid in blocks gridDimY : unsigned int Height of grid in blocks gridDimZ : unsigned int Depth of grid in blocks blockDimX : unsigned int X dimension of each thread block blockDimY : unsigned int Y dimension of each thread block blockDimZ : unsigned int Z dimension of each thread block sharedMemBytes : unsigned int Dynamic shared-memory size per thread block in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier kernelParams : Any Array of pointers to kernel parameters extra : List[Any] Extra options Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_LAUNCH_FAILED`, :py:obj:`~.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, :py:obj:`~.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cudaLaunchKernel`, :py:obj:`~.cuLibraryGetKernel`, :py:obj:`~.cuKernelSetCacheConfig`, :py:obj:`~.cuKernelGetAttribute`, :py:obj:`~.cuKernelSetAttribute` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUfunction cf if f is None: cf = 0 elif isinstance(f, (CUfunction,)): pf = int(f) cf = pf else: pf = int(CUfunction(f)) cf = pf ckernelParams = utils.HelperKernelParams(kernelParams) err = ccuda.cuLaunchKernel(cf, gridDimX, gridDimY, gridDimZ, blockDimX, blockDimY, blockDimZ, sharedMemBytes, chStream, ckernelParams.ckernelParams, extra) return (CUresult(err),) {{endif}} {{if 'cuLaunchKernelEx' in found_functions}} @cython.embedsignature(True) def cuLaunchKernelEx(config : Optional[CUlaunchConfig], f, kernelParams, void_ptr extra): """ Launches a CUDA function :py:obj:`~.CUfunction` or a CUDA kernel :py:obj:`~.CUkernel` with launch-time configuration. Invokes the function :py:obj:`~.CUfunction` or the kernel :py:obj:`~.CUkernel` `f` with the specified launch-time configuration `config`. The :py:obj:`~.CUlaunchConfig` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.CUlaunchConfig.gridDimX` is the width of the grid in blocks. - :py:obj:`~.CUlaunchConfig.gridDimY` is the height of the grid in blocks. - :py:obj:`~.CUlaunchConfig.gridDimZ` is the depth of the grid in blocks. - :py:obj:`~.CUlaunchConfig.blockDimX` is the X dimension of each thread block. - :py:obj:`~.CUlaunchConfig.blockDimX` is the Y dimension of each thread block. - :py:obj:`~.CUlaunchConfig.blockDimZ` is the Z dimension of each thread block. - :py:obj:`~.CUlaunchConfig.sharedMemBytes` is the dynamic shared- memory size per thread block in bytes. - :py:obj:`~.CUlaunchConfig.hStream` is the handle to the stream to perform the launch in. The CUDA context associated with this stream must match that associated with function f. - :py:obj:`~.CUlaunchConfig.attrs` is an array of :py:obj:`~.CUlaunchConfig.numAttrs` continguous :py:obj:`~.CUlaunchAttribute` elements. The value of this pointer is not considered if :py:obj:`~.CUlaunchConfig.numAttrs` is zero. However, in that case, it is recommended to set the pointer to NULL. - :py:obj:`~.CUlaunchConfig.numAttrs` is the number of attributes populating the first :py:obj:`~.CUlaunchConfig.numAttrs` positions of the :py:obj:`~.CUlaunchConfig.attrs` array. Launch-time configuration is specified by adding entries to :py:obj:`~.CUlaunchConfig.attrs`. Each entry is an attribute ID and a corresponding attribute value. The :py:obj:`~.CUlaunchAttribute` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.CUlaunchAttribute.id` is a unique enum identifying the attribute. - :py:obj:`~.CUlaunchAttribute.value` is a union that hold the attribute value. An example of using the `config` parameter: **View CUDA Toolkit Documentation for a C++ code example** The :py:obj:`~.CUlaunchAttributeID` enum is defined as: **View CUDA Toolkit Documentation for a C++ code example** and the corresponding :py:obj:`~.CUlaunchAttributeValue` union as : **View CUDA Toolkit Documentation for a C++ code example** Setting :py:obj:`~.CU_LAUNCH_ATTRIBUTE_COOPERATIVE` to a non-zero value causes the kernel launch to be a cooperative launch, with exactly the same usage and semantics of :py:obj:`~.cuLaunchCooperativeKernel`. Setting :py:obj:`~.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_STREAM_SERIALIZATION` to a non-zero values causes the kernel to use programmatic means to resolve its stream dependency -- enabling the CUDA runtime to opportunistically allow the grid's execution to overlap with the previous kernel in the stream, if that kernel requests the overlap. :py:obj:`~.CU_LAUNCH_ATTRIBUTE_PROGRAMMATIC_EVENT` records an event along with the kernel launch. Event recorded through this launch attribute is guaranteed to only trigger after all block in the associated kernel trigger the event. A block can trigger the event through PTX launchdep.release or CUDA builtin function cudaTriggerProgrammaticLaunchCompletion(). A trigger can also be inserted at the beginning of each block's execution if triggerAtBlockStart is set to non-0. Note that dependents (including the CPU thread calling :py:obj:`~.cuEventSynchronize()`) are not guaranteed to observe the release precisely when it is released. For example, :py:obj:`~.cuEventSynchronize()` may only observe the event trigger long after the associated kernel has completed. This recording type is primarily meant for establishing programmatic dependency between device tasks. The event supplied must not be an interprocess or interop event. The event must disable timing (i.e. created with :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). :py:obj:`~.CU_LAUNCH_ATTRIBUTE_LAUNCH_COMPLETION_EVENT` records an event along with the kernel launch. Nominally, the event is triggered once all blocks of the kernel have begun execution. Currently this is a best effort. If a kernel B has a launch completion dependency on a kernel A, B may wait until A is complete. Alternatively, blocks of B may begin before all blocks of A have begun, for example: - If B can claim execution resources unavaiable to A, for example if they run on different GPUs. - If B is a higher priority than A. Exercise caution if such an ordering inversion could lead to deadlock. The event supplied must not be an interprocess or interop event. The event must disable timing (i.e. must be created with the :py:obj:`~.CU_EVENT_DISABLE_TIMING` flag set). Setting :py:obj:`~.CU_LAUNCH_ATTRIBUTE_DEVICE_UPDATABLE_KERNEL_NODE` to 1 on a captured launch causes the resulting kernel node to be device- updatable. This attribute is specific to graphs, and passing it to a launch in a non-capturing stream results in an error. Passing a value other than 0 or 1 is not allowed. On success, a handle will be returned via :py:obj:`~.CUlaunchAttributeValue`::deviceUpdatableKernelNode::devNode which can be passed to the various device-side update functions to update the node's kernel parameters from within another kernel. For more information on the types of device updates that can be made, as well as the relevant limitations thereof, see :py:obj:`~.cudaGraphKernelNodeUpdatesApply`. Kernel nodes which are device-updatable have additional restrictions compared to regular kernel nodes. Firstly, device-updatable nodes cannot be removed from their graph via :py:obj:`~.cuGraphDestroyNode`. Additionally, once opted-in to this functionality, a node cannot opt out, and any attempt to set the attribute to 0 will result in an error. Graphs containing one or more device-updatable node also do not allow multiple instantiation. The effect of other attributes is consistent with their effect when set via persistent APIs. See :py:obj:`~.cuStreamSetAttribute` for - :py:obj:`~.CU_LAUNCH_ATTRIBUTE_ACCESS_POLICY_WINDOW` - :py:obj:`~.CU_LAUNCH_ATTRIBUTE_SYNCHRONIZATION_POLICY` See :py:obj:`~.cuFuncSetAttribute` for - :py:obj:`~.CU_LAUNCH_ATTRIBUTE_CLUSTER_DIMENSION` - :py:obj:`~.CU_LAUNCH_ATTRIBUTE_CLUSTER_SCHEDULING_POLICY_PREFERENCE` Kernel parameters to `f` can be specified in the same ways that they can be using :py:obj:`~.cuLaunchKernel`. Note that the API can also be used to launch context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to launch the kernel on will either be taken from the specified stream :py:obj:`~.CUlaunchConfig.hStream` or the current context in case of NULL stream. Parameters ---------- config : :py:obj:`~.CUlaunchConfig` Config to launch f : :py:obj:`~.CUfunction` Function :py:obj:`~.CUfunction` or Kernel :py:obj:`~.CUkernel` to launch kernelParams : Any Array of pointers to kernel parameters extra : List[Any] Extra options Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_LAUNCH_FAILED`, :py:obj:`~.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, :py:obj:`~.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING`, :py:obj:`~.CUDA_ERROR_COOPERATIVE_LAUNCH_TOO_LARGE`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cudaLaunchKernel`, :py:obj:`~.cudaLaunchKernelEx`, :py:obj:`~.cuLibraryGetKernel`, :py:obj:`~.cuKernelSetCacheConfig`, :py:obj:`~.cuKernelGetAttribute`, :py:obj:`~.cuKernelSetAttribute` """ cdef ccuda.CUfunction cf if f is None: cf = 0 elif isinstance(f, (CUfunction,)): pf = int(f) cf = pf else: pf = int(CUfunction(f)) cf = pf cdef ccuda.CUlaunchConfig* cconfig_ptr = config._ptr if config != None else NULL ckernelParams = utils.HelperKernelParams(kernelParams) err = ccuda.cuLaunchKernelEx(cconfig_ptr, cf, ckernelParams.ckernelParams, extra) return (CUresult(err),) {{endif}} {{if 'cuLaunchCooperativeKernel' in found_functions}} @cython.embedsignature(True) def cuLaunchCooperativeKernel(f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gridDimZ, unsigned int blockDimX, unsigned int blockDimY, unsigned int blockDimZ, unsigned int sharedMemBytes, hStream, kernelParams): """ Launches a CUDA function :py:obj:`~.CUfunction` or a CUDA kernel :py:obj:`~.CUkernel` where thread blocks can cooperate and synchronize as they execute. Invokes the function :py:obj:`~.CUfunction` or the kernel :py:obj:`~.CUkernel` `f` on a `gridDimX` x `gridDimY` x `gridDimZ` grid of blocks. Each block contains `blockDimX` x `blockDimY` x `blockDimZ` threads. `sharedMemBytes` sets the amount of dynamic shared memory that will be available to each thread block. The device on which this kernel is invoked must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_COOPERATIVE_LAUNCH`. The total number of blocks launched cannot exceed the maximum number of blocks per multiprocessor as returned by :py:obj:`~.cuOccupancyMaxActiveBlocksPerMultiprocessor` (or :py:obj:`~.cuOccupancyMaxActiveBlocksPerMultiprocessorWithFlags`) times the number of multiprocessors as specified by the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT`. The kernel cannot make use of CUDA dynamic parallelism. Kernel parameters must be specified via `kernelParams`. If `f` has N parameters, then `kernelParams` needs to be an array of N pointers. Each of `kernelParams`[0] through `kernelParams`[N-1] must point to a region of memory from which the actual kernel parameter will be copied. The number of kernel parameters and their offsets and sizes do not need to be specified as that information is retrieved directly from the kernel's image. Calling :py:obj:`~.cuLaunchCooperativeKernel()` sets persistent function state that is the same as function state set through :py:obj:`~.cuLaunchKernel` API When the kernel `f` is launched via :py:obj:`~.cuLaunchCooperativeKernel()`, the previous block shape, shared size and parameter info associated with `f` is overwritten. Note that to use :py:obj:`~.cuLaunchCooperativeKernel()`, the kernel `f` must either have been compiled with toolchain version 3.2 or later so that it will contain kernel parameter information, or have no kernel parameters. If either of these conditions is not met, then :py:obj:`~.cuLaunchCooperativeKernel()` will return :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`. Note that the API can also be used to launch context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to launch the kernel on will either be taken from the specified stream `hStream` or the current context in case of NULL stream. Parameters ---------- f : :py:obj:`~.CUfunction` Function :py:obj:`~.CUfunction` or Kernel :py:obj:`~.CUkernel` to launch gridDimX : unsigned int Width of grid in blocks gridDimY : unsigned int Height of grid in blocks gridDimZ : unsigned int Depth of grid in blocks blockDimX : unsigned int X dimension of each thread block blockDimY : unsigned int Y dimension of each thread block blockDimZ : unsigned int Z dimension of each thread block sharedMemBytes : unsigned int Dynamic shared-memory size per thread block in bytes hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier kernelParams : Any Array of pointers to kernel parameters Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_LAUNCH_FAILED`, :py:obj:`~.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, :py:obj:`~.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING`, :py:obj:`~.CUDA_ERROR_COOPERATIVE_LAUNCH_TOO_LARGE`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED`, :py:obj:`~.CUDA_ERROR_NOT_FOUND` See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuLaunchCooperativeKernelMultiDevice`, :py:obj:`~.cudaLaunchCooperativeKernel`, :py:obj:`~.cuLibraryGetKernel`, :py:obj:`~.cuKernelSetCacheConfig`, :py:obj:`~.cuKernelGetAttribute`, :py:obj:`~.cuKernelSetAttribute` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUfunction cf if f is None: cf = 0 elif isinstance(f, (CUfunction,)): pf = int(f) cf = pf else: pf = int(CUfunction(f)) cf = pf ckernelParams = utils.HelperKernelParams(kernelParams) err = ccuda.cuLaunchCooperativeKernel(cf, gridDimX, gridDimY, gridDimZ, blockDimX, blockDimY, blockDimZ, sharedMemBytes, chStream, ckernelParams.ckernelParams) return (CUresult(err),) {{endif}} {{if 'cuLaunchCooperativeKernelMultiDevice' in found_functions}} @cython.embedsignature(True) def cuLaunchCooperativeKernelMultiDevice(launchParamsList : Optional[Tuple[CUDA_LAUNCH_PARAMS] | List[CUDA_LAUNCH_PARAMS]], unsigned int numDevices, unsigned int flags): """ Launches CUDA functions on multiple devices where thread blocks can cooperate and synchronize as they execute. [Deprecated] Invokes kernels as specified in the `launchParamsList` array where each element of the array specifies all the parameters required to perform a single kernel launch. These kernels can cooperate and synchronize as they execute. The size of the array is specified by `numDevices`. No two kernels can be launched on the same device. All the devices targeted by this multi-device launch must be identical. All devices must have a non-zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_COOPERATIVE_MULTI_DEVICE_LAUNCH`. All kernels launched must be identical with respect to the compiled code. Note that any device, constant or managed variables present in the module that owns the kernel launched on each device, are independently instantiated on every device. It is the application's responsibility to ensure these variables are initialized and used appropriately. The size of the grids as specified in blocks, the size of the blocks themselves and the amount of shared memory used by each thread block must also match across all launched kernels. The streams used to launch these kernels must have been created via either :py:obj:`~.cuStreamCreate` or :py:obj:`~.cuStreamCreateWithPriority`. The NULL stream or :py:obj:`~.CU_STREAM_LEGACY` or :py:obj:`~.CU_STREAM_PER_THREAD` cannot be used. The total number of blocks launched per kernel cannot exceed the maximum number of blocks per multiprocessor as returned by :py:obj:`~.cuOccupancyMaxActiveBlocksPerMultiprocessor` (or :py:obj:`~.cuOccupancyMaxActiveBlocksPerMultiprocessorWithFlags`) times the number of multiprocessors as specified by the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_MULTIPROCESSOR_COUNT`. Since the total number of blocks launched per device has to match across all devices, the maximum number of blocks that can be launched per device will be limited by the device with the least number of multiprocessors. The kernels cannot make use of CUDA dynamic parallelism. The :py:obj:`~.CUDA_LAUNCH_PARAMS` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.CUDA_LAUNCH_PARAMS.function` specifies the kernel to be launched. All functions must be identical with respect to the compiled code. Note that you can also specify context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then casting to :py:obj:`~.CUfunction`. In this case, the context to launch the kernel on be taken from the specified stream :py:obj:`~.CUDA_LAUNCH_PARAMS.hStream`. - :py:obj:`~.CUDA_LAUNCH_PARAMS.gridDimX` is the width of the grid in blocks. This must match across all kernels launched. - :py:obj:`~.CUDA_LAUNCH_PARAMS.gridDimY` is the height of the grid in blocks. This must match across all kernels launched. - :py:obj:`~.CUDA_LAUNCH_PARAMS.gridDimZ` is the depth of the grid in blocks. This must match across all kernels launched. - :py:obj:`~.CUDA_LAUNCH_PARAMS.blockDimX` is the X dimension of each thread block. This must match across all kernels launched. - :py:obj:`~.CUDA_LAUNCH_PARAMS.blockDimX` is the Y dimension of each thread block. This must match across all kernels launched. - :py:obj:`~.CUDA_LAUNCH_PARAMS.blockDimZ` is the Z dimension of each thread block. This must match across all kernels launched. - :py:obj:`~.CUDA_LAUNCH_PARAMS.sharedMemBytes` is the dynamic shared- memory size per thread block in bytes. This must match across all kernels launched. - :py:obj:`~.CUDA_LAUNCH_PARAMS.hStream` is the handle to the stream to perform the launch in. This cannot be the NULL stream or :py:obj:`~.CU_STREAM_LEGACY` or :py:obj:`~.CU_STREAM_PER_THREAD`. The CUDA context associated with this stream must match that associated with :py:obj:`~.CUDA_LAUNCH_PARAMS.function`. - :py:obj:`~.CUDA_LAUNCH_PARAMS.kernelParams` is an array of pointers to kernel parameters. If :py:obj:`~.CUDA_LAUNCH_PARAMS.function` has N parameters, then :py:obj:`~.CUDA_LAUNCH_PARAMS.kernelParams` needs to be an array of N pointers. Each of :py:obj:`~.CUDA_LAUNCH_PARAMS.kernelParams`[0] through :py:obj:`~.CUDA_LAUNCH_PARAMS.kernelParams`[N-1] must point to a region of memory from which the actual kernel parameter will be copied. The number of kernel parameters and their offsets and sizes do not need to be specified as that information is retrieved directly from the kernel's image. By default, the kernel won't begin execution on any GPU until all prior work in all the specified streams has completed. This behavior can be overridden by specifying the flag :py:obj:`~.CUDA_COOPERATIVE_LAUNCH_MULTI_DEVICE_NO_PRE_LAUNCH_SYNC`. When this flag is specified, each kernel will only wait for prior work in the stream corresponding to that GPU to complete before it begins execution. Similarly, by default, any subsequent work pushed in any of the specified streams will not begin execution until the kernels on all GPUs have completed. This behavior can be overridden by specifying the flag :py:obj:`~.CUDA_COOPERATIVE_LAUNCH_MULTI_DEVICE_NO_POST_LAUNCH_SYNC`. When this flag is specified, any subsequent work pushed in any of the specified streams will only wait for the kernel launched on the GPU corresponding to that stream to complete before it begins execution. Calling :py:obj:`~.cuLaunchCooperativeKernelMultiDevice()` sets persistent function state that is the same as function state set through :py:obj:`~.cuLaunchKernel` API when called individually for each element in `launchParamsList`. When kernels are launched via :py:obj:`~.cuLaunchCooperativeKernelMultiDevice()`, the previous block shape, shared size and parameter info associated with each :py:obj:`~.CUDA_LAUNCH_PARAMS.function` in `launchParamsList` is overwritten. Note that to use :py:obj:`~.cuLaunchCooperativeKernelMultiDevice()`, the kernels must either have been compiled with toolchain version 3.2 or later so that it will contain kernel parameter information, or have no kernel parameters. If either of these conditions is not met, then :py:obj:`~.cuLaunchCooperativeKernelMultiDevice()` will return :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`. Parameters ---------- launchParamsList : List[:py:obj:`~.CUDA_LAUNCH_PARAMS`] List of launch parameters, one per device numDevices : unsigned int Size of the `launchParamsList` array flags : unsigned int Flags to control launch behavior Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_IMAGE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_LAUNCH_FAILED`, :py:obj:`~.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, :py:obj:`~.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING`, :py:obj:`~.CUDA_ERROR_COOPERATIVE_LAUNCH_TOO_LARGE`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED` See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuLaunchCooperativeKernel`, :py:obj:`~.cudaLaunchCooperativeKernelMultiDevice` """ launchParamsList = [] if launchParamsList is None else launchParamsList if not all(isinstance(_x, (CUDA_LAUNCH_PARAMS,)) for _x in launchParamsList): raise TypeError("Argument 'launchParamsList' is not instance of type (expected Tuple[ccuda.CUDA_LAUNCH_PARAMS,] or List[ccuda.CUDA_LAUNCH_PARAMS,]") cdef ccuda.CUDA_LAUNCH_PARAMS* claunchParamsList = NULL if len(launchParamsList) > 0: claunchParamsList = calloc(len(launchParamsList), sizeof(ccuda.CUDA_LAUNCH_PARAMS)) if claunchParamsList is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(launchParamsList)) + 'x' + str(sizeof(ccuda.CUDA_LAUNCH_PARAMS))) for idx in range(len(launchParamsList)): string.memcpy(&claunchParamsList[idx], (launchParamsList[idx])._ptr, sizeof(ccuda.CUDA_LAUNCH_PARAMS)) if numDevices > len(launchParamsList): raise RuntimeError("List is too small: " + str(len(launchParamsList)) + " < " + str(numDevices)) err = ccuda.cuLaunchCooperativeKernelMultiDevice((launchParamsList[0])._ptr if len(launchParamsList) == 1 else claunchParamsList, numDevices, flags) if claunchParamsList is not NULL: free(claunchParamsList) return (CUresult(err),) {{endif}} {{if 'cuLaunchHostFunc' in found_functions}} @cython.embedsignature(True) def cuLaunchHostFunc(hStream, fn, userData): """ Enqueues a host function call in a stream. Enqueues a host function to run in a stream. The function will be called after currently enqueued work and will block work added after it. The host function must not make any CUDA API calls. Attempting to use a CUDA API may result in :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, but this is not required. The host function must not perform any synchronization that may depend on outstanding CUDA work not mandated to run earlier. Host functions without a mandated order (such as in independent streams) execute in undefined order and may be serialized. For the purposes of Unified Memory, execution makes a number of guarantees: - The stream is considered idle for the duration of the function's execution. Thus, for example, the function may always use memory attached to the stream it was enqueued in. - The start of execution of the function has the same effect as synchronizing an event recorded in the same stream immediately prior to the function. It thus synchronizes streams which have been "joined" prior to the function. - Adding device work to any stream does not have the effect of making the stream active until all preceding host functions and stream callbacks have executed. Thus, for example, a function might use global attached memory even if work has been added to another stream, if the work has been ordered behind the function call with an event. - Completion of the function does not cause a stream to become active except as described above. The stream will remain idle if no device work follows the function, and will remain idle across consecutive host functions or stream callbacks without device work in between. Thus, for example, stream synchronization can be done by signaling from a host function at the end of the stream. Note that, in contrast to :py:obj:`~.cuStreamAddCallback`, the function will not be called in the event of an error in the CUDA context. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream to enqueue function call in fn : :py:obj:`~.CUhostFn` The function to call once preceding stream operations are complete userData : Any User-specified data to be passed to the function Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuMemAllocManaged`, :py:obj:`~.cuStreamAttachMemAsync`, :py:obj:`~.cuStreamAddCallback` """ cdef ccuda.CUhostFn cfn if fn is None: cfn = 0 elif isinstance(fn, (CUhostFn,)): pfn = int(fn) cfn = pfn else: pfn = int(CUhostFn(fn)) cfn = pfn cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cuserData = utils.HelperInputVoidPtr(userData) cdef void* cuserData_ptr = cuserData.cptr err = ccuda.cuLaunchHostFunc(chStream, cfn, cuserData_ptr) return (CUresult(err),) {{endif}} {{if 'cuFuncSetBlockShape' in found_functions}} @cython.embedsignature(True) def cuFuncSetBlockShape(hfunc, int x, int y, int z): """ Sets the block-dimensions for the function. [Deprecated] Specifies the `x`, `y`, and `z` dimensions of the thread blocks that are created when the kernel given by `hfunc` is launched. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to specify dimensions of x : int X dimension y : int Y dimension z : int Z dimension Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc err = ccuda.cuFuncSetBlockShape(chfunc, x, y, z) return (CUresult(err),) {{endif}} {{if 'cuFuncSetSharedSize' in found_functions}} @cython.embedsignature(True) def cuFuncSetSharedSize(hfunc, unsigned int numbytes): """ Sets the dynamic shared-memory size for the function. [Deprecated] Sets through `numbytes` the amount of dynamic shared memory that will be available to each thread block when the kernel given by `hfunc` is launched. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to specify dynamic shared-memory size for numbytes : unsigned int Dynamic shared-memory size per thread in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetCacheConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc err = ccuda.cuFuncSetSharedSize(chfunc, numbytes) return (CUresult(err),) {{endif}} {{if 'cuParamSetSize' in found_functions}} @cython.embedsignature(True) def cuParamSetSize(hfunc, unsigned int numbytes): """ Sets the parameter size for the function. [Deprecated] Sets through `numbytes` the total size in bytes needed by the function parameters of the kernel corresponding to `hfunc`. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to set parameter size for numbytes : unsigned int Size of parameter list in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc err = ccuda.cuParamSetSize(chfunc, numbytes) return (CUresult(err),) {{endif}} {{if 'cuParamSeti' in found_functions}} @cython.embedsignature(True) def cuParamSeti(hfunc, int offset, unsigned int value): """ Adds an integer parameter to the function's argument list. [Deprecated] Sets an integer parameter that will be specified the next time the kernel corresponding to `hfunc` will be invoked. `offset` is a byte offset. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to add parameter to offset : int Offset to add parameter to argument list value : unsigned int Value of parameter Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc err = ccuda.cuParamSeti(chfunc, offset, value) return (CUresult(err),) {{endif}} {{if 'cuParamSetf' in found_functions}} @cython.embedsignature(True) def cuParamSetf(hfunc, int offset, float value): """ Adds a floating-point parameter to the function's argument list. [Deprecated] Sets a floating-point parameter that will be specified the next time the kernel corresponding to `hfunc` will be invoked. `offset` is a byte offset. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to add parameter to offset : int Offset to add parameter to argument list value : float Value of parameter Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc err = ccuda.cuParamSetf(chfunc, offset, value) return (CUresult(err),) {{endif}} {{if 'cuParamSetv' in found_functions}} @cython.embedsignature(True) def cuParamSetv(hfunc, int offset, ptr, unsigned int numbytes): """ Adds arbitrary data to the function's argument list. [Deprecated] Copies an arbitrary amount of data (specified in `numbytes`) from `ptr` into the parameter space of the kernel corresponding to `hfunc`. `offset` is a byte offset. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to add data to offset : int Offset to add data to argument list ptr : Any Pointer to arbitrary data numbytes : unsigned int Size of data to copy in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc cptr = utils.HelperInputVoidPtr(ptr) cdef void* cptr_ptr = cptr.cptr err = ccuda.cuParamSetv(chfunc, offset, cptr_ptr, numbytes) return (CUresult(err),) {{endif}} {{if 'cuLaunch' in found_functions}} @cython.embedsignature(True) def cuLaunch(f): """ Launches a CUDA function. [Deprecated] Invokes the kernel `f` on a 1 x 1 x 1 grid of blocks. The block contains the number of threads specified by a previous call to :py:obj:`~.cuFuncSetBlockShape()`. The block shape, dynamic shared memory size, and parameter information must be set using :py:obj:`~.cuFuncSetBlockShape()`, :py:obj:`~.cuFuncSetSharedSize()`, :py:obj:`~.cuParamSetSize()`, :py:obj:`~.cuParamSeti()`, :py:obj:`~.cuParamSetf()`, and :py:obj:`~.cuParamSetv()` prior to calling this function. Launching a function via :py:obj:`~.cuLaunchKernel()` invalidates the function's block shape, dynamic shared memory size, and parameter information. After launching via cuLaunchKernel, this state must be re- initialized prior to calling this function. Failure to do so results in undefined behavior. Parameters ---------- f : :py:obj:`~.CUfunction` Kernel to launch Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_LAUNCH_FAILED`, :py:obj:`~.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, :py:obj:`~.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction cf if f is None: cf = 0 elif isinstance(f, (CUfunction,)): pf = int(f) cf = pf else: pf = int(CUfunction(f)) cf = pf err = ccuda.cuLaunch(cf) return (CUresult(err),) {{endif}} {{if 'cuLaunchGrid' in found_functions}} @cython.embedsignature(True) def cuLaunchGrid(f, int grid_width, int grid_height): """ Launches a CUDA function. [Deprecated] Invokes the kernel `f` on a `grid_width` x `grid_height` grid of blocks. Each block contains the number of threads specified by a previous call to :py:obj:`~.cuFuncSetBlockShape()`. The block shape, dynamic shared memory size, and parameter information must be set using :py:obj:`~.cuFuncSetBlockShape()`, :py:obj:`~.cuFuncSetSharedSize()`, :py:obj:`~.cuParamSetSize()`, :py:obj:`~.cuParamSeti()`, :py:obj:`~.cuParamSetf()`, and :py:obj:`~.cuParamSetv()` prior to calling this function. Launching a function via :py:obj:`~.cuLaunchKernel()` invalidates the function's block shape, dynamic shared memory size, and parameter information. After launching via cuLaunchKernel, this state must be re- initialized prior to calling this function. Failure to do so results in undefined behavior. Parameters ---------- f : :py:obj:`~.CUfunction` Kernel to launch grid_width : int Width of grid in blocks grid_height : int Height of grid in blocks Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_LAUNCH_FAILED`, :py:obj:`~.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, :py:obj:`~.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGridAsync`, :py:obj:`~.cuLaunchKernel` """ cdef ccuda.CUfunction cf if f is None: cf = 0 elif isinstance(f, (CUfunction,)): pf = int(f) cf = pf else: pf = int(CUfunction(f)) cf = pf err = ccuda.cuLaunchGrid(cf, grid_width, grid_height) return (CUresult(err),) {{endif}} {{if 'cuLaunchGridAsync' in found_functions}} @cython.embedsignature(True) def cuLaunchGridAsync(f, int grid_width, int grid_height, hStream): """ Launches a CUDA function. [Deprecated] Invokes the kernel `f` on a `grid_width` x `grid_height` grid of blocks. Each block contains the number of threads specified by a previous call to :py:obj:`~.cuFuncSetBlockShape()`. The block shape, dynamic shared memory size, and parameter information must be set using :py:obj:`~.cuFuncSetBlockShape()`, :py:obj:`~.cuFuncSetSharedSize()`, :py:obj:`~.cuParamSetSize()`, :py:obj:`~.cuParamSeti()`, :py:obj:`~.cuParamSetf()`, and :py:obj:`~.cuParamSetv()` prior to calling this function. Launching a function via :py:obj:`~.cuLaunchKernel()` invalidates the function's block shape, dynamic shared memory size, and parameter information. After launching via cuLaunchKernel, this state must be re- initialized prior to calling this function. Failure to do so results in undefined behavior. \note_null_stream Parameters ---------- f : :py:obj:`~.CUfunction` Kernel to launch grid_width : int Width of grid in blocks grid_height : int Height of grid in blocks hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream identifier Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_LAUNCH_FAILED`, :py:obj:`~.CUDA_ERROR_LAUNCH_OUT_OF_RESOURCES`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, :py:obj:`~.CUDA_ERROR_LAUNCH_INCOMPATIBLE_TEXTURING`, :py:obj:`~.CUDA_ERROR_SHARED_OBJECT_INIT_FAILED` See Also -------- :py:obj:`~.cuFuncSetBlockShape`, :py:obj:`~.cuFuncSetSharedSize`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuParamSetSize`, :py:obj:`~.cuParamSetf`, :py:obj:`~.cuParamSeti`, :py:obj:`~.cuParamSetv`, :py:obj:`~.cuLaunch`, :py:obj:`~.cuLaunchGrid`, :py:obj:`~.cuLaunchKernel` Notes ----- In certain cases where cubins are created with no ABI (i.e., using `ptxas` `None` `no`), this function may serialize kernel launches. The CUDA driver retains asynchronous behavior by growing the per-thread stack as needed per launch and not shrinking it afterwards. """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUfunction cf if f is None: cf = 0 elif isinstance(f, (CUfunction,)): pf = int(f) cf = pf else: pf = int(CUfunction(f)) cf = pf err = ccuda.cuLaunchGridAsync(cf, grid_width, grid_height, chStream) return (CUresult(err),) {{endif}} {{if 'cuParamSetTexRef' in found_functions}} @cython.embedsignature(True) def cuParamSetTexRef(hfunc, int texunit, hTexRef): """ Adds a texture-reference to the function's argument list. [Deprecated] Makes the CUDA array or linear memory bound to the texture reference `hTexRef` available to a device program as a texture. In this version of CUDA, the texture-reference must be obtained via :py:obj:`~.cuModuleGetTexRef()` and the `texunit` parameter must be set to :py:obj:`~.CU_PARAM_TR_DEFAULT`. Parameters ---------- hfunc : :py:obj:`~.CUfunction` Kernel to add texture-reference to texunit : int Texture unit (must be :py:obj:`~.CU_PARAM_TR_DEFAULT`) hTexRef : :py:obj:`~.CUtexref` Texture-reference to add to argument list Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc err = ccuda.cuParamSetTexRef(chfunc, texunit, chTexRef) return (CUresult(err),) {{endif}} {{if 'cuFuncSetSharedMemConfig' in found_functions}} @cython.embedsignature(True) def cuFuncSetSharedMemConfig(hfunc, config not None : CUsharedconfig): """ Sets the shared memory configuration for a device function. [Deprecated] On devices with configurable shared memory banks, this function will force all subsequent launches of the specified device function to have the given shared memory bank size configuration. On any given launch of the function, the shared memory configuration of the device will be temporarily changed if needed to suit the function's preferred configuration. Changes in shared memory configuration between subsequent launches of functions, may introduce a device side synchronization point. Any per-function setting of shared memory bank size set via :py:obj:`~.cuFuncSetSharedMemConfig` will override the context wide setting set with :py:obj:`~.cuCtxSetSharedMemConfig`. Changing the shared memory bank size will not increase shared memory usage or affect occupancy of kernels, but may have major effects on performance. Larger bank sizes will allow for greater potential bandwidth to shared memory, but will change what kinds of accesses to shared memory will result in bank conflicts. This function will do nothing on devices with fixed shared memory bank size. The supported bank configurations are: - :py:obj:`~.CU_SHARED_MEM_CONFIG_DEFAULT_BANK_SIZE`: use the context's shared memory configuration when launching this function. - :py:obj:`~.CU_SHARED_MEM_CONFIG_FOUR_BYTE_BANK_SIZE`: set shared memory bank width to be natively four bytes when launching this function. - :py:obj:`~.CU_SHARED_MEM_CONFIG_EIGHT_BYTE_BANK_SIZE`: set shared memory bank width to be natively eight bytes when launching this function. Parameters ---------- hfunc : :py:obj:`~.CUfunction` kernel to be given a shared memory config config : :py:obj:`~.CUsharedconfig` requested shared memory configuration Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuCtxGetCacheConfig`, :py:obj:`~.cuCtxSetCacheConfig`, :py:obj:`~.cuCtxGetSharedMemConfig`, :py:obj:`~.cuCtxSetSharedMemConfig`, :py:obj:`~.cuFuncGetAttribute`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cudaFuncSetSharedMemConfig` """ cdef ccuda.CUfunction chfunc if hfunc is None: chfunc = 0 elif isinstance(hfunc, (CUfunction,)): phfunc = int(hfunc) chfunc = phfunc else: phfunc = int(CUfunction(hfunc)) chfunc = phfunc cdef ccuda.CUsharedconfig cconfig = config.value err = ccuda.cuFuncSetSharedMemConfig(chfunc, cconfig) return (CUresult(err),) {{endif}} {{if 'cuGraphCreate' in found_functions}} @cython.embedsignature(True) def cuGraphCreate(unsigned int flags): """ Creates a graph. Creates an empty graph, which is returned via `phGraph`. Parameters ---------- flags : unsigned int Graph creation flags, must be 0 Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phGraph : :py:obj:`~.CUgraph` Returns newly created graph See Also -------- :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode`, :py:obj:`~.cuGraphInstantiate`, :py:obj:`~.cuGraphDestroy`, :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphClone` """ cdef CUgraph phGraph = CUgraph() err = ccuda.cuGraphCreate(phGraph._ptr, flags) return (CUresult(err), phGraph) {{endif}} {{if 'cuGraphAddKernelNode_v2' in found_functions}} @cython.embedsignature(True) def cuGraphAddKernelNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, nodeParams : Optional[CUDA_KERNEL_NODE_PARAMS]): """ Creates a kernel execution node and adds it to a graph. Creates a new kernel execution node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and arguments specified in `nodeParams`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. The CUDA_KERNEL_NODE_PARAMS structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** When the graph is launched, the node will invoke kernel `func` on a (`gridDimX` x `gridDimY` x `gridDimZ`) grid of blocks. Each block contains (`blockDimX` x `blockDimY` x `blockDimZ`) threads. `sharedMemBytes` sets the amount of dynamic shared memory that will be available to each thread block. Kernel parameters to `func` can be specified in one of two ways: 1) Kernel parameters can be specified via `kernelParams`. If the kernel has N parameters, then `kernelParams` needs to be an array of N pointers. Each pointer, from `kernelParams`[0] to `kernelParams`[N-1], points to the region of memory from which the actual parameter will be copied. The number of kernel parameters and their offsets and sizes do not need to be specified as that information is retrieved directly from the kernel's image. 2) Kernel parameters for non-cooperative kernels can also be packaged by the application into a single buffer that is passed in via `extra`. This places the burden on the application of knowing each kernel parameter's size and alignment/padding within the buffer. The `extra` parameter exists to allow this function to take additional less commonly used arguments. `extra` specifies a list of names of extra settings and their corresponding values. Each extra setting name is immediately followed by the corresponding value. The list must be terminated with either NULL or CU_LAUNCH_PARAM_END. - :py:obj:`~.CU_LAUNCH_PARAM_END`, which indicates the end of the `extra` array; - :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_POINTER`, which specifies that the next value in `extra` will be a pointer to a buffer containing all the kernel parameters for launching kernel `func`; - :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_SIZE`, which specifies that the next value in `extra` will be a pointer to a size_t containing the size of the buffer specified with :py:obj:`~.CU_LAUNCH_PARAM_BUFFER_POINTER`; The error :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned if kernel parameters are specified with both `kernelParams` and `extra` (i.e. both `kernelParams` and `extra` are non-NULL). :py:obj:`~.CUDA_ERROR_INVALID_VALUE` will be returned if `extra` is used for a cooperative kernel. The `kernelParams` or `extra` array, as well as the argument values it points to, are copied during this call. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUDA_KERNEL_NODE_PARAMS` Parameters for the GPU execution node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cuLaunchCooperativeKernel`, :py:obj:`~.cuGraphKernelNodeGetParams`, :py:obj:`~.cuGraphKernelNodeSetParams`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` Notes ----- Kernels launched using graphs must not use texture and surface references. Reading or writing through any texture or surface reference is undefined behavior. This restriction does not apply to texture and surface objects. """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_KERNEL_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddKernelNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphKernelNodeGetParams_v2' in found_functions}} @cython.embedsignature(True) def cuGraphKernelNodeGetParams(hNode): """ Returns a kernel node's parameters. Returns the parameters of kernel node `hNode` in `nodeParams`. The `kernelParams` or `extra` array returned in `nodeParams`, as well as the argument values it points to, are owned by the node. This memory remains valid until the node is destroyed or its parameters are modified, and should not be modified directly. Use :py:obj:`~.cuGraphKernelNodeSetParams` to update the parameters of this node. The params will contain either `kernelParams` or `extra`, according to which of these was most recently set on the node. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` nodeParams : :py:obj:`~.CUDA_KERNEL_NODE_PARAMS` Pointer to return the parameters See Also -------- :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphKernelNodeSetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_KERNEL_NODE_PARAMS nodeParams = CUDA_KERNEL_NODE_PARAMS() err = ccuda.cuGraphKernelNodeGetParams(chNode, nodeParams._ptr) return (CUresult(err), nodeParams) {{endif}} {{if 'cuGraphKernelNodeSetParams_v2' in found_functions}} @cython.embedsignature(True) def cuGraphKernelNodeSetParams(hNode, nodeParams : Optional[CUDA_KERNEL_NODE_PARAMS]): """ Sets a kernel node's parameters. Sets the parameters of kernel node `hNode` to `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUDA_KERNEL_NODE_PARAMS` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphKernelNodeGetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUDA_KERNEL_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphKernelNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphAddMemcpyNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddMemcpyNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, copyParams : Optional[CUDA_MEMCPY3D], ctx): """ Creates a memcpy node and adds it to a graph. Creates a new memcpy node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. When the graph is launched, the node will perform the memcpy described by `copyParams`. See :py:obj:`~.cuMemcpy3D()` for a description of the structure and its restrictions. Memcpy nodes have some additional restrictions with regards to managed memory, if the system contains at least one device which has a zero value for the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_CONCURRENT_MANAGED_ACCESS`. If one or more of the operands refer to managed memory, then using the memory type :py:obj:`~.CU_MEMORYTYPE_UNIFIED` is disallowed for those operand(s). The managed memory will be treated as residing on either the host or the device, depending on which memory type is specified. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies copyParams : :py:obj:`~.CUDA_MEMCPY3D` Parameters for the memory copy ctx : :py:obj:`~.CUcontext` Context on which to run the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuGraphMemcpyNodeGetParams`, :py:obj:`~.cuGraphMemcpyNodeSetParams`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphAddMemsetNode` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_MEMCPY3D* ccopyParams_ptr = copyParams._ptr if copyParams != None else NULL err = ccuda.cuGraphAddMemcpyNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, ccopyParams_ptr, cctx) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphMemcpyNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphMemcpyNodeGetParams(hNode): """ Returns a memcpy node's parameters. Returns the parameters of memcpy node `hNode` in `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` nodeParams : :py:obj:`~.CUDA_MEMCPY3D` Pointer to return the parameters See Also -------- :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphMemcpyNodeSetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_MEMCPY3D nodeParams = CUDA_MEMCPY3D() err = ccuda.cuGraphMemcpyNodeGetParams(chNode, nodeParams._ptr) return (CUresult(err), nodeParams) {{endif}} {{if 'cuGraphMemcpyNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphMemcpyNodeSetParams(hNode, nodeParams : Optional[CUDA_MEMCPY3D]): """ Sets a memcpy node's parameters. Sets the parameters of memcpy node `hNode` to `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUDA_MEMCPY3D` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuMemcpy3D`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphMemcpyNodeGetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUDA_MEMCPY3D* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphMemcpyNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphAddMemsetNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddMemsetNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, memsetParams : Optional[CUDA_MEMSET_NODE_PARAMS], ctx): """ Creates a memset node and adds it to a graph. Creates a new memset node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. The element size must be 1, 2, or 4 bytes. When the graph is launched, the node will perform the memset described by `memsetParams`. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies memsetParams : :py:obj:`~.CUDA_MEMSET_NODE_PARAMS` Parameters for the memory set ctx : :py:obj:`~.CUcontext` Context on which to run the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuGraphMemsetNodeGetParams`, :py:obj:`~.cuGraphMemsetNodeSetParams`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphAddMemcpyNode` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_MEMSET_NODE_PARAMS* cmemsetParams_ptr = memsetParams._ptr if memsetParams != None else NULL err = ccuda.cuGraphAddMemsetNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cmemsetParams_ptr, cctx) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphMemsetNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphMemsetNodeGetParams(hNode): """ Returns a memset node's parameters. Returns the parameters of memset node `hNode` in `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` nodeParams : :py:obj:`~.CUDA_MEMSET_NODE_PARAMS` Pointer to return the parameters See Also -------- :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuGraphAddMemsetNode`, :py:obj:`~.cuGraphMemsetNodeSetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_MEMSET_NODE_PARAMS nodeParams = CUDA_MEMSET_NODE_PARAMS() err = ccuda.cuGraphMemsetNodeGetParams(chNode, nodeParams._ptr) return (CUresult(err), nodeParams) {{endif}} {{if 'cuGraphMemsetNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphMemsetNodeSetParams(hNode, nodeParams : Optional[CUDA_MEMSET_NODE_PARAMS]): """ Sets a memset node's parameters. Sets the parameters of memset node `hNode` to `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUDA_MEMSET_NODE_PARAMS` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuMemsetD2D32`, :py:obj:`~.cuGraphAddMemsetNode`, :py:obj:`~.cuGraphMemsetNodeGetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUDA_MEMSET_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphMemsetNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphAddHostNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddHostNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, nodeParams : Optional[CUDA_HOST_NODE_PARAMS]): """ Creates a host execution node and adds it to a graph. Creates a new CPU execution node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and arguments specified in `nodeParams`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. When the graph is launched, the node will invoke the specified CPU function. Host nodes are not supported under MPS with pre-Volta GPUs. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUDA_HOST_NODE_PARAMS` Parameters for the host node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuLaunchHostFunc`, :py:obj:`~.cuGraphHostNodeGetParams`, :py:obj:`~.cuGraphHostNodeSetParams`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_HOST_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddHostNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphHostNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphHostNodeGetParams(hNode): """ Returns a host node's parameters. Returns the parameters of host node `hNode` in `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` nodeParams : :py:obj:`~.CUDA_HOST_NODE_PARAMS` Pointer to return the parameters See Also -------- :py:obj:`~.cuLaunchHostFunc`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphHostNodeSetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_HOST_NODE_PARAMS nodeParams = CUDA_HOST_NODE_PARAMS() err = ccuda.cuGraphHostNodeGetParams(chNode, nodeParams._ptr) return (CUresult(err), nodeParams) {{endif}} {{if 'cuGraphHostNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphHostNodeSetParams(hNode, nodeParams : Optional[CUDA_HOST_NODE_PARAMS]): """ Sets a host node's parameters. Sets the parameters of host node `hNode` to `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUDA_HOST_NODE_PARAMS` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuLaunchHostFunc`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphHostNodeGetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUDA_HOST_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphHostNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphAddChildGraphNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddChildGraphNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, childGraph): """ Creates a child graph node and adds it to a graph. Creates a new node which executes an embedded graph, and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. If `hGraph` contains allocation or free nodes, this call will return an error. The node executes an embedded child graph. The child graph is cloned in this call. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies childGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` The graph to clone into this node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphChildGraphNodeGetGraph`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode`, :py:obj:`~.cuGraphClone` """ cdef ccuda.CUgraph cchildGraph if childGraph is None: cchildGraph = 0 elif isinstance(childGraph, (CUgraph,)): pchildGraph = int(childGraph) cchildGraph = pchildGraph else: pchildGraph = int(CUgraph(childGraph)) cchildGraph = pchildGraph dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) err = ccuda.cuGraphAddChildGraphNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cchildGraph) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphChildGraphNodeGetGraph' in found_functions}} @cython.embedsignature(True) def cuGraphChildGraphNodeGetGraph(hNode): """ Gets a handle to the embedded graph of a child graph node. Gets a handle to the embedded graph in a child graph node. This call does not clone the graph. Changes to the graph will be reflected in the node, and the node retains ownership of the graph. Allocation and free nodes cannot be added to the returned graph. Attempting to do so will return an error. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the embedded graph for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, phGraph : :py:obj:`~.CUgraph` Location to store a handle to the graph See Also -------- :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphNodeFindInClone` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUgraph phGraph = CUgraph() err = ccuda.cuGraphChildGraphNodeGetGraph(chNode, phGraph._ptr) return (CUresult(err), phGraph) {{endif}} {{if 'cuGraphAddEmptyNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddEmptyNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies): """ Creates an empty node and adds it to a graph. Creates a new node which performs no operation, and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. An empty node performs no operation during execution, but can be used for transitive ordering. For example, a phased execution graph with 2 groups of n nodes with a barrier between them can be represented using an empty node and 2*n dependency edges, rather than no empty node and n^2 dependency edges. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) err = ccuda.cuGraphAddEmptyNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphAddEventRecordNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddEventRecordNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, event): """ Creates an event record node and adds it to a graph. Creates a new event record node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and event specified in `event`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. Each launch of the graph will record `event` to capture execution of the node's dependencies. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies event : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event for the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) err = ccuda.cuGraphAddEventRecordNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cevent) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphEventRecordNodeGetEvent' in found_functions}} @cython.embedsignature(True) def cuGraphEventRecordNodeGetEvent(hNode): """ Returns the event associated with an event record node. Returns the event of event record node `hNode` in `event_out`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the event for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` event_out : :py:obj:`~.CUevent` Pointer to return the event See Also -------- :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuGraphEventRecordNodeSetEvent`, :py:obj:`~.cuGraphEventWaitNodeGetEvent`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUevent event_out = CUevent() err = ccuda.cuGraphEventRecordNodeGetEvent(chNode, event_out._ptr) return (CUresult(err), event_out) {{endif}} {{if 'cuGraphEventRecordNodeSetEvent' in found_functions}} @cython.embedsignature(True) def cuGraphEventRecordNodeSetEvent(hNode, event): """ Sets an event record node's event. Sets the event of event record node `hNode` to `event`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the event for event : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to use Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuGraphEventRecordNodeGetEvent`, :py:obj:`~.cuGraphEventWaitNodeSetEvent`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent` """ cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode err = ccuda.cuGraphEventRecordNodeSetEvent(chNode, cevent) return (CUresult(err),) {{endif}} {{if 'cuGraphAddEventWaitNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddEventWaitNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, event): """ Creates an event wait node and adds it to a graph. Creates a new event wait node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and event specified in `event`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. The graph node will wait for all work captured in `event`. See :py:obj:`~.cuEventRecord()` for details on what is captured by an event. `event` may be from a different context or device than the launch stream. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies event : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event for the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) err = ccuda.cuGraphAddEventWaitNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cevent) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphEventWaitNodeGetEvent' in found_functions}} @cython.embedsignature(True) def cuGraphEventWaitNodeGetEvent(hNode): """ Returns the event associated with an event wait node. Returns the event of event wait node `hNode` in `event_out`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the event for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` event_out : :py:obj:`~.CUevent` Pointer to return the event See Also -------- :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuGraphEventWaitNodeSetEvent`, :py:obj:`~.cuGraphEventRecordNodeGetEvent`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUevent event_out = CUevent() err = ccuda.cuGraphEventWaitNodeGetEvent(chNode, event_out._ptr) return (CUresult(err), event_out) {{endif}} {{if 'cuGraphEventWaitNodeSetEvent' in found_functions}} @cython.embedsignature(True) def cuGraphEventWaitNodeSetEvent(hNode, event): """ Sets an event wait node's event. Sets the event of event wait node `hNode` to `event`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the event for event : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to use Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuGraphEventWaitNodeGetEvent`, :py:obj:`~.cuGraphEventRecordNodeSetEvent`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent` """ cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode err = ccuda.cuGraphEventWaitNodeSetEvent(chNode, cevent) return (CUresult(err),) {{endif}} {{if 'cuGraphAddExternalSemaphoresSignalNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddExternalSemaphoresSignalNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, nodeParams : Optional[CUDA_EXT_SEM_SIGNAL_NODE_PARAMS]): """ Creates an external semaphore signal node and adds it to a graph. Creates a new external semaphore signal node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and arguments specified in `nodeParams`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. Performs a signal operation on a set of externally allocated semaphore objects when the node is launched. The operation(s) will occur after all of the node's dependencies have completed. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS` Parameters for the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphExternalSemaphoresSignalNodeGetParams`, :py:obj:`~.cuGraphExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddExternalSemaphoresSignalNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphExternalSemaphoresSignalNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExternalSemaphoresSignalNodeGetParams(hNode): """ Returns an external semaphore signal node's parameters. Returns the parameters of an external semaphore signal node `hNode` in `params_out`. The `extSemArray` and `paramsArray` returned in `params_out`, are owned by the node. This memory remains valid until the node is destroyed or its parameters are modified, and should not be modified directly. Use :py:obj:`~.cuGraphExternalSemaphoresSignalNodeSetParams` to update the parameters of this node. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` params_out : :py:obj:`~.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS` Pointer to return the parameters See Also -------- :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cuGraphAddExternalSemaphoresSignalNode`, :py:obj:`~.cuGraphExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_EXT_SEM_SIGNAL_NODE_PARAMS params_out = CUDA_EXT_SEM_SIGNAL_NODE_PARAMS() err = ccuda.cuGraphExternalSemaphoresSignalNodeGetParams(chNode, params_out._ptr) return (CUresult(err), params_out) {{endif}} {{if 'cuGraphExternalSemaphoresSignalNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExternalSemaphoresSignalNodeSetParams(hNode, nodeParams : Optional[CUDA_EXT_SEM_SIGNAL_NODE_PARAMS]): """ Sets an external semaphore signal node's parameters. Sets the parameters of an external semaphore signal node `hNode` to `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresSignalNode`, :py:obj:`~.cuGraphExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExternalSemaphoresSignalNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphAddExternalSemaphoresWaitNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddExternalSemaphoresWaitNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, nodeParams : Optional[CUDA_EXT_SEM_WAIT_NODE_PARAMS]): """ Creates an external semaphore wait node and adds it to a graph. Creates a new external semaphore wait node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and arguments specified in `nodeParams`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. Performs a wait operation on a set of externally allocated semaphore objects when the node is launched. The node's dependencies will not be launched until the wait operation has completed. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUDA_EXT_SEM_WAIT_NODE_PARAMS` Parameters for the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphExternalSemaphoresWaitNodeGetParams`, :py:obj:`~.cuGraphExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresSignalNode`, :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddExternalSemaphoresWaitNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphExternalSemaphoresWaitNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExternalSemaphoresWaitNodeGetParams(hNode): """ Returns an external semaphore wait node's parameters. Returns the parameters of an external semaphore wait node `hNode` in `params_out`. The `extSemArray` and `paramsArray` returned in `params_out`, are owned by the node. This memory remains valid until the node is destroyed or its parameters are modified, and should not be modified directly. Use :py:obj:`~.cuGraphExternalSemaphoresSignalNodeSetParams` to update the parameters of this node. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` params_out : :py:obj:`~.CUDA_EXT_SEM_WAIT_NODE_PARAMS` Pointer to return the parameters See Also -------- :py:obj:`~.cuLaunchKernel`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuGraphExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_EXT_SEM_WAIT_NODE_PARAMS params_out = CUDA_EXT_SEM_WAIT_NODE_PARAMS() err = ccuda.cuGraphExternalSemaphoresWaitNodeGetParams(chNode, params_out._ptr) return (CUresult(err), params_out) {{endif}} {{if 'cuGraphExternalSemaphoresWaitNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExternalSemaphoresWaitNodeSetParams(hNode, nodeParams : Optional[CUDA_EXT_SEM_WAIT_NODE_PARAMS]): """ Sets an external semaphore wait node's parameters. Sets the parameters of an external semaphore wait node `hNode` to `nodeParams`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUDA_EXT_SEM_WAIT_NODE_PARAMS` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuGraphExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExternalSemaphoresWaitNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphAddBatchMemOpNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddBatchMemOpNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, nodeParams : Optional[CUDA_BATCH_MEM_OP_NODE_PARAMS]): """ Creates a batch memory operation node and adds it to a graph. Creates a new batch memory operation node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and arguments specified in `nodeParams`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. When the node is added, the paramArray inside `nodeParams` is copied and therefore it can be freed after the call returns. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUDA_BATCH_MEM_OP_NODE_PARAMS` Parameters for the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuStreamWaitValue32`, :py:obj:`~.cuStreamWriteValue32`, :py:obj:`~.cuStreamWaitValue64`, :py:obj:`~.cuStreamWriteValue64`, :py:obj:`~.cuGraphBatchMemOpNodeGetParams`, :py:obj:`~.cuGraphBatchMemOpNodeSetParams`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` Notes ----- Warning: Improper use of this API may deadlock the application. Synchronization ordering established through this API is not visible to CUDA. CUDA tasks that are (even indirectly) ordered by this API should also have that order expressed with CUDA-visible dependencies such as events. This ensures that the scheduler does not serialize them in an improper order. For more information, see the Stream Memory Operations section in the programming guide(https://docs.nvidia.com/cuda/cuda-c-programming-guide/index.html). """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddBatchMemOpNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphBatchMemOpNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphBatchMemOpNodeGetParams(hNode): """ Returns a batch mem op node's parameters. Returns the parameters of batch mem op node `hNode` in `nodeParams_out`. The `paramArray` returned in `nodeParams_out` is owned by the node. This memory remains valid until the node is destroyed or its parameters are modified, and should not be modified directly. Use :py:obj:`~.cuGraphBatchMemOpNodeSetParams` to update the parameters of this node. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` nodeParams_out : :py:obj:`~.CUDA_BATCH_MEM_OP_NODE_PARAMS` Pointer to return the parameters See Also -------- :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuGraphAddBatchMemOpNode`, :py:obj:`~.cuGraphBatchMemOpNodeSetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_BATCH_MEM_OP_NODE_PARAMS nodeParams_out = CUDA_BATCH_MEM_OP_NODE_PARAMS() err = ccuda.cuGraphBatchMemOpNodeGetParams(chNode, nodeParams_out._ptr) return (CUresult(err), nodeParams_out) {{endif}} {{if 'cuGraphBatchMemOpNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphBatchMemOpNodeSetParams(hNode, nodeParams : Optional[CUDA_BATCH_MEM_OP_NODE_PARAMS]): """ Sets a batch mem op node's parameters. Sets the parameters of batch mem op node `hNode` to `nodeParams`. The paramArray inside `nodeParams` is copied and therefore it can be freed after the call returns. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUDA_BATCH_MEM_OP_NODE_PARAMS` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` See Also -------- :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuGraphAddBatchMemOpNode`, :py:obj:`~.cuGraphBatchMemOpNodeGetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphBatchMemOpNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphExecBatchMemOpNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecBatchMemOpNodeSetParams(hGraphExec, hNode, nodeParams : Optional[CUDA_BATCH_MEM_OP_NODE_PARAMS]): """ Sets the parameters for a batch mem op node in the given graphExec. Sets the parameters of a batch mem op node in an executable graph `hGraphExec`. The node is identified by the corresponding node `hNode` in the non-executable graph, from which the executable graph was instantiated. The following fields on operations may be modified on an executable graph: op.waitValue.address op.waitValue.value[64] op.waitValue.flags bits corresponding to wait type (i.e. CU_STREAM_WAIT_VALUE_FLUSH bit cannot be modified) op.writeValue.address op.writeValue.value[64] Other fields, such as the context, count or type of operations, and other types of operations such as membars, may not be modified. `hNode` must not have been removed from the original graph. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. The paramArray inside `nodeParams` is copied and therefore it can be freed after the call returns. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Batch mem op node from the graph from which graphExec was instantiated nodeParams : :py:obj:`~.CUDA_BATCH_MEM_OP_NODE_PARAMS` Updated Parameters to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuStreamBatchMemOp`, :py:obj:`~.cuGraphAddBatchMemOpNode`, :py:obj:`~.cuGraphBatchMemOpNodeGetParams`, :py:obj:`~.cuGraphBatchMemOpNodeSetParams`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExecBatchMemOpNodeSetParams(chGraphExec, chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphAddMemAllocNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddMemAllocNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, nodeParams : Optional[CUDA_MEM_ALLOC_NODE_PARAMS]): """ Creates an allocation node and adds it to a graph. Creates a new allocation node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and arguments specified in `nodeParams`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. When :py:obj:`~.cuGraphAddMemAllocNode` creates an allocation node, it returns the address of the allocation in `nodeParams.dptr`. The allocation's address remains fixed across instantiations and launches. If the allocation is freed in the same graph, by creating a free node using :py:obj:`~.cuGraphAddMemFreeNode`, the allocation can be accessed by nodes ordered after the allocation node but before the free node. These allocations cannot be freed outside the owning graph, and they can only be freed once in the owning graph. If the allocation is not freed in the same graph, then it can be accessed not only by nodes in the graph which are ordered after the allocation node, but also by stream operations ordered after the graph's execution but before the allocation is freed. Allocations which are not freed in the same graph can be freed by: - passing the allocation to :py:obj:`~.cuMemFreeAsync` or :py:obj:`~.cuMemFree`; - launching a graph with a free node for that allocation; or - specifying :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH` during instantiation, which makes each launch behave as though it called :py:obj:`~.cuMemFreeAsync` for every unfreed allocation. It is not possible to free an allocation in both the owning graph and another graph. If the allocation is freed in the same graph, a free node cannot be added to another graph. If the allocation is freed in another graph, a free node can no longer be added to the owning graph. The following restrictions apply to graphs which contain allocation and/or memory free nodes: - Nodes and edges of the graph cannot be deleted. - The graph cannot be used in a child node. - Only one instantiation of the graph may exist at any point in time. - The graph cannot be cloned. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUDA_MEM_ALLOC_NODE_PARAMS` Parameters for the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphAddMemFreeNode`, :py:obj:`~.cuGraphMemAllocNodeGetParams`, :py:obj:`~.cuDeviceGraphMemTrim`, :py:obj:`~.cuDeviceGetGraphMemAttribute`, :py:obj:`~.cuDeviceSetGraphMemAttribute`, :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuGraphAddExternalSemaphoresSignalNode`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUDA_MEM_ALLOC_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddMemAllocNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphMemAllocNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphMemAllocNodeGetParams(hNode): """ Returns a memory alloc node's parameters. Returns the parameters of a memory alloc node `hNode` in `params_out`. The `poolProps` and `accessDescs` returned in `params_out`, are owned by the node. This memory remains valid until the node is destroyed. The returned parameters must not be modified. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` params_out : :py:obj:`~.CUDA_MEM_ALLOC_NODE_PARAMS` Pointer to return the parameters See Also -------- :py:obj:`~.cuGraphAddMemAllocNode`, :py:obj:`~.cuGraphMemFreeNodeGetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUDA_MEM_ALLOC_NODE_PARAMS params_out = CUDA_MEM_ALLOC_NODE_PARAMS() err = ccuda.cuGraphMemAllocNodeGetParams(chNode, params_out._ptr) return (CUresult(err), params_out) {{endif}} {{if 'cuGraphAddMemFreeNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddMemFreeNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, dptr): """ Creates a memory free node and adds it to a graph. Creates a new memory free node and adds it to `hGraph` with `numDependencies` dependencies specified via `dependencies` and arguments specified in `nodeParams`. It is possible for `numDependencies` to be 0, in which case the node will be placed at the root of the graph. `dependencies` may not have any duplicate entries. A handle to the new node will be returned in `phGraphNode`. :py:obj:`~.cuGraphAddMemFreeNode` will return :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if the user attempts to free: - an allocation twice in the same graph. - an address that was not returned by an allocation node. - an invalid address. The following restrictions apply to graphs which contain allocation and/or memory free nodes: - Nodes and edges of the graph cannot be deleted. - The graph cannot be used in a child node. - Only one instantiation of the graph may exist at any point in time. - The graph cannot be cloned. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies dptr : :py:obj:`~.CUdeviceptr` Address of memory to free Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphAddMemAllocNode`, :py:obj:`~.cuGraphMemFreeNodeGetParams`, :py:obj:`~.cuDeviceGraphMemTrim`, :py:obj:`~.cuDeviceGetGraphMemAttribute`, :py:obj:`~.cuDeviceSetGraphMemAttribute`, :py:obj:`~.cuMemAllocAsync`, :py:obj:`~.cuMemFreeAsync`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphDestroyNode`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuGraphAddExternalSemaphoresSignalNode`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) err = ccuda.cuGraphAddMemFreeNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cdptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphMemFreeNodeGetParams' in found_functions}} @cython.embedsignature(True) def cuGraphMemFreeNodeGetParams(hNode): """ Returns a memory free node's parameters. Returns the address of a memory free node `hNode` in `dptr_out`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to get the parameters for Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` dptr_out : :py:obj:`~.CUdeviceptr` Pointer to return the device address See Also -------- :py:obj:`~.cuGraphAddMemFreeNode`, :py:obj:`~.cuGraphMemAllocNodeGetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef CUdeviceptr dptr_out = CUdeviceptr() err = ccuda.cuGraphMemFreeNodeGetParams(chNode, dptr_out._ptr) return (CUresult(err), dptr_out) {{endif}} {{if 'cuDeviceGraphMemTrim' in found_functions}} @cython.embedsignature(True) def cuDeviceGraphMemTrim(device): """ Free unused memory that was cached on the specified device for use with graphs back to the OS. Blocks which are not in use by a graph that is either currently executing or scheduled to execute are freed back to the operating system. Parameters ---------- device : :py:obj:`~.CUdevice` The device for which cached memory should be freed. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` See Also -------- :py:obj:`~.cuGraphAddMemAllocNode`, :py:obj:`~.cuGraphAddMemFreeNode`, :py:obj:`~.cuDeviceSetGraphMemAttribute`, :py:obj:`~.cuDeviceGetGraphMemAttribute` """ cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice err = ccuda.cuDeviceGraphMemTrim(cdevice) return (CUresult(err),) {{endif}} {{if 'cuDeviceGetGraphMemAttribute' in found_functions}} @cython.embedsignature(True) def cuDeviceGetGraphMemAttribute(device, attr not None : CUgraphMem_attribute): """ Query asynchronous allocation attributes related to graphs. Valid attributes are: - :py:obj:`~.CU_GRAPH_MEM_ATTR_USED_MEM_CURRENT`: Amount of memory, in bytes, currently associated with graphs - :py:obj:`~.CU_GRAPH_MEM_ATTR_USED_MEM_HIGH`: High watermark of memory, in bytes, associated with graphs since the last time it was reset. High watermark can only be reset to zero. - :py:obj:`~.CU_GRAPH_MEM_ATTR_RESERVED_MEM_CURRENT`: Amount of memory, in bytes, currently allocated for use by the CUDA graphs asynchronous allocator. - :py:obj:`~.CU_GRAPH_MEM_ATTR_RESERVED_MEM_HIGH`: High watermark of memory, in bytes, currently allocated for use by the CUDA graphs asynchronous allocator. Parameters ---------- device : :py:obj:`~.CUdevice` Specifies the scope of the query attr : :py:obj:`~.CUgraphMem_attribute` attribute to get Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` value : Any retrieved value See Also -------- :py:obj:`~.cuDeviceSetGraphMemAttribute`, :py:obj:`~.cuGraphAddMemAllocNode`, :py:obj:`~.cuGraphAddMemFreeNode` """ cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cdef ccuda.CUgraphMem_attribute cattr = attr.value cdef utils.HelperCUgraphMem_attribute cvalue = utils.HelperCUgraphMem_attribute(attr, 0, is_getter=True) cdef void* cvalue_ptr = cvalue.cptr err = ccuda.cuDeviceGetGraphMemAttribute(cdevice, cattr, cvalue_ptr) return (CUresult(err), cvalue.pyObj()) {{endif}} {{if 'cuDeviceSetGraphMemAttribute' in found_functions}} @cython.embedsignature(True) def cuDeviceSetGraphMemAttribute(device, attr not None : CUgraphMem_attribute, value): """ Set asynchronous allocation attributes related to graphs. Valid attributes are: - :py:obj:`~.CU_GRAPH_MEM_ATTR_USED_MEM_HIGH`: High watermark of memory, in bytes, associated with graphs since the last time it was reset. High watermark can only be reset to zero. - :py:obj:`~.CU_GRAPH_MEM_ATTR_RESERVED_MEM_HIGH`: High watermark of memory, in bytes, currently allocated for use by the CUDA graphs asynchronous allocator. Parameters ---------- device : :py:obj:`~.CUdevice` Specifies the scope of the query attr : :py:obj:`~.CUgraphMem_attribute` attribute to get value : Any pointer to value to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` See Also -------- :py:obj:`~.cuDeviceGetGraphMemAttribute`, :py:obj:`~.cuGraphAddMemAllocNode`, :py:obj:`~.cuGraphAddMemFreeNode` """ cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cdef ccuda.CUgraphMem_attribute cattr = attr.value cdef utils.HelperCUgraphMem_attribute cvalue = utils.HelperCUgraphMem_attribute(attr, value, is_getter=False) cdef void* cvalue_ptr = cvalue.cptr err = ccuda.cuDeviceSetGraphMemAttribute(cdevice, cattr, cvalue_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphClone' in found_functions}} @cython.embedsignature(True) def cuGraphClone(originalGraph): """ Clones a graph. This function creates a copy of `originalGraph` and returns it in `phGraphClone`. All parameters are copied into the cloned graph. The original graph may be modified after this call without affecting the clone. Child graph nodes in the original graph are recursively copied into the clone. Parameters ---------- originalGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to clone Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phGraphClone : :py:obj:`~.CUgraph` Returns newly created cloned graph See Also -------- :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphNodeFindInClone` """ cdef ccuda.CUgraph coriginalGraph if originalGraph is None: coriginalGraph = 0 elif isinstance(originalGraph, (CUgraph,)): poriginalGraph = int(originalGraph) coriginalGraph = poriginalGraph else: poriginalGraph = int(CUgraph(originalGraph)) coriginalGraph = poriginalGraph cdef CUgraph phGraphClone = CUgraph() err = ccuda.cuGraphClone(phGraphClone._ptr, coriginalGraph) return (CUresult(err), phGraphClone) {{endif}} {{if 'cuGraphNodeFindInClone' in found_functions}} @cython.embedsignature(True) def cuGraphNodeFindInClone(hOriginalNode, hClonedGraph): """ Finds a cloned version of a node. This function returns the node in `hClonedGraph` corresponding to `hOriginalNode` in the original graph. `hClonedGraph` must have been cloned from `hOriginalGraph` via :py:obj:`~.cuGraphClone`. `hOriginalNode` must have been in `hOriginalGraph` at the time of the call to :py:obj:`~.cuGraphClone`, and the corresponding cloned node in `hClonedGraph` must not have been removed. The cloned node is then returned via `phClonedNode`. Parameters ---------- hOriginalNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Handle to the original node hClonedGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Cloned graph to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, phNode : :py:obj:`~.CUgraphNode` Returns handle to the cloned node See Also -------- :py:obj:`~.cuGraphClone` """ cdef ccuda.CUgraph chClonedGraph if hClonedGraph is None: chClonedGraph = 0 elif isinstance(hClonedGraph, (CUgraph,)): phClonedGraph = int(hClonedGraph) chClonedGraph = phClonedGraph else: phClonedGraph = int(CUgraph(hClonedGraph)) chClonedGraph = phClonedGraph cdef ccuda.CUgraphNode chOriginalNode if hOriginalNode is None: chOriginalNode = 0 elif isinstance(hOriginalNode, (CUgraphNode,)): phOriginalNode = int(hOriginalNode) chOriginalNode = phOriginalNode else: phOriginalNode = int(CUgraphNode(hOriginalNode)) chOriginalNode = phOriginalNode cdef CUgraphNode phNode = CUgraphNode() err = ccuda.cuGraphNodeFindInClone(phNode._ptr, chOriginalNode, chClonedGraph) return (CUresult(err), phNode) {{endif}} {{if 'cuGraphNodeGetType' in found_functions}} @cython.embedsignature(True) def cuGraphNodeGetType(hNode): """ Returns a node's type. Returns the node type of `hNode` in `typename`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` typename : :py:obj:`~.CUgraphNodeType` Pointer to return the node type See Also -------- :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphChildGraphNodeGetGraph`, :py:obj:`~.cuGraphKernelNodeGetParams`, :py:obj:`~.cuGraphKernelNodeSetParams`, :py:obj:`~.cuGraphHostNodeGetParams`, :py:obj:`~.cuGraphHostNodeSetParams`, :py:obj:`~.cuGraphMemcpyNodeGetParams`, :py:obj:`~.cuGraphMemcpyNodeSetParams`, :py:obj:`~.cuGraphMemsetNodeGetParams`, :py:obj:`~.cuGraphMemsetNodeSetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphNodeType typename err = ccuda.cuGraphNodeGetType(chNode, &typename) return (CUresult(err), CUgraphNodeType(typename)) {{endif}} {{if 'cuGraphGetNodes' in found_functions}} @cython.embedsignature(True) def cuGraphGetNodes(hGraph, size_t numNodes = 0): """ Returns a graph's nodes. Returns a list of `hGraph's` nodes. `nodes` may be NULL, in which case this function will return the number of nodes in `numNodes`. Otherwise, `numNodes` entries will be filled in. If `numNodes` is higher than the actual number of nodes, the remaining entries in `nodes` will be set to NULL, and the number of nodes actually obtained will be returned in `numNodes`. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to query numNodes : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` nodes : List[:py:obj:`~.CUgraphNode`] Pointer to return the nodes numNodes : int See description See Also -------- :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphNodeGetType`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ cdef size_t _graph_length = numNodes cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* cnodes = NULL pynodes = [] if _graph_length != 0: cnodes = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cnodes is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) err = ccuda.cuGraphGetNodes(chGraph, cnodes, &numNodes) if CUresult(err) == CUresult(0): pynodes = [CUgraphNode(init_value=cnodes[idx]) for idx in range(_graph_length)] if cnodes is not NULL: free(cnodes) return (CUresult(err), pynodes, numNodes) {{endif}} {{if 'cuGraphGetRootNodes' in found_functions}} @cython.embedsignature(True) def cuGraphGetRootNodes(hGraph, size_t numRootNodes = 0): """ Returns a graph's root nodes. Returns a list of `hGraph's` root nodes. `rootNodes` may be NULL, in which case this function will return the number of root nodes in `numRootNodes`. Otherwise, `numRootNodes` entries will be filled in. If `numRootNodes` is higher than the actual number of root nodes, the remaining entries in `rootNodes` will be set to NULL, and the number of nodes actually obtained will be returned in `numRootNodes`. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to query numRootNodes : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` rootNodes : List[:py:obj:`~.CUgraphNode`] Pointer to return the root nodes numRootNodes : int See description See Also -------- :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphNodeGetType`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ cdef size_t _graph_length = numRootNodes cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* crootNodes = NULL pyrootNodes = [] if _graph_length != 0: crootNodes = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if crootNodes is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) err = ccuda.cuGraphGetRootNodes(chGraph, crootNodes, &numRootNodes) if CUresult(err) == CUresult(0): pyrootNodes = [CUgraphNode(init_value=crootNodes[idx]) for idx in range(_graph_length)] if crootNodes is not NULL: free(crootNodes) return (CUresult(err), pyrootNodes, numRootNodes) {{endif}} {{if 'cuGraphGetEdges' in found_functions}} @cython.embedsignature(True) def cuGraphGetEdges(hGraph, size_t numEdges = 0): """ Returns a graph's dependency edges. Returns a list of `hGraph's` dependency edges. Edges are returned via corresponding indices in `from` and `to`; that is, the node in `to`[i] has a dependency on the node in `from`[i]. `from` and `to` may both be NULL, in which case this function only returns the number of edges in `numEdges`. Otherwise, `numEdges` entries will be filled in. If `numEdges` is higher than the actual number of edges, the remaining entries in `from` and `to` will be set to NULL, and the number of edges actually returned will be written to `numEdges`. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to get the edges from numEdges : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` from : List[:py:obj:`~.CUgraphNode`] Location to return edge endpoints to : List[:py:obj:`~.CUgraphNode`] Location to return edge endpoints numEdges : int See description See Also -------- :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphRemoveDependencies`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ cdef size_t _graph_length = numEdges cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* cfrom_ = NULL pyfrom_ = [] if _graph_length != 0: cfrom_ = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cfrom_ is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) cdef ccuda.CUgraphNode* cto = NULL pyto = [] if _graph_length != 0: cto = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cto is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) err = ccuda.cuGraphGetEdges(chGraph, cfrom_, cto, &numEdges) if CUresult(err) == CUresult(0): pyfrom_ = [CUgraphNode(init_value=cfrom_[idx]) for idx in range(_graph_length)] if cfrom_ is not NULL: free(cfrom_) if CUresult(err) == CUresult(0): pyto = [CUgraphNode(init_value=cto[idx]) for idx in range(_graph_length)] if cto is not NULL: free(cto) return (CUresult(err), pyfrom_, pyto, numEdges) {{endif}} {{if 'cuGraphGetEdges_v2' in found_functions}} @cython.embedsignature(True) def cuGraphGetEdges_v2(hGraph, size_t numEdges = 0): """ Returns a graph's dependency edges (12.3+) Returns a list of `hGraph's` dependency edges. Edges are returned via corresponding indices in `from`, `to` and `edgeData`; that is, the node in `to`[i] has a dependency on the node in `from`[i] with data `edgeData`[i]. `from` and `to` may both be NULL, in which case this function only returns the number of edges in `numEdges`. Otherwise, `numEdges` entries will be filled in. If `numEdges` is higher than the actual number of edges, the remaining entries in `from` and `to` will be set to NULL, and the number of edges actually returned will be written to `numEdges`. `edgeData` may alone be NULL, in which case the edges must all have default (zeroed) edge data. Attempting a lossy query via NULL `edgeData` will result in :py:obj:`~.CUDA_ERROR_LOSSY_QUERY`. If `edgeData` is non-NULL then `from` and `to` must be as well. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to get the edges from numEdges : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_LOSSY_QUERY`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` from : List[:py:obj:`~.CUgraphNode`] Location to return edge endpoints to : List[:py:obj:`~.CUgraphNode`] Location to return edge endpoints edgeData : List[:py:obj:`~.CUgraphEdgeData`] Optional location to return edge data numEdges : int See description See Also -------- :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphRemoveDependencies`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ cdef size_t _graph_length = numEdges cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* cfrom_ = NULL pyfrom_ = [] if _graph_length != 0: cfrom_ = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cfrom_ is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) cdef ccuda.CUgraphNode* cto = NULL pyto = [] if _graph_length != 0: cto = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cto is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) cdef ccuda.CUgraphEdgeData* cedgeData = NULL pyedgeData = [] if _graph_length != 0: cedgeData = calloc(_graph_length, sizeof(ccuda.CUgraphEdgeData)) if cedgeData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) err = ccuda.cuGraphGetEdges_v2(chGraph, cfrom_, cto, cedgeData, &numEdges) if CUresult(err) == CUresult(0): pyfrom_ = [CUgraphNode(init_value=cfrom_[idx]) for idx in range(_graph_length)] if cfrom_ is not NULL: free(cfrom_) if CUresult(err) == CUresult(0): pyto = [CUgraphNode(init_value=cto[idx]) for idx in range(_graph_length)] if cto is not NULL: free(cto) if CUresult(err) == CUresult(0): pyedgeData = [CUgraphEdgeData(_ptr=&cedgeData[idx]) for idx in range(_graph_length)] if cedgeData is not NULL: free(cedgeData) return (CUresult(err), pyfrom_, pyto, pyedgeData, numEdges) {{endif}} {{if 'cuGraphNodeGetDependencies' in found_functions}} @cython.embedsignature(True) def cuGraphNodeGetDependencies(hNode, size_t numDependencies = 0): """ Returns a node's dependencies. Returns a list of `node's` dependencies. `dependencies` may be NULL, in which case this function will return the number of dependencies in `numDependencies`. Otherwise, `numDependencies` entries will be filled in. If `numDependencies` is higher than the actual number of dependencies, the remaining entries in `dependencies` will be set to NULL, and the number of nodes actually obtained will be returned in `numDependencies`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to query numDependencies : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` dependencies : List[:py:obj:`~.CUgraphNode`] Pointer to return the dependencies numDependencies : int See description See Also -------- :py:obj:`~.cuGraphNodeGetDependentNodes`, :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphRemoveDependencies` """ cdef size_t _graph_length = numDependencies cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphNode* cdependencies = NULL pydependencies = [] if _graph_length != 0: cdependencies = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) err = ccuda.cuGraphNodeGetDependencies(chNode, cdependencies, &numDependencies) if CUresult(err) == CUresult(0): pydependencies = [CUgraphNode(init_value=cdependencies[idx]) for idx in range(_graph_length)] if cdependencies is not NULL: free(cdependencies) return (CUresult(err), pydependencies, numDependencies) {{endif}} {{if 'cuGraphNodeGetDependencies_v2' in found_functions}} @cython.embedsignature(True) def cuGraphNodeGetDependencies_v2(hNode, size_t numDependencies = 0): """ Returns a node's dependencies (12.3+) Returns a list of `node's` dependencies. `dependencies` may be NULL, in which case this function will return the number of dependencies in `numDependencies`. Otherwise, `numDependencies` entries will be filled in. If `numDependencies` is higher than the actual number of dependencies, the remaining entries in `dependencies` will be set to NULL, and the number of nodes actually obtained will be returned in `numDependencies`. Note that if an edge has non-zero (non-default) edge data and `edgeData` is NULL, this API will return :py:obj:`~.CUDA_ERROR_LOSSY_QUERY`. If `edgeData` is non-NULL, then `dependencies` must be as well. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to query numDependencies : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_LOSSY_QUERY`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` dependencies : List[:py:obj:`~.CUgraphNode`] Pointer to return the dependencies edgeData : List[:py:obj:`~.CUgraphEdgeData`] Optional array to return edge data for each dependency numDependencies : int See description See Also -------- :py:obj:`~.cuGraphNodeGetDependentNodes`, :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphRemoveDependencies` """ cdef size_t _graph_length = numDependencies cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphNode* cdependencies = NULL pydependencies = [] if _graph_length != 0: cdependencies = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) cdef ccuda.CUgraphEdgeData* cedgeData = NULL pyedgeData = [] if _graph_length != 0: cedgeData = calloc(_graph_length, sizeof(ccuda.CUgraphEdgeData)) if cedgeData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) err = ccuda.cuGraphNodeGetDependencies_v2(chNode, cdependencies, cedgeData, &numDependencies) if CUresult(err) == CUresult(0): pydependencies = [CUgraphNode(init_value=cdependencies[idx]) for idx in range(_graph_length)] if cdependencies is not NULL: free(cdependencies) if CUresult(err) == CUresult(0): pyedgeData = [CUgraphEdgeData(_ptr=&cedgeData[idx]) for idx in range(_graph_length)] if cedgeData is not NULL: free(cedgeData) return (CUresult(err), pydependencies, pyedgeData, numDependencies) {{endif}} {{if 'cuGraphNodeGetDependentNodes' in found_functions}} @cython.embedsignature(True) def cuGraphNodeGetDependentNodes(hNode, size_t numDependentNodes = 0): """ Returns a node's dependent nodes. Returns a list of `node's` dependent nodes. `dependentNodes` may be NULL, in which case this function will return the number of dependent nodes in `numDependentNodes`. Otherwise, `numDependentNodes` entries will be filled in. If `numDependentNodes` is higher than the actual number of dependent nodes, the remaining entries in `dependentNodes` will be set to NULL, and the number of nodes actually obtained will be returned in `numDependentNodes`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to query numDependentNodes : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` dependentNodes : List[:py:obj:`~.CUgraphNode`] Pointer to return the dependent nodes numDependentNodes : int See description See Also -------- :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphRemoveDependencies` """ cdef size_t _graph_length = numDependentNodes cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphNode* cdependentNodes = NULL pydependentNodes = [] if _graph_length != 0: cdependentNodes = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cdependentNodes is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) err = ccuda.cuGraphNodeGetDependentNodes(chNode, cdependentNodes, &numDependentNodes) if CUresult(err) == CUresult(0): pydependentNodes = [CUgraphNode(init_value=cdependentNodes[idx]) for idx in range(_graph_length)] if cdependentNodes is not NULL: free(cdependentNodes) return (CUresult(err), pydependentNodes, numDependentNodes) {{endif}} {{if 'cuGraphNodeGetDependentNodes_v2' in found_functions}} @cython.embedsignature(True) def cuGraphNodeGetDependentNodes_v2(hNode, size_t numDependentNodes = 0): """ Returns a node's dependent nodes (12.3+) Returns a list of `node's` dependent nodes. `dependentNodes` may be NULL, in which case this function will return the number of dependent nodes in `numDependentNodes`. Otherwise, `numDependentNodes` entries will be filled in. If `numDependentNodes` is higher than the actual number of dependent nodes, the remaining entries in `dependentNodes` will be set to NULL, and the number of nodes actually obtained will be returned in `numDependentNodes`. Note that if an edge has non-zero (non-default) edge data and `edgeData` is NULL, this API will return :py:obj:`~.CUDA_ERROR_LOSSY_QUERY`. If `edgeData` is non-NULL, then `dependentNodes` must be as well. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to query numDependentNodes : int See description Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_LOSSY_QUERY`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` dependentNodes : List[:py:obj:`~.CUgraphNode`] Pointer to return the dependent nodes edgeData : List[:py:obj:`~.CUgraphEdgeData`] Optional pointer to return edge data for dependent nodes numDependentNodes : int See description See Also -------- :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphGetNodes`, :py:obj:`~.cuGraphGetRootNodes`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphRemoveDependencies` """ cdef size_t _graph_length = numDependentNodes cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphNode* cdependentNodes = NULL pydependentNodes = [] if _graph_length != 0: cdependentNodes = calloc(_graph_length, sizeof(ccuda.CUgraphNode)) if cdependentNodes is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphNode))) cdef ccuda.CUgraphEdgeData* cedgeData = NULL pyedgeData = [] if _graph_length != 0: cedgeData = calloc(_graph_length, sizeof(ccuda.CUgraphEdgeData)) if cedgeData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(_graph_length) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) err = ccuda.cuGraphNodeGetDependentNodes_v2(chNode, cdependentNodes, cedgeData, &numDependentNodes) if CUresult(err) == CUresult(0): pydependentNodes = [CUgraphNode(init_value=cdependentNodes[idx]) for idx in range(_graph_length)] if cdependentNodes is not NULL: free(cdependentNodes) if CUresult(err) == CUresult(0): pyedgeData = [CUgraphEdgeData(_ptr=&cedgeData[idx]) for idx in range(_graph_length)] if cedgeData is not NULL: free(cedgeData) return (CUresult(err), pydependentNodes, pyedgeData, numDependentNodes) {{endif}} {{if 'cuGraphAddDependencies' in found_functions}} @cython.embedsignature(True) def cuGraphAddDependencies(hGraph, from_ : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], to : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies): """ Adds dependency edges to a graph. The number of dependencies to be added is defined by `numDependencies` Elements in `from` and `to` at corresponding indices define a dependency. Each node in `from` and `to` must belong to `hGraph`. If `numDependencies` is 0, elements in `from` and `to` will be ignored. Specifying an existing dependency will return an error. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which dependencies are added from : List[:py:obj:`~.CUgraphNode`] Array of nodes that provide the dependencies to : List[:py:obj:`~.CUgraphNode`] Array of dependent nodes numDependencies : size_t Number of dependencies to be added Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphRemoveDependencies`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ to = [] if to is None else to if not all(isinstance(_x, (CUgraphNode,)) for _x in to): raise TypeError("Argument 'to' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") from_ = [] if from_ is None else from_ if not all(isinstance(_x, (CUgraphNode,)) for _x in from_): raise TypeError("Argument 'from_' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* cfrom_ = NULL if len(from_) > 0: cfrom_ = calloc(len(from_), sizeof(ccuda.CUgraphNode)) if cfrom_ is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(from_)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(from_)): cfrom_[idx] = (from_[idx])._ptr[0] cdef ccuda.CUgraphNode* cto = NULL if len(to) > 0: cto = calloc(len(to), sizeof(ccuda.CUgraphNode)) if cto is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(to)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(to)): cto[idx] = (to[idx])._ptr[0] err = ccuda.cuGraphAddDependencies(chGraph, (from_[0])._ptr if len(from_) == 1 else cfrom_, (to[0])._ptr if len(to) == 1 else cto, numDependencies) if cfrom_ is not NULL: free(cfrom_) if cto is not NULL: free(cto) return (CUresult(err),) {{endif}} {{if 'cuGraphAddDependencies_v2' in found_functions}} @cython.embedsignature(True) def cuGraphAddDependencies_v2(hGraph, from_ : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], to : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], edgeData : Optional[Tuple[CUgraphEdgeData] | List[CUgraphEdgeData]], size_t numDependencies): """ Adds dependency edges to a graph (12.3+) The number of dependencies to be added is defined by `numDependencies` Elements in `from` and `to` at corresponding indices define a dependency. Each node in `from` and `to` must belong to `hGraph`. If `numDependencies` is 0, elements in `from` and `to` will be ignored. Specifying an existing dependency will return an error. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which dependencies are added from : List[:py:obj:`~.CUgraphNode`] Array of nodes that provide the dependencies to : List[:py:obj:`~.CUgraphNode`] Array of dependent nodes edgeData : List[:py:obj:`~.CUgraphEdgeData`] Optional array of edge data. If NULL, default (zeroed) edge data is assumed. numDependencies : size_t Number of dependencies to be added Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphRemoveDependencies`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ edgeData = [] if edgeData is None else edgeData if not all(isinstance(_x, (CUgraphEdgeData,)) for _x in edgeData): raise TypeError("Argument 'edgeData' is not instance of type (expected Tuple[ccuda.CUgraphEdgeData,] or List[ccuda.CUgraphEdgeData,]") to = [] if to is None else to if not all(isinstance(_x, (CUgraphNode,)) for _x in to): raise TypeError("Argument 'to' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") from_ = [] if from_ is None else from_ if not all(isinstance(_x, (CUgraphNode,)) for _x in from_): raise TypeError("Argument 'from_' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* cfrom_ = NULL if len(from_) > 0: cfrom_ = calloc(len(from_), sizeof(ccuda.CUgraphNode)) if cfrom_ is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(from_)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(from_)): cfrom_[idx] = (from_[idx])._ptr[0] cdef ccuda.CUgraphNode* cto = NULL if len(to) > 0: cto = calloc(len(to), sizeof(ccuda.CUgraphNode)) if cto is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(to)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(to)): cto[idx] = (to[idx])._ptr[0] cdef ccuda.CUgraphEdgeData* cedgeData = NULL if len(edgeData) > 0: cedgeData = calloc(len(edgeData), sizeof(ccuda.CUgraphEdgeData)) if cedgeData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(edgeData)) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) for idx in range(len(edgeData)): string.memcpy(&cedgeData[idx], (edgeData[idx])._ptr, sizeof(ccuda.CUgraphEdgeData)) err = ccuda.cuGraphAddDependencies_v2(chGraph, (from_[0])._ptr if len(from_) == 1 else cfrom_, (to[0])._ptr if len(to) == 1 else cto, (edgeData[0])._ptr if len(edgeData) == 1 else cedgeData, numDependencies) if cfrom_ is not NULL: free(cfrom_) if cto is not NULL: free(cto) if cedgeData is not NULL: free(cedgeData) return (CUresult(err),) {{endif}} {{if 'cuGraphRemoveDependencies' in found_functions}} @cython.embedsignature(True) def cuGraphRemoveDependencies(hGraph, from_ : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], to : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies): """ Removes dependency edges from a graph. The number of `dependencies` to be removed is defined by `numDependencies`. Elements in `from` and `to` at corresponding indices define a dependency. Each node in `from` and `to` must belong to `hGraph`. If `numDependencies` is 0, elements in `from` and `to` will be ignored. Specifying a non-existing dependency will return an error. Dependencies cannot be removed from graphs which contain allocation or free nodes. Any attempt to do so will return an error. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph from which to remove dependencies from : List[:py:obj:`~.CUgraphNode`] Array of nodes that provide the dependencies to : List[:py:obj:`~.CUgraphNode`] Array of dependent nodes numDependencies : size_t Number of dependencies to be removed Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ to = [] if to is None else to if not all(isinstance(_x, (CUgraphNode,)) for _x in to): raise TypeError("Argument 'to' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") from_ = [] if from_ is None else from_ if not all(isinstance(_x, (CUgraphNode,)) for _x in from_): raise TypeError("Argument 'from_' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* cfrom_ = NULL if len(from_) > 0: cfrom_ = calloc(len(from_), sizeof(ccuda.CUgraphNode)) if cfrom_ is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(from_)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(from_)): cfrom_[idx] = (from_[idx])._ptr[0] cdef ccuda.CUgraphNode* cto = NULL if len(to) > 0: cto = calloc(len(to), sizeof(ccuda.CUgraphNode)) if cto is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(to)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(to)): cto[idx] = (to[idx])._ptr[0] err = ccuda.cuGraphRemoveDependencies(chGraph, (from_[0])._ptr if len(from_) == 1 else cfrom_, (to[0])._ptr if len(to) == 1 else cto, numDependencies) if cfrom_ is not NULL: free(cfrom_) if cto is not NULL: free(cto) return (CUresult(err),) {{endif}} {{if 'cuGraphRemoveDependencies_v2' in found_functions}} @cython.embedsignature(True) def cuGraphRemoveDependencies_v2(hGraph, from_ : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], to : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], edgeData : Optional[Tuple[CUgraphEdgeData] | List[CUgraphEdgeData]], size_t numDependencies): """ Removes dependency edges from a graph (12.3+) The number of `dependencies` to be removed is defined by `numDependencies`. Elements in `from` and `to` at corresponding indices define a dependency. Each node in `from` and `to` must belong to `hGraph`. If `numDependencies` is 0, elements in `from` and `to` will be ignored. Specifying an edge that does not exist in the graph, with data matching `edgeData`, results in an error. `edgeData` is nullable, which is equivalent to passing default (zeroed) data for each edge. Dependencies cannot be removed from graphs which contain allocation or free nodes. Any attempt to do so will return an error. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph from which to remove dependencies from : List[:py:obj:`~.CUgraphNode`] Array of nodes that provide the dependencies to : List[:py:obj:`~.CUgraphNode`] Array of dependent nodes edgeData : List[:py:obj:`~.CUgraphEdgeData`] Optional array of edge data. If NULL, edge data is assumed to be default (zeroed). numDependencies : size_t Number of dependencies to be removed Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphAddDependencies`, :py:obj:`~.cuGraphGetEdges`, :py:obj:`~.cuGraphNodeGetDependencies`, :py:obj:`~.cuGraphNodeGetDependentNodes` """ edgeData = [] if edgeData is None else edgeData if not all(isinstance(_x, (CUgraphEdgeData,)) for _x in edgeData): raise TypeError("Argument 'edgeData' is not instance of type (expected Tuple[ccuda.CUgraphEdgeData,] or List[ccuda.CUgraphEdgeData,]") to = [] if to is None else to if not all(isinstance(_x, (CUgraphNode,)) for _x in to): raise TypeError("Argument 'to' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") from_ = [] if from_ is None else from_ if not all(isinstance(_x, (CUgraphNode,)) for _x in from_): raise TypeError("Argument 'from_' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphNode* cfrom_ = NULL if len(from_) > 0: cfrom_ = calloc(len(from_), sizeof(ccuda.CUgraphNode)) if cfrom_ is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(from_)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(from_)): cfrom_[idx] = (from_[idx])._ptr[0] cdef ccuda.CUgraphNode* cto = NULL if len(to) > 0: cto = calloc(len(to), sizeof(ccuda.CUgraphNode)) if cto is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(to)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(to)): cto[idx] = (to[idx])._ptr[0] cdef ccuda.CUgraphEdgeData* cedgeData = NULL if len(edgeData) > 0: cedgeData = calloc(len(edgeData), sizeof(ccuda.CUgraphEdgeData)) if cedgeData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(edgeData)) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) for idx in range(len(edgeData)): string.memcpy(&cedgeData[idx], (edgeData[idx])._ptr, sizeof(ccuda.CUgraphEdgeData)) err = ccuda.cuGraphRemoveDependencies_v2(chGraph, (from_[0])._ptr if len(from_) == 1 else cfrom_, (to[0])._ptr if len(to) == 1 else cto, (edgeData[0])._ptr if len(edgeData) == 1 else cedgeData, numDependencies) if cfrom_ is not NULL: free(cfrom_) if cto is not NULL: free(cto) if cedgeData is not NULL: free(cedgeData) return (CUresult(err),) {{endif}} {{if 'cuGraphDestroyNode' in found_functions}} @cython.embedsignature(True) def cuGraphDestroyNode(hNode): """ Remove a node from the graph. Removes `hNode` from its graph. This operation also severs any dependencies of other nodes on `hNode` and vice versa. Nodes which belong to a graph which contains allocation or free nodes cannot be destroyed. Any attempt to do so will return an error. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to remove Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphAddEmptyNode`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphAddMemsetNode` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode err = ccuda.cuGraphDestroyNode(chNode) return (CUresult(err),) {{endif}} {{if 'cuGraphInstantiateWithFlags' in found_functions}} @cython.embedsignature(True) def cuGraphInstantiate(hGraph, unsigned long long flags): """ Creates an executable graph from a graph. Instantiates `hGraph` as an executable graph. The graph is validated for any structural constraints or intra-node constraints which were not previously validated. If instantiation is successful, a handle to the instantiated graph is returned in `phGraphExec`. The `flags` parameter controls the behavior of instantiation and subsequent graph launches. Valid flags are: - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH`, which configures a graph containing memory allocation nodes to automatically free any unfreed memory allocations before the graph is relaunched. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_DEVICE_LAUNCH`, which configures the graph for launch from the device. If this flag is passed, the executable graph handle returned can be used to launch the graph from both the host and device. This flag can only be used on platforms which support unified addressing. This flag cannot be used in conjunction with :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH`. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_USE_NODE_PRIORITY`, which causes the graph to use the priorities from the per-node attributes rather than the priority of the launch stream during execution. Note that priorities are only available on kernel nodes, and are copied from stream priority during stream capture. If `hGraph` contains any allocation or free nodes, there can be at most one executable graph in existence for that graph at a time. An attempt to instantiate a second executable graph before destroying the first with :py:obj:`~.cuGraphExecDestroy` will result in an error. The same also applies if `hGraph` contains any device-updatable kernel nodes. If `hGraph` contains kernels which call device-side cudaGraphLaunch() from multiple contexts, this will result in an error. Graphs instantiated for launch on the device have additional restrictions which do not apply to host graphs: - The graph's nodes must reside on a single context. - The graph can only contain kernel nodes, memcpy nodes, memset nodes, and child graph nodes. - The graph cannot be empty and must contain at least one kernel, memcpy, or memset node. Operation-specific restrictions are outlined below. - Kernel nodes: - Use of CUDA Dynamic Parallelism is not permitted. - Cooperative launches are permitted as long as MPS is not in use. - Memcpy nodes: - Only copies involving device memory and/or pinned device-mapped host memory are permitted. - Copies involving CUDA arrays are not permitted. - Both operands must be accessible from the current context, and the current context must match the context of other nodes in the graph. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to instantiate flags : unsigned long long Flags to control instantiation. See :py:obj:`~.CUgraphInstantiate_flags`. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phGraphExec : :py:obj:`~.CUgraphExec` Returns instantiated graph See Also -------- :py:obj:`~.cuGraphInstantiate`, :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphUpload`, :py:obj:`~.cuGraphLaunch`, :py:obj:`~.cuGraphExecDestroy` """ cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphExec phGraphExec = CUgraphExec() err = ccuda.cuGraphInstantiate(phGraphExec._ptr, chGraph, flags) return (CUresult(err), phGraphExec) {{endif}} {{if 'cuGraphInstantiateWithParams' in found_functions}} @cython.embedsignature(True) def cuGraphInstantiateWithParams(hGraph, instantiateParams : Optional[CUDA_GRAPH_INSTANTIATE_PARAMS]): """ Creates an executable graph from a graph. Instantiates `hGraph` as an executable graph according to the `instantiateParams` structure. The graph is validated for any structural constraints or intra-node constraints which were not previously validated. If instantiation is successful, a handle to the instantiated graph is returned in `phGraphExec`. `instantiateParams` controls the behavior of instantiation and subsequent graph launches, as well as returning more detailed information in the event of an error. :py:obj:`~.CUDA_GRAPH_INSTANTIATE_PARAMS` is defined as: **View CUDA Toolkit Documentation for a C++ code example** The `flags` field controls the behavior of instantiation and subsequent graph launches. Valid flags are: - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH`, which configures a graph containing memory allocation nodes to automatically free any unfreed memory allocations before the graph is relaunched. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_UPLOAD`, which will perform an upload of the graph into `hUploadStream` once the graph has been instantiated. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_DEVICE_LAUNCH`, which configures the graph for launch from the device. If this flag is passed, the executable graph handle returned can be used to launch the graph from both the host and device. This flag can only be used on platforms which support unified addressing. This flag cannot be used in conjunction with :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH`. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_USE_NODE_PRIORITY`, which causes the graph to use the priorities from the per-node attributes rather than the priority of the launch stream during execution. Note that priorities are only available on kernel nodes, and are copied from stream priority during stream capture. If `hGraph` contains any allocation or free nodes, there can be at most one executable graph in existence for that graph at a time. An attempt to instantiate a second executable graph before destroying the first with :py:obj:`~.cuGraphExecDestroy` will result in an error. The same also applies if `hGraph` contains any device-updatable kernel nodes. If `hGraph` contains kernels which call device-side cudaGraphLaunch() from multiple contexts, this will result in an error. Graphs instantiated for launch on the device have additional restrictions which do not apply to host graphs: - The graph's nodes must reside on a single context. - The graph can only contain kernel nodes, memcpy nodes, memset nodes, and child graph nodes. - The graph cannot be empty and must contain at least one kernel, memcpy, or memset node. Operation-specific restrictions are outlined below. - Kernel nodes: - Use of CUDA Dynamic Parallelism is not permitted. - Cooperative launches are permitted as long as MPS is not in use. - Memcpy nodes: - Only copies involving device memory and/or pinned device-mapped host memory are permitted. - Copies involving CUDA arrays are not permitted. - Both operands must be accessible from the current context, and the current context must match the context of other nodes in the graph. In the event of an error, the `result_out` and `hErrNode_out` fields will contain more information about the nature of the error. Possible error reporting includes: - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_ERROR`, if passed an invalid value or if an unexpected error occurred which is described by the return value of the function. `hErrNode_out` will be set to NULL. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_INVALID_STRUCTURE`, if the graph structure is invalid. `hErrNode_out` will be set to one of the offending nodes. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_NODE_OPERATION_NOT_SUPPORTED`, if the graph is instantiated for device launch but contains a node of an unsupported node type, or a node which performs unsupported operations, such as use of CUDA dynamic parallelism within a kernel node. `hErrNode_out` will be set to this node. - :py:obj:`~.CUDA_GRAPH_INSTANTIATE_MULTIPLE_CTXS_NOT_SUPPORTED`, if the graph is instantiated for device launch but a node’s context differs from that of another node. This error can also be returned if a graph is not instantiated for device launch and it contains kernels which call device-side cudaGraphLaunch() from multiple contexts. `hErrNode_out` will be set to this node. If instantiation is successful, `result_out` will be set to :py:obj:`~.CUDA_GRAPH_INSTANTIATE_SUCCESS`, and `hErrNode_out` will be set to NULL. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to instantiate instantiateParams : :py:obj:`~.CUDA_GRAPH_INSTANTIATE_PARAMS` Instantiation parameters Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, phGraphExec : :py:obj:`~.CUgraphExec` Returns instantiated graph See Also -------- :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphInstantiate`, :py:obj:`~.cuGraphExecDestroy` """ cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphExec phGraphExec = CUgraphExec() cdef ccuda.CUDA_GRAPH_INSTANTIATE_PARAMS* cinstantiateParams_ptr = instantiateParams._ptr if instantiateParams != None else NULL err = ccuda.cuGraphInstantiateWithParams(phGraphExec._ptr, chGraph, cinstantiateParams_ptr) return (CUresult(err), phGraphExec) {{endif}} {{if 'cuGraphExecGetFlags' in found_functions}} @cython.embedsignature(True) def cuGraphExecGetFlags(hGraphExec): """ Query the instantiation flags of an executable graph. Returns the flags that were passed to instantiation for the given executable graph. :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_UPLOAD` will not be returned by this API as it does not affect the resulting executable graph. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph to query Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, flags : :py:obj:`~.cuuint64_t` Returns the instantiation flags See Also -------- :py:obj:`~.cuGraphInstantiate`, :py:obj:`~.cuGraphInstantiateWithParams` """ cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef cuuint64_t flags = cuuint64_t() err = ccuda.cuGraphExecGetFlags(chGraphExec, flags._ptr) return (CUresult(err), flags) {{endif}} {{if 'cuGraphExecKernelNodeSetParams_v2' in found_functions}} @cython.embedsignature(True) def cuGraphExecKernelNodeSetParams(hGraphExec, hNode, nodeParams : Optional[CUDA_KERNEL_NODE_PARAMS]): """ Sets the parameters for a kernel node in the given graphExec. Sets the parameters of a kernel node in an executable graph `hGraphExec`. The node is identified by the corresponding node `hNode` in the non-executable graph, from which the executable graph was instantiated. `hNode` must not have been removed from the original graph. All `nodeParams` fields may change, but the following restrictions apply to `func` updates: - The owning context of the function cannot change. - A node whose function originally did not use CUDA dynamic parallelism cannot be updated to a function which uses CDP - A node whose function originally did not make device-side update calls cannot be updated to a function which makes device-side update calls. - If `hGraphExec` was not instantiated for device launch, a node whose function originally did not use device-side cudaGraphLaunch() cannot be updated to a function which uses device-side cudaGraphLaunch() unless the node resides on the same context as nodes which contained such calls at instantiate-time. If no such calls were present at instantiation, these updates cannot be performed at all. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. If `hNode` is a device-updatable kernel node, the next upload/launch of `hGraphExec` will overwrite any previous device-side updates. Additionally, applying host updates to a device-updatable kernel node while it is being updated from the device will result in undefined behavior. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` kernel node from the graph from which graphExec was instantiated nodeParams : :py:obj:`~.CUDA_KERNEL_NODE_PARAMS` Updated Parameters to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddKernelNode`, :py:obj:`~.cuGraphKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUDA_KERNEL_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExecKernelNodeSetParams(chGraphExec, chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphExecMemcpyNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecMemcpyNodeSetParams(hGraphExec, hNode, copyParams : Optional[CUDA_MEMCPY3D], ctx): """ Sets the parameters for a memcpy node in the given graphExec. Updates the work represented by `hNode` in `hGraphExec` as though `hNode` had contained `copyParams` at instantiation. hNode must remain in the graph which was used to instantiate `hGraphExec`. Changed edges to and from hNode are ignored. The source and destination memory in `copyParams` must be allocated from the same contexts as the original source and destination memory. Both the instantiation-time memory operands and the memory operands in `copyParams` must be 1-dimensional. Zero-length operations are not supported. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. hNode is also not modified by this call. Returns CUDA_ERROR_INVALID_VALUE if the memory operands' mappings changed or either the original or new memory operands are multidimensional. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Memcpy node from the graph which was used to instantiate graphExec copyParams : :py:obj:`~.CUDA_MEMCPY3D` The updated parameters to set ctx : :py:obj:`~.CUcontext` Context on which to run the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddMemcpyNode`, :py:obj:`~.cuGraphMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUDA_MEMCPY3D* ccopyParams_ptr = copyParams._ptr if copyParams != None else NULL err = ccuda.cuGraphExecMemcpyNodeSetParams(chGraphExec, chNode, ccopyParams_ptr, cctx) return (CUresult(err),) {{endif}} {{if 'cuGraphExecMemsetNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecMemsetNodeSetParams(hGraphExec, hNode, memsetParams : Optional[CUDA_MEMSET_NODE_PARAMS], ctx): """ Sets the parameters for a memset node in the given graphExec. Updates the work represented by `hNode` in `hGraphExec` as though `hNode` had contained `memsetParams` at instantiation. hNode must remain in the graph which was used to instantiate `hGraphExec`. Changed edges to and from hNode are ignored. Zero sized operations are not supported. The new destination pointer in memsetParams must be to the same kind of allocation as the original destination pointer and have the same context association and device mapping as the original destination pointer. Both the value and pointer address may be updated. Changing other aspects of the memset (width, height, element size or pitch) may cause the update to be rejected. Specifically, for 2d memsets, all dimension changes are rejected. For 1d memsets, changes in height are explicitly rejected and other changes are oportunistically allowed if the resulting work maps onto the work resources already allocated for the node. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. hNode is also not modified by this call. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Memset node from the graph which was used to instantiate graphExec memsetParams : :py:obj:`~.CUDA_MEMSET_NODE_PARAMS` The updated parameters to set ctx : :py:obj:`~.CUcontext` Context on which to run the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddMemsetNode`, :py:obj:`~.cuGraphMemsetNodeSetParams`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUDA_MEMSET_NODE_PARAMS* cmemsetParams_ptr = memsetParams._ptr if memsetParams != None else NULL err = ccuda.cuGraphExecMemsetNodeSetParams(chGraphExec, chNode, cmemsetParams_ptr, cctx) return (CUresult(err),) {{endif}} {{if 'cuGraphExecHostNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecHostNodeSetParams(hGraphExec, hNode, nodeParams : Optional[CUDA_HOST_NODE_PARAMS]): """ Sets the parameters for a host node in the given graphExec. Updates the work represented by `hNode` in `hGraphExec` as though `hNode` had contained `nodeParams` at instantiation. hNode must remain in the graph which was used to instantiate `hGraphExec`. Changed edges to and from hNode are ignored. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. hNode is also not modified by this call. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Host node from the graph which was used to instantiate graphExec nodeParams : :py:obj:`~.CUDA_HOST_NODE_PARAMS` The updated parameters to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddHostNode`, :py:obj:`~.cuGraphHostNodeSetParams`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUDA_HOST_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExecHostNodeSetParams(chGraphExec, chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphExecChildGraphNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecChildGraphNodeSetParams(hGraphExec, hNode, childGraph): """ Updates node parameters in the child graph node in the given graphExec. Updates the work represented by `hNode` in `hGraphExec` as though the nodes contained in `hNode's` graph had the parameters contained in `childGraph's` nodes at instantiation. `hNode` must remain in the graph which was used to instantiate `hGraphExec`. Changed edges to and from `hNode` are ignored. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. The topology of `childGraph`, as well as the node insertion order, must match that of the graph contained in `hNode`. See :py:obj:`~.cuGraphExecUpdate()` for a list of restrictions on what can be updated in an instantiated graph. The update is recursive, so child graph nodes contained within the top level child graph will also be updated. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Host node from the graph which was used to instantiate graphExec childGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` The graph supplying the updated parameters Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddChildGraphNode`, :py:obj:`~.cuGraphChildGraphNodeGetGraph`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraph cchildGraph if childGraph is None: cchildGraph = 0 elif isinstance(childGraph, (CUgraph,)): pchildGraph = int(childGraph) cchildGraph = pchildGraph else: pchildGraph = int(CUgraph(childGraph)) cchildGraph = pchildGraph cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec err = ccuda.cuGraphExecChildGraphNodeSetParams(chGraphExec, chNode, cchildGraph) return (CUresult(err),) {{endif}} {{if 'cuGraphExecEventRecordNodeSetEvent' in found_functions}} @cython.embedsignature(True) def cuGraphExecEventRecordNodeSetEvent(hGraphExec, hNode, event): """ Sets the event for an event record node in the given graphExec. Sets the event of an event record node in an executable graph `hGraphExec`. The node is identified by the corresponding node `hNode` in the non-executable graph, from which the executable graph was instantiated. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` event record node from the graph from which graphExec was instantiated event : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Updated event to use Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddEventRecordNode`, :py:obj:`~.cuGraphEventRecordNodeGetEvent`, :py:obj:`~.cuGraphEventWaitNodeSetEvent`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec err = ccuda.cuGraphExecEventRecordNodeSetEvent(chGraphExec, chNode, cevent) return (CUresult(err),) {{endif}} {{if 'cuGraphExecEventWaitNodeSetEvent' in found_functions}} @cython.embedsignature(True) def cuGraphExecEventWaitNodeSetEvent(hGraphExec, hNode, event): """ Sets the event for an event wait node in the given graphExec. Sets the event of an event wait node in an executable graph `hGraphExec`. The node is identified by the corresponding node `hNode` in the non-executable graph, from which the executable graph was instantiated. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` event wait node from the graph from which graphExec was instantiated event : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Updated event to use Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddEventWaitNode`, :py:obj:`~.cuGraphEventWaitNodeGetEvent`, :py:obj:`~.cuGraphEventRecordNodeSetEvent`, :py:obj:`~.cuEventRecordWithFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUevent cevent if event is None: cevent = 0 elif isinstance(event, (CUevent,)): pevent = int(event) cevent = pevent else: pevent = int(CUevent(event)) cevent = pevent cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec err = ccuda.cuGraphExecEventWaitNodeSetEvent(chGraphExec, chNode, cevent) return (CUresult(err),) {{endif}} {{if 'cuGraphExecExternalSemaphoresSignalNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecExternalSemaphoresSignalNodeSetParams(hGraphExec, hNode, nodeParams : Optional[CUDA_EXT_SEM_SIGNAL_NODE_PARAMS]): """ Sets the parameters for an external semaphore signal node in the given graphExec. Sets the parameters of an external semaphore signal node in an executable graph `hGraphExec`. The node is identified by the corresponding node `hNode` in the non-executable graph, from which the executable graph was instantiated. `hNode` must not have been removed from the original graph. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. Changing `nodeParams->numExtSems` is not supported. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` semaphore signal node from the graph from which graphExec was instantiated nodeParams : :py:obj:`~.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS` Updated Parameters to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresSignalNode`, :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresWaitNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExecExternalSemaphoresSignalNodeSetParams(chGraphExec, chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphExecExternalSemaphoresWaitNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecExternalSemaphoresWaitNodeSetParams(hGraphExec, hNode, nodeParams : Optional[CUDA_EXT_SEM_WAIT_NODE_PARAMS]): """ Sets the parameters for an external semaphore wait node in the given graphExec. Sets the parameters of an external semaphore wait node in an executable graph `hGraphExec`. The node is identified by the corresponding node `hNode` in the non-executable graph, from which the executable graph was instantiated. `hNode` must not have been removed from the original graph. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. Changing `nodeParams->numExtSems` is not supported. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` semaphore wait node from the graph from which graphExec was instantiated nodeParams : :py:obj:`~.CUDA_EXT_SEM_WAIT_NODE_PARAMS` Updated Parameters to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphExecNodeSetParams`, :py:obj:`~.cuGraphAddExternalSemaphoresWaitNode`, :py:obj:`~.cuImportExternalSemaphore`, :py:obj:`~.cuSignalExternalSemaphoresAsync`, :py:obj:`~.cuWaitExternalSemaphoresAsync`, :py:obj:`~.cuGraphExecKernelNodeSetParams`, :py:obj:`~.cuGraphExecMemcpyNodeSetParams`, :py:obj:`~.cuGraphExecMemsetNodeSetParams`, :py:obj:`~.cuGraphExecHostNodeSetParams`, :py:obj:`~.cuGraphExecChildGraphNodeSetParams`, :py:obj:`~.cuGraphExecEventRecordNodeSetEvent`, :py:obj:`~.cuGraphExecEventWaitNodeSetEvent`, :py:obj:`~.cuGraphExecExternalSemaphoresSignalNodeSetParams`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExecExternalSemaphoresWaitNodeSetParams(chGraphExec, chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphNodeSetEnabled' in found_functions}} @cython.embedsignature(True) def cuGraphNodeSetEnabled(hGraphExec, hNode, unsigned int isEnabled): """ Enables or disables the specified node in the given graphExec. Sets `hNode` to be either enabled or disabled. Disabled nodes are functionally equivalent to empty nodes until they are reenabled. Existing node parameters are not affected by disabling/enabling the node. The node is identified by the corresponding node `hNode` in the non- executable graph, from which the executable graph was instantiated. `hNode` must not have been removed from the original graph. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. If `hNode` is a device-updatable kernel node, the next upload/launch of `hGraphExec` will overwrite any previous device-side updates. Additionally, applying host updates to a device-updatable kernel node while it is being updated from the device will result in undefined behavior. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node from the graph from which graphExec was instantiated isEnabled : unsigned int Node is enabled if != 0, otherwise the node is disabled Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, See Also -------- :py:obj:`~.cuGraphNodeGetEnabled`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` :py:obj:`~.cuGraphLaunch` Notes ----- Currently only kernel, memset and memcpy nodes are supported. """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec err = ccuda.cuGraphNodeSetEnabled(chGraphExec, chNode, isEnabled) return (CUresult(err),) {{endif}} {{if 'cuGraphNodeGetEnabled' in found_functions}} @cython.embedsignature(True) def cuGraphNodeGetEnabled(hGraphExec, hNode): """ Query whether a node in the given graphExec is enabled. Sets isEnabled to 1 if `hNode` is enabled, or 0 if `hNode` is disabled. The node is identified by the corresponding node `hNode` in the non- executable graph, from which the executable graph was instantiated. `hNode` must not have been removed from the original graph. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to set the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node from the graph from which graphExec was instantiated Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, isEnabled : unsigned int Location to return the enabled status of the node See Also -------- :py:obj:`~.cuGraphNodeSetEnabled`, :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` :py:obj:`~.cuGraphLaunch` Notes ----- Currently only kernel, memset and memcpy nodes are supported. This function will not reflect device-side updates for device-updatable kernel nodes. """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef unsigned int isEnabled = 0 err = ccuda.cuGraphNodeGetEnabled(chGraphExec, chNode, &isEnabled) return (CUresult(err), isEnabled) {{endif}} {{if 'cuGraphUpload' in found_functions}} @cython.embedsignature(True) def cuGraphUpload(hGraphExec, hStream): """ Uploads an executable graph in a stream. Uploads `hGraphExec` to the device in `hStream` without executing it. Uploads of the same `hGraphExec` will be serialized. Each upload is ordered behind both any previous work in `hStream` and any previous launches of `hGraphExec`. Uses memory cached by `stream` to back the allocations owned by `hGraphExec`. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` Executable graph to upload hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream in which to upload the graph Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphInstantiate`, :py:obj:`~.cuGraphLaunch`, :py:obj:`~.cuGraphExecDestroy` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec err = ccuda.cuGraphUpload(chGraphExec, chStream) return (CUresult(err),) {{endif}} {{if 'cuGraphLaunch' in found_functions}} @cython.embedsignature(True) def cuGraphLaunch(hGraphExec, hStream): """ Launches an executable graph in a stream. Executes `hGraphExec` in `hStream`. Only one instance of `hGraphExec` may be executing at a time. Each launch is ordered behind both any previous work in `hStream` and any previous launches of `hGraphExec`. To execute a graph concurrently, it must be instantiated multiple times into multiple executable graphs. If any allocations created by `hGraphExec` remain unfreed (from a previous launch) and `hGraphExec` was not instantiated with :py:obj:`~.CUDA_GRAPH_INSTANTIATE_FLAG_AUTO_FREE_ON_LAUNCH`, the launch will fail with :py:obj:`~.CUDA_ERROR_INVALID_VALUE`. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` Executable graph to launch hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream in which to launch the graph Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphInstantiate`, :py:obj:`~.cuGraphUpload`, :py:obj:`~.cuGraphExecDestroy` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec err = ccuda.cuGraphLaunch(chGraphExec, chStream) return (CUresult(err),) {{endif}} {{if 'cuGraphExecDestroy' in found_functions}} @cython.embedsignature(True) def cuGraphExecDestroy(hGraphExec): """ Destroys an executable graph. Destroys the executable graph specified by `hGraphExec`, as well as all of its executable nodes. If the executable graph is in-flight, it will not be terminated, but rather freed asynchronously on completion. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` Executable graph to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphInstantiate`, :py:obj:`~.cuGraphUpload`, :py:obj:`~.cuGraphLaunch` """ cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec err = ccuda.cuGraphExecDestroy(chGraphExec) return (CUresult(err),) {{endif}} {{if 'cuGraphDestroy' in found_functions}} @cython.embedsignature(True) def cuGraphDestroy(hGraph): """ Destroys a graph. Destroys the graph specified by `hGraph`, as well as all of its nodes. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuGraphCreate` """ cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph err = ccuda.cuGraphDestroy(chGraph) return (CUresult(err),) {{endif}} {{if 'cuGraphExecUpdate_v2' in found_functions}} @cython.embedsignature(True) def cuGraphExecUpdate(hGraphExec, hGraph): """ Check whether an executable graph can be updated with a graph and perform the update if possible. Updates the node parameters in the instantiated graph specified by `hGraphExec` with the node parameters in a topologically identical graph specified by `hGraph`. Limitations: - Kernel nodes: - The owning context of the function cannot change. - A node whose function originally did not use CUDA dynamic parallelism cannot be updated to a function which uses CDP. - A node whose function originally did not make device-side update calls cannot be updated to a function which makes device-side update calls. - A cooperative node cannot be updated to a non-cooperative node, and vice-versa. - If the graph was instantiated with CUDA_GRAPH_INSTANTIATE_FLAG_USE_NODE_PRIORITY, the priority attribute cannot change. Equality is checked on the originally requested priority values, before they are clamped to the device's supported range. - If `hGraphExec` was not instantiated for device launch, a node whose function originally did not use device-side cudaGraphLaunch() cannot be updated to a function which uses device-side cudaGraphLaunch() unless the node resides on the same context as nodes which contained such calls at instantiate-time. If no such calls were present at instantiation, these updates cannot be performed at all. - Neither `hGraph` nor `hGraphExec` may contain device-updatable kernel nodes. - Memset and memcpy nodes: - The CUDA device(s) to which the operand(s) was allocated/mapped cannot change. - The source/destination memory must be allocated from the same contexts as the original source/destination memory. - For 2d memsets, only address and assinged value may be updated. - For 1d memsets, updating dimensions is also allowed, but may fail if the resulting operation doesn't map onto the work resources already allocated for the node. - Additional memcpy node restrictions: - Changing either the source or destination memory type(i.e. CU_MEMORYTYPE_DEVICE, CU_MEMORYTYPE_ARRAY, etc.) is not supported. - External semaphore wait nodes and record nodes: - Changing the number of semaphores is not supported. - Conditional nodes: - Changing node parameters is not supported. - Changeing parameters of nodes within the conditional body graph is subject to the rules above. - Conditional handle flags and default values are updated as part of the graph update. Note: The API may add further restrictions in future releases. The return code should always be checked. cuGraphExecUpdate sets the result member of `resultInfo` to CU_GRAPH_EXEC_UPDATE_ERROR_TOPOLOGY_CHANGED under the following conditions: - The count of nodes directly in `hGraphExec` and `hGraph` differ, in which case resultInfo->errorNode is set to NULL. - `hGraph` has more exit nodes than `hGraph`, in which case resultInfo->errorNode is set to one of the exit nodes in hGraph. - A node in `hGraph` has a different number of dependencies than the node from `hGraphExec` it is paired with, in which case resultInfo->errorNode is set to the node from `hGraph`. - A node in `hGraph` has a dependency that does not match with the corresponding dependency of the paired node from `hGraphExec`. resultInfo->errorNode will be set to the node from `hGraph`. resultInfo->errorFromNode will be set to the mismatched dependency. The dependencies are paired based on edge order and a dependency does not match when the nodes are already paired based on other edges examined in the graph. cuGraphExecUpdate sets the result member of `resultInfo` to: - CU_GRAPH_EXEC_UPDATE_ERROR if passed an invalid value. - CU_GRAPH_EXEC_UPDATE_ERROR_TOPOLOGY_CHANGED if the graph topology changed - CU_GRAPH_EXEC_UPDATE_ERROR_NODE_TYPE_CHANGED if the type of a node changed, in which case `hErrorNode_out` is set to the node from `hGraph`. - CU_GRAPH_EXEC_UPDATE_ERROR_UNSUPPORTED_FUNCTION_CHANGE if the function changed in an unsupported way(see note above), in which case `hErrorNode_out` is set to the node from `hGraph` - CU_GRAPH_EXEC_UPDATE_ERROR_PARAMETERS_CHANGED if any parameters to a node changed in a way that is not supported, in which case `hErrorNode_out` is set to the node from `hGraph`. - CU_GRAPH_EXEC_UPDATE_ERROR_ATTRIBUTES_CHANGED if any attributes of a node changed in a way that is not supported, in which case `hErrorNode_out` is set to the node from `hGraph`. - CU_GRAPH_EXEC_UPDATE_ERROR_NOT_SUPPORTED if something about a node is unsupported, like the node's type or configuration, in which case `hErrorNode_out` is set to the node from `hGraph` If the update fails for a reason not listed above, the result member of `resultInfo` will be set to CU_GRAPH_EXEC_UPDATE_ERROR. If the update succeeds, the result member will be set to CU_GRAPH_EXEC_UPDATE_SUCCESS. cuGraphExecUpdate returns CUDA_SUCCESS when the updated was performed successfully. It returns CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE if the graph update was not performed because it included changes which violated constraints specific to instantiated graph update. Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The instantiated graph to be updated hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` The graph containing the updated parameters Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_GRAPH_EXEC_UPDATE_FAILURE`, resultInfo : :py:obj:`~.CUgraphExecUpdateResultInfo` the error info structure See Also -------- :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef CUgraphExecUpdateResultInfo resultInfo = CUgraphExecUpdateResultInfo() err = ccuda.cuGraphExecUpdate(chGraphExec, chGraph, resultInfo._ptr) return (CUresult(err), resultInfo) {{endif}} {{if 'cuGraphKernelNodeCopyAttributes' in found_functions}} @cython.embedsignature(True) def cuGraphKernelNodeCopyAttributes(dst, src): """ Copies attributes from source node to destination node. Copies attributes from source node `src` to destination node `dst`. Both node must have the same context. Parameters ---------- dst : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Destination node src : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Source node For list of attributes see :py:obj:`~.CUkernelNodeAttrID` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.CUaccessPolicyWindow` """ cdef ccuda.CUgraphNode csrc if src is None: csrc = 0 elif isinstance(src, (CUgraphNode,)): psrc = int(src) csrc = psrc else: psrc = int(CUgraphNode(src)) csrc = psrc cdef ccuda.CUgraphNode cdst if dst is None: cdst = 0 elif isinstance(dst, (CUgraphNode,)): pdst = int(dst) cdst = pdst else: pdst = int(CUgraphNode(dst)) cdst = pdst err = ccuda.cuGraphKernelNodeCopyAttributes(cdst, csrc) return (CUresult(err),) {{endif}} {{if 'cuGraphKernelNodeGetAttribute' in found_functions}} @cython.embedsignature(True) def cuGraphKernelNodeGetAttribute(hNode, attr not None : CUkernelNodeAttrID): """ Queries node attribute. Queries attribute `attr` from node `hNode` and stores it in corresponding member of `value_out`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` attr : :py:obj:`~.CUkernelNodeAttrID` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` value_out : :py:obj:`~.CUkernelNodeAttrValue` See Also -------- :py:obj:`~.CUaccessPolicyWindow` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUkernelNodeAttrID cattr = attr.value cdef CUkernelNodeAttrValue value_out = CUkernelNodeAttrValue() err = ccuda.cuGraphKernelNodeGetAttribute(chNode, cattr, value_out._ptr) return (CUresult(err), value_out) {{endif}} {{if 'cuGraphKernelNodeSetAttribute' in found_functions}} @cython.embedsignature(True) def cuGraphKernelNodeSetAttribute(hNode, attr not None : CUkernelNodeAttrID, value : Optional[CUkernelNodeAttrValue]): """ Sets node attribute. Sets attribute `attr` on node `hNode` from corresponding attribute of `value`. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` attr : :py:obj:`~.CUkernelNodeAttrID` value : :py:obj:`~.CUkernelNodeAttrValue` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` See Also -------- :py:obj:`~.CUaccessPolicyWindow` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUkernelNodeAttrID cattr = attr.value cdef ccuda.CUkernelNodeAttrValue* cvalue_ptr = value._ptr if value != None else NULL err = ccuda.cuGraphKernelNodeSetAttribute(chNode, cattr, cvalue_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphDebugDotPrint' in found_functions}} @cython.embedsignature(True) def cuGraphDebugDotPrint(hGraph, char* path, unsigned int flags): """ Write a DOT file describing graph structure. Using the provided `hGraph`, write to `path` a DOT formatted description of the graph. By default this includes the graph topology, node types, node id, kernel names and memcpy direction. `flags` can be specified to write more detailed information about each node type such as parameter values, kernel attributes, node and function handles. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` The graph to create a DOT file from path : bytes The path to write the DOT file to flags : unsigned int Flags from CUgraphDebugDot_flags for specifying which additional node information to write Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OPERATING_SYSTEM` """ cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph err = ccuda.cuGraphDebugDotPrint(chGraph, path, flags) return (CUresult(err),) {{endif}} {{if 'cuUserObjectCreate' in found_functions}} @cython.embedsignature(True) def cuUserObjectCreate(ptr, destroy, unsigned int initialRefcount, unsigned int flags): """ Create a user object. Create a user object with the specified destructor callback and initial reference count. The initial references are owned by the caller. Destructor callbacks cannot make CUDA API calls and should avoid blocking behavior, as they are executed by a shared internal thread. Another thread may be signaled to perform such actions, if it does not block forward progress of tasks scheduled through CUDA. See CUDA User Objects in the CUDA C++ Programming Guide for more information on user objects. Parameters ---------- ptr : Any The pointer to pass to the destroy function destroy : :py:obj:`~.CUhostFn` Callback to free the user object when it is no longer in use initialRefcount : unsigned int The initial refcount to create the object with, typically 1. The initial references are owned by the calling thread. flags : unsigned int Currently it is required to pass :py:obj:`~.CU_USER_OBJECT_NO_DESTRUCTOR_SYNC`, which is the only defined flag. This indicates that the destroy callback cannot be waited on by any CUDA API. Users requiring synchronization of the callback should signal its completion manually. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` object_out : :py:obj:`~.CUuserObject` Location to return the user object handle See Also -------- :py:obj:`~.cuUserObjectRetain`, :py:obj:`~.cuUserObjectRelease`, :py:obj:`~.cuGraphRetainUserObject`, :py:obj:`~.cuGraphReleaseUserObject`, :py:obj:`~.cuGraphCreate` """ cdef ccuda.CUhostFn cdestroy if destroy is None: cdestroy = 0 elif isinstance(destroy, (CUhostFn,)): pdestroy = int(destroy) cdestroy = pdestroy else: pdestroy = int(CUhostFn(destroy)) cdestroy = pdestroy cdef CUuserObject object_out = CUuserObject() cptr = utils.HelperInputVoidPtr(ptr) cdef void* cptr_ptr = cptr.cptr err = ccuda.cuUserObjectCreate(object_out._ptr, cptr_ptr, cdestroy, initialRefcount, flags) return (CUresult(err), object_out) {{endif}} {{if 'cuUserObjectRetain' in found_functions}} @cython.embedsignature(True) def cuUserObjectRetain(object, unsigned int count): """ Retain a reference to a user object. Retains new references to a user object. The new references are owned by the caller. See CUDA User Objects in the CUDA C++ Programming Guide for more information on user objects. Parameters ---------- object : :py:obj:`~.CUuserObject` The object to retain count : unsigned int The number of references to retain, typically 1. Must be nonzero and not larger than INT_MAX. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuUserObjectCreate`, :py:obj:`~.cuUserObjectRelease`, :py:obj:`~.cuGraphRetainUserObject`, :py:obj:`~.cuGraphReleaseUserObject`, :py:obj:`~.cuGraphCreate` """ cdef ccuda.CUuserObject cobject if object is None: cobject = 0 elif isinstance(object, (CUuserObject,)): pobject = int(object) cobject = pobject else: pobject = int(CUuserObject(object)) cobject = pobject err = ccuda.cuUserObjectRetain(cobject, count) return (CUresult(err),) {{endif}} {{if 'cuUserObjectRelease' in found_functions}} @cython.embedsignature(True) def cuUserObjectRelease(object, unsigned int count): """ Release a reference to a user object. Releases user object references owned by the caller. The object's destructor is invoked if the reference count reaches zero. It is undefined behavior to release references not owned by the caller, or to use a user object handle after all references are released. See CUDA User Objects in the CUDA C++ Programming Guide for more information on user objects. Parameters ---------- object : :py:obj:`~.CUuserObject` The object to release count : unsigned int The number of references to release, typically 1. Must be nonzero and not larger than INT_MAX. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuUserObjectCreate`, :py:obj:`~.cuUserObjectRetain`, :py:obj:`~.cuGraphRetainUserObject`, :py:obj:`~.cuGraphReleaseUserObject`, :py:obj:`~.cuGraphCreate` """ cdef ccuda.CUuserObject cobject if object is None: cobject = 0 elif isinstance(object, (CUuserObject,)): pobject = int(object) cobject = pobject else: pobject = int(CUuserObject(object)) cobject = pobject err = ccuda.cuUserObjectRelease(cobject, count) return (CUresult(err),) {{endif}} {{if 'cuGraphRetainUserObject' in found_functions}} @cython.embedsignature(True) def cuGraphRetainUserObject(graph, object, unsigned int count, unsigned int flags): """ Retain a reference to a user object from a graph. Creates or moves user object references that will be owned by a CUDA graph. See CUDA User Objects in the CUDA C++ Programming Guide for more information on user objects. Parameters ---------- graph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` The graph to associate the reference with object : :py:obj:`~.CUuserObject` The user object to retain a reference for count : unsigned int The number of references to add to the graph, typically 1. Must be nonzero and not larger than INT_MAX. flags : unsigned int The optional flag :py:obj:`~.CU_GRAPH_USER_OBJECT_MOVE` transfers references from the calling thread, rather than create new references. Pass 0 to create new references. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuUserObjectCreate`, :py:obj:`~.cuUserObjectRetain`, :py:obj:`~.cuUserObjectRelease`, :py:obj:`~.cuGraphReleaseUserObject`, :py:obj:`~.cuGraphCreate` """ cdef ccuda.CUuserObject cobject if object is None: cobject = 0 elif isinstance(object, (CUuserObject,)): pobject = int(object) cobject = pobject else: pobject = int(CUuserObject(object)) cobject = pobject cdef ccuda.CUgraph cgraph if graph is None: cgraph = 0 elif isinstance(graph, (CUgraph,)): pgraph = int(graph) cgraph = pgraph else: pgraph = int(CUgraph(graph)) cgraph = pgraph err = ccuda.cuGraphRetainUserObject(cgraph, cobject, count, flags) return (CUresult(err),) {{endif}} {{if 'cuGraphReleaseUserObject' in found_functions}} @cython.embedsignature(True) def cuGraphReleaseUserObject(graph, object, unsigned int count): """ Release a user object reference from a graph. Releases user object references owned by a graph. See CUDA User Objects in the CUDA C++ Programming Guide for more information on user objects. Parameters ---------- graph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` The graph that will release the reference object : :py:obj:`~.CUuserObject` The user object to release a reference for count : unsigned int The number of references to release, typically 1. Must be nonzero and not larger than INT_MAX. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuUserObjectCreate`, :py:obj:`~.cuUserObjectRetain`, :py:obj:`~.cuUserObjectRelease`, :py:obj:`~.cuGraphRetainUserObject`, :py:obj:`~.cuGraphCreate` """ cdef ccuda.CUuserObject cobject if object is None: cobject = 0 elif isinstance(object, (CUuserObject,)): pobject = int(object) cobject = pobject else: pobject = int(CUuserObject(object)) cobject = pobject cdef ccuda.CUgraph cgraph if graph is None: cgraph = 0 elif isinstance(graph, (CUgraph,)): pgraph = int(graph) cgraph = pgraph else: pgraph = int(CUgraph(graph)) cgraph = pgraph err = ccuda.cuGraphReleaseUserObject(cgraph, cobject, count) return (CUresult(err),) {{endif}} {{if 'cuGraphAddNode' in found_functions}} @cython.embedsignature(True) def cuGraphAddNode(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], size_t numDependencies, nodeParams : Optional[CUgraphNodeParams]): """ Adds a node of arbitrary type to a graph. Creates a new node in `hGraph` described by `nodeParams` with `numDependencies` dependencies specified via `dependencies`. `numDependencies` may be 0. `dependencies` may be null if `numDependencies` is 0. `dependencies` may not have any duplicate entries. `nodeParams` is a tagged union. The node type should be specified in the `typename` field, and type-specific parameters in the corresponding union member. All unused bytes - that is, `reserved0` and all bytes past the utilized union member - must be set to zero. It is recommended to use brace initialization or memset to ensure all bytes are initialized. Note that for some node types, `nodeParams` may contain "out parameters" which are modified during the call, such as `nodeParams->alloc.dptr`. A handle to the new node will be returned in `phGraphNode`. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUgraphNodeParams` Specification of the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuGraphExecNodeSetParams` """ dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) cdef ccuda.CUgraphNodeParams* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddNode(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphAddNode_v2' in found_functions}} @cython.embedsignature(True) def cuGraphAddNode_v2(hGraph, dependencies : Optional[Tuple[CUgraphNode] | List[CUgraphNode]], dependencyData : Optional[Tuple[CUgraphEdgeData] | List[CUgraphEdgeData]], size_t numDependencies, nodeParams : Optional[CUgraphNodeParams]): """ Adds a node of arbitrary type to a graph (12.3+) Creates a new node in `hGraph` described by `nodeParams` with `numDependencies` dependencies specified via `dependencies`. `numDependencies` may be 0. `dependencies` may be null if `numDependencies` is 0. `dependencies` may not have any duplicate entries. `nodeParams` is a tagged union. The node type should be specified in the `typename` field, and type-specific parameters in the corresponding union member. All unused bytes - that is, `reserved0` and all bytes past the utilized union member - must be set to zero. It is recommended to use brace initialization or memset to ensure all bytes are initialized. Note that for some node types, `nodeParams` may contain "out parameters" which are modified during the call, such as `nodeParams->alloc.dptr`. A handle to the new node will be returned in `phGraphNode`. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph to which to add the node dependencies : List[:py:obj:`~.CUgraphNode`] Dependencies of the node dependencyData : List[:py:obj:`~.CUgraphEdgeData`] Optional edge data for the dependencies. If NULL, the data is assumed to be default (zeroed) for all dependencies. numDependencies : size_t Number of dependencies nodeParams : :py:obj:`~.CUgraphNodeParams` Specification of the node Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` phGraphNode : :py:obj:`~.CUgraphNode` Returns newly created node See Also -------- :py:obj:`~.cuGraphCreate`, :py:obj:`~.cuGraphNodeSetParams`, :py:obj:`~.cuGraphExecNodeSetParams` """ dependencyData = [] if dependencyData is None else dependencyData if not all(isinstance(_x, (CUgraphEdgeData,)) for _x in dependencyData): raise TypeError("Argument 'dependencyData' is not instance of type (expected Tuple[ccuda.CUgraphEdgeData,] or List[ccuda.CUgraphEdgeData,]") dependencies = [] if dependencies is None else dependencies if not all(isinstance(_x, (CUgraphNode,)) for _x in dependencies): raise TypeError("Argument 'dependencies' is not instance of type (expected Tuple[ccuda.CUgraphNode,] or List[ccuda.CUgraphNode,]") cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphNode phGraphNode = CUgraphNode() cdef ccuda.CUgraphNode* cdependencies = NULL if len(dependencies) > 0: cdependencies = calloc(len(dependencies), sizeof(ccuda.CUgraphNode)) if cdependencies is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencies)) + 'x' + str(sizeof(ccuda.CUgraphNode))) else: for idx in range(len(dependencies)): cdependencies[idx] = (dependencies[idx])._ptr[0] cdef ccuda.CUgraphEdgeData* cdependencyData = NULL if len(dependencyData) > 0: cdependencyData = calloc(len(dependencyData), sizeof(ccuda.CUgraphEdgeData)) if cdependencyData is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(dependencyData)) + 'x' + str(sizeof(ccuda.CUgraphEdgeData))) for idx in range(len(dependencyData)): string.memcpy(&cdependencyData[idx], (dependencyData[idx])._ptr, sizeof(ccuda.CUgraphEdgeData)) if numDependencies > len(dependencies): raise RuntimeError("List is too small: " + str(len(dependencies)) + " < " + str(numDependencies)) if numDependencies > len(dependencyData): raise RuntimeError("List is too small: " + str(len(dependencyData)) + " < " + str(numDependencies)) cdef ccuda.CUgraphNodeParams* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphAddNode_v2(phGraphNode._ptr, chGraph, (dependencies[0])._ptr if len(dependencies) == 1 else cdependencies, (dependencyData[0])._ptr if len(dependencyData) == 1 else cdependencyData, numDependencies, cnodeParams_ptr) if cdependencies is not NULL: free(cdependencies) if cdependencyData is not NULL: free(cdependencyData) return (CUresult(err), phGraphNode) {{endif}} {{if 'cuGraphNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphNodeSetParams(hNode, nodeParams : Optional[CUgraphNodeParams]): """ Update's a graph node's parameters. Sets the parameters of graph node `hNode` to `nodeParams`. The node type specified by `nodeParams->type` must match the type of `hNode`. `nodeParams` must be fully initialized and all unused bytes (reserved, padding) zeroed. Modifying parameters is not supported for node types CU_GRAPH_NODE_TYPE_MEM_ALLOC and CU_GRAPH_NODE_TYPE_MEM_FREE. Parameters ---------- hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Node to set the parameters for nodeParams : :py:obj:`~.CUgraphNodeParams` Parameters to copy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphExecNodeSetParams` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphNodeParams* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphNodeSetParams(chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphExecNodeSetParams' in found_functions}} @cython.embedsignature(True) def cuGraphExecNodeSetParams(hGraphExec, hNode, nodeParams : Optional[CUgraphNodeParams]): """ Update's a graph node's parameters in an instantiated graph. Sets the parameters of a node in an executable graph `hGraphExec`. The node is identified by the corresponding node `hNode` in the non- executable graph from which the executable graph was instantiated. `hNode` must not have been removed from the original graph. The modifications only affect future launches of `hGraphExec`. Already enqueued or running launches of `hGraphExec` are not affected by this call. `hNode` is also not modified by this call. Allowed changes to parameters on executable graphs are as follows: **View CUDA Toolkit Documentation for a table example** Parameters ---------- hGraphExec : :py:obj:`~.CUgraphExec` or :py:obj:`~.cudaGraphExec_t` The executable graph in which to update the specified node hNode : :py:obj:`~.CUgraphNode` or :py:obj:`~.cudaGraphNode_t` Corresponding node from the graph from which graphExec was instantiated nodeParams : :py:obj:`~.CUgraphNodeParams` Updated Parameters to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` See Also -------- :py:obj:`~.cuGraphAddNode`, :py:obj:`~.cuGraphNodeSetParams` :py:obj:`~.cuGraphExecUpdate`, :py:obj:`~.cuGraphInstantiate` """ cdef ccuda.CUgraphNode chNode if hNode is None: chNode = 0 elif isinstance(hNode, (CUgraphNode,)): phNode = int(hNode) chNode = phNode else: phNode = int(CUgraphNode(hNode)) chNode = phNode cdef ccuda.CUgraphExec chGraphExec if hGraphExec is None: chGraphExec = 0 elif isinstance(hGraphExec, (CUgraphExec,)): phGraphExec = int(hGraphExec) chGraphExec = phGraphExec else: phGraphExec = int(CUgraphExec(hGraphExec)) chGraphExec = phGraphExec cdef ccuda.CUgraphNodeParams* cnodeParams_ptr = nodeParams._ptr if nodeParams != None else NULL err = ccuda.cuGraphExecNodeSetParams(chGraphExec, chNode, cnodeParams_ptr) return (CUresult(err),) {{endif}} {{if 'cuGraphConditionalHandleCreate' in found_functions}} @cython.embedsignature(True) def cuGraphConditionalHandleCreate(hGraph, ctx, unsigned int defaultLaunchValue, unsigned int flags): """ Create a conditional handle. Creates a conditional handle associated with `hGraph`. The conditional handle must be associated with a conditional node in this graph or one of its children. Handles not associated with a conditional node may cause graph instantiation to fail. Handles can only be set from the context with which they are associated. Parameters ---------- hGraph : :py:obj:`~.CUgraph` or :py:obj:`~.cudaGraph_t` Graph which will contain the conditional node using this handle. ctx : :py:obj:`~.CUcontext` Context for the handle and associated conditional node. defaultLaunchValue : unsigned int Optional initial value for the conditional variable. flags : unsigned int Currently must be CU_GRAPH_COND_ASSIGN_DEFAULT or 0. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` pHandle_out : :py:obj:`~.CUgraphConditionalHandle` Pointer used to return the handle to the caller. See Also -------- :py:obj:`~.cuGraphAddNode` """ cdef ccuda.CUcontext cctx if ctx is None: cctx = 0 elif isinstance(ctx, (CUcontext,)): pctx = int(ctx) cctx = pctx else: pctx = int(CUcontext(ctx)) cctx = pctx cdef ccuda.CUgraph chGraph if hGraph is None: chGraph = 0 elif isinstance(hGraph, (CUgraph,)): phGraph = int(hGraph) chGraph = phGraph else: phGraph = int(CUgraph(hGraph)) chGraph = phGraph cdef CUgraphConditionalHandle pHandle_out = CUgraphConditionalHandle() err = ccuda.cuGraphConditionalHandleCreate(pHandle_out._ptr, chGraph, cctx, defaultLaunchValue, flags) return (CUresult(err), pHandle_out) {{endif}} {{if 'cuOccupancyMaxActiveBlocksPerMultiprocessor' in found_functions}} @cython.embedsignature(True) def cuOccupancyMaxActiveBlocksPerMultiprocessor(func, int blockSize, size_t dynamicSMemSize): """ Returns occupancy of a function. Returns in `*numBlocks` the number of the maximum active blocks per streaming multiprocessor. Note that the API can also be used with context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to use for calculations will be the current context. Parameters ---------- func : :py:obj:`~.CUfunction` Kernel for which occupancy is calculated blockSize : int Block size the kernel is intended to be launched with dynamicSMemSize : size_t Per-block dynamic shared memory usage intended, in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` numBlocks : int Returned occupancy See Also -------- :py:obj:`~.cudaOccupancyMaxActiveBlocksPerMultiprocessor` """ cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef int numBlocks = 0 err = ccuda.cuOccupancyMaxActiveBlocksPerMultiprocessor(&numBlocks, cfunc, blockSize, dynamicSMemSize) return (CUresult(err), numBlocks) {{endif}} {{if 'cuOccupancyMaxActiveBlocksPerMultiprocessorWithFlags' in found_functions}} @cython.embedsignature(True) def cuOccupancyMaxActiveBlocksPerMultiprocessorWithFlags(func, int blockSize, size_t dynamicSMemSize, unsigned int flags): """ Returns occupancy of a function. Returns in `*numBlocks` the number of the maximum active blocks per streaming multiprocessor. The `Flags` parameter controls how special cases are handled. The valid flags are: - :py:obj:`~.CU_OCCUPANCY_DEFAULT`, which maintains the default behavior as :py:obj:`~.cuOccupancyMaxActiveBlocksPerMultiprocessor`; - :py:obj:`~.CU_OCCUPANCY_DISABLE_CACHING_OVERRIDE`, which suppresses the default behavior on platform where global caching affects occupancy. On such platforms, if caching is enabled, but per-block SM resource usage would result in zero occupancy, the occupancy calculator will calculate the occupancy as if caching is disabled. Setting :py:obj:`~.CU_OCCUPANCY_DISABLE_CACHING_OVERRIDE` makes the occupancy calculator to return 0 in such cases. More information can be found about this feature in the "Unified L1/Texture Cache" section of the Maxwell tuning guide. Note that the API can also be with launch context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to use for calculations will be the current context. Parameters ---------- func : :py:obj:`~.CUfunction` Kernel for which occupancy is calculated blockSize : int Block size the kernel is intended to be launched with dynamicSMemSize : size_t Per-block dynamic shared memory usage intended, in bytes flags : unsigned int Requested behavior for the occupancy calculator Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` numBlocks : int Returned occupancy See Also -------- :py:obj:`~.cudaOccupancyMaxActiveBlocksPerMultiprocessorWithFlags` """ cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef int numBlocks = 0 err = ccuda.cuOccupancyMaxActiveBlocksPerMultiprocessorWithFlags(&numBlocks, cfunc, blockSize, dynamicSMemSize, flags) return (CUresult(err), numBlocks) {{endif}} {{if 'cuOccupancyMaxPotentialBlockSize' in found_functions}} @cython.embedsignature(True) def cuOccupancyMaxPotentialBlockSize(func, blockSizeToDynamicSMemSize, size_t dynamicSMemSize, int blockSizeLimit): """ Suggest a launch configuration with reasonable occupancy. Returns in `*blockSize` a reasonable block size that can achieve the maximum occupancy (or, the maximum number of active warps with the fewest blocks per multiprocessor), and in `*minGridSize` the minimum grid size to achieve the maximum occupancy. If `blockSizeLimit` is 0, the configurator will use the maximum block size permitted by the device / function instead. If per-block dynamic shared memory allocation is not needed, the user should leave both `blockSizeToDynamicSMemSize` and `dynamicSMemSize` as 0. If per-block dynamic shared memory allocation is needed, then if the dynamic shared memory size is constant regardless of block size, the size should be passed through `dynamicSMemSize`, and `blockSizeToDynamicSMemSize` should be NULL. Otherwise, if the per-block dynamic shared memory size varies with different block sizes, the user needs to provide a unary function through `blockSizeToDynamicSMemSize` that computes the dynamic shared memory needed by `func` for any given block size. `dynamicSMemSize` is ignored. An example signature is: **View CUDA Toolkit Documentation for a C++ code example** Note that the API can also be used with context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to use for calculations will be the current context. Parameters ---------- func : :py:obj:`~.CUfunction` Kernel for which launch configuration is calculated blockSizeToDynamicSMemSize : :py:obj:`~.CUoccupancyB2DSize` A function that calculates how much per-block dynamic shared memory `func` uses based on the block size dynamicSMemSize : size_t Dynamic shared memory usage intended, in bytes blockSizeLimit : int The maximum block size `func` is designed to handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` minGridSize : int Returned minimum grid size needed to achieve the maximum occupancy blockSize : int Returned maximum block size that can achieve the maximum occupancy See Also -------- :py:obj:`~.cudaOccupancyMaxPotentialBlockSize` """ cdef ccuda.CUoccupancyB2DSize cblockSizeToDynamicSMemSize if blockSizeToDynamicSMemSize is None: cblockSizeToDynamicSMemSize = 0 elif isinstance(blockSizeToDynamicSMemSize, (CUoccupancyB2DSize,)): pblockSizeToDynamicSMemSize = int(blockSizeToDynamicSMemSize) cblockSizeToDynamicSMemSize = pblockSizeToDynamicSMemSize else: pblockSizeToDynamicSMemSize = int(CUoccupancyB2DSize(blockSizeToDynamicSMemSize)) cblockSizeToDynamicSMemSize = pblockSizeToDynamicSMemSize cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef int minGridSize = 0 cdef int blockSize = 0 err = ccuda.cuOccupancyMaxPotentialBlockSize(&minGridSize, &blockSize, cfunc, cblockSizeToDynamicSMemSize, dynamicSMemSize, blockSizeLimit) return (CUresult(err), minGridSize, blockSize) {{endif}} {{if 'cuOccupancyMaxPotentialBlockSizeWithFlags' in found_functions}} @cython.embedsignature(True) def cuOccupancyMaxPotentialBlockSizeWithFlags(func, blockSizeToDynamicSMemSize, size_t dynamicSMemSize, int blockSizeLimit, unsigned int flags): """ Suggest a launch configuration with reasonable occupancy. An extended version of :py:obj:`~.cuOccupancyMaxPotentialBlockSize`. In addition to arguments passed to :py:obj:`~.cuOccupancyMaxPotentialBlockSize`, :py:obj:`~.cuOccupancyMaxPotentialBlockSizeWithFlags` also takes a `Flags` parameter. The `Flags` parameter controls how special cases are handled. The valid flags are: - :py:obj:`~.CU_OCCUPANCY_DEFAULT`, which maintains the default behavior as :py:obj:`~.cuOccupancyMaxPotentialBlockSize`; - :py:obj:`~.CU_OCCUPANCY_DISABLE_CACHING_OVERRIDE`, which suppresses the default behavior on platform where global caching affects occupancy. On such platforms, the launch configurations that produces maximal occupancy might not support global caching. Setting :py:obj:`~.CU_OCCUPANCY_DISABLE_CACHING_OVERRIDE` guarantees that the the produced launch configuration is global caching compatible at a potential cost of occupancy. More information can be found about this feature in the "Unified L1/Texture Cache" section of the Maxwell tuning guide. Note that the API can also be used with context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to use for calculations will be the current context. Parameters ---------- func : :py:obj:`~.CUfunction` Kernel for which launch configuration is calculated blockSizeToDynamicSMemSize : :py:obj:`~.CUoccupancyB2DSize` A function that calculates how much per-block dynamic shared memory `func` uses based on the block size dynamicSMemSize : size_t Dynamic shared memory usage intended, in bytes blockSizeLimit : int The maximum block size `func` is designed to handle flags : unsigned int Options Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` minGridSize : int Returned minimum grid size needed to achieve the maximum occupancy blockSize : int Returned maximum block size that can achieve the maximum occupancy See Also -------- :py:obj:`~.cudaOccupancyMaxPotentialBlockSizeWithFlags` """ cdef ccuda.CUoccupancyB2DSize cblockSizeToDynamicSMemSize if blockSizeToDynamicSMemSize is None: cblockSizeToDynamicSMemSize = 0 elif isinstance(blockSizeToDynamicSMemSize, (CUoccupancyB2DSize,)): pblockSizeToDynamicSMemSize = int(blockSizeToDynamicSMemSize) cblockSizeToDynamicSMemSize = pblockSizeToDynamicSMemSize else: pblockSizeToDynamicSMemSize = int(CUoccupancyB2DSize(blockSizeToDynamicSMemSize)) cblockSizeToDynamicSMemSize = pblockSizeToDynamicSMemSize cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef int minGridSize = 0 cdef int blockSize = 0 err = ccuda.cuOccupancyMaxPotentialBlockSizeWithFlags(&minGridSize, &blockSize, cfunc, cblockSizeToDynamicSMemSize, dynamicSMemSize, blockSizeLimit, flags) return (CUresult(err), minGridSize, blockSize) {{endif}} {{if 'cuOccupancyAvailableDynamicSMemPerBlock' in found_functions}} @cython.embedsignature(True) def cuOccupancyAvailableDynamicSMemPerBlock(func, int numBlocks, int blockSize): """ Returns dynamic shared memory available per block when launching `numBlocks` blocks on SM. Returns in `*dynamicSmemSize` the maximum size of dynamic shared memory to allow `numBlocks` blocks per SM. Note that the API can also be used with context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to use for calculations will be the current context. Parameters ---------- func : :py:obj:`~.CUfunction` Kernel function for which occupancy is calculated numBlocks : int Number of blocks to fit on SM blockSize : int Size of the blocks Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` dynamicSmemSize : int Returned maximum dynamic shared memory """ cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef size_t dynamicSmemSize = 0 err = ccuda.cuOccupancyAvailableDynamicSMemPerBlock(&dynamicSmemSize, cfunc, numBlocks, blockSize) return (CUresult(err), dynamicSmemSize) {{endif}} {{if 'cuOccupancyMaxPotentialClusterSize' in found_functions}} @cython.embedsignature(True) def cuOccupancyMaxPotentialClusterSize(func, config : Optional[CUlaunchConfig]): """ Given the kernel function (`func`) and launch configuration (`config`), return the maximum cluster size in `*clusterSize`. The cluster dimensions in `config` are ignored. If func has a required cluster size set (see :py:obj:`~.cudaFuncGetAttributes` / :py:obj:`~.cuFuncGetAttribute`),`*clusterSize` will reflect the required cluster size. By default this function will always return a value that's portable on future hardware. A higher value may be returned if the kernel function allows non-portable cluster sizes. This function will respect the compile time launch bounds. Note that the API can also be used with context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to use for calculations will either be taken from the specified stream `config->hStream` or the current context in case of NULL stream. Parameters ---------- func : :py:obj:`~.CUfunction` Kernel function for which maximum cluster size is calculated config : :py:obj:`~.CUlaunchConfig` Launch configuration for the given kernel function Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` clusterSize : int Returned maximum cluster size that can be launched for the given kernel function and launch configuration See Also -------- :py:obj:`~.cudaFuncGetAttributes`, :py:obj:`~.cuFuncGetAttribute` """ cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef int clusterSize = 0 cdef ccuda.CUlaunchConfig* cconfig_ptr = config._ptr if config != None else NULL err = ccuda.cuOccupancyMaxPotentialClusterSize(&clusterSize, cfunc, cconfig_ptr) return (CUresult(err), clusterSize) {{endif}} {{if 'cuOccupancyMaxActiveClusters' in found_functions}} @cython.embedsignature(True) def cuOccupancyMaxActiveClusters(func, config : Optional[CUlaunchConfig]): """ Given the kernel function (`func`) and launch configuration (`config`), return the maximum number of clusters that could co-exist on the target device in `*numClusters`. If the function has required cluster size already set (see :py:obj:`~.cudaFuncGetAttributes` / :py:obj:`~.cuFuncGetAttribute`), the cluster size from config must either be unspecified or match the required size. Without required sizes, the cluster size must be specified in config, else the function will return an error. Note that various attributes of the kernel function may affect occupancy calculation. Runtime environment may affect how the hardware schedules the clusters, so the calculated occupancy is not guaranteed to be achievable. Note that the API can also be used with context-less kernel :py:obj:`~.CUkernel` by querying the handle using :py:obj:`~.cuLibraryGetKernel()` and then passing it to the API by casting to :py:obj:`~.CUfunction`. Here, the context to use for calculations will either be taken from the specified stream `config->hStream` or the current context in case of NULL stream. Parameters ---------- func : :py:obj:`~.CUfunction` Kernel function for which maximum number of clusters are calculated config : :py:obj:`~.CUlaunchConfig` Launch configuration for the given kernel function Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_CLUSTER_SIZE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` numClusters : int Returned maximum number of clusters that could co-exist on the target device See Also -------- :py:obj:`~.cudaFuncGetAttributes`, :py:obj:`~.cuFuncGetAttribute` """ cdef ccuda.CUfunction cfunc if func is None: cfunc = 0 elif isinstance(func, (CUfunction,)): pfunc = int(func) cfunc = pfunc else: pfunc = int(CUfunction(func)) cfunc = pfunc cdef int numClusters = 0 cdef ccuda.CUlaunchConfig* cconfig_ptr = config._ptr if config != None else NULL err = ccuda.cuOccupancyMaxActiveClusters(&numClusters, cfunc, cconfig_ptr) return (CUresult(err), numClusters) {{endif}} {{if 'cuTexRefSetArray' in found_functions}} @cython.embedsignature(True) def cuTexRefSetArray(hTexRef, hArray, unsigned int Flags): """ Binds an array as a texture reference. [Deprecated] Binds the CUDA array `hArray` to the texture reference `hTexRef`. Any previous address or CUDA array state associated with the texture reference is superseded by this function. `Flags` must be set to :py:obj:`~.CU_TRSA_OVERRIDE_FORMAT`. Any CUDA array previously bound to `hTexRef` is unbound. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference to bind hArray : :py:obj:`~.CUarray` Array to bind Flags : unsigned int Options (must be :py:obj:`~.CU_TRSA_OVERRIDE_FORMAT`) Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUarray chArray if hArray is None: chArray = 0 elif isinstance(hArray, (CUarray,)): phArray = int(hArray) chArray = phArray else: phArray = int(CUarray(hArray)) chArray = phArray cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefSetArray(chTexRef, chArray, Flags) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetMipmappedArray' in found_functions}} @cython.embedsignature(True) def cuTexRefSetMipmappedArray(hTexRef, hMipmappedArray, unsigned int Flags): """ Binds a mipmapped array to a texture reference. [Deprecated] Binds the CUDA mipmapped array `hMipmappedArray` to the texture reference `hTexRef`. Any previous address or CUDA array state associated with the texture reference is superseded by this function. `Flags` must be set to :py:obj:`~.CU_TRSA_OVERRIDE_FORMAT`. Any CUDA array previously bound to `hTexRef` is unbound. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference to bind hMipmappedArray : :py:obj:`~.CUmipmappedArray` Mipmapped array to bind Flags : unsigned int Options (must be :py:obj:`~.CU_TRSA_OVERRIDE_FORMAT`) Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUmipmappedArray chMipmappedArray if hMipmappedArray is None: chMipmappedArray = 0 elif isinstance(hMipmappedArray, (CUmipmappedArray,)): phMipmappedArray = int(hMipmappedArray) chMipmappedArray = phMipmappedArray else: phMipmappedArray = int(CUmipmappedArray(hMipmappedArray)) chMipmappedArray = phMipmappedArray cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefSetMipmappedArray(chTexRef, chMipmappedArray, Flags) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetAddress_v2' in found_functions}} @cython.embedsignature(True) def cuTexRefSetAddress(hTexRef, dptr, size_t numbytes): """ Binds an address as a texture reference. [Deprecated] Binds a linear address range to the texture reference `hTexRef`. Any previous address or CUDA array state associated with the texture reference is superseded by this function. Any memory previously bound to `hTexRef` is unbound. Since the hardware enforces an alignment requirement on texture base addresses, :py:obj:`~.cuTexRefSetAddress()` passes back a byte offset in `*ByteOffset` that must be applied to texture fetches in order to read from the desired memory. This offset must be divided by the texel size and passed to kernels that read from the texture so they can be applied to the :py:obj:`~.tex1Dfetch()` function. If the device memory pointer was returned from :py:obj:`~.cuMemAlloc()`, the offset is guaranteed to be 0 and NULL may be passed as the `ByteOffset` parameter. The total number of elements (or texels) in the linear address range cannot exceed :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LINEAR_WIDTH`. The number of elements is computed as (`numbytes` / bytesPerElement), where bytesPerElement is determined from the data format and number of components set using :py:obj:`~.cuTexRefSetFormat()`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference to bind dptr : :py:obj:`~.CUdeviceptr` Device pointer to bind numbytes : size_t Size of memory to bind in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` ByteOffset : int Returned byte offset See Also -------- :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef size_t ByteOffset = 0 err = ccuda.cuTexRefSetAddress(&ByteOffset, chTexRef, cdptr, numbytes) return (CUresult(err), ByteOffset) {{endif}} {{if 'cuTexRefSetAddress2D_v3' in found_functions}} @cython.embedsignature(True) def cuTexRefSetAddress2D(hTexRef, desc : Optional[CUDA_ARRAY_DESCRIPTOR], dptr, size_t Pitch): """ Binds an address as a 2D texture reference. [Deprecated] Binds a linear address range to the texture reference `hTexRef`. Any previous address or CUDA array state associated with the texture reference is superseded by this function. Any memory previously bound to `hTexRef` is unbound. Using a :py:obj:`~.tex2D()` function inside a kernel requires a call to either :py:obj:`~.cuTexRefSetArray()` to bind the corresponding texture reference to an array, or :py:obj:`~.cuTexRefSetAddress2D()` to bind the texture reference to linear memory. Function calls to :py:obj:`~.cuTexRefSetFormat()` cannot follow calls to :py:obj:`~.cuTexRefSetAddress2D()` for the same texture reference. It is required that `dptr` be aligned to the appropriate hardware- specific texture alignment. You can query this value using the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT`. If an unaligned `dptr` is supplied, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. `Pitch` has to be aligned to the hardware-specific texture pitch alignment. This value can be queried using the device attribute :py:obj:`~.CU_DEVICE_ATTRIBUTE_TEXTURE_PITCH_ALIGNMENT`. If an unaligned `Pitch` is supplied, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. Width and Height, which are specified in elements (or texels), cannot exceed :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_WIDTH` and :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_HEIGHT` respectively. `Pitch`, which is specified in bytes, cannot exceed :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_PITCH`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference to bind desc : :py:obj:`~.CUDA_ARRAY_DESCRIPTOR` Descriptor of CUDA array dptr : :py:obj:`~.CUdeviceptr` Device pointer to bind Pitch : size_t Line pitch in bytes Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUdeviceptr cdptr if dptr is None: cdptr = 0 elif isinstance(dptr, (CUdeviceptr,)): pdptr = int(dptr) cdptr = pdptr else: pdptr = int(CUdeviceptr(dptr)) cdptr = pdptr cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUDA_ARRAY_DESCRIPTOR* cdesc_ptr = desc._ptr if desc != None else NULL err = ccuda.cuTexRefSetAddress2D(chTexRef, cdesc_ptr, cdptr, Pitch) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetFormat' in found_functions}} @cython.embedsignature(True) def cuTexRefSetFormat(hTexRef, fmt not None : CUarray_format, int NumPackedComponents): """ Sets the format for a texture reference. [Deprecated] Specifies the format of the data to be read by the texture reference `hTexRef`. `fmt` and `NumPackedComponents` are exactly analogous to the :py:obj:`~.Format` and :py:obj:`~.NumChannels` members of the :py:obj:`~.CUDA_ARRAY_DESCRIPTOR` structure: They specify the format of each component and the number of components per array element. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference fmt : :py:obj:`~.CUarray_format` Format to set NumPackedComponents : int Number of components per array element Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat`, :py:obj:`~.cudaCreateChannelDesc` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUarray_format cfmt = fmt.value err = ccuda.cuTexRefSetFormat(chTexRef, cfmt, NumPackedComponents) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetAddressMode' in found_functions}} @cython.embedsignature(True) def cuTexRefSetAddressMode(hTexRef, int dim, am not None : CUaddress_mode): """ Sets the addressing mode for a texture reference. [Deprecated] Specifies the addressing mode `am` for the given dimension `dim` of the texture reference `hTexRef`. If `dim` is zero, the addressing mode is applied to the first parameter of the functions used to fetch from the texture; if `dim` is 1, the second, and so on. :py:obj:`~.CUaddress_mode` is defined as: **View CUDA Toolkit Documentation for a C++ code example** Note that this call has no effect if `hTexRef` is bound to linear memory. Also, if the flag, :py:obj:`~.CU_TRSF_NORMALIZED_COORDINATES`, is not set, the only supported address mode is :py:obj:`~.CU_TR_ADDRESS_MODE_CLAMP`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference dim : int Dimension am : :py:obj:`~.CUaddress_mode` Addressing mode to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUaddress_mode cam = am.value err = ccuda.cuTexRefSetAddressMode(chTexRef, dim, cam) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetFilterMode' in found_functions}} @cython.embedsignature(True) def cuTexRefSetFilterMode(hTexRef, fm not None : CUfilter_mode): """ Sets the filtering mode for a texture reference. [Deprecated] Specifies the filtering mode `fm` to be used when reading memory through the texture reference `hTexRef`. :py:obj:`~.CUfilter_mode_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** Note that this call has no effect if `hTexRef` is bound to linear memory. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference fm : :py:obj:`~.CUfilter_mode` Filtering mode to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUfilter_mode cfm = fm.value err = ccuda.cuTexRefSetFilterMode(chTexRef, cfm) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetMipmapFilterMode' in found_functions}} @cython.embedsignature(True) def cuTexRefSetMipmapFilterMode(hTexRef, fm not None : CUfilter_mode): """ Sets the mipmap filtering mode for a texture reference. [Deprecated] Specifies the mipmap filtering mode `fm` to be used when reading memory through the texture reference `hTexRef`. :py:obj:`~.CUfilter_mode_enum` is defined as: **View CUDA Toolkit Documentation for a C++ code example** Note that this call has no effect if `hTexRef` is not bound to a mipmapped array. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference fm : :py:obj:`~.CUfilter_mode` Filtering mode to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUfilter_mode cfm = fm.value err = ccuda.cuTexRefSetMipmapFilterMode(chTexRef, cfm) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetMipmapLevelBias' in found_functions}} @cython.embedsignature(True) def cuTexRefSetMipmapLevelBias(hTexRef, float bias): """ Sets the mipmap level bias for a texture reference. [Deprecated] Specifies the mipmap level bias `bias` to be added to the specified mipmap level when reading memory through the texture reference `hTexRef`. Note that this call has no effect if `hTexRef` is not bound to a mipmapped array. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference bias : float Mipmap level bias Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefSetMipmapLevelBias(chTexRef, bias) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetMipmapLevelClamp' in found_functions}} @cython.embedsignature(True) def cuTexRefSetMipmapLevelClamp(hTexRef, float minMipmapLevelClamp, float maxMipmapLevelClamp): """ Sets the mipmap min/max mipmap level clamps for a texture reference. [Deprecated] Specifies the min/max mipmap level clamps, `minMipmapLevelClamp` and `maxMipmapLevelClamp` respectively, to be used when reading memory through the texture reference `hTexRef`. Note that this call has no effect if `hTexRef` is not bound to a mipmapped array. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference minMipmapLevelClamp : float Mipmap min level clamp maxMipmapLevelClamp : float Mipmap max level clamp Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefSetMipmapLevelClamp(chTexRef, minMipmapLevelClamp, maxMipmapLevelClamp) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetMaxAnisotropy' in found_functions}} @cython.embedsignature(True) def cuTexRefSetMaxAnisotropy(hTexRef, unsigned int maxAniso): """ Sets the maximum anisotropy for a texture reference. [Deprecated] Specifies the maximum anisotropy `maxAniso` to be used when reading memory through the texture reference `hTexRef`. Note that this call has no effect if `hTexRef` is bound to linear memory. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference maxAniso : unsigned int Maximum anisotropy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefSetMaxAnisotropy(chTexRef, maxAniso) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetBorderColor' in found_functions}} @cython.embedsignature(True) def cuTexRefSetBorderColor(hTexRef, float pBorderColor): """ Sets the border color for a texture reference. [Deprecated] Specifies the value of the RGBA color via the `pBorderColor` to the texture reference `hTexRef`. The color value supports only float type and holds color components in the following sequence: pBorderColor[0] holds 'R' component pBorderColor[1] holds 'G' component pBorderColor[2] holds 'B' component pBorderColor[3] holds 'A' component Note that the color values can be set only when the Address mode is set to CU_TR_ADDRESS_MODE_BORDER using :py:obj:`~.cuTexRefSetAddressMode`. Applications using integer border color values have to "reinterpret_cast" their values to float. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference pBorderColor : float RGBA color Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetBorderColor` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefSetBorderColor(chTexRef, &pBorderColor) return (CUresult(err),) {{endif}} {{if 'cuTexRefSetFlags' in found_functions}} @cython.embedsignature(True) def cuTexRefSetFlags(hTexRef, unsigned int Flags): """ Sets the flags for a texture reference. [Deprecated] Specifies optional flags via `Flags` to specify the behavior of data returned through the texture reference `hTexRef`. The valid flags are: - :py:obj:`~.CU_TRSF_READ_AS_INTEGER`, which suppresses the default behavior of having the texture promote integer data to floating point data in the range [0, 1]. Note that texture with 32-bit integer format would not be promoted, regardless of whether or not this flag is specified; - :py:obj:`~.CU_TRSF_NORMALIZED_COORDINATES`, which suppresses the default behavior of having the texture coordinates range from [0, Dim) where Dim is the width or height of the CUDA array. Instead, the texture coordinates [0, 1.0) reference the entire breadth of the array dimension; - :py:obj:`~.CU_TRSF_DISABLE_TRILINEAR_OPTIMIZATION`, which disables any trilinear filtering optimizations. Trilinear optimizations improve texture filtering performance by allowing bilinear filtering on textures in scenarios where it can closely approximate the expected results. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Flags : unsigned int Optional flags to set Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefSetFlags(chTexRef, Flags) return (CUresult(err),) {{endif}} {{if 'cuTexRefGetAddress_v2' in found_functions}} @cython.embedsignature(True) def cuTexRefGetAddress(hTexRef): """ Gets the address associated with a texture reference. [Deprecated] Returns in `*pdptr` the base address bound to the texture reference `hTexRef`, or returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if the texture reference is not bound to any device memory range. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pdptr : :py:obj:`~.CUdeviceptr` Returned device address See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef CUdeviceptr pdptr = CUdeviceptr() err = ccuda.cuTexRefGetAddress(pdptr._ptr, chTexRef) return (CUresult(err), pdptr) {{endif}} {{if 'cuTexRefGetArray' in found_functions}} @cython.embedsignature(True) def cuTexRefGetArray(hTexRef): """ Gets the array bound to a texture reference. [Deprecated] Returns in `*phArray` the CUDA array bound to the texture reference `hTexRef`, or returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if the texture reference is not bound to any CUDA array. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phArray : :py:obj:`~.CUarray` Returned array See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef CUarray phArray = CUarray() err = ccuda.cuTexRefGetArray(phArray._ptr, chTexRef) return (CUresult(err), phArray) {{endif}} {{if 'cuTexRefGetMipmappedArray' in found_functions}} @cython.embedsignature(True) def cuTexRefGetMipmappedArray(hTexRef): """ Gets the mipmapped array bound to a texture reference. [Deprecated] Returns in `*phMipmappedArray` the CUDA mipmapped array bound to the texture reference `hTexRef`, or returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if the texture reference is not bound to any CUDA mipmapped array. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phMipmappedArray : :py:obj:`~.CUmipmappedArray` Returned mipmapped array See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef CUmipmappedArray phMipmappedArray = CUmipmappedArray() err = ccuda.cuTexRefGetMipmappedArray(phMipmappedArray._ptr, chTexRef) return (CUresult(err), phMipmappedArray) {{endif}} {{if 'cuTexRefGetAddressMode' in found_functions}} @cython.embedsignature(True) def cuTexRefGetAddressMode(hTexRef, int dim): """ Gets the addressing mode used by a texture reference. [Deprecated] Returns in `*pam` the addressing mode corresponding to the dimension `dim` of the texture reference `hTexRef`. Currently, the only valid value for `dim` are 0 and 1. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference dim : int Dimension Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pam : :py:obj:`~.CUaddress_mode` Returned addressing mode See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUaddress_mode pam err = ccuda.cuTexRefGetAddressMode(&pam, chTexRef, dim) return (CUresult(err), CUaddress_mode(pam)) {{endif}} {{if 'cuTexRefGetFilterMode' in found_functions}} @cython.embedsignature(True) def cuTexRefGetFilterMode(hTexRef): """ Gets the filter-mode used by a texture reference. [Deprecated] Returns in `*pfm` the filtering mode of the texture reference `hTexRef`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pfm : :py:obj:`~.CUfilter_mode` Returned filtering mode See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUfilter_mode pfm err = ccuda.cuTexRefGetFilterMode(&pfm, chTexRef) return (CUresult(err), CUfilter_mode(pfm)) {{endif}} {{if 'cuTexRefGetFormat' in found_functions}} @cython.embedsignature(True) def cuTexRefGetFormat(hTexRef): """ Gets the format used by a texture reference. [Deprecated] Returns in `*pFormat` and `*pNumChannels` the format and number of components of the CUDA array bound to the texture reference `hTexRef`. If `pFormat` or `pNumChannels` is NULL, it will be ignored. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pFormat : :py:obj:`~.CUarray_format` Returned format pNumChannels : int Returned number of components See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUarray_format pFormat cdef int pNumChannels = 0 err = ccuda.cuTexRefGetFormat(&pFormat, &pNumChannels, chTexRef) return (CUresult(err), CUarray_format(pFormat), pNumChannels) {{endif}} {{if 'cuTexRefGetMipmapFilterMode' in found_functions}} @cython.embedsignature(True) def cuTexRefGetMipmapFilterMode(hTexRef): """ Gets the mipmap filtering mode for a texture reference. [Deprecated] Returns the mipmap filtering mode in `pfm` that's used when reading memory through the texture reference `hTexRef`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pfm : :py:obj:`~.CUfilter_mode` Returned mipmap filtering mode See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef ccuda.CUfilter_mode pfm err = ccuda.cuTexRefGetMipmapFilterMode(&pfm, chTexRef) return (CUresult(err), CUfilter_mode(pfm)) {{endif}} {{if 'cuTexRefGetMipmapLevelBias' in found_functions}} @cython.embedsignature(True) def cuTexRefGetMipmapLevelBias(hTexRef): """ Gets the mipmap level bias for a texture reference. [Deprecated] Returns the mipmap level bias in `pBias` that's added to the specified mipmap level when reading memory through the texture reference `hTexRef`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pbias : float Returned mipmap level bias See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef float pbias = 0 err = ccuda.cuTexRefGetMipmapLevelBias(&pbias, chTexRef) return (CUresult(err), pbias) {{endif}} {{if 'cuTexRefGetMipmapLevelClamp' in found_functions}} @cython.embedsignature(True) def cuTexRefGetMipmapLevelClamp(hTexRef): """ Gets the min/max mipmap level clamps for a texture reference. [Deprecated] Returns the min/max mipmap level clamps in `pminMipmapLevelClamp` and `pmaxMipmapLevelClamp` that's used when reading memory through the texture reference `hTexRef`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pminMipmapLevelClamp : float Returned mipmap min level clamp pmaxMipmapLevelClamp : float Returned mipmap max level clamp See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef float pminMipmapLevelClamp = 0 cdef float pmaxMipmapLevelClamp = 0 err = ccuda.cuTexRefGetMipmapLevelClamp(&pminMipmapLevelClamp, &pmaxMipmapLevelClamp, chTexRef) return (CUresult(err), pminMipmapLevelClamp, pmaxMipmapLevelClamp) {{endif}} {{if 'cuTexRefGetMaxAnisotropy' in found_functions}} @cython.embedsignature(True) def cuTexRefGetMaxAnisotropy(hTexRef): """ Gets the maximum anisotropy for a texture reference. [Deprecated] Returns the maximum anisotropy in `pmaxAniso` that's used when reading memory through the texture reference `hTexRef`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pmaxAniso : int Returned maximum anisotropy See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFlags`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef int pmaxAniso = 0 err = ccuda.cuTexRefGetMaxAnisotropy(&pmaxAniso, chTexRef) return (CUresult(err), pmaxAniso) {{endif}} {{if 'cuTexRefGetBorderColor' in found_functions}} @cython.embedsignature(True) def cuTexRefGetBorderColor(hTexRef): """ Gets the border color used by a texture reference. [Deprecated] Returns in `pBorderColor`, values of the RGBA color used by the texture reference `hTexRef`. The color value is of type float and holds color components in the following sequence: pBorderColor[0] holds 'R' component pBorderColor[1] holds 'G' component pBorderColor[2] holds 'B' component pBorderColor[3] holds 'A' component Parameters ---------- pBorderColor : :py:obj:`~.CUtexref` Returned Type and Value of RGBA color Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` hTexRef : float Texture reference See Also -------- :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetBorderColor` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef float pBorderColor = 0 err = ccuda.cuTexRefGetBorderColor(&pBorderColor, chTexRef) return (CUresult(err), pBorderColor) {{endif}} {{if 'cuTexRefGetFlags' in found_functions}} @cython.embedsignature(True) def cuTexRefGetFlags(hTexRef): """ Gets the flags used by a texture reference. [Deprecated] Returns in `*pFlags` the flags of the texture reference `hTexRef`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pFlags : unsigned int Returned flags See Also -------- :py:obj:`~.cuTexRefSetAddress`, :py:obj:`~.cuTexRefSetAddress2D`, :py:obj:`~.cuTexRefSetAddressMode`, :py:obj:`~.cuTexRefSetArray`, :py:obj:`~.cuTexRefSetFilterMode`, :py:obj:`~.cuTexRefSetFlags`, :py:obj:`~.cuTexRefSetFormat`, :py:obj:`~.cuTexRefGetAddress`, :py:obj:`~.cuTexRefGetAddressMode`, :py:obj:`~.cuTexRefGetArray`, :py:obj:`~.cuTexRefGetFilterMode`, :py:obj:`~.cuTexRefGetFormat` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef cdef unsigned int pFlags = 0 err = ccuda.cuTexRefGetFlags(&pFlags, chTexRef) return (CUresult(err), pFlags) {{endif}} {{if 'cuTexRefCreate' in found_functions}} @cython.embedsignature(True) def cuTexRefCreate(): """ Creates a texture reference. [Deprecated] Creates a texture reference and returns its handle in `*pTexRef`. Once created, the application must call :py:obj:`~.cuTexRefSetArray()` or :py:obj:`~.cuTexRefSetAddress()` to associate the reference with allocated memory. Other texture reference functions are used to specify the format and interpretation (addressing, filtering, etc.) to be used when the memory is read through this texture reference. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pTexRef : :py:obj:`~.CUtexref` Returned texture reference See Also -------- :py:obj:`~.cuTexRefDestroy` """ cdef CUtexref pTexRef = CUtexref() err = ccuda.cuTexRefCreate(pTexRef._ptr) return (CUresult(err), pTexRef) {{endif}} {{if 'cuTexRefDestroy' in found_functions}} @cython.embedsignature(True) def cuTexRefDestroy(hTexRef): """ Destroys a texture reference. [Deprecated] Destroys the texture reference specified by `hTexRef`. Parameters ---------- hTexRef : :py:obj:`~.CUtexref` Texture reference to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexRefCreate` """ cdef ccuda.CUtexref chTexRef if hTexRef is None: chTexRef = 0 elif isinstance(hTexRef, (CUtexref,)): phTexRef = int(hTexRef) chTexRef = phTexRef else: phTexRef = int(CUtexref(hTexRef)) chTexRef = phTexRef err = ccuda.cuTexRefDestroy(chTexRef) return (CUresult(err),) {{endif}} {{if 'cuSurfRefSetArray' in found_functions}} @cython.embedsignature(True) def cuSurfRefSetArray(hSurfRef, hArray, unsigned int Flags): """ Sets the CUDA array for a surface reference. [Deprecated] Sets the CUDA array `hArray` to be read and written by the surface reference `hSurfRef`. Any previous CUDA array state associated with the surface reference is superseded by this function. `Flags` must be set to 0. The :py:obj:`~.CUDA_ARRAY3D_SURFACE_LDST` flag must have been set for the CUDA array. Any CUDA array previously bound to `hSurfRef` is unbound. Parameters ---------- hSurfRef : :py:obj:`~.CUsurfref` Surface reference handle hArray : :py:obj:`~.CUarray` CUDA array handle Flags : unsigned int set to 0 Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuModuleGetSurfRef`, :py:obj:`~.cuSurfRefGetArray` """ cdef ccuda.CUarray chArray if hArray is None: chArray = 0 elif isinstance(hArray, (CUarray,)): phArray = int(hArray) chArray = phArray else: phArray = int(CUarray(hArray)) chArray = phArray cdef ccuda.CUsurfref chSurfRef if hSurfRef is None: chSurfRef = 0 elif isinstance(hSurfRef, (CUsurfref,)): phSurfRef = int(hSurfRef) chSurfRef = phSurfRef else: phSurfRef = int(CUsurfref(hSurfRef)) chSurfRef = phSurfRef err = ccuda.cuSurfRefSetArray(chSurfRef, chArray, Flags) return (CUresult(err),) {{endif}} {{if 'cuSurfRefGetArray' in found_functions}} @cython.embedsignature(True) def cuSurfRefGetArray(hSurfRef): """ Passes back the CUDA array bound to a surface reference. [Deprecated] Returns in `*phArray` the CUDA array bound to the surface reference `hSurfRef`, or returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if the surface reference is not bound to any CUDA array. Parameters ---------- hSurfRef : :py:obj:`~.CUsurfref` Surface reference handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` phArray : :py:obj:`~.CUarray` Surface reference handle See Also -------- :py:obj:`~.cuModuleGetSurfRef`, :py:obj:`~.cuSurfRefSetArray` """ cdef ccuda.CUsurfref chSurfRef if hSurfRef is None: chSurfRef = 0 elif isinstance(hSurfRef, (CUsurfref,)): phSurfRef = int(hSurfRef) chSurfRef = phSurfRef else: phSurfRef = int(CUsurfref(hSurfRef)) chSurfRef = phSurfRef cdef CUarray phArray = CUarray() err = ccuda.cuSurfRefGetArray(phArray._ptr, chSurfRef) return (CUresult(err), phArray) {{endif}} {{if 'cuTexObjectCreate' in found_functions}} @cython.embedsignature(True) def cuTexObjectCreate(pResDesc : Optional[CUDA_RESOURCE_DESC], pTexDesc : Optional[CUDA_TEXTURE_DESC], pResViewDesc : Optional[CUDA_RESOURCE_VIEW_DESC]): """ Creates a texture object. Creates a texture object and returns it in `pTexObject`. `pResDesc` describes the data to texture from. `pTexDesc` describes how the data should be sampled. `pResViewDesc` is an optional argument that specifies an alternate format for the data described by `pResDesc`, and also describes the subresource region to restrict access to when texturing. `pResViewDesc` can only be specified if the type of resource is a CUDA array or a CUDA mipmapped array not in a block compressed format. Texture objects are only supported on devices of compute capability 3.0 or higher. Additionally, a texture object is an opaque value, and, as such, should only be accessed through CUDA API calls. The :py:obj:`~.CUDA_RESOURCE_DESC` structure is defined as: **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.CUDA_RESOURCE_DESC.resType` specifies the type of resource to texture from. CUresourceType is defined as: - **View CUDA Toolkit Documentation for a C++ code example** If :py:obj:`~.CUDA_RESOURCE_DESC.resType` is set to :py:obj:`~.CU_RESOURCE_TYPE_ARRAY`, :py:obj:`~.CUDA_RESOURCE_DESC`::res::array::hArray must be set to a valid CUDA array handle. If :py:obj:`~.CUDA_RESOURCE_DESC.resType` is set to :py:obj:`~.CU_RESOURCE_TYPE_MIPMAPPED_ARRAY`, :py:obj:`~.CUDA_RESOURCE_DESC`::res::mipmap::hMipmappedArray must be set to a valid CUDA mipmapped array handle. If :py:obj:`~.CUDA_RESOURCE_DESC.resType` is set to :py:obj:`~.CU_RESOURCE_TYPE_LINEAR`, :py:obj:`~.CUDA_RESOURCE_DESC`::res::linear::devPtr must be set to a valid device pointer, that is aligned to :py:obj:`~.CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT`. :py:obj:`~.CUDA_RESOURCE_DESC`::res::linear::format and :py:obj:`~.CUDA_RESOURCE_DESC`::res::linear::numChannels describe the format of each component and the number of components per array element. :py:obj:`~.CUDA_RESOURCE_DESC`::res::linear::sizeInBytes specifies the size of the array in bytes. The total number of elements in the linear address range cannot exceed :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE1D_LINEAR_WIDTH`. The number of elements is computed as (sizeInBytes / (sizeof(format) * numChannels)). If :py:obj:`~.CUDA_RESOURCE_DESC.resType` is set to :py:obj:`~.CU_RESOURCE_TYPE_PITCH2D`, :py:obj:`~.CUDA_RESOURCE_DESC`::res::pitch2D::devPtr must be set to a valid device pointer, that is aligned to :py:obj:`~.CU_DEVICE_ATTRIBUTE_TEXTURE_ALIGNMENT`. :py:obj:`~.CUDA_RESOURCE_DESC`::res::pitch2D::format and :py:obj:`~.CUDA_RESOURCE_DESC`::res::pitch2D::numChannels describe the format of each component and the number of components per array element. :py:obj:`~.CUDA_RESOURCE_DESC`::res::pitch2D::width and :py:obj:`~.CUDA_RESOURCE_DESC`::res::pitch2D::height specify the width and height of the array in elements, and cannot exceed :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_WIDTH` and :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_HEIGHT` respectively. :py:obj:`~.CUDA_RESOURCE_DESC`::res::pitch2D::pitchInBytes specifies the pitch between two rows in bytes and has to be aligned to :py:obj:`~.CU_DEVICE_ATTRIBUTE_TEXTURE_PITCH_ALIGNMENT`. Pitch cannot exceed :py:obj:`~.CU_DEVICE_ATTRIBUTE_MAXIMUM_TEXTURE2D_LINEAR_PITCH`. - :py:obj:`~.flags` must be set to zero. The :py:obj:`~.CUDA_TEXTURE_DESC` struct is defined as **View CUDA Toolkit Documentation for a C++ code example** where - :py:obj:`~.CUDA_TEXTURE_DESC.addressMode` specifies the addressing mode for each dimension of the texture data. :py:obj:`~.CUaddress_mode` is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - This is ignored if :py:obj:`~.CUDA_RESOURCE_DESC.resType` is :py:obj:`~.CU_RESOURCE_TYPE_LINEAR`. Also, if the flag, :py:obj:`~.CU_TRSF_NORMALIZED_COORDINATES` is not set, the only supported address mode is :py:obj:`~.CU_TR_ADDRESS_MODE_CLAMP`. - :py:obj:`~.CUDA_TEXTURE_DESC.filterMode` specifies the filtering mode to be used when fetching from the texture. CUfilter_mode is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - This is ignored if :py:obj:`~.CUDA_RESOURCE_DESC.resType` is :py:obj:`~.CU_RESOURCE_TYPE_LINEAR`. - :py:obj:`~.CUDA_TEXTURE_DESC.flags` can be any combination of the following: - :py:obj:`~.CU_TRSF_READ_AS_INTEGER`, which suppresses the default behavior of having the texture promote integer data to floating point data in the range [0, 1]. Note that texture with 32-bit integer format would not be promoted, regardless of whether or not this flag is specified. - :py:obj:`~.CU_TRSF_NORMALIZED_COORDINATES`, which suppresses the default behavior of having the texture coordinates range from [0, Dim) where Dim is the width or height of the CUDA array. Instead, the texture coordinates [0, 1.0) reference the entire breadth of the array dimension; Note that for CUDA mipmapped arrays, this flag has to be set. - :py:obj:`~.CU_TRSF_DISABLE_TRILINEAR_OPTIMIZATION`, which disables any trilinear filtering optimizations. Trilinear optimizations improve texture filtering performance by allowing bilinear filtering on textures in scenarios where it can closely approximate the expected results. - :py:obj:`~.CU_TRSF_SEAMLESS_CUBEMAP`, which enables seamless cube map filtering. This flag can only be specified if the underlying resource is a CUDA array or a CUDA mipmapped array that was created with the flag :py:obj:`~.CUDA_ARRAY3D_CUBEMAP`. When seamless cube map filtering is enabled, texture address modes specified by :py:obj:`~.CUDA_TEXTURE_DESC.addressMode` are ignored. Instead, if the :py:obj:`~.CUDA_TEXTURE_DESC.filterMode` is set to :py:obj:`~.CU_TR_FILTER_MODE_POINT` the address mode :py:obj:`~.CU_TR_ADDRESS_MODE_CLAMP` will be applied for all dimensions. If the :py:obj:`~.CUDA_TEXTURE_DESC.filterMode` is set to :py:obj:`~.CU_TR_FILTER_MODE_LINEAR` seamless cube map filtering will be performed when sampling along the cube face borders. - :py:obj:`~.CUDA_TEXTURE_DESC.maxAnisotropy` specifies the maximum anisotropy ratio to be used when doing anisotropic filtering. This value will be clamped to the range [1,16]. - :py:obj:`~.CUDA_TEXTURE_DESC.mipmapFilterMode` specifies the filter mode when the calculated mipmap level lies between two defined mipmap levels. - :py:obj:`~.CUDA_TEXTURE_DESC.mipmapLevelBias` specifies the offset to be applied to the calculated mipmap level. - :py:obj:`~.CUDA_TEXTURE_DESC.minMipmapLevelClamp` specifies the lower end of the mipmap level range to clamp access to. - :py:obj:`~.CUDA_TEXTURE_DESC.maxMipmapLevelClamp` specifies the upper end of the mipmap level range to clamp access to. The :py:obj:`~.CUDA_RESOURCE_VIEW_DESC` struct is defined as **View CUDA Toolkit Documentation for a C++ code example** where: - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.format` specifies how the data contained in the CUDA array or CUDA mipmapped array should be interpreted. Note that this can incur a change in size of the texture data. If the resource view format is a block compressed format, then the underlying CUDA array or CUDA mipmapped array has to have a base of format :py:obj:`~.CU_AD_FORMAT_UNSIGNED_INT32`. with 2 or 4 channels, depending on the block compressed format. For ex., BC1 and BC4 require the underlying CUDA array to have a format of :py:obj:`~.CU_AD_FORMAT_UNSIGNED_INT32` with 2 channels. The other BC formats require the underlying resource to have the same base format but with 4 channels. - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.width` specifies the new width of the texture data. If the resource view format is a block compressed format, this value has to be 4 times the original width of the resource. For non block compressed formats, this value has to be equal to that of the original resource. - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.height` specifies the new height of the texture data. If the resource view format is a block compressed format, this value has to be 4 times the original height of the resource. For non block compressed formats, this value has to be equal to that of the original resource. - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.depth` specifies the new depth of the texture data. This value has to be equal to that of the original resource. - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.firstMipmapLevel` specifies the most detailed mipmap level. This will be the new mipmap level zero. For non-mipmapped resources, this value has to be zero.:py:obj:`~.CUDA_TEXTURE_DESC.minMipmapLevelClamp` and :py:obj:`~.CUDA_TEXTURE_DESC.maxMipmapLevelClamp` will be relative to this value. For ex., if the firstMipmapLevel is set to 2, and a minMipmapLevelClamp of 1.2 is specified, then the actual minimum mipmap level clamp will be 3.2. - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.lastMipmapLevel` specifies the least detailed mipmap level. For non-mipmapped resources, this value has to be zero. - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.firstLayer` specifies the first layer index for layered textures. This will be the new layer zero. For non-layered resources, this value has to be zero. - :py:obj:`~.CUDA_RESOURCE_VIEW_DESC.lastLayer` specifies the last layer index for layered textures. For non-layered resources, this value has to be zero. Parameters ---------- pResDesc : :py:obj:`~.CUDA_RESOURCE_DESC` Resource descriptor pTexDesc : :py:obj:`~.CUDA_TEXTURE_DESC` Texture descriptor pResViewDesc : :py:obj:`~.CUDA_RESOURCE_VIEW_DESC` Resource view descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pTexObject : :py:obj:`~.CUtexObject` Texture object to create See Also -------- :py:obj:`~.cuTexObjectDestroy`, :py:obj:`~.cudaCreateTextureObject` """ cdef CUtexObject pTexObject = CUtexObject() cdef ccuda.CUDA_RESOURCE_DESC* cpResDesc_ptr = pResDesc._ptr if pResDesc != None else NULL cdef ccuda.CUDA_TEXTURE_DESC* cpTexDesc_ptr = pTexDesc._ptr if pTexDesc != None else NULL cdef ccuda.CUDA_RESOURCE_VIEW_DESC* cpResViewDesc_ptr = pResViewDesc._ptr if pResViewDesc != None else NULL err = ccuda.cuTexObjectCreate(pTexObject._ptr, cpResDesc_ptr, cpTexDesc_ptr, cpResViewDesc_ptr) return (CUresult(err), pTexObject) {{endif}} {{if 'cuTexObjectDestroy' in found_functions}} @cython.embedsignature(True) def cuTexObjectDestroy(texObject): """ Destroys a texture object. Destroys the texture object specified by `texObject`. Parameters ---------- texObject : :py:obj:`~.CUtexObject` Texture object to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTexObjectCreate`, :py:obj:`~.cudaDestroyTextureObject` """ cdef ccuda.CUtexObject ctexObject if texObject is None: ctexObject = 0 elif isinstance(texObject, (CUtexObject,)): ptexObject = int(texObject) ctexObject = ptexObject else: ptexObject = int(CUtexObject(texObject)) ctexObject = ptexObject err = ccuda.cuTexObjectDestroy(ctexObject) return (CUresult(err),) {{endif}} {{if 'cuTexObjectGetResourceDesc' in found_functions}} @cython.embedsignature(True) def cuTexObjectGetResourceDesc(texObject): """ Returns a texture object's resource descriptor. Returns the resource descriptor for the texture object specified by `texObject`. Parameters ---------- texObject : :py:obj:`~.CUtexObject` Texture object Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pResDesc : :py:obj:`~.CUDA_RESOURCE_DESC` Resource descriptor See Also -------- :py:obj:`~.cuTexObjectCreate`, :py:obj:`~.cudaGetTextureObjectResourceDesc`, """ cdef ccuda.CUtexObject ctexObject if texObject is None: ctexObject = 0 elif isinstance(texObject, (CUtexObject,)): ptexObject = int(texObject) ctexObject = ptexObject else: ptexObject = int(CUtexObject(texObject)) ctexObject = ptexObject cdef CUDA_RESOURCE_DESC pResDesc = CUDA_RESOURCE_DESC() err = ccuda.cuTexObjectGetResourceDesc(pResDesc._ptr, ctexObject) return (CUresult(err), pResDesc) {{endif}} {{if 'cuTexObjectGetTextureDesc' in found_functions}} @cython.embedsignature(True) def cuTexObjectGetTextureDesc(texObject): """ Returns a texture object's texture descriptor. Returns the texture descriptor for the texture object specified by `texObject`. Parameters ---------- texObject : :py:obj:`~.CUtexObject` Texture object Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pTexDesc : :py:obj:`~.CUDA_TEXTURE_DESC` Texture descriptor See Also -------- :py:obj:`~.cuTexObjectCreate`, :py:obj:`~.cudaGetTextureObjectTextureDesc` """ cdef ccuda.CUtexObject ctexObject if texObject is None: ctexObject = 0 elif isinstance(texObject, (CUtexObject,)): ptexObject = int(texObject) ctexObject = ptexObject else: ptexObject = int(CUtexObject(texObject)) ctexObject = ptexObject cdef CUDA_TEXTURE_DESC pTexDesc = CUDA_TEXTURE_DESC() err = ccuda.cuTexObjectGetTextureDesc(pTexDesc._ptr, ctexObject) return (CUresult(err), pTexDesc) {{endif}} {{if 'cuTexObjectGetResourceViewDesc' in found_functions}} @cython.embedsignature(True) def cuTexObjectGetResourceViewDesc(texObject): """ Returns a texture object's resource view descriptor. Returns the resource view descriptor for the texture object specified by `texObject`. If no resource view was set for `texObject`, the :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. Parameters ---------- texObject : :py:obj:`~.CUtexObject` Texture object Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pResViewDesc : :py:obj:`~.CUDA_RESOURCE_VIEW_DESC` Resource view descriptor See Also -------- :py:obj:`~.cuTexObjectCreate`, :py:obj:`~.cudaGetTextureObjectResourceViewDesc` """ cdef ccuda.CUtexObject ctexObject if texObject is None: ctexObject = 0 elif isinstance(texObject, (CUtexObject,)): ptexObject = int(texObject) ctexObject = ptexObject else: ptexObject = int(CUtexObject(texObject)) ctexObject = ptexObject cdef CUDA_RESOURCE_VIEW_DESC pResViewDesc = CUDA_RESOURCE_VIEW_DESC() err = ccuda.cuTexObjectGetResourceViewDesc(pResViewDesc._ptr, ctexObject) return (CUresult(err), pResViewDesc) {{endif}} {{if 'cuSurfObjectCreate' in found_functions}} @cython.embedsignature(True) def cuSurfObjectCreate(pResDesc : Optional[CUDA_RESOURCE_DESC]): """ Creates a surface object. Creates a surface object and returns it in `pSurfObject`. `pResDesc` describes the data to perform surface load/stores on. :py:obj:`~.CUDA_RESOURCE_DESC.resType` must be :py:obj:`~.CU_RESOURCE_TYPE_ARRAY` and :py:obj:`~.CUDA_RESOURCE_DESC`::res::array::hArray must be set to a valid CUDA array handle. :py:obj:`~.CUDA_RESOURCE_DESC.flags` must be set to zero. Surface objects are only supported on devices of compute capability 3.0 or higher. Additionally, a surface object is an opaque value, and, as such, should only be accessed through CUDA API calls. Parameters ---------- pResDesc : :py:obj:`~.CUDA_RESOURCE_DESC` Resource descriptor Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pSurfObject : :py:obj:`~.CUsurfObject` Surface object to create See Also -------- :py:obj:`~.cuSurfObjectDestroy`, :py:obj:`~.cudaCreateSurfaceObject` """ cdef CUsurfObject pSurfObject = CUsurfObject() cdef ccuda.CUDA_RESOURCE_DESC* cpResDesc_ptr = pResDesc._ptr if pResDesc != None else NULL err = ccuda.cuSurfObjectCreate(pSurfObject._ptr, cpResDesc_ptr) return (CUresult(err), pSurfObject) {{endif}} {{if 'cuSurfObjectDestroy' in found_functions}} @cython.embedsignature(True) def cuSurfObjectDestroy(surfObject): """ Destroys a surface object. Destroys the surface object specified by `surfObject`. Parameters ---------- surfObject : :py:obj:`~.CUsurfObject` Surface object to destroy Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuSurfObjectCreate`, :py:obj:`~.cudaDestroySurfaceObject` """ cdef ccuda.CUsurfObject csurfObject if surfObject is None: csurfObject = 0 elif isinstance(surfObject, (CUsurfObject,)): psurfObject = int(surfObject) csurfObject = psurfObject else: psurfObject = int(CUsurfObject(surfObject)) csurfObject = psurfObject err = ccuda.cuSurfObjectDestroy(csurfObject) return (CUresult(err),) {{endif}} {{if 'cuSurfObjectGetResourceDesc' in found_functions}} @cython.embedsignature(True) def cuSurfObjectGetResourceDesc(surfObject): """ Returns a surface object's resource descriptor. Returns the resource descriptor for the surface object specified by `surfObject`. Parameters ---------- surfObject : :py:obj:`~.CUsurfObject` Surface object Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pResDesc : :py:obj:`~.CUDA_RESOURCE_DESC` Resource descriptor See Also -------- :py:obj:`~.cuSurfObjectCreate`, :py:obj:`~.cudaGetSurfaceObjectResourceDesc` """ cdef ccuda.CUsurfObject csurfObject if surfObject is None: csurfObject = 0 elif isinstance(surfObject, (CUsurfObject,)): psurfObject = int(surfObject) csurfObject = psurfObject else: psurfObject = int(CUsurfObject(surfObject)) csurfObject = psurfObject cdef CUDA_RESOURCE_DESC pResDesc = CUDA_RESOURCE_DESC() err = ccuda.cuSurfObjectGetResourceDesc(pResDesc._ptr, csurfObject) return (CUresult(err), pResDesc) {{endif}} {{if 'cuTensorMapEncodeTiled' in found_functions}} @cython.embedsignature(True) def cuTensorMapEncodeTiled(tensorDataType not None : CUtensorMapDataType, tensorRank, globalAddress, globalDim : Optional[Tuple[cuuint64_t] | List[cuuint64_t]], globalStrides : Optional[Tuple[cuuint64_t] | List[cuuint64_t]], boxDim : Optional[Tuple[cuuint32_t] | List[cuuint32_t]], elementStrides : Optional[Tuple[cuuint32_t] | List[cuuint32_t]], interleave not None : CUtensorMapInterleave, swizzle not None : CUtensorMapSwizzle, l2Promotion not None : CUtensorMapL2promotion, oobFill not None : CUtensorMapFloatOOBfill): """ Create a tensor map descriptor object representing tiled memory region. Creates a descriptor for Tensor Memory Access (TMA) object specified by the parameters describing a tiled region and returns it in `tensorMap`. Tensor map objects are only supported on devices of compute capability 9.0 or higher. Additionally, a tensor map object is an opaque value, and, as such, should only be accessed through CUDA API calls. The parameters passed are bound to the following requirements: - `tensorMap` address must be aligned to 64 bytes. - `tensorDataType` has to be an enum from :py:obj:`~.CUtensorMapDataType` which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - `tensorRank` must be non-zero and less than or equal to the maximum supported dimensionality of 5. If `interleave` is not :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_NONE`, then `tensorRank` must additionally be greater than or equal to 3. - `globalAddress`, which specifies the starting address of the memory region described, must be 32 byte aligned when `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_32B` and 16 byte aligned otherwise. - `globalDim` array, which specifies tensor size of each of the `tensorRank` dimensions, must be non-zero and less than or equal to 2^32. - `globalStrides` array, which specifies tensor stride of each of the lower `tensorRank` - 1 dimensions in bytes, must be a multiple of 16 and less than 2^40. Additionally, the stride must be a multiple of 32 when `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_32B`. Each following dimension specified includes previous dimension stride: - **View CUDA Toolkit Documentation for a C++ code example** - `boxDim` array, which specifies number of elements to be traversed along each of the `tensorRank` dimensions, must be non-zero and less than or equal to 256. When `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_NONE`, { `boxDim`[0] * elementSizeInBytes( `tensorDataType` ) } must be a multiple of 16 bytes. - `elementStrides` array, which specifies the iteration step along each of the `tensorRank` dimensions, must be non-zero and less than or equal to 8. Note that when `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_NONE`, the first element of this array is ignored since TMA doesn’t support the stride for dimension zero. When all elements of `elementStrides` array is one, `boxDim` specifies the number of elements to load. However, if the `elementStrides`[i] is not equal to one, then TMA loads ceil( `boxDim`[i] / `elementStrides`[i]) number of elements along i-th dimension. To load N elements along i-th dimension, `boxDim`[i] must be set to N * `elementStrides`[i]. - `interleave` specifies the interleaved layout of type :py:obj:`~.CUtensorMapInterleave`, which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - TMA supports interleaved layouts like NC/8HWC8 where C8 utilizes 16 bytes in memory assuming 2 byte per channel or NC/16HWC16 where C16 uses 32 bytes. When `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_NONE` and `swizzle` is not :py:obj:`~.CU_TENSOR_MAP_SWIZZLE_NONE`, the bounding box inner dimension (computed as `boxDim`[0] multiplied by element size derived from `tensorDataType`) must be less than or equal to the swizzle size. - CU_TENSOR_MAP_SWIZZLE_32B implies the bounding box inner dimension will be <= 32. - CU_TENSOR_MAP_SWIZZLE_64B implies the bounding box inner dimension will be <= 64. - CU_TENSOR_MAP_SWIZZLE_128B implies the bounding box inner dimension will be <= 128. - `swizzle`, which specifies the shared memory bank swizzling pattern, has to be of type :py:obj:`~.CUtensorMapSwizzle` which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - Data are organized in a specific order in global memory; however, this may not match the order in which the application accesses data in shared memory. This difference in data organization may cause bank conflicts when shared memory is accessed. In order to avoid this problem, data can be loaded to shared memory with shuffling across shared memory banks. When `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_32B`, `swizzle` must be :py:obj:`~.CU_TENSOR_MAP_SWIZZLE_32B`. Other interleave modes can have any swizzling pattern. - `l2Promotion` specifies L2 fetch size which indicates the byte granurality at which L2 requests is filled from DRAM. It must be of type :py:obj:`~.CUtensorMapL2promotion`, which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - `oobFill`, which indicates whether zero or a special NaN constant should be used to fill out-of-bound elements, must be of type :py:obj:`~.CUtensorMapFloatOOBfill` which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - Note that :py:obj:`~.CU_TENSOR_MAP_FLOAT_OOB_FILL_NAN_REQUEST_ZERO_FMA` can only be used when `tensorDataType` represents a floating-point data type. Parameters ---------- tensorDataType : :py:obj:`~.CUtensorMapDataType` Tensor data type tensorRank : Any Dimensionality of tensor globalAddress : Any Starting address of memory region described by tensor globalDim : List[:py:obj:`~.cuuint64_t`] Array containing tensor size (number of elements) along each of the `tensorRank` dimensions globalStrides : List[:py:obj:`~.cuuint64_t`] Array containing stride size (in bytes) along each of the `tensorRank` - 1 dimensions boxDim : List[:py:obj:`~.cuuint32_t`] Array containing traversal box size (number of elments) along each of the `tensorRank` dimensions. Specifies how many elements to be traversed along each tensor dimension. elementStrides : List[:py:obj:`~.cuuint32_t`] Array containing traversal stride in each of the `tensorRank` dimensions interleave : :py:obj:`~.CUtensorMapInterleave` Type of interleaved layout the tensor addresses swizzle : :py:obj:`~.CUtensorMapSwizzle` Bank swizzling pattern inside shared memory l2Promotion : :py:obj:`~.CUtensorMapL2promotion` L2 promotion size oobFill : :py:obj:`~.CUtensorMapFloatOOBfill` Indicate whether zero or special NaN constant must be used to fill out-of-bound elements Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` tensorMap : :py:obj:`~.CUtensorMap` Tensor map object to create See Also -------- :py:obj:`~.cuTensorMapEncodeIm2col`, :py:obj:`~.cuTensorMapReplaceAddress` """ elementStrides = [] if elementStrides is None else elementStrides if not all(isinstance(_x, (cuuint32_t,)) for _x in elementStrides): raise TypeError("Argument 'elementStrides' is not instance of type (expected Tuple[ccuda.cuuint32_t,] or List[ccuda.cuuint32_t,]") boxDim = [] if boxDim is None else boxDim if not all(isinstance(_x, (cuuint32_t,)) for _x in boxDim): raise TypeError("Argument 'boxDim' is not instance of type (expected Tuple[ccuda.cuuint32_t,] or List[ccuda.cuuint32_t,]") globalStrides = [] if globalStrides is None else globalStrides if not all(isinstance(_x, (cuuint64_t,)) for _x in globalStrides): raise TypeError("Argument 'globalStrides' is not instance of type (expected Tuple[ccuda.cuuint64_t,] or List[ccuda.cuuint64_t,]") globalDim = [] if globalDim is None else globalDim if not all(isinstance(_x, (cuuint64_t,)) for _x in globalDim): raise TypeError("Argument 'globalDim' is not instance of type (expected Tuple[ccuda.cuuint64_t,] or List[ccuda.cuuint64_t,]") cdef ccuda.cuuint32_t ctensorRank if tensorRank is None: ctensorRank = 0 elif isinstance(tensorRank, (cuuint32_t,)): ptensorRank = int(tensorRank) ctensorRank = ptensorRank else: ptensorRank = int(cuuint32_t(tensorRank)) ctensorRank = ptensorRank cdef CUtensorMap tensorMap = CUtensorMap() cdef ccuda.CUtensorMapDataType ctensorDataType = tensorDataType.value cglobalAddress = utils.HelperInputVoidPtr(globalAddress) cdef void* cglobalAddress_ptr = cglobalAddress.cptr cdef ccuda.cuuint64_t* cglobalDim = NULL if len(globalDim) > 0: cglobalDim = calloc(len(globalDim), sizeof(ccuda.cuuint64_t)) if cglobalDim is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(globalDim)) + 'x' + str(sizeof(ccuda.cuuint64_t))) else: for idx in range(len(globalDim)): cglobalDim[idx] = (globalDim[idx])._ptr[0] cdef ccuda.cuuint64_t* cglobalStrides = NULL if len(globalStrides) > 0: cglobalStrides = calloc(len(globalStrides), sizeof(ccuda.cuuint64_t)) if cglobalStrides is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(globalStrides)) + 'x' + str(sizeof(ccuda.cuuint64_t))) else: for idx in range(len(globalStrides)): cglobalStrides[idx] = (globalStrides[idx])._ptr[0] cdef ccuda.cuuint32_t* cboxDim = NULL if len(boxDim) > 0: cboxDim = calloc(len(boxDim), sizeof(ccuda.cuuint32_t)) if cboxDim is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(boxDim)) + 'x' + str(sizeof(ccuda.cuuint32_t))) else: for idx in range(len(boxDim)): cboxDim[idx] = (boxDim[idx])._ptr[0] cdef ccuda.cuuint32_t* celementStrides = NULL if len(elementStrides) > 0: celementStrides = calloc(len(elementStrides), sizeof(ccuda.cuuint32_t)) if celementStrides is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(elementStrides)) + 'x' + str(sizeof(ccuda.cuuint32_t))) else: for idx in range(len(elementStrides)): celementStrides[idx] = (elementStrides[idx])._ptr[0] cdef ccuda.CUtensorMapInterleave cinterleave = interleave.value cdef ccuda.CUtensorMapSwizzle cswizzle = swizzle.value cdef ccuda.CUtensorMapL2promotion cl2Promotion = l2Promotion.value cdef ccuda.CUtensorMapFloatOOBfill coobFill = oobFill.value err = ccuda.cuTensorMapEncodeTiled(tensorMap._ptr, ctensorDataType, ctensorRank, cglobalAddress_ptr, (globalDim[0])._ptr if len(globalDim) == 1 else cglobalDim, (globalStrides[0])._ptr if len(globalStrides) == 1 else cglobalStrides, (boxDim[0])._ptr if len(boxDim) == 1 else cboxDim, (elementStrides[0])._ptr if len(elementStrides) == 1 else celementStrides, cinterleave, cswizzle, cl2Promotion, coobFill) if cglobalDim is not NULL: free(cglobalDim) if cglobalStrides is not NULL: free(cglobalStrides) if cboxDim is not NULL: free(cboxDim) if celementStrides is not NULL: free(celementStrides) return (CUresult(err), tensorMap) {{endif}} {{if 'cuTensorMapEncodeIm2col' in found_functions}} @cython.embedsignature(True) def cuTensorMapEncodeIm2col(tensorDataType not None : CUtensorMapDataType, tensorRank, globalAddress, globalDim : Optional[Tuple[cuuint64_t] | List[cuuint64_t]], globalStrides : Optional[Tuple[cuuint64_t] | List[cuuint64_t]], pixelBoxLowerCorner : Optional[Tuple[int] | List[int]], pixelBoxUpperCorner : Optional[Tuple[int] | List[int]], channelsPerPixel, pixelsPerColumn, elementStrides : Optional[Tuple[cuuint32_t] | List[cuuint32_t]], interleave not None : CUtensorMapInterleave, swizzle not None : CUtensorMapSwizzle, l2Promotion not None : CUtensorMapL2promotion, oobFill not None : CUtensorMapFloatOOBfill): """ Create a tensor map descriptor object representing im2col memory region. Creates a descriptor for Tensor Memory Access (TMA) object specified by the parameters describing a im2col memory layout and returns it in `tensorMap`. Tensor map objects are only supported on devices of compute capability 9.0 or higher. Additionally, a tensor map object is an opaque value, and, as such, should only be accessed through CUDA API calls. The parameters passed are bound to the following requirements: - `tensorMap` address must be aligned to 64 bytes. - `tensorDataType` has to be an enum from :py:obj:`~.CUtensorMapDataType` which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - `tensorRank`, which specifies the number of tensor dimensions, must be 3, 4, or 5. - `globalAddress`, which specifies the starting address of the memory region described, must be 32 byte aligned when `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_32B` and 16 byte aligned otherwise. - `globalDim` array, which specifies tensor size of each of the `tensorRank` dimensions, must be non-zero and less than or equal to 2^32. - `globalStrides` array, which specifies tensor stride of each of the lower `tensorRank` - 1 dimensions in bytes, must be a multiple of 16 and less than 2^40. Additionally, the stride must be a multiple of 32 when `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_32B`. Each following dimension specified includes previous dimension stride: - **View CUDA Toolkit Documentation for a C++ code example** - `pixelBoxLowerCorner` array specifies the coordinate offsets {D, H, W} of the bounding box from top/left/front corner. The number of offsets and their precision depend on the tensor dimensionality: - When `tensorRank` is 3, one signed offset within range [-32768, 32767] is supported. - When `tensorRank` is 4, two signed offsets each within range [-128, 127] are supported. - When `tensorRank` is 5, three offsets each within range [-16, 15] are supported. - `pixelBoxUpperCorner` array specifies the coordinate offsets {D, H, W} of the bounding box from bottom/right/back corner. The number of offsets and their precision depend on the tensor dimensionality: - When `tensorRank` is 3, one signed offset within range [-32768, 32767] is supported. - When `tensorRank` is 4, two signed offsets each within range [-128, 127] are supported. - When `tensorRank` is 5, three offsets each within range [-16, 15] are supported. The bounding box specified by `pixelBoxLowerCorner` and `pixelBoxUpperCorner` must have non-zero area. - `channelsPerPixel`, which specifies the number of elements which must be accessed along C dimension, must be less than or equal to 256. - `pixelsPerColumn`, which specifies the number of elements that must be accessed along the {N, D, H, W} dimensions, must be less than or equal to 1024. - `elementStrides` array, which specifies the iteration step along each of the `tensorRank` dimensions, must be non-zero and less than or equal to 8. Note that when `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_NONE`, the first element of this array is ignored since TMA doesn’t support the stride for dimension zero. When all elements of the `elementStrides` array are one, `boxDim` specifies the number of elements to load. However, if `elementStrides`[i] is not equal to one for some `i`, then TMA loads ceil( `boxDim`[i] / `elementStrides`[i]) number of elements along i-th dimension. To load N elements along i-th dimension, `boxDim`[i] must be set to N * `elementStrides`[i]. - `interleave` specifies the interleaved layout of type :py:obj:`~.CUtensorMapInterleave`, which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - TMA supports interleaved layouts like NC/8HWC8 where C8 utilizes 16 bytes in memory assuming 2 byte per channel or NC/16HWC16 where C16 uses 32 bytes. When `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_NONE` and `swizzle` is not :py:obj:`~.CU_TENSOR_MAP_SWIZZLE_NONE`, the bounding box inner dimension (computed as `boxDim`[0] multiplied by element size derived from `tensorDataType`) must be less than or equal to the swizzle size. - CU_TENSOR_MAP_SWIZZLE_32B implies the bounding box inner dimension will be <= 32. - CU_TENSOR_MAP_SWIZZLE_64B implies the bounding box inner dimension will be <= 64. - CU_TENSOR_MAP_SWIZZLE_128B implies the bounding box inner dimension will be <= 128. - `swizzle`, which specifies the shared memory bank swizzling pattern, has to be of type :py:obj:`~.CUtensorMapSwizzle` which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - Data are organized in a specific order in global memory; however, this may not match the order in which the application accesses data in shared memory. This difference in data organization may cause bank conflicts when shared memory is accessed. In order to avoid this problem, data can be loaded to shared memory with shuffling across shared memory banks. When `interleave` is :py:obj:`~.CU_TENSOR_MAP_INTERLEAVE_32B`, `swizzle` must be :py:obj:`~.CU_TENSOR_MAP_SWIZZLE_32B`. Other interleave modes can have any swizzling pattern. - `l2Promotion` specifies L2 fetch size which indicates the byte granularity at which L2 requests are filled from DRAM. It must be of type :py:obj:`~.CUtensorMapL2promotion`, which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - `oobFill`, which indicates whether zero or a special NaN constant should be used to fill out-of-bound elements, must be of type :py:obj:`~.CUtensorMapFloatOOBfill` which is defined as: - **View CUDA Toolkit Documentation for a C++ code example** - Note that :py:obj:`~.CU_TENSOR_MAP_FLOAT_OOB_FILL_NAN_REQUEST_ZERO_FMA` can only be used when `tensorDataType` represents a floating-point data type. Parameters ---------- tensorDataType : :py:obj:`~.CUtensorMapDataType` Tensor data type tensorRank : Any Dimensionality of tensor; must be at least 3 globalAddress : Any Starting address of memory region described by tensor globalDim : List[:py:obj:`~.cuuint64_t`] Array containing tensor size (number of elements) along each of the `tensorRank` dimensions globalStrides : List[:py:obj:`~.cuuint64_t`] Array containing stride size (in bytes) along each of the `tensorRank` - 1 dimensions pixelBoxLowerCorner : List[int] Array containing DHW dimensions of lower box corner pixelBoxUpperCorner : List[int] Array containing DHW dimensions of upper box corner channelsPerPixel : Any Number of channels per pixel pixelsPerColumn : Any Number of pixels per column elementStrides : List[:py:obj:`~.cuuint32_t`] Array containing traversal stride in each of the `tensorRank` dimensions interleave : :py:obj:`~.CUtensorMapInterleave` Type of interleaved layout the tensor addresses swizzle : :py:obj:`~.CUtensorMapSwizzle` Bank swizzling pattern inside shared memory l2Promotion : :py:obj:`~.CUtensorMapL2promotion` L2 promotion size oobFill : :py:obj:`~.CUtensorMapFloatOOBfill` Indicate whether zero or special NaN constant will be used to fill out-of-bound elements Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` tensorMap : :py:obj:`~.CUtensorMap` Tensor map object to create See Also -------- :py:obj:`~.cuTensorMapEncodeTiled`, :py:obj:`~.cuTensorMapReplaceAddress` """ elementStrides = [] if elementStrides is None else elementStrides if not all(isinstance(_x, (cuuint32_t,)) for _x in elementStrides): raise TypeError("Argument 'elementStrides' is not instance of type (expected Tuple[ccuda.cuuint32_t,] or List[ccuda.cuuint32_t,]") cdef ccuda.cuuint32_t cpixelsPerColumn if pixelsPerColumn is None: cpixelsPerColumn = 0 elif isinstance(pixelsPerColumn, (cuuint32_t,)): ppixelsPerColumn = int(pixelsPerColumn) cpixelsPerColumn = ppixelsPerColumn else: ppixelsPerColumn = int(cuuint32_t(pixelsPerColumn)) cpixelsPerColumn = ppixelsPerColumn cdef ccuda.cuuint32_t cchannelsPerPixel if channelsPerPixel is None: cchannelsPerPixel = 0 elif isinstance(channelsPerPixel, (cuuint32_t,)): pchannelsPerPixel = int(channelsPerPixel) cchannelsPerPixel = pchannelsPerPixel else: pchannelsPerPixel = int(cuuint32_t(channelsPerPixel)) cchannelsPerPixel = pchannelsPerPixel pixelBoxUpperCorner = [] if pixelBoxUpperCorner is None else pixelBoxUpperCorner if not all(isinstance(_x, (int)) for _x in pixelBoxUpperCorner): raise TypeError("Argument 'pixelBoxUpperCorner' is not instance of type (expected Tuple[int] or List[int]") pixelBoxLowerCorner = [] if pixelBoxLowerCorner is None else pixelBoxLowerCorner if not all(isinstance(_x, (int)) for _x in pixelBoxLowerCorner): raise TypeError("Argument 'pixelBoxLowerCorner' is not instance of type (expected Tuple[int] or List[int]") globalStrides = [] if globalStrides is None else globalStrides if not all(isinstance(_x, (cuuint64_t,)) for _x in globalStrides): raise TypeError("Argument 'globalStrides' is not instance of type (expected Tuple[ccuda.cuuint64_t,] or List[ccuda.cuuint64_t,]") globalDim = [] if globalDim is None else globalDim if not all(isinstance(_x, (cuuint64_t,)) for _x in globalDim): raise TypeError("Argument 'globalDim' is not instance of type (expected Tuple[ccuda.cuuint64_t,] or List[ccuda.cuuint64_t,]") cdef ccuda.cuuint32_t ctensorRank if tensorRank is None: ctensorRank = 0 elif isinstance(tensorRank, (cuuint32_t,)): ptensorRank = int(tensorRank) ctensorRank = ptensorRank else: ptensorRank = int(cuuint32_t(tensorRank)) ctensorRank = ptensorRank cdef CUtensorMap tensorMap = CUtensorMap() cdef ccuda.CUtensorMapDataType ctensorDataType = tensorDataType.value cglobalAddress = utils.HelperInputVoidPtr(globalAddress) cdef void* cglobalAddress_ptr = cglobalAddress.cptr cdef ccuda.cuuint64_t* cglobalDim = NULL if len(globalDim) > 0: cglobalDim = calloc(len(globalDim), sizeof(ccuda.cuuint64_t)) if cglobalDim is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(globalDim)) + 'x' + str(sizeof(ccuda.cuuint64_t))) else: for idx in range(len(globalDim)): cglobalDim[idx] = (globalDim[idx])._ptr[0] cdef ccuda.cuuint64_t* cglobalStrides = NULL if len(globalStrides) > 0: cglobalStrides = calloc(len(globalStrides), sizeof(ccuda.cuuint64_t)) if cglobalStrides is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(globalStrides)) + 'x' + str(sizeof(ccuda.cuuint64_t))) else: for idx in range(len(globalStrides)): cglobalStrides[idx] = (globalStrides[idx])._ptr[0] cdef vector[int] cpixelBoxLowerCorner = pixelBoxLowerCorner cdef vector[int] cpixelBoxUpperCorner = pixelBoxUpperCorner cdef ccuda.cuuint32_t* celementStrides = NULL if len(elementStrides) > 0: celementStrides = calloc(len(elementStrides), sizeof(ccuda.cuuint32_t)) if celementStrides is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(elementStrides)) + 'x' + str(sizeof(ccuda.cuuint32_t))) else: for idx in range(len(elementStrides)): celementStrides[idx] = (elementStrides[idx])._ptr[0] cdef ccuda.CUtensorMapInterleave cinterleave = interleave.value cdef ccuda.CUtensorMapSwizzle cswizzle = swizzle.value cdef ccuda.CUtensorMapL2promotion cl2Promotion = l2Promotion.value cdef ccuda.CUtensorMapFloatOOBfill coobFill = oobFill.value err = ccuda.cuTensorMapEncodeIm2col(tensorMap._ptr, ctensorDataType, ctensorRank, cglobalAddress_ptr, (globalDim[0])._ptr if len(globalDim) == 1 else cglobalDim, (globalStrides[0])._ptr if len(globalStrides) == 1 else cglobalStrides, cpixelBoxLowerCorner.data(), cpixelBoxUpperCorner.data(), cchannelsPerPixel, cpixelsPerColumn, (elementStrides[0])._ptr if len(elementStrides) == 1 else celementStrides, cinterleave, cswizzle, cl2Promotion, coobFill) if cglobalDim is not NULL: free(cglobalDim) if cglobalStrides is not NULL: free(cglobalStrides) if celementStrides is not NULL: free(celementStrides) return (CUresult(err), tensorMap) {{endif}} {{if 'cuTensorMapReplaceAddress' in found_functions}} @cython.embedsignature(True) def cuTensorMapReplaceAddress(tensorMap : Optional[CUtensorMap], globalAddress): """ Modify an existing tensor map descriptor with an updated global address. Modifies the descriptor for Tensor Memory Access (TMA) object passed in `tensorMap` with an updated `globalAddress`. Tensor map objects are only supported on devices of compute capability 9.0 or higher. Additionally, a tensor map object is an opaque value, and, as such, should only be accessed through CUDA API calls. Parameters ---------- tensorMap : :py:obj:`~.CUtensorMap` Tensor map object to modify globalAddress : Any Starting address of memory region described by tensor, must follow previous alignment requirements Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuTensorMapEncodeTiled`, :py:obj:`~.cuTensorMapEncodeIm2col` """ cdef ccuda.CUtensorMap* ctensorMap_ptr = tensorMap._ptr if tensorMap != None else NULL cglobalAddress = utils.HelperInputVoidPtr(globalAddress) cdef void* cglobalAddress_ptr = cglobalAddress.cptr err = ccuda.cuTensorMapReplaceAddress(ctensorMap_ptr, cglobalAddress_ptr) return (CUresult(err),) {{endif}} {{if 'cuDeviceCanAccessPeer' in found_functions}} @cython.embedsignature(True) def cuDeviceCanAccessPeer(dev, peerDev): """ Queries if a device may directly access a peer device's memory. Returns in `*canAccessPeer` a value of 1 if contexts on `dev` are capable of directly accessing memory from contexts on `peerDev` and 0 otherwise. If direct access of `peerDev` from `dev` is possible, then access may be enabled on two specific contexts by calling :py:obj:`~.cuCtxEnablePeerAccess()`. Parameters ---------- dev : :py:obj:`~.CUdevice` Device from which allocations on `peerDev` are to be directly accessed. peerDev : :py:obj:`~.CUdevice` Device on which the allocations to be directly accessed by `dev` reside. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` canAccessPeer : int Returned access capability See Also -------- :py:obj:`~.cuCtxEnablePeerAccess`, :py:obj:`~.cuCtxDisablePeerAccess`, :py:obj:`~.cudaDeviceCanAccessPeer` """ cdef ccuda.CUdevice cpeerDev if peerDev is None: cpeerDev = 0 elif isinstance(peerDev, (CUdevice,)): ppeerDev = int(peerDev) cpeerDev = ppeerDev else: ppeerDev = int(CUdevice(peerDev)) cpeerDev = ppeerDev cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef int canAccessPeer = 0 err = ccuda.cuDeviceCanAccessPeer(&canAccessPeer, cdev, cpeerDev) return (CUresult(err), canAccessPeer) {{endif}} {{if 'cuCtxEnablePeerAccess' in found_functions}} @cython.embedsignature(True) def cuCtxEnablePeerAccess(peerContext, unsigned int Flags): """ Enables direct access to memory allocations in a peer context. If both the current context and `peerContext` are on devices which support unified addressing (as may be queried using :py:obj:`~.CU_DEVICE_ATTRIBUTE_UNIFIED_ADDRESSING`) and same major compute capability, then on success all allocations from `peerContext` will immediately be accessible by the current context. See :py:obj:`~.Unified Addressing` for additional details. Note that access granted by this call is unidirectional and that in order to access memory from the current context in `peerContext`, a separate symmetric call to :py:obj:`~.cuCtxEnablePeerAccess()` is required. Note that there are both device-wide and system-wide limitations per system configuration, as noted in the CUDA Programming Guide under the section "Peer-to-Peer Memory Access". Returns :py:obj:`~.CUDA_ERROR_PEER_ACCESS_UNSUPPORTED` if :py:obj:`~.cuDeviceCanAccessPeer()` indicates that the :py:obj:`~.CUdevice` of the current context cannot directly access memory from the :py:obj:`~.CUdevice` of `peerContext`. Returns :py:obj:`~.CUDA_ERROR_PEER_ACCESS_ALREADY_ENABLED` if direct access of `peerContext` from the current context has already been enabled. Returns :py:obj:`~.CUDA_ERROR_TOO_MANY_PEERS` if direct peer access is not possible because hardware resources required for peer access have been exhausted. Returns :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` if there is no current context, `peerContext` is not a valid context, or if the current context is `peerContext`. Returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if `Flags` is not 0. Parameters ---------- peerContext : :py:obj:`~.CUcontext` Peer context to enable direct access to from the current context Flags : unsigned int Reserved for future use and must be set to 0 Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_PEER_ACCESS_ALREADY_ENABLED`, :py:obj:`~.CUDA_ERROR_TOO_MANY_PEERS`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_PEER_ACCESS_UNSUPPORTED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` See Also -------- :py:obj:`~.cuDeviceCanAccessPeer`, :py:obj:`~.cuCtxDisablePeerAccess`, :py:obj:`~.cudaDeviceEnablePeerAccess` """ cdef ccuda.CUcontext cpeerContext if peerContext is None: cpeerContext = 0 elif isinstance(peerContext, (CUcontext,)): ppeerContext = int(peerContext) cpeerContext = ppeerContext else: ppeerContext = int(CUcontext(peerContext)) cpeerContext = ppeerContext err = ccuda.cuCtxEnablePeerAccess(cpeerContext, Flags) return (CUresult(err),) {{endif}} {{if 'cuCtxDisablePeerAccess' in found_functions}} @cython.embedsignature(True) def cuCtxDisablePeerAccess(peerContext): """ Disables direct access to memory allocations in a peer context and unregisters any registered allocations. Returns :py:obj:`~.CUDA_ERROR_PEER_ACCESS_NOT_ENABLED` if direct peer access has not yet been enabled from `peerContext` to the current context. Returns :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` if there is no current context, or if `peerContext` is not a valid context. Parameters ---------- peerContext : :py:obj:`~.CUcontext` Peer context to disable direct access to Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_PEER_ACCESS_NOT_ENABLED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, See Also -------- :py:obj:`~.cuDeviceCanAccessPeer`, :py:obj:`~.cuCtxEnablePeerAccess`, :py:obj:`~.cudaDeviceDisablePeerAccess` """ cdef ccuda.CUcontext cpeerContext if peerContext is None: cpeerContext = 0 elif isinstance(peerContext, (CUcontext,)): ppeerContext = int(peerContext) cpeerContext = ppeerContext else: ppeerContext = int(CUcontext(peerContext)) cpeerContext = ppeerContext err = ccuda.cuCtxDisablePeerAccess(cpeerContext) return (CUresult(err),) {{endif}} {{if 'cuDeviceGetP2PAttribute' in found_functions}} @cython.embedsignature(True) def cuDeviceGetP2PAttribute(attrib not None : CUdevice_P2PAttribute, srcDevice, dstDevice): """ Queries attributes of the link between two devices. Returns in `*value` the value of the requested attribute `attrib` of the link between `srcDevice` and `dstDevice`. The supported attributes are: - :py:obj:`~.CU_DEVICE_P2P_ATTRIBUTE_PERFORMANCE_RANK`: A relative value indicating the performance of the link between two devices. - :py:obj:`~.CU_DEVICE_P2P_ATTRIBUTE_ACCESS_SUPPORTED` P2P: 1 if P2P Access is enable. - :py:obj:`~.CU_DEVICE_P2P_ATTRIBUTE_NATIVE_ATOMIC_SUPPORTED`: 1 if Atomic operations over the link are supported. - :py:obj:`~.CU_DEVICE_P2P_ATTRIBUTE_CUDA_ARRAY_ACCESS_SUPPORTED`: 1 if cudaArray can be accessed over the link. Returns :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` if `srcDevice` or `dstDevice` are not valid or if they represent the same device. Returns :py:obj:`~.CUDA_ERROR_INVALID_VALUE` if `attrib` is not valid or if `value` is a null pointer. Parameters ---------- attrib : :py:obj:`~.CUdevice_P2PAttribute` The requested attribute of the link between `srcDevice` and `dstDevice`. srcDevice : :py:obj:`~.CUdevice` The source device of the target link. dstDevice : :py:obj:`~.CUdevice` The destination device of the target link. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` value : int Returned value of the requested attribute See Also -------- :py:obj:`~.cuCtxEnablePeerAccess`, :py:obj:`~.cuCtxDisablePeerAccess`, :py:obj:`~.cuDeviceCanAccessPeer`, :py:obj:`~.cudaDeviceGetP2PAttribute` """ cdef ccuda.CUdevice cdstDevice if dstDevice is None: cdstDevice = 0 elif isinstance(dstDevice, (CUdevice,)): pdstDevice = int(dstDevice) cdstDevice = pdstDevice else: pdstDevice = int(CUdevice(dstDevice)) cdstDevice = pdstDevice cdef ccuda.CUdevice csrcDevice if srcDevice is None: csrcDevice = 0 elif isinstance(srcDevice, (CUdevice,)): psrcDevice = int(srcDevice) csrcDevice = psrcDevice else: psrcDevice = int(CUdevice(srcDevice)) csrcDevice = psrcDevice cdef int value = 0 cdef ccuda.CUdevice_P2PAttribute cattrib = attrib.value err = ccuda.cuDeviceGetP2PAttribute(&value, cattrib, csrcDevice, cdstDevice) return (CUresult(err), value) {{endif}} {{if 'cuGraphicsUnregisterResource' in found_functions}} @cython.embedsignature(True) def cuGraphicsUnregisterResource(resource): """ Unregisters a graphics resource for access by CUDA. Unregisters the graphics resource `resource` so it is not accessible by CUDA unless registered again. If `resource` is invalid then :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` is returned. Parameters ---------- resource : :py:obj:`~.CUgraphicsResource` Resource to unregister Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_UNKNOWN` See Also -------- :py:obj:`~.cuGraphicsD3D9RegisterResource`, :py:obj:`~.cuGraphicsD3D10RegisterResource`, :py:obj:`~.cuGraphicsD3D11RegisterResource`, :py:obj:`~.cuGraphicsGLRegisterBuffer`, :py:obj:`~.cuGraphicsGLRegisterImage`, :py:obj:`~.cudaGraphicsUnregisterResource` """ cdef ccuda.CUgraphicsResource cresource if resource is None: cresource = 0 elif isinstance(resource, (CUgraphicsResource,)): presource = int(resource) cresource = presource else: presource = int(CUgraphicsResource(resource)) cresource = presource err = ccuda.cuGraphicsUnregisterResource(cresource) return (CUresult(err),) {{endif}} {{if 'cuGraphicsSubResourceGetMappedArray' in found_functions}} @cython.embedsignature(True) def cuGraphicsSubResourceGetMappedArray(resource, unsigned int arrayIndex, unsigned int mipLevel): """ Get an array through which to access a subresource of a mapped graphics resource. Returns in `*pArray` an array through which the subresource of the mapped graphics resource `resource` which corresponds to array index `arrayIndex` and mipmap level `mipLevel` may be accessed. The value set in `*pArray` may change every time that `resource` is mapped. If `resource` is not a texture then it cannot be accessed via an array and :py:obj:`~.CUDA_ERROR_NOT_MAPPED_AS_ARRAY` is returned. If `arrayIndex` is not a valid array index for `resource` then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. If `mipLevel` is not a valid mipmap level for `resource` then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. If `resource` is not mapped then :py:obj:`~.CUDA_ERROR_NOT_MAPPED` is returned. Parameters ---------- resource : :py:obj:`~.CUgraphicsResource` Mapped resource to access arrayIndex : unsigned int Array index for array textures or cubemap face index as defined by :py:obj:`~.CUarray_cubemap_face` for cubemap textures for the subresource to access mipLevel : unsigned int Mipmap level for the subresource to access Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_MAPPED`, :py:obj:`~.CUDA_ERROR_NOT_MAPPED_AS_ARRAY` pArray : :py:obj:`~.CUarray` Returned array through which a subresource of `resource` may be accessed See Also -------- :py:obj:`~.cuGraphicsResourceGetMappedPointer`, :py:obj:`~.cudaGraphicsSubResourceGetMappedArray` """ cdef ccuda.CUgraphicsResource cresource if resource is None: cresource = 0 elif isinstance(resource, (CUgraphicsResource,)): presource = int(resource) cresource = presource else: presource = int(CUgraphicsResource(resource)) cresource = presource cdef CUarray pArray = CUarray() err = ccuda.cuGraphicsSubResourceGetMappedArray(pArray._ptr, cresource, arrayIndex, mipLevel) return (CUresult(err), pArray) {{endif}} {{if 'cuGraphicsResourceGetMappedMipmappedArray' in found_functions}} @cython.embedsignature(True) def cuGraphicsResourceGetMappedMipmappedArray(resource): """ Get a mipmapped array through which to access a mapped graphics resource. Returns in `*pMipmappedArray` a mipmapped array through which the mapped graphics resource `resource`. The value set in `*pMipmappedArray` may change every time that `resource` is mapped. If `resource` is not a texture then it cannot be accessed via a mipmapped array and :py:obj:`~.CUDA_ERROR_NOT_MAPPED_AS_ARRAY` is returned. If `resource` is not mapped then :py:obj:`~.CUDA_ERROR_NOT_MAPPED` is returned. Parameters ---------- resource : :py:obj:`~.CUgraphicsResource` Mapped resource to access Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_MAPPED`, :py:obj:`~.CUDA_ERROR_NOT_MAPPED_AS_ARRAY` pMipmappedArray : :py:obj:`~.CUmipmappedArray` Returned mipmapped array through which `resource` may be accessed See Also -------- :py:obj:`~.cuGraphicsResourceGetMappedPointer`, :py:obj:`~.cudaGraphicsResourceGetMappedMipmappedArray` """ cdef ccuda.CUgraphicsResource cresource if resource is None: cresource = 0 elif isinstance(resource, (CUgraphicsResource,)): presource = int(resource) cresource = presource else: presource = int(CUgraphicsResource(resource)) cresource = presource cdef CUmipmappedArray pMipmappedArray = CUmipmappedArray() err = ccuda.cuGraphicsResourceGetMappedMipmappedArray(pMipmappedArray._ptr, cresource) return (CUresult(err), pMipmappedArray) {{endif}} {{if 'cuGraphicsResourceGetMappedPointer_v2' in found_functions}} @cython.embedsignature(True) def cuGraphicsResourceGetMappedPointer(resource): """ Get a device pointer through which to access a mapped graphics resource. Returns in `*pDevPtr` a pointer through which the mapped graphics resource `resource` may be accessed. Returns in `pSize` the size of the memory in bytes which may be accessed from that pointer. The value set in `pPointer` may change every time that `resource` is mapped. If `resource` is not a buffer then it cannot be accessed via a pointer and :py:obj:`~.CUDA_ERROR_NOT_MAPPED_AS_POINTER` is returned. If `resource` is not mapped then :py:obj:`~.CUDA_ERROR_NOT_MAPPED` is returned. Parameters ---------- resource : :py:obj:`~.CUgraphicsResource` None Returns ------- CUresult pDevPtr : :py:obj:`~.CUdeviceptr` None pSize : int None """ cdef ccuda.CUgraphicsResource cresource if resource is None: cresource = 0 elif isinstance(resource, (CUgraphicsResource,)): presource = int(resource) cresource = presource else: presource = int(CUgraphicsResource(resource)) cresource = presource cdef CUdeviceptr pDevPtr = CUdeviceptr() cdef size_t pSize = 0 err = ccuda.cuGraphicsResourceGetMappedPointer(pDevPtr._ptr, &pSize, cresource) return (CUresult(err), pDevPtr, pSize) {{endif}} {{if 'cuGraphicsResourceSetMapFlags_v2' in found_functions}} @cython.embedsignature(True) def cuGraphicsResourceSetMapFlags(resource, unsigned int flags): """ Set usage flags for mapping a graphics resource. Set `flags` for mapping the graphics resource `resource`. Changes to `flags` will take effect the next time `resource` is mapped. The `flags` argument may be any of the following: - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE`: Specifies no hints about how this resource will be used. It is therefore assumed that this resource will be read from and written to by CUDA kernels. This is the default value. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_READONLY`: Specifies that CUDA kernels which access this resource will not write to this resource. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITEDISCARD`: Specifies that CUDA kernels which access this resource will not read from this resource and will write over the entire contents of the resource, so none of the data previously stored in the resource will be preserved. If `resource` is presently mapped for access by CUDA then :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED` is returned. If `flags` is not one of the above values then :py:obj:`~.CUDA_ERROR_INVALID_VALUE` is returned. Parameters ---------- resource : :py:obj:`~.CUgraphicsResource` Registered resource to set flags for flags : unsigned int Parameters for resource mapping Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED` See Also -------- :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cudaGraphicsResourceSetMapFlags` """ cdef ccuda.CUgraphicsResource cresource if resource is None: cresource = 0 elif isinstance(resource, (CUgraphicsResource,)): presource = int(resource) cresource = presource else: presource = int(CUgraphicsResource(resource)) cresource = presource err = ccuda.cuGraphicsResourceSetMapFlags(cresource, flags) return (CUresult(err),) {{endif}} {{if 'cuGraphicsMapResources' in found_functions}} @cython.embedsignature(True) def cuGraphicsMapResources(unsigned int count, resources, hStream): """ Map graphics resources for access by CUDA. Maps the `count` graphics resources in `resources` for access by CUDA. The resources in `resources` may be accessed by CUDA until they are unmapped. The graphics API from which `resources` were registered should not access any resources while they are mapped by CUDA. If an application does so, the results are undefined. This function provides the synchronization guarantee that any graphics calls issued before :py:obj:`~.cuGraphicsMapResources()` will complete before any subsequent CUDA work issued in `stream` begins. If `resources` includes any duplicate entries then :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` is returned. If any of `resources` are presently mapped for access by CUDA then :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED` is returned. Parameters ---------- count : unsigned int Number of resources to map resources : :py:obj:`~.CUgraphicsResource` Resources to map for CUDA usage hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream with which to synchronize Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED`, :py:obj:`~.CUDA_ERROR_UNKNOWN` See Also -------- :py:obj:`~.cuGraphicsResourceGetMappedPointer`, :py:obj:`~.cuGraphicsSubResourceGetMappedArray`, :py:obj:`~.cuGraphicsUnmapResources`, :py:obj:`~.cudaGraphicsMapResources` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUgraphicsResource *cresources if resources is None: cresources = NULL elif isinstance(resources, (CUgraphicsResource,)): presources = resources.getPtr() cresources = presources elif isinstance(resources, (int)): cresources = resources else: raise TypeError("Argument 'resources' is not instance of type (expected , found " + str(type(resources))) err = ccuda.cuGraphicsMapResources(count, cresources, chStream) return (CUresult(err),) {{endif}} {{if 'cuGraphicsUnmapResources' in found_functions}} @cython.embedsignature(True) def cuGraphicsUnmapResources(unsigned int count, resources, hStream): """ Unmap graphics resources. Unmaps the `count` graphics resources in `resources`. Once unmapped, the resources in `resources` may not be accessed by CUDA until they are mapped again. This function provides the synchronization guarantee that any CUDA work issued in `stream` before :py:obj:`~.cuGraphicsUnmapResources()` will complete before any subsequently issued graphics work begins. If `resources` includes any duplicate entries then :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` is returned. If any of `resources` are not presently mapped for access by CUDA then :py:obj:`~.CUDA_ERROR_NOT_MAPPED` is returned. Parameters ---------- count : unsigned int Number of resources to unmap resources : :py:obj:`~.CUgraphicsResource` Resources to unmap hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Stream with which to synchronize Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_NOT_MAPPED`, :py:obj:`~.CUDA_ERROR_UNKNOWN` See Also -------- :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cudaGraphicsUnmapResources` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef ccuda.CUgraphicsResource *cresources if resources is None: cresources = NULL elif isinstance(resources, (CUgraphicsResource,)): presources = resources.getPtr() cresources = presources elif isinstance(resources, (int)): cresources = resources else: raise TypeError("Argument 'resources' is not instance of type (expected , found " + str(type(resources))) err = ccuda.cuGraphicsUnmapResources(count, cresources, chStream) return (CUresult(err),) {{endif}} {{if 'cuGetProcAddress_v2' in found_functions}} @cython.embedsignature(True) def cuGetProcAddress(char* symbol, int cudaVersion, flags): """ Returns the requested driver API function pointer. Returns in `**pfn` the address of the CUDA driver function for the requested CUDA version and flags. The CUDA version is specified as (1000 * major + 10 * minor), so CUDA 11.2 should be specified as 11020. For a requested driver symbol, if the specified CUDA version is greater than or equal to the CUDA version in which the driver symbol was introduced, this API will return the function pointer to the corresponding versioned function. The pointer returned by the API should be cast to a function pointer matching the requested driver function's definition in the API header file. The function pointer typedef can be picked up from the corresponding typedefs header file. For example, cudaTypedefs.h consists of function pointer typedefs for driver APIs defined in :py:obj:`~.cuda.h`. The API will return :py:obj:`~.CUDA_SUCCESS` and set the returned `pfn` to NULL if the requested driver function is not supported on the platform, no ABI compatible driver function exists for the specified `cudaVersion` or if the driver symbol is invalid. It will also set the optional `symbolStatus` to one of the values in :py:obj:`~.CUdriverProcAddressQueryResult` with the following meanings: - :py:obj:`~.CU_GET_PROC_ADDRESS_SUCCESS` - The requested symbol was succesfully found based on input arguments and `pfn` is valid - :py:obj:`~.CU_GET_PROC_ADDRESS_SYMBOL_NOT_FOUND` - The requested symbol was not found - :py:obj:`~.CU_GET_PROC_ADDRESS_VERSION_NOT_SUFFICIENT` - The requested symbol was found but is not supported by cudaVersion specified The requested flags can be: - :py:obj:`~.CU_GET_PROC_ADDRESS_DEFAULT`: This is the default mode. This is equivalent to :py:obj:`~.CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM` if the code is compiled with --default-stream per-thread compilation flag or the macro CUDA_API_PER_THREAD_DEFAULT_STREAM is defined; :py:obj:`~.CU_GET_PROC_ADDRESS_LEGACY_STREAM` otherwise. - :py:obj:`~.CU_GET_PROC_ADDRESS_LEGACY_STREAM`: This will enable the search for all driver symbols that match the requested driver symbol name except the corresponding per-thread versions. - :py:obj:`~.CU_GET_PROC_ADDRESS_PER_THREAD_DEFAULT_STREAM`: This will enable the search for all driver symbols that match the requested driver symbol name including the per-thread versions. If a per-thread version is not found, the API will return the legacy version of the driver function. Parameters ---------- symbol : bytes The base name of the driver API function to look for. As an example, for the driver API :py:obj:`~.cuMemAlloc_v2`, `symbol` would be cuMemAlloc and `cudaVersion` would be the ABI compatible CUDA version for the _v2 variant. cudaVersion : int The CUDA version to look for the requested driver symbol flags : Any Flags to specify search options. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` pfn : Any Location to return the function pointer to the requested driver function symbolStatus : :py:obj:`~.CUdriverProcAddressQueryResult` Optional location to store the status of the search for `symbol` based on `cudaVersion`. See :py:obj:`~.CUdriverProcAddressQueryResult` for possible values. See Also -------- :py:obj:`~.cudaGetDriverEntryPoint` """ cdef ccuda.cuuint64_t cflags if flags is None: cflags = 0 elif isinstance(flags, (cuuint64_t,)): pflags = int(flags) cflags = pflags else: pflags = int(cuuint64_t(flags)) cflags = pflags cdef void_ptr pfn = 0 cdef ccuda.CUdriverProcAddressQueryResult symbolStatus err = ccuda.cuGetProcAddress(symbol, &pfn, cudaVersion, cflags, &symbolStatus) return (CUresult(err), pfn, CUdriverProcAddressQueryResult(symbolStatus)) {{endif}} {{if 'cuCoredumpGetAttribute' in found_functions}} @cython.embedsignature(True) def cuCoredumpGetAttribute(attrib not None : CUcoredumpSettings): """ Allows caller to fetch a coredump attribute value for the current context. Returns in `*value` the requested value specified by `attrib`. It is up to the caller to ensure that the data type and size of `*value` matches the request. If the caller calls this function with `*value` equal to NULL, the size of the memory region (in bytes) expected for `attrib` will be placed in `size`. The supported attributes are: - :py:obj:`~.CU_COREDUMP_ENABLE_ON_EXCEPTION`: Bool where :py:obj:`~.true` means that GPU exceptions from this context will create a coredump at the location specified by :py:obj:`~.CU_COREDUMP_FILE`. The default value is :py:obj:`~.false` unless set to :py:obj:`~.true` globally or locally, or the CU_CTX_USER_COREDUMP_ENABLE flag was set during context creation. - :py:obj:`~.CU_COREDUMP_TRIGGER_HOST`: Bool where :py:obj:`~.true` means that the host CPU will also create a coredump. The default value is :py:obj:`~.true` unless set to :py:obj:`~.false` globally or or locally. This value is deprecated as of CUDA 12.5 - raise the :py:obj:`~.CU_COREDUMP_SKIP_ABORT` flag to disable host device abort() if needed. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT`: Bool where :py:obj:`~.true` means that any resulting coredumps will not have a dump of GPU memory or non-reloc ELF images. The default value is :py:obj:`~.false` unless set to :py:obj:`~.true` globally or locally. This attribute is deprecated as of CUDA 12.5, please use :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS` instead. - :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER`: Bool where :py:obj:`~.true` means that a coredump can be created by writing to the system pipe specified by :py:obj:`~.CU_COREDUMP_PIPE`. The default value is :py:obj:`~.false` unless set to :py:obj:`~.true` globally or locally. - :py:obj:`~.CU_COREDUMP_FILE`: String of up to 1023 characters that defines the location where any coredumps generated by this context will be written. The default value is :py:obj:`~.core`.cuda.HOSTNAME.PID where :py:obj:`~.HOSTNAME` is the host name of the machine running the CUDA applications and :py:obj:`~.PID` is the process ID of the CUDA application. - :py:obj:`~.CU_COREDUMP_PIPE`: String of up to 1023 characters that defines the name of the pipe that will be monitored if user-triggered coredumps are enabled. The default value is :py:obj:`~.corepipe`.cuda.HOSTNAME.PID where :py:obj:`~.HOSTNAME` is the host name of the machine running the CUDA application and :py:obj:`~.PID` is the process ID of the CUDA application. - :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS`: An integer with values to allow granular control the data contained in a coredump specified as a bitwise OR combination of the following values: - :py:obj:`~.CU_COREDUMP_DEFAULT_FLAGS` - if set by itself, coredump generation returns to its default settings of including all memory regions that it is able to access - :py:obj:`~.CU_COREDUMP_SKIP_NONRELOCATED_ELF_IMAGES` - Coredump will not include the data from CUDA source modules that are not relocated at runtime. - :py:obj:`~.CU_COREDUMP_SKIP_GLOBAL_MEMORY` - Coredump will not include device-side global data that does not belong to any context. - :py:obj:`~.CU_COREDUMP_SKIP_SHARED_MEMORY` - Coredump will not include grid-scale shared memory for the warp that the dumped kernel belonged to. - :py:obj:`~.CU_COREDUMP_SKIP_LOCAL_MEMORY` - Coredump will not include local memory from the kernel. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT_FLAGS` - Enables all of the above options. Equiavlent to setting the :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT` attribute to :py:obj:`~.true`. - :py:obj:`~.CU_COREDUMP_SKIP_ABORT` - If set, GPU exceptions will not raise an abort() in the host CPU process. Same functional goal as :py:obj:`~.CU_COREDUMP_TRIGGER_HOST` but better reflects the default behavior. Parameters ---------- attrib : :py:obj:`~.CUcoredumpSettings` The enum defining which value to fetch. size : int The size of the memory region `value` points to. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` value : Any void* containing the requested data. size : int The size of the memory region `value` points to. See Also -------- :py:obj:`~.cuCoredumpGetAttributeGlobal`, :py:obj:`~.cuCoredumpSetAttribute`, :py:obj:`~.cuCoredumpSetAttributeGlobal` """ cdef ccuda.CUcoredumpSettings cattrib = attrib.value cdef utils.HelperCUcoredumpSettings cvalue = utils.HelperCUcoredumpSettings(attrib, 0, is_getter=True) cdef void* cvalue_ptr = cvalue.cptr cdef size_t size = cvalue.size() err = ccuda.cuCoredumpGetAttribute(cattrib, cvalue_ptr, &size) return (CUresult(err), cvalue.pyObj()) {{endif}} {{if 'cuCoredumpGetAttributeGlobal' in found_functions}} @cython.embedsignature(True) def cuCoredumpGetAttributeGlobal(attrib not None : CUcoredumpSettings): """ Allows caller to fetch a coredump attribute value for the entire application. Returns in `*value` the requested value specified by `attrib`. It is up to the caller to ensure that the data type and size of `*value` matches the request. If the caller calls this function with `*value` equal to NULL, the size of the memory region (in bytes) expected for `attrib` will be placed in `size`. The supported attributes are: - :py:obj:`~.CU_COREDUMP_ENABLE_ON_EXCEPTION`: Bool where :py:obj:`~.true` means that GPU exceptions from this context will create a coredump at the location specified by :py:obj:`~.CU_COREDUMP_FILE`. The default value is :py:obj:`~.false`. - :py:obj:`~.CU_COREDUMP_TRIGGER_HOST`: Bool where :py:obj:`~.true` means that the host CPU will also create a coredump. The default value is :py:obj:`~.true` unless set to :py:obj:`~.false` globally or or locally. This value is deprecated as of CUDA 12.5 - raise the :py:obj:`~.CU_COREDUMP_SKIP_ABORT` flag to disable host device abort() if needed. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT`: Bool where :py:obj:`~.true` means that any resulting coredumps will not have a dump of GPU memory or non-reloc ELF images. The default value is :py:obj:`~.false`. This attribute is deprecated as of CUDA 12.5, please use :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS` instead. - :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER`: Bool where :py:obj:`~.true` means that a coredump can be created by writing to the system pipe specified by :py:obj:`~.CU_COREDUMP_PIPE`. The default value is :py:obj:`~.false`. - :py:obj:`~.CU_COREDUMP_FILE`: String of up to 1023 characters that defines the location where any coredumps generated by this context will be written. The default value is :py:obj:`~.core`.cuda.HOSTNAME.PID where :py:obj:`~.HOSTNAME` is the host name of the machine running the CUDA applications and :py:obj:`~.PID` is the process ID of the CUDA application. - :py:obj:`~.CU_COREDUMP_PIPE`: String of up to 1023 characters that defines the name of the pipe that will be monitored if user-triggered coredumps are enabled. The default value is :py:obj:`~.corepipe`.cuda.HOSTNAME.PID where :py:obj:`~.HOSTNAME` is the host name of the machine running the CUDA application and :py:obj:`~.PID` is the process ID of the CUDA application. - :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS`: An integer with values to allow granular control the data contained in a coredump specified as a bitwise OR combination of the following values: - :py:obj:`~.CU_COREDUMP_DEFAULT_FLAGS` - if set by itself, coredump generation returns to its default settings of including all memory regions that it is able to access - :py:obj:`~.CU_COREDUMP_SKIP_NONRELOCATED_ELF_IMAGES` - Coredump will not include the data from CUDA source modules that are not relocated at runtime. - :py:obj:`~.CU_COREDUMP_SKIP_GLOBAL_MEMORY` - Coredump will not include device-side global data that does not belong to any context. - :py:obj:`~.CU_COREDUMP_SKIP_SHARED_MEMORY` - Coredump will not include grid-scale shared memory for the warp that the dumped kernel belonged to. - :py:obj:`~.CU_COREDUMP_SKIP_LOCAL_MEMORY` - Coredump will not include local memory from the kernel. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT_FLAGS` - Enables all of the above options. Equiavlent to setting the :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT` attribute to :py:obj:`~.true`. - :py:obj:`~.CU_COREDUMP_SKIP_ABORT` - If set, GPU exceptions will not raise an abort() in the host CPU process. Same functional goal as :py:obj:`~.CU_COREDUMP_TRIGGER_HOST` but better reflects the default behavior. Parameters ---------- attrib : :py:obj:`~.CUcoredumpSettings` The enum defining which value to fetch. size : int The size of the memory region `value` points to. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` value : Any void* containing the requested data. size : int The size of the memory region `value` points to. See Also -------- :py:obj:`~.cuCoredumpGetAttribute`, :py:obj:`~.cuCoredumpSetAttribute`, :py:obj:`~.cuCoredumpSetAttributeGlobal` """ cdef ccuda.CUcoredumpSettings cattrib = attrib.value cdef utils.HelperCUcoredumpSettings cvalue = utils.HelperCUcoredumpSettings(attrib, 0, is_getter=True) cdef void* cvalue_ptr = cvalue.cptr cdef size_t size = cvalue.size() err = ccuda.cuCoredumpGetAttributeGlobal(cattrib, cvalue_ptr, &size) return (CUresult(err), cvalue.pyObj()) {{endif}} {{if 'cuCoredumpSetAttribute' in found_functions}} @cython.embedsignature(True) def cuCoredumpSetAttribute(attrib not None : CUcoredumpSettings, value): """ Allows caller to set a coredump attribute value for the current context. This function should be considered an alternate interface to the CUDA- GDB environment variables defined in this document: https://docs.nvidia.com/cuda/cuda-gdb/index.html#gpu-coredump An important design decision to note is that any coredump environment variable values set before CUDA initializes will take permanent precedence over any values set with this function. This decision was made to ensure no change in behavior for any users that may be currently using these variables to get coredumps. `*value` shall contain the requested value specified by `set`. It is up to the caller to ensure that the data type and size of `*value` matches the request. If the caller calls this function with `*value` equal to NULL, the size of the memory region (in bytes) expected for `set` will be placed in `size`. /note This function will return :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` if the caller attempts to set :py:obj:`~.CU_COREDUMP_ENABLE_ON_EXCEPTION` on a GPU of with Compute Capability < 6.0. :py:obj:`~.cuCoredumpSetAttributeGlobal` works on those platforms as an alternative. /note :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER` and :py:obj:`~.CU_COREDUMP_PIPE` cannot be set on a per-context basis. The supported attributes are: - :py:obj:`~.CU_COREDUMP_ENABLE_ON_EXCEPTION`: Bool where :py:obj:`~.true` means that GPU exceptions from this context will create a coredump at the location specified by :py:obj:`~.CU_COREDUMP_FILE`. The default value is :py:obj:`~.false`. - :py:obj:`~.CU_COREDUMP_TRIGGER_HOST`: Bool where :py:obj:`~.true` means that the host CPU will also create a coredump. The default value is :py:obj:`~.true` unless set to :py:obj:`~.false` globally or or locally. This value is deprecated as of CUDA 12.5 - raise the :py:obj:`~.CU_COREDUMP_SKIP_ABORT` flag to disable host device abort() if needed. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT`: Bool where :py:obj:`~.true` means that any resulting coredumps will not have a dump of GPU memory or non-reloc ELF images. The default value is :py:obj:`~.false`. This attribute is deprecated as of CUDA 12.5, please use :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS` instead. - :py:obj:`~.CU_COREDUMP_FILE`: String of up to 1023 characters that defines the location where any coredumps generated by this context will be written. The default value is :py:obj:`~.core`.cuda.HOSTNAME.PID where :py:obj:`~.HOSTNAME` is the host name of the machine running the CUDA applications and :py:obj:`~.PID` is the process ID of the CUDA application. - :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS`: An integer with values to allow granular control the data contained in a coredump specified as a bitwise OR combination of the following values: - :py:obj:`~.CU_COREDUMP_DEFAULT_FLAGS` - if set by itself, coredump generation returns to its default settings of including all memory regions that it is able to access - :py:obj:`~.CU_COREDUMP_SKIP_NONRELOCATED_ELF_IMAGES` - Coredump will not include the data from CUDA source modules that are not relocated at runtime. - :py:obj:`~.CU_COREDUMP_SKIP_GLOBAL_MEMORY` - Coredump will not include device-side global data that does not belong to any context. - :py:obj:`~.CU_COREDUMP_SKIP_SHARED_MEMORY` - Coredump will not include grid-scale shared memory for the warp that the dumped kernel belonged to. - :py:obj:`~.CU_COREDUMP_SKIP_LOCAL_MEMORY` - Coredump will not include local memory from the kernel. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT_FLAGS` - Enables all of the above options. Equiavlent to setting the :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT` attribute to :py:obj:`~.true`. - :py:obj:`~.CU_COREDUMP_SKIP_ABORT` - If set, GPU exceptions will not raise an abort() in the host CPU process. Same functional goal as :py:obj:`~.CU_COREDUMP_TRIGGER_HOST` but better reflects the default behavior. Parameters ---------- attrib : :py:obj:`~.CUcoredumpSettings` The enum defining which value to set. value : Any void* containing the requested data. size : int The size of the memory region `value` points to. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED` size : int The size of the memory region `value` points to. See Also -------- :py:obj:`~.cuCoredumpGetAttributeGlobal`, :py:obj:`~.cuCoredumpGetAttribute`, :py:obj:`~.cuCoredumpSetAttributeGlobal` """ cdef ccuda.CUcoredumpSettings cattrib = attrib.value cdef utils.HelperCUcoredumpSettings cvalue = utils.HelperCUcoredumpSettings(attrib, value, is_getter=False) cdef void* cvalue_ptr = cvalue.cptr cdef size_t size = cvalue.size() err = ccuda.cuCoredumpSetAttribute(cattrib, cvalue_ptr, &size) return (CUresult(err),) {{endif}} {{if 'cuCoredumpSetAttributeGlobal' in found_functions}} @cython.embedsignature(True) def cuCoredumpSetAttributeGlobal(attrib not None : CUcoredumpSettings, value): """ Allows caller to set a coredump attribute value globally. This function should be considered an alternate interface to the CUDA- GDB environment variables defined in this document: https://docs.nvidia.com/cuda/cuda-gdb/index.html#gpu-coredump An important design decision to note is that any coredump environment variable values set before CUDA initializes will take permanent precedence over any values set with this function. This decision was made to ensure no change in behavior for any users that may be currently using these variables to get coredumps. `*value` shall contain the requested value specified by `set`. It is up to the caller to ensure that the data type and size of `*value` matches the request. If the caller calls this function with `*value` equal to NULL, the size of the memory region (in bytes) expected for `set` will be placed in `size`. The supported attributes are: - :py:obj:`~.CU_COREDUMP_ENABLE_ON_EXCEPTION`: Bool where :py:obj:`~.true` means that GPU exceptions from this context will create a coredump at the location specified by :py:obj:`~.CU_COREDUMP_FILE`. The default value is :py:obj:`~.false`. - :py:obj:`~.CU_COREDUMP_TRIGGER_HOST`: Bool where :py:obj:`~.true` means that the host CPU will also create a coredump. The default value is :py:obj:`~.true` unless set to :py:obj:`~.false` globally or or locally. This value is deprecated as of CUDA 12.5 - raise the :py:obj:`~.CU_COREDUMP_SKIP_ABORT` flag to disable host device abort() if needed. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT`: Bool where :py:obj:`~.true` means that any resulting coredumps will not have a dump of GPU memory or non-reloc ELF images. The default value is :py:obj:`~.false`. This attribute is deprecated as of CUDA 12.5, please use :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS` instead. - :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER`: Bool where :py:obj:`~.true` means that a coredump can be created by writing to the system pipe specified by :py:obj:`~.CU_COREDUMP_PIPE`. The default value is :py:obj:`~.false`. - :py:obj:`~.CU_COREDUMP_FILE`: String of up to 1023 characters that defines the location where any coredumps generated by this context will be written. The default value is :py:obj:`~.core`.cuda.HOSTNAME.PID where :py:obj:`~.HOSTNAME` is the host name of the machine running the CUDA applications and :py:obj:`~.PID` is the process ID of the CUDA application. - :py:obj:`~.CU_COREDUMP_PIPE`: String of up to 1023 characters that defines the name of the pipe that will be monitored if user-triggered coredumps are enabled. This value may not be changed after :py:obj:`~.CU_COREDUMP_ENABLE_USER_TRIGGER` is set to :py:obj:`~.true`. The default value is :py:obj:`~.corepipe`.cuda.HOSTNAME.PID where :py:obj:`~.HOSTNAME` is the host name of the machine running the CUDA application and :py:obj:`~.PID` is the process ID of the CUDA application. - :py:obj:`~.CU_COREDUMP_GENERATION_FLAGS`: An integer with values to allow granular control the data contained in a coredump specified as a bitwise OR combination of the following values: - :py:obj:`~.CU_COREDUMP_DEFAULT_FLAGS` - if set by itself, coredump generation returns to its default settings of including all memory regions that it is able to access - :py:obj:`~.CU_COREDUMP_SKIP_NONRELOCATED_ELF_IMAGES` - Coredump will not include the data from CUDA source modules that are not relocated at runtime. - :py:obj:`~.CU_COREDUMP_SKIP_GLOBAL_MEMORY` - Coredump will not include device-side global data that does not belong to any context. - :py:obj:`~.CU_COREDUMP_SKIP_SHARED_MEMORY` - Coredump will not include grid-scale shared memory for the warp that the dumped kernel belonged to. - :py:obj:`~.CU_COREDUMP_SKIP_LOCAL_MEMORY` - Coredump will not include local memory from the kernel. - :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT_FLAGS` - Enables all of the above options. Equiavlent to setting the :py:obj:`~.CU_COREDUMP_LIGHTWEIGHT` attribute to :py:obj:`~.true`. - :py:obj:`~.CU_COREDUMP_SKIP_ABORT` - If set, GPU exceptions will not raise an abort() in the host CPU process. Same functional goal as :py:obj:`~.CU_COREDUMP_TRIGGER_HOST` but better reflects the default behavior. Parameters ---------- attrib : :py:obj:`~.CUcoredumpSettings` The enum defining which value to set. value : Any void* containing the requested data. size : int The size of the memory region `value` points to. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_PERMITTED` size : int The size of the memory region `value` points to. See Also -------- :py:obj:`~.cuCoredumpGetAttribute`, :py:obj:`~.cuCoredumpGetAttributeGlobal`, :py:obj:`~.cuCoredumpSetAttribute` """ cdef ccuda.CUcoredumpSettings cattrib = attrib.value cdef utils.HelperCUcoredumpSettings cvalue = utils.HelperCUcoredumpSettings(attrib, value, is_getter=False) cdef void* cvalue_ptr = cvalue.cptr cdef size_t size = cvalue.size() err = ccuda.cuCoredumpSetAttributeGlobal(cattrib, cvalue_ptr, &size) return (CUresult(err),) {{endif}} {{if 'cuGetExportTable' in found_functions}} @cython.embedsignature(True) def cuGetExportTable(pExportTableId : Optional[CUuuid]): """ Parameters ---------- pExportTableId : :py:obj:`~.CUuuid` None Returns ------- CUresult ppExportTable : Any None """ cdef void_ptr ppExportTable = 0 cdef ccuda.CUuuid* cpExportTableId_ptr = pExportTableId._ptr if pExportTableId != None else NULL err = ccuda.cuGetExportTable(&ppExportTable, cpExportTableId_ptr) return (CUresult(err), ppExportTable) {{endif}} {{if 'cuGreenCtxCreate' in found_functions}} @cython.embedsignature(True) def cuGreenCtxCreate(desc, dev, unsigned int flags): """ Creates a green context with a specified set of resources. This API creates a green context with the resources specified in the descriptor `desc` and returns it in the handle represented by `phCtx`. This API will retain the primary context on device `dev`, which will is released when the green context is destroyed. It is advised to have the primary context active before calling this API to avoid the heavy cost of triggering primary context initialization and deinitialization multiple times. The API does not set the green context current. In order to set it current, you need to explicitly set it current by first converting the green context to a CUcontext using :py:obj:`~.cuCtxFromGreenCtx` and subsequently calling :py:obj:`~.cuCtxSetCurrent` / :py:obj:`~.cuCtxPushCurrent`. It should be noted that a green context can be current to only one thread at a time. There is no internal synchronization to make API calls accessing the same green context from multiple threads work. Note: The API is not supported on 32-bit platforms. The supported flags are: - `CU_GREEN_CTX_DEFAULT_STREAM` : Creates a default stream to use inside the green context. Required. Parameters ---------- desc : :py:obj:`~.CUdevResourceDesc` Descriptor generated via :py:obj:`~.cuDevResourceGenerateDesc` which contains the set of resources to be used dev : :py:obj:`~.CUdevice` Device on which to create the green context. flags : unsigned int One of the supported green context creation flags. `CU_GREEN_CTX_DEFAULT_STREAM` is required. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_NOT_SUPPORTED`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phCtx : :py:obj:`~.CUgreenCtx` Pointer for the output handle to the green context See Also -------- :py:obj:`~.cuGreenCtxDestroy`, :py:obj:`~.cuCtxFromGreenCtx`, :py:obj:`~.cuCtxSetCurrent`, :py:obj:`~.cuCtxPushCurrent`, :py:obj:`~.cuDevResourceGenerateDesc`, :py:obj:`~.cuDevicePrimaryCtxRetain`, :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuCtxCreate_v3` """ cdef ccuda.CUdevice cdev if dev is None: cdev = 0 elif isinstance(dev, (CUdevice,)): pdev = int(dev) cdev = pdev else: pdev = int(CUdevice(dev)) cdev = pdev cdef ccuda.CUdevResourceDesc cdesc if desc is None: cdesc = 0 elif isinstance(desc, (CUdevResourceDesc,)): pdesc = int(desc) cdesc = pdesc else: pdesc = int(CUdevResourceDesc(desc)) cdesc = pdesc cdef CUgreenCtx phCtx = CUgreenCtx() err = ccuda.cuGreenCtxCreate(phCtx._ptr, cdesc, cdev, flags) return (CUresult(err), phCtx) {{endif}} {{if 'cuGreenCtxDestroy' in found_functions}} @cython.embedsignature(True) def cuGreenCtxDestroy(hCtx): """ Destroys a green context. Destroys the green context, releasing the primary context of the device that this green context was created for. Any resources provisioned for this green context (that were initially available via the resource descriptor) are released as well. Parameters ---------- hCtx : :py:obj:`~.CUgreenCtx` Green context to be destroyed Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_CONTEXT_IS_DESTROYED` See Also -------- :py:obj:`~.cuGreenCtxCreate`, :py:obj:`~.cuCtxDestroy` """ cdef ccuda.CUgreenCtx chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUgreenCtx,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUgreenCtx(hCtx)) chCtx = phCtx err = ccuda.cuGreenCtxDestroy(chCtx) return (CUresult(err),) {{endif}} {{if 'cuCtxFromGreenCtx' in found_functions}} @cython.embedsignature(True) def cuCtxFromGreenCtx(hCtx): """ Converts a green context into the primary context. The API converts a green context into the primary context returned in `pContext`. It is important to note that the converted context `pContext` is a normal primary context but with the resources of the specified green context `hCtx`. Once converted, it can then be used to set the context current with :py:obj:`~.cuCtxSetCurrent` or with any of the CUDA APIs that accept a CUcontext parameter. Users are expected to call this API before calling any CUDA APIs that accept a CUcontext. Failing to do so will result in the APIs returning :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`. Parameters ---------- hCtx : :py:obj:`~.CUgreenCtx` Green context to convert Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pContext : :py:obj:`~.CUcontext` Returned primary context with green context resources See Also -------- :py:obj:`~.cuGreenCtxCreate` """ cdef ccuda.CUgreenCtx chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUgreenCtx,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUgreenCtx(hCtx)) chCtx = phCtx cdef CUcontext pContext = CUcontext() err = ccuda.cuCtxFromGreenCtx(pContext._ptr, chCtx) return (CUresult(err), pContext) {{endif}} {{if 'cuDeviceGetDevResource' in found_functions}} @cython.embedsignature(True) def cuDeviceGetDevResource(device, typename not None : CUdevResourceType): """ Get device resources. Get the `typename` resources available to the `device`. This may often be the starting point for further partitioning or configuring of resources. Note: The API is not supported on 32-bit platforms. Parameters ---------- device : :py:obj:`~.CUdevice` Device to get resource for typename : :py:obj:`~.CUdevResourceType` Type of resource to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_RESOURCE_TYPE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE` resource : :py:obj:`~.CUdevResource` Output pointer to a :py:obj:`~.CUdevResource` structure See Also -------- :py:obj:`~.cuDevResourceGenerateDesc` """ cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cdef CUdevResource resource = CUdevResource() cdef ccuda.CUdevResourceType ctypename = typename.value err = ccuda.cuDeviceGetDevResource(cdevice, resource._ptr, ctypename) return (CUresult(err), resource) {{endif}} {{if 'cuCtxGetDevResource' in found_functions}} @cython.embedsignature(True) def cuCtxGetDevResource(hCtx, typename not None : CUdevResourceType): """ Get context resources. Get the `typename` resources available to the context represented by `hCtx` Note: The API is not supported on 32-bit platforms. Parameters ---------- hCtx : :py:obj:`~.CUcontext` Context to get resource for typename : :py:obj:`~.CUdevResourceType` Type of resource to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_RESOURCE_TYPE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` resource : :py:obj:`~.CUdevResource` Output pointer to a :py:obj:`~.CUdevResource` structure See Also -------- :py:obj:`~.cuDevResourceGenerateDesc` """ cdef ccuda.CUcontext chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUcontext,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUcontext(hCtx)) chCtx = phCtx cdef CUdevResource resource = CUdevResource() cdef ccuda.CUdevResourceType ctypename = typename.value err = ccuda.cuCtxGetDevResource(chCtx, resource._ptr, ctypename) return (CUresult(err), resource) {{endif}} {{if 'cuGreenCtxGetDevResource' in found_functions}} @cython.embedsignature(True) def cuGreenCtxGetDevResource(hCtx, typename not None : CUdevResourceType): """ Get green context resources. Get the `typename` resources available to the green context represented by `hCtx` Parameters ---------- hCtx : :py:obj:`~.CUgreenCtx` Green context to get resource for typename : :py:obj:`~.CUdevResourceType` Type of resource to retrieve Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_RESOURCE_TYPE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` resource : :py:obj:`~.CUdevResource` Output pointer to a :py:obj:`~.CUdevResource` structure See Also -------- :py:obj:`~.cuDevResourceGenerateDesc` """ cdef ccuda.CUgreenCtx chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUgreenCtx,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUgreenCtx(hCtx)) chCtx = phCtx cdef CUdevResource resource = CUdevResource() cdef ccuda.CUdevResourceType ctypename = typename.value err = ccuda.cuGreenCtxGetDevResource(chCtx, resource._ptr, ctypename) return (CUresult(err), resource) {{endif}} {{if 'cuDevSmResourceSplitByCount' in found_functions}} @cython.embedsignature(True) def cuDevSmResourceSplitByCount(unsigned int nbGroups, input_ : Optional[CUdevResource], unsigned int useFlags, unsigned int minCount): """ Splits `CU_DEV_RESOURCE_TYPE_SM` resources. Splits `CU_DEV_RESOURCE_TYPE_SM` resources into `nbGroups`, adhering to the minimum SM count specified in `minCount` and the usage flags in `useFlags`. If `result` is NULL, the API simulates a split and provides the amount of groups that would be created in `nbGroups`. Otherwise, `nbGroups` must point to the amount of elements in `result` and on return, the API will overwrite `nbGroups` with the amount actually created. The groups are written to the array in `result`. `nbGroups` can be less than the total amount if a smaller number of groups is needed. This API is used to spatially partition the input resource. The input resource needs to come from one of :py:obj:`~.cuDeviceGetDevResource`, :py:obj:`~.cuCtxGetDevResource`, or :py:obj:`~.cuGreenCtxGetDevResource`. A limitation of the API is that the output results cannot be split again without first creating a descriptor and a green context with that descriptor. When creating the groups, the API will take into account the performance and functional characteristics of the input resource, and guarantee a split that will create a disjoint set of symmetrical partitions. This may lead to fewer groups created than purely dividing the total SM count by the `minCount` due to cluster requirements or alignment and granularity requirements for the minCount. The `remainder` set does not have the same functional or performance guarantees as the groups in `result`. Its use should be carefully planned and future partitions of the `remainder` set are discouraged. The following flags are supported: - `CU_DEV_SM_RESOURCE_SPLIT_IGNORE_SM_COSCHEDULING` : Lower the minimum SM count and alignment, and treat each SM independent of its hierarchy. This allows more fine grained partitions but at the cost of advanced features (such as large clusters on compute capability 9.0+). - `CU_DEV_SM_RESOURCE_SPLIT_MAX_POTENTIAL_CLUSTER_SIZE` : Compute Capability 9.0+ only. Attempt to create groups that may allow for maximally sized thread clusters. This can be queried post green context creation using :py:obj:`~.cuOccupancyMaxPotentialClusterSize`. A successful API call must either have: - A valid array of `result` pointers of size passed in `nbGroups`, with `input` of type `CU_DEV_RESOURCE_TYPE_SM`. Value of `minCount` must be between 0 and the SM count specified in `input`. `remaining` may be NULL. - NULL passed in for `result`, with a valid integer pointer in `nbGroups` and `input` of type `CU_DEV_RESOURCE_TYPE_SM`. Value of `minCount` must be between 0 and the SM count specified in `input`. `remaining` may be NULL. This queries the number of groups that would be created by the API. Note: The API is not supported on 32-bit platforms. Parameters ---------- nbGroups : unsigned int This is a pointer, specifying the number of groups that would be or should be created as described below. input : :py:obj:`~.CUdevResource` Input SM resource to be split. Must be a valid `CU_DEV_RESOURCE_TYPE_SM` resource. useFlags : unsigned int Flags specifying how these partitions are used or which constraints to abide by when splitting the input. Zero is valid for default behavior. minCount : unsigned int Minimum number of SMs required Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_DEVICE`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_RESOURCE_TYPE`, :py:obj:`~.CUDA_ERROR_INVALID_RESOURCE_CONFIGURATION` result : List[:py:obj:`~.CUdevResource`] Output array of `None` resources. Can be NULL to query the number of groups. nbGroups : unsigned int This is a pointer, specifying the number of groups that would be or should be created as described below. remaining : :py:obj:`~.CUdevResource` If the input resource cannot be cleanly split among `nbGroups`, the remaining is placed in here. Can be ommitted (NULL) if the user does not need the remaining set. See Also -------- :py:obj:`~.cuGreenCtxGetDevResource`, :py:obj:`~.cuCtxGetDevResource`, :py:obj:`~.cuDeviceGetDevResource` """ cdef ccuda.CUdevResource* cresult = NULL pyresult = [CUdevResource() for idx in range(nbGroups)] if nbGroups != 0: cresult = calloc(nbGroups, sizeof(ccuda.CUdevResource)) if cresult is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(nbGroups) + 'x' + str(sizeof(ccuda.CUdevResource))) cdef unsigned int cnbGroups = nbGroups cdef ccuda.CUdevResource* cinput__ptr = input_._ptr if input_ != None else NULL cdef CUdevResource remaining = CUdevResource() err = ccuda.cuDevSmResourceSplitByCount(cresult, &cnbGroups, cinput__ptr, remaining._ptr, useFlags, minCount) if CUresult(err) == CUresult(0): for idx in range(nbGroups): string.memcpy((pyresult[idx])._ptr, &cresult[idx], sizeof(ccuda.CUdevResource)) if cresult is not NULL: free(cresult) return (CUresult(err), pyresult, cnbGroups, remaining) {{endif}} {{if 'cuDevResourceGenerateDesc' in found_functions}} @cython.embedsignature(True) def cuDevResourceGenerateDesc(resources : Optional[Tuple[CUdevResource] | List[CUdevResource]], unsigned int nbResources): """ Generate a resource descriptor. Generates a single resource descriptor with the set of resources specified in `resources`. The generated resource descriptor is necessary for the creation of green contexts via the :py:obj:`~.cuGreenCtxCreate` API. Resources of the same type can be passed in, provided they meet the requirements as noted below. A successful API call must have: - A valid output pointer for the `phDesc` descriptor as well as a valid array of `resources` pointers, with the array size passed in `nbResources`. If multiple resources are provided in `resources`, the device they came from must be the same, otherwise CUDA_ERROR_INVALID_RESOURCE_CONFIGURATION is returned. If multiple resources are provided in `resources` and they are of type :py:obj:`~.CU_DEV_RESOURCE_TYPE_SM`, they must be outputs (whether `result` or `remaining`) from the same split API instance, otherwise CUDA_ERROR_INVALID_RESOURCE_CONFIGURATION is returned. Note: The API is not supported on 32-bit platforms. Parameters ---------- resources : List[:py:obj:`~.CUdevResource`] Array of resources to be included in the descriptor nbResources : unsigned int Number of resources passed in `resources` Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_INVALID_RESOURCE_TYPE`, :py:obj:`~.CUDA_ERROR_INVALID_RESOURCE_CONFIGURATION` phDesc : :py:obj:`~.CUdevResourceDesc` Output descriptor See Also -------- :py:obj:`~.cuDevSmResourceSplitByCount` """ resources = [] if resources is None else resources if not all(isinstance(_x, (CUdevResource,)) for _x in resources): raise TypeError("Argument 'resources' is not instance of type (expected Tuple[ccuda.CUdevResource,] or List[ccuda.CUdevResource,]") cdef CUdevResourceDesc phDesc = CUdevResourceDesc() cdef ccuda.CUdevResource* cresources = NULL if len(resources) > 0: cresources = calloc(len(resources), sizeof(ccuda.CUdevResource)) if cresources is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(len(resources)) + 'x' + str(sizeof(ccuda.CUdevResource))) for idx in range(len(resources)): string.memcpy(&cresources[idx], (resources[idx])._ptr, sizeof(ccuda.CUdevResource)) if nbResources > len(resources): raise RuntimeError("List is too small: " + str(len(resources)) + " < " + str(nbResources)) err = ccuda.cuDevResourceGenerateDesc(phDesc._ptr, (resources[0])._ptr if len(resources) == 1 else cresources, nbResources) if cresources is not NULL: free(cresources) return (CUresult(err), phDesc) {{endif}} {{if 'cuGreenCtxRecordEvent' in found_functions}} @cython.embedsignature(True) def cuGreenCtxRecordEvent(hCtx, hEvent): """ Records an event. Captures in `hEvent` all the activities of the green context of `hCtx` at the time of this call. `hEvent` and `hCtx` must be from the same primary context otherwise :py:obj:`~.CUDA_ERROR_INVALID_HANDLE` is returned. Calls such as :py:obj:`~.cuEventQuery()` or :py:obj:`~.cuGreenCtxWaitEvent()` will then examine or wait for completion of the work that was captured. Uses of `hCtx` after this call do not modify `hEvent`. Parameters ---------- hCtx : :py:obj:`~.CUgreenCtx` Green context to record event for hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to record Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS` :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` See Also -------- :py:obj:`~.cuGreenCtxWaitEvent`, :py:obj:`~.cuEventRecord`, :py:obj:`~.cuCtxRecordEvent`, :py:obj:`~.cuCtxWaitEvent` Notes ----- The API will return :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` if the specified green context `hCtx` has a stream in the capture mode. In such a case, the call will invalidate all the conflicting captures. """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent cdef ccuda.CUgreenCtx chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUgreenCtx,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUgreenCtx(hCtx)) chCtx = phCtx err = ccuda.cuGreenCtxRecordEvent(chCtx, chEvent) return (CUresult(err),) {{endif}} {{if 'cuGreenCtxWaitEvent' in found_functions}} @cython.embedsignature(True) def cuGreenCtxWaitEvent(hCtx, hEvent): """ Make a green context wait on an event. Makes all future work submitted to green context `hCtx` wait for all work captured in `hEvent`. The synchronization will be performed on the device and will not block the calling CPU thread. See :py:obj:`~.cuGreenCtxRecordEvent()` or :py:obj:`~.cuEventRecord()`, for details on what is captured by an event. Parameters ---------- hCtx : :py:obj:`~.CUgreenCtx` Green context to wait hEvent : :py:obj:`~.CUevent` or :py:obj:`~.cudaEvent_t` Event to wait on Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` See Also -------- :py:obj:`~.cuGreenCtxRecordEvent`, :py:obj:`~.cuStreamWaitEvent` :py:obj:`~.cuCtxRecordEvent`, :py:obj:`~.cuCtxWaitEvent` Notes ----- `hEvent` may be from a different context or device than `hCtx`. The API will return :py:obj:`~.CUDA_ERROR_STREAM_CAPTURE_UNSUPPORTED` and invalidate the capture if the specified event `hEvent` is part of an ongoing capture sequence or if the specified green context `hCtx` has a stream in the capture mode. """ cdef ccuda.CUevent chEvent if hEvent is None: chEvent = 0 elif isinstance(hEvent, (CUevent,)): phEvent = int(hEvent) chEvent = phEvent else: phEvent = int(CUevent(hEvent)) chEvent = phEvent cdef ccuda.CUgreenCtx chCtx if hCtx is None: chCtx = 0 elif isinstance(hCtx, (CUgreenCtx,)): phCtx = int(hCtx) chCtx = phCtx else: phCtx = int(CUgreenCtx(hCtx)) chCtx = phCtx err = ccuda.cuGreenCtxWaitEvent(chCtx, chEvent) return (CUresult(err),) {{endif}} {{if 'cuStreamGetGreenCtx' in found_functions}} @cython.embedsignature(True) def cuStreamGetGreenCtx(hStream): """ Query the green context associated with a stream. Returns the CUDA green context that the stream is associated with, or NULL if the stream is not associated with any green context. The stream handle `hStream` can refer to any of the following: - a stream created via any of the CUDA driver APIs such as :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamCreateWithPriority` and :py:obj:`~.cuGreenCtxStreamCreate`, or their runtime API equivalents such as :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cudaStreamCreateWithFlags` and :py:obj:`~.cudaStreamCreateWithPriority`. If during stream creation the context that was active in the calling thread was obtained with cuCtxFromGreenCtx, that green context is returned in `phCtx`. Otherwise, `*phCtx` is set to NULL instead. - special stream such as the NULL stream or :py:obj:`~.CU_STREAM_LEGACY`. In that case if context that is active in the calling thread was obtained with cuCtxFromGreenCtx, that green context is returned. Otherwise, `*phCtx` is set to NULL instead. Passing an invalid handle will result in undefined behavior. Parameters ---------- hStream : :py:obj:`~.CUstream` or :py:obj:`~.cudaStream_t` Handle to the stream to be queried Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, phCtx : :py:obj:`~.CUgreenCtx` Returned green context associated with the stream See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamCreateWithPriority`, :py:obj:`~.cuStreamGetCtx_v2`, :py:obj:`~.cuGreenCtxStreamCreate`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cuStreamGetFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamCreate`, :py:obj:`~.cudaStreamCreateWithFlags` """ cdef ccuda.CUstream chStream if hStream is None: chStream = 0 elif isinstance(hStream, (CUstream,)): phStream = int(hStream) chStream = phStream else: phStream = int(CUstream(hStream)) chStream = phStream cdef CUgreenCtx phCtx = CUgreenCtx() err = ccuda.cuStreamGetGreenCtx(chStream, phCtx._ptr) return (CUresult(err), phCtx) {{endif}} {{if 'cuGreenCtxStreamCreate' in found_functions}} @cython.embedsignature(True) def cuGreenCtxStreamCreate(greenCtx, unsigned int flags, int priority): """ Create a stream for use in the green context. Creates a stream for use in the specified green context `greenCtx` and returns a handle in `phStream`. The stream can be destroyed by calling :py:obj:`~.cuStreamDestroy()`. Note that the API ignores the context that is current to the calling thread and creates a stream in the specified green context `greenCtx`. The supported values for `flags` are: - :py:obj:`~.CU_STREAM_NON_BLOCKING`: This must be specified. It indicates that work running in the created stream may run concurrently with work in the default stream, and that the created stream should perform no implicit synchronization with the default stream. Specifying `priority` affects the scheduling priority of work in the stream. Priorities provide a hint to preferentially run work with higher priority when possible, but do not preempt already-running work or provide any other functional guarantee on execution order. `priority` follows a convention where lower numbers represent higher priorities. '0' represents default priority. The range of meaningful numerical priorities can be queried using :py:obj:`~.cuCtxGetStreamPriorityRange`. If the specified priority is outside the numerical range returned by :py:obj:`~.cuCtxGetStreamPriorityRange`, it will automatically be clamped to the lowest or the highest number in the range. Parameters ---------- greenCtx : :py:obj:`~.CUgreenCtx` Green context for which to create the stream for flags : unsigned int Flags for stream creation. `CU_STREAM_NON_BLOCKING` must be specified. priority : int Stream priority. Lower numbers represent higher priorities. See :py:obj:`~.cuCtxGetStreamPriorityRange` for more information about meaningful stream priorities that can be passed. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phStream : :py:obj:`~.CUstream` Returned newly created stream See Also -------- :py:obj:`~.cuStreamDestroy`, :py:obj:`~.cuGreenCtxCreate` :py:obj:`~.cuStreamCreate`, :py:obj:`~.cuStreamGetPriority`, :py:obj:`~.cuCtxGetStreamPriorityRange`, :py:obj:`~.cuStreamGetFlags`, :py:obj:`~.cuStreamWaitEvent`, :py:obj:`~.cuStreamQuery`, :py:obj:`~.cuStreamSynchronize`, :py:obj:`~.cuStreamAddCallback`, :py:obj:`~.cudaStreamCreateWithPriority` Notes ----- In the current implementation, only compute kernels launched in priority streams are affected by the stream's priority. Stream priorities have no effect on host-to-device and device-to-host memory operations. """ cdef ccuda.CUgreenCtx cgreenCtx if greenCtx is None: cgreenCtx = 0 elif isinstance(greenCtx, (CUgreenCtx,)): pgreenCtx = int(greenCtx) cgreenCtx = pgreenCtx else: pgreenCtx = int(CUgreenCtx(greenCtx)) cgreenCtx = pgreenCtx cdef CUstream phStream = CUstream() err = ccuda.cuGreenCtxStreamCreate(phStream._ptr, cgreenCtx, flags, priority) return (CUresult(err), phStream) {{endif}} {{if 'cuProfilerStart' in found_functions}} @cython.embedsignature(True) def cuProfilerStart(): """ Enable profiling. Enables profile collection by the active profiling tool for the current context. If profiling is already enabled, then :py:obj:`~.cuProfilerStart()` has no effect. cuProfilerStart and cuProfilerStop APIs are used to programmatically control the profiling granularity by allowing profiling to be done only on selective pieces of code. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuProfilerInitialize`, :py:obj:`~.cuProfilerStop`, :py:obj:`~.cudaProfilerStart` """ err = ccuda.cuProfilerStart() return (CUresult(err),) {{endif}} {{if 'cuProfilerStop' in found_functions}} @cython.embedsignature(True) def cuProfilerStop(): """ Disable profiling. Disables profile collection by the active profiling tool for the current context. If profiling is already disabled, then :py:obj:`~.cuProfilerStop()` has no effect. cuProfilerStart and cuProfilerStop APIs are used to programmatically control the profiling granularity by allowing profiling to be done only on selective pieces of code. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT` See Also -------- :py:obj:`~.cuProfilerInitialize`, :py:obj:`~.cuProfilerStart`, :py:obj:`~.cudaProfilerStop` """ err = ccuda.cuProfilerStop() return (CUresult(err),) {{endif}} {{if True}} @cython.embedsignature(True) def cuGraphicsEGLRegisterImage(image, unsigned int flags): """ Registers an EGL image. Registers the EGLImageKHR specified by `image` for access by CUDA. A handle to the registered object is returned as `pCudaResource`. Additional Mapping/Unmapping is not required for the registered resource and :py:obj:`~.cuGraphicsResourceGetMappedEglFrame` can be directly called on the `pCudaResource`. The application will be responsible for synchronizing access to shared objects. The application must ensure that any pending operation which access the objects have completed before passing control to CUDA. This may be accomplished by issuing and waiting for glFinish command on all GLcontexts (for OpenGL and likewise for other APIs). The application will be also responsible for ensuring that any pending operation on the registered CUDA resource has completed prior to executing subsequent commands in other APIs accesing the same memory objects. This can be accomplished by calling cuCtxSynchronize or cuEventSynchronize (preferably). The surface's intended usage is specified using `flags`, as follows: - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE`: Specifies no hints about how this resource will be used. It is therefore assumed that this resource will be read from and written to by CUDA. This is the default value. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_READ_ONLY`: Specifies that CUDA will not write to this resource. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITE_DISCARD`: Specifies that CUDA will not read from this resource and will write over the entire contents of the resource, so none of the data previously stored in the resource will be preserved. The EGLImageKHR is an object which can be used to create EGLImage target resource. It is defined as a void pointer. typedef void* EGLImageKHR Parameters ---------- image : :py:obj:`~.EGLImageKHR` An EGLImageKHR image which can be used to create target resource. flags : unsigned int Map flags Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, pCudaResource : :py:obj:`~.CUgraphicsResource` Pointer to the returned object handle See Also -------- :py:obj:`~.cuGraphicsEGLRegisterImage`, :py:obj:`~.cuGraphicsUnregisterResource`, :py:obj:`~.cuGraphicsResourceSetMapFlags`, :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cuGraphicsUnmapResources`, :py:obj:`~.cudaGraphicsEGLRegisterImage` """ cdef ccuda.EGLImageKHR cimage if image is None: cimage = 0 elif isinstance(image, (EGLImageKHR,)): pimage = int(image) cimage = pimage else: pimage = int(EGLImageKHR(image)) cimage = pimage cdef CUgraphicsResource pCudaResource = CUgraphicsResource() err = ccuda.cuGraphicsEGLRegisterImage(pCudaResource._ptr, cimage, flags) return (CUresult(err), pCudaResource) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamConsumerConnect(stream): """ Connect CUDA to EGLStream as a consumer. Connect CUDA as a consumer to EGLStreamKHR specified by `stream`. The EGLStreamKHR is an EGL object that transfers a sequence of image frames from one API to another. Parameters ---------- stream : :py:obj:`~.EGLStreamKHR` EGLStreamKHR handle Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, conn : :py:obj:`~.CUeglStreamConnection` Pointer to the returned connection handle See Also -------- :py:obj:`~.cuEGLStreamConsumerConnect`, :py:obj:`~.cuEGLStreamConsumerDisconnect`, :py:obj:`~.cuEGLStreamConsumerAcquireFrame`, :py:obj:`~.cuEGLStreamConsumerReleaseFrame`, :py:obj:`~.cudaEGLStreamConsumerConnect` """ cdef ccuda.EGLStreamKHR cstream if stream is None: cstream = 0 elif isinstance(stream, (EGLStreamKHR,)): pstream = int(stream) cstream = pstream else: pstream = int(EGLStreamKHR(stream)) cstream = pstream cdef CUeglStreamConnection conn = CUeglStreamConnection() err = ccuda.cuEGLStreamConsumerConnect(conn._ptr, cstream) return (CUresult(err), conn) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamConsumerConnectWithFlags(stream, unsigned int flags): """ Connect CUDA to EGLStream as a consumer with given flags. Connect CUDA as a consumer to EGLStreamKHR specified by `stream` with specified `flags` defined by CUeglResourceLocationFlags. The flags specify whether the consumer wants to access frames from system memory or video memory. Default is :py:obj:`~.CU_EGL_RESOURCE_LOCATION_VIDMEM`. Parameters ---------- stream : :py:obj:`~.EGLStreamKHR` EGLStreamKHR handle flags : unsigned int Flags denote intended location - system or video. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, conn : :py:obj:`~.CUeglStreamConnection` Pointer to the returned connection handle See Also -------- :py:obj:`~.cuEGLStreamConsumerConnect`, :py:obj:`~.cuEGLStreamConsumerDisconnect`, :py:obj:`~.cuEGLStreamConsumerAcquireFrame`, :py:obj:`~.cuEGLStreamConsumerReleaseFrame`, :py:obj:`~.cudaEGLStreamConsumerConnectWithFlags` """ cdef ccuda.EGLStreamKHR cstream if stream is None: cstream = 0 elif isinstance(stream, (EGLStreamKHR,)): pstream = int(stream) cstream = pstream else: pstream = int(EGLStreamKHR(stream)) cstream = pstream cdef CUeglStreamConnection conn = CUeglStreamConnection() err = ccuda.cuEGLStreamConsumerConnectWithFlags(conn._ptr, cstream, flags) return (CUresult(err), conn) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamConsumerDisconnect(conn): """ Disconnect CUDA as a consumer to EGLStream . Disconnect CUDA as a consumer to EGLStreamKHR. Parameters ---------- conn : :py:obj:`~.CUeglStreamConnection` Conection to disconnect. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, See Also -------- :py:obj:`~.cuEGLStreamConsumerConnect`, :py:obj:`~.cuEGLStreamConsumerDisconnect`, :py:obj:`~.cuEGLStreamConsumerAcquireFrame`, :py:obj:`~.cuEGLStreamConsumerReleaseFrame`, :py:obj:`~.cudaEGLStreamConsumerDisconnect` """ cdef ccuda.CUeglStreamConnection *cconn if conn is None: cconn = NULL elif isinstance(conn, (CUeglStreamConnection,)): pconn = conn.getPtr() cconn = pconn elif isinstance(conn, (int)): cconn = conn else: raise TypeError("Argument 'conn' is not instance of type (expected , found " + str(type(conn))) err = ccuda.cuEGLStreamConsumerDisconnect(cconn) return (CUresult(err),) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamConsumerAcquireFrame(conn, pCudaResource, pStream, unsigned int timeout): """ Acquire an image frame from the EGLStream with CUDA as a consumer. Acquire an image frame from EGLStreamKHR. This API can also acquire an old frame presented by the producer unless explicitly disabled by setting EGL_SUPPORT_REUSE_NV flag to EGL_FALSE during stream initialization. By default, EGLStream is created with this flag set to EGL_TRUE. :py:obj:`~.cuGraphicsResourceGetMappedEglFrame` can be called on `pCudaResource` to get :py:obj:`~.CUeglFrame`. Parameters ---------- conn : :py:obj:`~.CUeglStreamConnection` Connection on which to acquire pCudaResource : :py:obj:`~.CUgraphicsResource` CUDA resource on which the stream frame will be mapped for use. pStream : :py:obj:`~.CUstream` CUDA stream for synchronization and any data migrations implied by :py:obj:`~.CUeglResourceLocationFlags`. timeout : unsigned int Desired timeout in usec for a new frame to be acquired. If set as :py:obj:`~.CUDA_EGL_INFINITE_TIMEOUT`, acquire waits infinitely. After timeout occurs CUDA consumer tries to acquire an old frame if available and EGL_SUPPORT_REUSE_NV flag is set. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT`, See Also -------- :py:obj:`~.cuEGLStreamConsumerConnect`, :py:obj:`~.cuEGLStreamConsumerDisconnect`, :py:obj:`~.cuEGLStreamConsumerAcquireFrame`, :py:obj:`~.cuEGLStreamConsumerReleaseFrame`, :py:obj:`~.cudaEGLStreamConsumerAcquireFrame` """ cdef ccuda.CUstream *cpStream if pStream is None: cpStream = NULL elif isinstance(pStream, (CUstream,)): ppStream = pStream.getPtr() cpStream = ppStream elif isinstance(pStream, (int)): cpStream = pStream else: raise TypeError("Argument 'pStream' is not instance of type (expected , found " + str(type(pStream))) cdef ccuda.CUgraphicsResource *cpCudaResource if pCudaResource is None: cpCudaResource = NULL elif isinstance(pCudaResource, (CUgraphicsResource,)): ppCudaResource = pCudaResource.getPtr() cpCudaResource = ppCudaResource elif isinstance(pCudaResource, (int)): cpCudaResource = pCudaResource else: raise TypeError("Argument 'pCudaResource' is not instance of type (expected , found " + str(type(pCudaResource))) cdef ccuda.CUeglStreamConnection *cconn if conn is None: cconn = NULL elif isinstance(conn, (CUeglStreamConnection,)): pconn = conn.getPtr() cconn = pconn elif isinstance(conn, (int)): cconn = conn else: raise TypeError("Argument 'conn' is not instance of type (expected , found " + str(type(conn))) err = ccuda.cuEGLStreamConsumerAcquireFrame(cconn, cpCudaResource, cpStream, timeout) return (CUresult(err),) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamConsumerReleaseFrame(conn, pCudaResource, pStream): """ Releases the last frame acquired from the EGLStream. Release the acquired image frame specified by `pCudaResource` to EGLStreamKHR. If EGL_SUPPORT_REUSE_NV flag is set to EGL_TRUE, at the time of EGL creation this API doesn't release the last frame acquired on the EGLStream. By default, EGLStream is created with this flag set to EGL_TRUE. Parameters ---------- conn : :py:obj:`~.CUeglStreamConnection` Connection on which to release pCudaResource : :py:obj:`~.CUgraphicsResource` CUDA resource whose corresponding frame is to be released pStream : :py:obj:`~.CUstream` CUDA stream on which release will be done. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, See Also -------- :py:obj:`~.cuEGLStreamConsumerConnect`, :py:obj:`~.cuEGLStreamConsumerDisconnect`, :py:obj:`~.cuEGLStreamConsumerAcquireFrame`, :py:obj:`~.cuEGLStreamConsumerReleaseFrame`, :py:obj:`~.cudaEGLStreamConsumerReleaseFrame` """ cdef ccuda.CUstream *cpStream if pStream is None: cpStream = NULL elif isinstance(pStream, (CUstream,)): ppStream = pStream.getPtr() cpStream = ppStream elif isinstance(pStream, (int)): cpStream = pStream else: raise TypeError("Argument 'pStream' is not instance of type (expected , found " + str(type(pStream))) cdef ccuda.CUgraphicsResource cpCudaResource if pCudaResource is None: cpCudaResource = 0 elif isinstance(pCudaResource, (CUgraphicsResource,)): ppCudaResource = int(pCudaResource) cpCudaResource = ppCudaResource else: ppCudaResource = int(CUgraphicsResource(pCudaResource)) cpCudaResource = ppCudaResource cdef ccuda.CUeglStreamConnection *cconn if conn is None: cconn = NULL elif isinstance(conn, (CUeglStreamConnection,)): pconn = conn.getPtr() cconn = pconn elif isinstance(conn, (int)): cconn = conn else: raise TypeError("Argument 'conn' is not instance of type (expected , found " + str(type(conn))) err = ccuda.cuEGLStreamConsumerReleaseFrame(cconn, cpCudaResource, cpStream) return (CUresult(err),) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamProducerConnect(stream, width, height): """ Connect CUDA to EGLStream as a producer. Connect CUDA as a producer to EGLStreamKHR specified by `stream`. The EGLStreamKHR is an EGL object that transfers a sequence of image frames from one API to another. Parameters ---------- stream : :py:obj:`~.EGLStreamKHR` EGLStreamKHR handle width : :py:obj:`~.EGLint` width of the image to be submitted to the stream height : :py:obj:`~.EGLint` height of the image to be submitted to the stream Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, conn : :py:obj:`~.CUeglStreamConnection` Pointer to the returned connection handle See Also -------- :py:obj:`~.cuEGLStreamProducerConnect`, :py:obj:`~.cuEGLStreamProducerDisconnect`, :py:obj:`~.cuEGLStreamProducerPresentFrame`, :py:obj:`~.cudaEGLStreamProducerConnect` """ cdef ccuda.EGLint cheight if height is None: cheight = 0 elif isinstance(height, (EGLint,)): pheight = int(height) cheight = pheight else: pheight = int(EGLint(height)) cheight = pheight cdef ccuda.EGLint cwidth if width is None: cwidth = 0 elif isinstance(width, (EGLint,)): pwidth = int(width) cwidth = pwidth else: pwidth = int(EGLint(width)) cwidth = pwidth cdef ccuda.EGLStreamKHR cstream if stream is None: cstream = 0 elif isinstance(stream, (EGLStreamKHR,)): pstream = int(stream) cstream = pstream else: pstream = int(EGLStreamKHR(stream)) cstream = pstream cdef CUeglStreamConnection conn = CUeglStreamConnection() err = ccuda.cuEGLStreamProducerConnect(conn._ptr, cstream, cwidth, cheight) return (CUresult(err), conn) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamProducerDisconnect(conn): """ Disconnect CUDA as a producer to EGLStream . Disconnect CUDA as a producer to EGLStreamKHR. Parameters ---------- conn : :py:obj:`~.CUeglStreamConnection` Conection to disconnect. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, See Also -------- :py:obj:`~.cuEGLStreamProducerConnect`, :py:obj:`~.cuEGLStreamProducerDisconnect`, :py:obj:`~.cuEGLStreamProducerPresentFrame`, :py:obj:`~.cudaEGLStreamProducerDisconnect` """ cdef ccuda.CUeglStreamConnection *cconn if conn is None: cconn = NULL elif isinstance(conn, (CUeglStreamConnection,)): pconn = conn.getPtr() cconn = pconn elif isinstance(conn, (int)): cconn = conn else: raise TypeError("Argument 'conn' is not instance of type (expected , found " + str(type(conn))) err = ccuda.cuEGLStreamProducerDisconnect(cconn) return (CUresult(err),) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamProducerPresentFrame(conn, eglframe not None : CUeglFrame, pStream): """ Present a CUDA eglFrame to the EGLStream with CUDA as a producer. When a frame is presented by the producer, it gets associated with the EGLStream and thus it is illegal to free the frame before the producer is disconnected. If a frame is freed and reused it may lead to undefined behavior. If producer and consumer are on different GPUs (iGPU and dGPU) then frametype :py:obj:`~.CU_EGL_FRAME_TYPE_ARRAY` is not supported. :py:obj:`~.CU_EGL_FRAME_TYPE_PITCH` can be used for such cross-device applications. The :py:obj:`~.CUeglFrame` is defined as: **View CUDA Toolkit Documentation for a C++ code example** For :py:obj:`~.CUeglFrame` of type :py:obj:`~.CU_EGL_FRAME_TYPE_PITCH`, the application may present sub-region of a memory allocation. In that case, the pitched pointer will specify the start address of the sub- region in the allocation and corresponding :py:obj:`~.CUeglFrame` fields will specify the dimensions of the sub-region. Parameters ---------- conn : :py:obj:`~.CUeglStreamConnection` Connection on which to present the CUDA array eglframe : :py:obj:`~.CUeglFrame` CUDA Eglstream Proucer Frame handle to be sent to the consumer over EglStream. pStream : :py:obj:`~.CUstream` CUDA stream on which to present the frame. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, See Also -------- :py:obj:`~.cuEGLStreamProducerConnect`, :py:obj:`~.cuEGLStreamProducerDisconnect`, :py:obj:`~.cuEGLStreamProducerReturnFrame`, :py:obj:`~.cudaEGLStreamProducerPresentFrame` """ cdef ccuda.CUstream *cpStream if pStream is None: cpStream = NULL elif isinstance(pStream, (CUstream,)): ppStream = pStream.getPtr() cpStream = ppStream elif isinstance(pStream, (int)): cpStream = pStream else: raise TypeError("Argument 'pStream' is not instance of type (expected , found " + str(type(pStream))) cdef ccuda.CUeglStreamConnection *cconn if conn is None: cconn = NULL elif isinstance(conn, (CUeglStreamConnection,)): pconn = conn.getPtr() cconn = pconn elif isinstance(conn, (int)): cconn = conn else: raise TypeError("Argument 'conn' is not instance of type (expected , found " + str(type(conn))) err = ccuda.cuEGLStreamProducerPresentFrame(cconn, eglframe._ptr[0], cpStream) return (CUresult(err),) {{endif}} {{if True}} @cython.embedsignature(True) def cuEGLStreamProducerReturnFrame(conn, eglframe : Optional[CUeglFrame], pStream): """ Return the CUDA eglFrame to the EGLStream released by the consumer. This API can potentially return CUDA_ERROR_LAUNCH_TIMEOUT if the consumer has not returned a frame to EGL stream. If timeout is returned the application can retry. Parameters ---------- conn : :py:obj:`~.CUeglStreamConnection` Connection on which to return eglframe : :py:obj:`~.CUeglFrame` CUDA Eglstream Proucer Frame handle returned from the consumer over EglStream. pStream : :py:obj:`~.CUstream` CUDA stream on which to return the frame. Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_LAUNCH_TIMEOUT` See Also -------- :py:obj:`~.cuEGLStreamProducerConnect`, :py:obj:`~.cuEGLStreamProducerDisconnect`, :py:obj:`~.cuEGLStreamProducerPresentFrame`, :py:obj:`~.cudaEGLStreamProducerReturnFrame` """ cdef ccuda.CUstream *cpStream if pStream is None: cpStream = NULL elif isinstance(pStream, (CUstream,)): ppStream = pStream.getPtr() cpStream = ppStream elif isinstance(pStream, (int)): cpStream = pStream else: raise TypeError("Argument 'pStream' is not instance of type (expected , found " + str(type(pStream))) cdef ccuda.CUeglStreamConnection *cconn if conn is None: cconn = NULL elif isinstance(conn, (CUeglStreamConnection,)): pconn = conn.getPtr() cconn = pconn elif isinstance(conn, (int)): cconn = conn else: raise TypeError("Argument 'conn' is not instance of type (expected , found " + str(type(conn))) cdef ccuda.CUeglFrame* ceglframe_ptr = eglframe._ptr if eglframe != None else NULL err = ccuda.cuEGLStreamProducerReturnFrame(cconn, ceglframe_ptr, cpStream) return (CUresult(err),) {{endif}} {{if True}} @cython.embedsignature(True) def cuGraphicsResourceGetMappedEglFrame(resource, unsigned int index, unsigned int mipLevel): """ Get an eglFrame through which to access a registered EGL graphics resource. Returns in `*eglFrame` an eglFrame pointer through which the registered graphics resource `resource` may be accessed. This API can only be called for registered EGL graphics resources. The :py:obj:`~.CUeglFrame` is defined as: **View CUDA Toolkit Documentation for a C++ code example** If `resource` is not registered then :py:obj:`~.CUDA_ERROR_NOT_MAPPED` is returned. Parameters ---------- resource : :py:obj:`~.CUgraphicsResource` None index : unsigned int None mipLevel : unsigned int None Returns ------- CUresult eglFrame : :py:obj:`~.CUeglFrame` None """ cdef ccuda.CUgraphicsResource cresource if resource is None: cresource = 0 elif isinstance(resource, (CUgraphicsResource,)): presource = int(resource) cresource = presource else: presource = int(CUgraphicsResource(resource)) cresource = presource cdef CUeglFrame eglFrame = CUeglFrame() err = ccuda.cuGraphicsResourceGetMappedEglFrame(eglFrame._ptr, cresource, index, mipLevel) return (CUresult(err), eglFrame) {{endif}} {{if True}} @cython.embedsignature(True) def cuEventCreateFromEGLSync(eglSync, unsigned int flags): """ Creates an event from EGLSync object. Creates an event *phEvent from an EGLSyncKHR eglSync with the flags specified via `flags`. Valid flags include: - :py:obj:`~.CU_EVENT_DEFAULT`: Default event creation flag. - :py:obj:`~.CU_EVENT_BLOCKING_SYNC`: Specifies that the created event should use blocking synchronization. A CPU thread that uses :py:obj:`~.cuEventSynchronize()` to wait on an event created with this flag will block until the event has actually been completed. Once the `eglSync` gets destroyed, :py:obj:`~.cuEventDestroy` is the only API that can be invoked on the event. :py:obj:`~.cuEventRecord` and TimingData are not supported for events created from EGLSync. The EGLSyncKHR is an opaque handle to an EGL sync object. typedef void* EGLSyncKHR Parameters ---------- eglSync : :py:obj:`~.EGLSyncKHR` Opaque handle to EGLSync object flags : unsigned int Event creation flags Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` phEvent : :py:obj:`~.CUevent` Returns newly created event See Also -------- :py:obj:`~.cuEventQuery`, :py:obj:`~.cuEventSynchronize`, :py:obj:`~.cuEventDestroy` """ cdef ccuda.EGLSyncKHR ceglSync if eglSync is None: ceglSync = 0 elif isinstance(eglSync, (EGLSyncKHR,)): peglSync = int(eglSync) ceglSync = peglSync else: peglSync = int(EGLSyncKHR(eglSync)) ceglSync = peglSync cdef CUevent phEvent = CUevent() err = ccuda.cuEventCreateFromEGLSync(phEvent._ptr, ceglSync, flags) return (CUresult(err), phEvent) {{endif}} {{if True}} @cython.embedsignature(True) def cuGraphicsGLRegisterBuffer(buffer, unsigned int Flags): """ Registers an OpenGL buffer object. Registers the buffer object specified by `buffer` for access by CUDA. A handle to the registered object is returned as `pCudaResource`. The register flags `Flags` specify the intended usage, as follows: - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_NONE`: Specifies no hints about how this resource will be used. It is therefore assumed that this resource will be read from and written to by CUDA. This is the default value. - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_READ_ONLY`: Specifies that CUDA will not write to this resource. - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD`: Specifies that CUDA will not read from this resource and will write over the entire contents of the resource, so none of the data previously stored in the resource will be preserved. Parameters ---------- buffer : :py:obj:`~.GLuint` name of buffer object to be registered Flags : unsigned int Register flags Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_OPERATING_SYSTEM` pCudaResource : :py:obj:`~.CUgraphicsResource` Pointer to the returned object handle See Also -------- :py:obj:`~.cuGraphicsUnregisterResource`, :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cuGraphicsResourceGetMappedPointer`, :py:obj:`~.cudaGraphicsGLRegisterBuffer` """ cdef ccuda.GLuint cbuffer if buffer is None: cbuffer = 0 elif isinstance(buffer, (GLuint,)): pbuffer = int(buffer) cbuffer = pbuffer else: pbuffer = int(GLuint(buffer)) cbuffer = pbuffer cdef CUgraphicsResource pCudaResource = CUgraphicsResource() err = ccuda.cuGraphicsGLRegisterBuffer(pCudaResource._ptr, cbuffer, Flags) return (CUresult(err), pCudaResource) {{endif}} {{if True}} @cython.embedsignature(True) def cuGraphicsGLRegisterImage(image, target, unsigned int Flags): """ Register an OpenGL texture or renderbuffer object. Registers the texture or renderbuffer object specified by `image` for access by CUDA. A handle to the registered object is returned as `pCudaResource`. `target` must match the type of the object, and must be one of :py:obj:`~.GL_TEXTURE_2D`, :py:obj:`~.GL_TEXTURE_RECTANGLE`, :py:obj:`~.GL_TEXTURE_CUBE_MAP`, :py:obj:`~.GL_TEXTURE_3D`, :py:obj:`~.GL_TEXTURE_2D_ARRAY`, or :py:obj:`~.GL_RENDERBUFFER`. The register flags `Flags` specify the intended usage, as follows: - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_NONE`: Specifies no hints about how this resource will be used. It is therefore assumed that this resource will be read from and written to by CUDA. This is the default value. - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_READ_ONLY`: Specifies that CUDA will not write to this resource. - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_WRITE_DISCARD`: Specifies that CUDA will not read from this resource and will write over the entire contents of the resource, so none of the data previously stored in the resource will be preserved. - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_SURFACE_LDST`: Specifies that CUDA will bind this resource to a surface reference. - :py:obj:`~.CU_GRAPHICS_REGISTER_FLAGS_TEXTURE_GATHER`: Specifies that CUDA will perform texture gather operations on this resource. The following image formats are supported. For brevity's sake, the list is abbreviated. For ex., {GL_R, GL_RG} X {8, 16} would expand to the following 4 formats {GL_R8, GL_R16, GL_RG8, GL_RG16} : - GL_RED, GL_RG, GL_RGBA, GL_LUMINANCE, GL_ALPHA, GL_LUMINANCE_ALPHA, GL_INTENSITY - {GL_R, GL_RG, GL_RGBA} X {8, 16, 16F, 32F, 8UI, 16UI, 32UI, 8I, 16I, 32I} - {GL_LUMINANCE, GL_ALPHA, GL_LUMINANCE_ALPHA, GL_INTENSITY} X {8, 16, 16F_ARB, 32F_ARB, 8UI_EXT, 16UI_EXT, 32UI_EXT, 8I_EXT, 16I_EXT, 32I_EXT} The following image classes are currently disallowed: - Textures with borders - Multisampled renderbuffers Parameters ---------- image : :py:obj:`~.GLuint` name of texture or renderbuffer object to be registered target : :py:obj:`~.GLenum` Identifies the type of object specified by `image` Flags : unsigned int Register flags Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_OPERATING_SYSTEM` pCudaResource : :py:obj:`~.CUgraphicsResource` Pointer to the returned object handle See Also -------- :py:obj:`~.cuGraphicsUnregisterResource`, :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cuGraphicsSubResourceGetMappedArray`, :py:obj:`~.cudaGraphicsGLRegisterImage` """ cdef ccuda.GLenum ctarget if target is None: ctarget = 0 elif isinstance(target, (GLenum,)): ptarget = int(target) ctarget = ptarget else: ptarget = int(GLenum(target)) ctarget = ptarget cdef ccuda.GLuint cimage if image is None: cimage = 0 elif isinstance(image, (GLuint,)): pimage = int(image) cimage = pimage else: pimage = int(GLuint(image)) cimage = pimage cdef CUgraphicsResource pCudaResource = CUgraphicsResource() err = ccuda.cuGraphicsGLRegisterImage(pCudaResource._ptr, cimage, ctarget, Flags) return (CUresult(err), pCudaResource) {{endif}} {{if True}} @cython.embedsignature(True) def cuGLGetDevices(unsigned int cudaDeviceCount, deviceList not None : CUGLDeviceList): """ Gets the CUDA devices associated with the current OpenGL context. Returns in `*pCudaDeviceCount` the number of CUDA-compatible devices corresponding to the current OpenGL context. Also returns in `*pCudaDevices` at most cudaDeviceCount of the CUDA-compatible devices corresponding to the current OpenGL context. If any of the GPUs being used by the current OpenGL context are not CUDA capable then the call will return CUDA_ERROR_NO_DEVICE. The `deviceList` argument may be any of the following: CU_GL_DEVICE_LIST_ALL: Query all devices used by the current OpenGL context. CU_GL_DEVICE_LIST_CURRENT_FRAME: Query the devices used by the current OpenGL context to render the current frame (in SLI). CU_GL_DEVICE_LIST_NEXT_FRAME: Query the devices used by the current OpenGL context to render the next frame (in SLI). Note that this is a prediction, it can't be guaranteed that this is correct in all cases. Parameters ---------- cudaDeviceCount : unsigned int The size of the output device array pCudaDevices. deviceList : CUGLDeviceList The set of devices to return. Returns ------- CUresult CUDA_SUCCESS CUDA_ERROR_NO_DEVICE CUDA_ERROR_INVALID_VALUE CUDA_ERROR_INVALID_CONTEXT CUDA_ERROR_INVALID_GRAPHICS_CONTEXT pCudaDeviceCount : unsigned int Returned number of CUDA devices. pCudaDevices : List[CUdevice] Returned CUDA devices. See Also -------- ~.cudaGLGetDevices Notes ----- This function is not supported on Mac OS X. """ cdef unsigned int pCudaDeviceCount = 0 cdef ccuda.CUdevice* cpCudaDevices = NULL pypCudaDevices = [] if cudaDeviceCount != 0: cpCudaDevices = calloc(cudaDeviceCount, sizeof(ccuda.CUdevice)) if cpCudaDevices is NULL: raise MemoryError('Failed to allocate length x size memory: ' + str(cudaDeviceCount) + 'x' + str(sizeof(ccuda.CUdevice))) cdef ccuda.CUGLDeviceList cdeviceList = deviceList.value err = ccuda.cuGLGetDevices(&pCudaDeviceCount, cpCudaDevices, cudaDeviceCount, cdeviceList) if CUresult(err) == CUresult(0): pypCudaDevices = [CUdevice(init_value=cpCudaDevices[idx]) for idx in range(cudaDeviceCount)] if cpCudaDevices is not NULL: free(cpCudaDevices) return (CUresult(err), pCudaDeviceCount, pypCudaDevices) {{endif}} {{if True}} @cython.embedsignature(True) def cuVDPAUGetDevice(vdpDevice, vdpGetProcAddress): """ Gets the CUDA device associated with a VDPAU device. Returns in `*pDevice` the CUDA device associated with a `vdpDevice`, if applicable. Parameters ---------- vdpDevice : :py:obj:`~.VdpDevice` A VdpDevice handle vdpGetProcAddress : :py:obj:`~.VdpGetProcAddress` VDPAU's VdpGetProcAddress function pointer Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE` pDevice : :py:obj:`~.CUdevice` Device associated with vdpDevice See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuVDPAUCtxCreate`, :py:obj:`~.cuGraphicsVDPAURegisterVideoSurface`, :py:obj:`~.cuGraphicsVDPAURegisterOutputSurface`, :py:obj:`~.cuGraphicsUnregisterResource`, :py:obj:`~.cuGraphicsResourceSetMapFlags`, :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cuGraphicsUnmapResources`, :py:obj:`~.cuGraphicsSubResourceGetMappedArray`, :py:obj:`~.cudaVDPAUGetDevice` """ cdef ccuda.VdpGetProcAddress *cvdpGetProcAddress if vdpGetProcAddress is None: cvdpGetProcAddress = NULL elif isinstance(vdpGetProcAddress, (VdpGetProcAddress,)): pvdpGetProcAddress = vdpGetProcAddress.getPtr() cvdpGetProcAddress = pvdpGetProcAddress elif isinstance(vdpGetProcAddress, (int)): cvdpGetProcAddress = vdpGetProcAddress else: raise TypeError("Argument 'vdpGetProcAddress' is not instance of type (expected , found " + str(type(vdpGetProcAddress))) cdef ccuda.VdpDevice cvdpDevice if vdpDevice is None: cvdpDevice = 0 elif isinstance(vdpDevice, (VdpDevice,)): pvdpDevice = int(vdpDevice) cvdpDevice = pvdpDevice else: pvdpDevice = int(VdpDevice(vdpDevice)) cvdpDevice = pvdpDevice cdef CUdevice pDevice = CUdevice() err = ccuda.cuVDPAUGetDevice(pDevice._ptr, cvdpDevice, cvdpGetProcAddress) return (CUresult(err), pDevice) {{endif}} {{if True}} @cython.embedsignature(True) def cuVDPAUCtxCreate(unsigned int flags, device, vdpDevice, vdpGetProcAddress): """ Create a CUDA context for interoperability with VDPAU. Creates a new CUDA context, initializes VDPAU interoperability, and associates the CUDA context with the calling thread. It must be called before performing any other VDPAU interoperability operations. It may fail if the needed VDPAU driver facilities are not available. For usage of the `flags` parameter, see :py:obj:`~.cuCtxCreate()`. Parameters ---------- flags : unsigned int Options for CUDA context creation device : :py:obj:`~.CUdevice` Device on which to create the context vdpDevice : :py:obj:`~.VdpDevice` The VdpDevice to interop with vdpGetProcAddress : :py:obj:`~.VdpGetProcAddress` VDPAU's VdpGetProcAddress function pointer Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_DEINITIALIZED`, :py:obj:`~.CUDA_ERROR_NOT_INITIALIZED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, :py:obj:`~.CUDA_ERROR_INVALID_VALUE`, :py:obj:`~.CUDA_ERROR_OUT_OF_MEMORY` pCtx : :py:obj:`~.CUcontext` Returned CUDA context See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuGraphicsVDPAURegisterVideoSurface`, :py:obj:`~.cuGraphicsVDPAURegisterOutputSurface`, :py:obj:`~.cuGraphicsUnregisterResource`, :py:obj:`~.cuGraphicsResourceSetMapFlags`, :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cuGraphicsUnmapResources`, :py:obj:`~.cuGraphicsSubResourceGetMappedArray`, :py:obj:`~.cuVDPAUGetDevice` """ cdef ccuda.VdpGetProcAddress *cvdpGetProcAddress if vdpGetProcAddress is None: cvdpGetProcAddress = NULL elif isinstance(vdpGetProcAddress, (VdpGetProcAddress,)): pvdpGetProcAddress = vdpGetProcAddress.getPtr() cvdpGetProcAddress = pvdpGetProcAddress elif isinstance(vdpGetProcAddress, (int)): cvdpGetProcAddress = vdpGetProcAddress else: raise TypeError("Argument 'vdpGetProcAddress' is not instance of type (expected , found " + str(type(vdpGetProcAddress))) cdef ccuda.VdpDevice cvdpDevice if vdpDevice is None: cvdpDevice = 0 elif isinstance(vdpDevice, (VdpDevice,)): pvdpDevice = int(vdpDevice) cvdpDevice = pvdpDevice else: pvdpDevice = int(VdpDevice(vdpDevice)) cvdpDevice = pvdpDevice cdef ccuda.CUdevice cdevice if device is None: cdevice = 0 elif isinstance(device, (CUdevice,)): pdevice = int(device) cdevice = pdevice else: pdevice = int(CUdevice(device)) cdevice = pdevice cdef CUcontext pCtx = CUcontext() err = ccuda.cuVDPAUCtxCreate(pCtx._ptr, flags, cdevice, cvdpDevice, cvdpGetProcAddress) return (CUresult(err), pCtx) {{endif}} {{if True}} @cython.embedsignature(True) def cuGraphicsVDPAURegisterVideoSurface(vdpSurface, unsigned int flags): """ Registers a VDPAU VdpVideoSurface object. Registers the VdpVideoSurface specified by `vdpSurface` for access by CUDA. A handle to the registered object is returned as `pCudaResource`. The surface's intended usage is specified using `flags`, as follows: - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE`: Specifies no hints about how this resource will be used. It is therefore assumed that this resource will be read from and written to by CUDA. This is the default value. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_READ_ONLY`: Specifies that CUDA will not write to this resource. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITE_DISCARD`: Specifies that CUDA will not read from this resource and will write over the entire contents of the resource, so none of the data previously stored in the resource will be preserved. The VdpVideoSurface is presented as an array of subresources that may be accessed using pointers returned by :py:obj:`~.cuGraphicsSubResourceGetMappedArray`. The exact number of valid `arrayIndex` values depends on the VDPAU surface format. The mapping is shown in the table below. `mipLevel` must be 0. Parameters ---------- vdpSurface : :py:obj:`~.VdpVideoSurface` The VdpVideoSurface to be registered flags : unsigned int Map flags Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, pCudaResource : :py:obj:`~.CUgraphicsResource` Pointer to the returned object handle See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuVDPAUCtxCreate`, :py:obj:`~.cuGraphicsVDPAURegisterOutputSurface`, :py:obj:`~.cuGraphicsUnregisterResource`, :py:obj:`~.cuGraphicsResourceSetMapFlags`, :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cuGraphicsUnmapResources`, :py:obj:`~.cuGraphicsSubResourceGetMappedArray`, :py:obj:`~.cuVDPAUGetDevice`, :py:obj:`~.cudaGraphicsVDPAURegisterVideoSurface` """ cdef ccuda.VdpVideoSurface cvdpSurface if vdpSurface is None: cvdpSurface = 0 elif isinstance(vdpSurface, (VdpVideoSurface,)): pvdpSurface = int(vdpSurface) cvdpSurface = pvdpSurface else: pvdpSurface = int(VdpVideoSurface(vdpSurface)) cvdpSurface = pvdpSurface cdef CUgraphicsResource pCudaResource = CUgraphicsResource() err = ccuda.cuGraphicsVDPAURegisterVideoSurface(pCudaResource._ptr, cvdpSurface, flags) return (CUresult(err), pCudaResource) {{endif}} {{if True}} @cython.embedsignature(True) def cuGraphicsVDPAURegisterOutputSurface(vdpSurface, unsigned int flags): """ Registers a VDPAU VdpOutputSurface object. Registers the VdpOutputSurface specified by `vdpSurface` for access by CUDA. A handle to the registered object is returned as `pCudaResource`. The surface's intended usage is specified using `flags`, as follows: - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_NONE`: Specifies no hints about how this resource will be used. It is therefore assumed that this resource will be read from and written to by CUDA. This is the default value. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_READ_ONLY`: Specifies that CUDA will not write to this resource. - :py:obj:`~.CU_GRAPHICS_MAP_RESOURCE_FLAGS_WRITE_DISCARD`: Specifies that CUDA will not read from this resource and will write over the entire contents of the resource, so none of the data previously stored in the resource will be preserved. The VdpOutputSurface is presented as an array of subresources that may be accessed using pointers returned by :py:obj:`~.cuGraphicsSubResourceGetMappedArray`. The exact number of valid `arrayIndex` values depends on the VDPAU surface format. The mapping is shown in the table below. `mipLevel` must be 0. Parameters ---------- vdpSurface : :py:obj:`~.VdpOutputSurface` The VdpOutputSurface to be registered flags : unsigned int Map flags Returns ------- CUresult :py:obj:`~.CUDA_SUCCESS`, :py:obj:`~.CUDA_ERROR_INVALID_HANDLE`, :py:obj:`~.CUDA_ERROR_ALREADY_MAPPED`, :py:obj:`~.CUDA_ERROR_INVALID_CONTEXT`, pCudaResource : :py:obj:`~.CUgraphicsResource` Pointer to the returned object handle See Also -------- :py:obj:`~.cuCtxCreate`, :py:obj:`~.cuVDPAUCtxCreate`, :py:obj:`~.cuGraphicsVDPAURegisterVideoSurface`, :py:obj:`~.cuGraphicsUnregisterResource`, :py:obj:`~.cuGraphicsResourceSetMapFlags`, :py:obj:`~.cuGraphicsMapResources`, :py:obj:`~.cuGraphicsUnmapResources`, :py:obj:`~.cuGraphicsSubResourceGetMappedArray`, :py:obj:`~.cuVDPAUGetDevice`, :py:obj:`~.cudaGraphicsVDPAURegisterOutputSurface` """ cdef ccuda.VdpOutputSurface cvdpSurface if vdpSurface is None: cvdpSurface = 0 elif isinstance(vdpSurface, (VdpOutputSurface,)): pvdpSurface = int(vdpSurface) cvdpSurface = pvdpSurface else: pvdpSurface = int(VdpOutputSurface(vdpSurface)) cvdpSurface = pvdpSurface cdef CUgraphicsResource pCudaResource = CUgraphicsResource() err = ccuda.cuGraphicsVDPAURegisterOutputSurface(pCudaResource._ptr, cvdpSurface, flags) return (CUresult(err), pCudaResource) {{endif}} @cython.embedsignature(True) def sizeof(objType): """ Returns the size of provided CUDA Python structure in bytes Parameters ---------- objType : Any CUDA Python object Returns ------- lowered_name : int The size of `objType` in bytes """ {{if 'cuuint32_t' in found_types}} if objType == cuuint32_t: return sizeof(ccuda.cuuint32_t){{endif}} {{if 'cuuint64_t' in found_types}} if objType == cuuint64_t: return sizeof(ccuda.cuuint64_t){{endif}} {{if 'CUdeviceptr_v2' in found_types}} if objType == CUdeviceptr_v2: return sizeof(ccuda.CUdeviceptr_v2){{endif}} {{if 'CUdeviceptr' in found_types}} if objType == CUdeviceptr: return sizeof(ccuda.CUdeviceptr){{endif}} {{if 'CUdevice_v1' in found_types}} if objType == CUdevice_v1: return sizeof(ccuda.CUdevice_v1){{endif}} {{if 'CUdevice' in found_types}} if objType == CUdevice: return sizeof(ccuda.CUdevice){{endif}} {{if 'CUcontext' in found_types}} if objType == CUcontext: return sizeof(ccuda.CUcontext){{endif}} {{if 'CUmodule' in found_types}} if objType == CUmodule: return sizeof(ccuda.CUmodule){{endif}} {{if 'CUfunction' in found_types}} if objType == CUfunction: return sizeof(ccuda.CUfunction){{endif}} {{if 'CUlibrary' in found_types}} if objType == CUlibrary: return sizeof(ccuda.CUlibrary){{endif}} {{if 'CUkernel' in found_types}} if objType == CUkernel: return sizeof(ccuda.CUkernel){{endif}} {{if 'CUarray' in found_types}} if objType == CUarray: return sizeof(ccuda.CUarray){{endif}} {{if 'CUmipmappedArray' in found_types}} if objType == CUmipmappedArray: return sizeof(ccuda.CUmipmappedArray){{endif}} {{if 'CUtexref' in found_types}} if objType == CUtexref: return sizeof(ccuda.CUtexref){{endif}} {{if 'CUsurfref' in found_types}} if objType == CUsurfref: return sizeof(ccuda.CUsurfref){{endif}} {{if 'CUevent' in found_types}} if objType == CUevent: return sizeof(ccuda.CUevent){{endif}} {{if 'CUstream' in found_types}} if objType == CUstream: return sizeof(ccuda.CUstream){{endif}} {{if 'CUgraphicsResource' in found_types}} if objType == CUgraphicsResource: return sizeof(ccuda.CUgraphicsResource){{endif}} {{if 'CUtexObject_v1' in found_types}} if objType == CUtexObject_v1: return sizeof(ccuda.CUtexObject_v1){{endif}} {{if 'CUtexObject' in found_types}} if objType == CUtexObject: return sizeof(ccuda.CUtexObject){{endif}} {{if 'CUsurfObject_v1' in found_types}} if objType == CUsurfObject_v1: return sizeof(ccuda.CUsurfObject_v1){{endif}} {{if 'CUsurfObject' in found_types}} if objType == CUsurfObject: return sizeof(ccuda.CUsurfObject){{endif}} {{if 'CUexternalMemory' in found_types}} if objType == CUexternalMemory: return sizeof(ccuda.CUexternalMemory){{endif}} {{if 'CUexternalSemaphore' in found_types}} if objType == CUexternalSemaphore: return sizeof(ccuda.CUexternalSemaphore){{endif}} {{if 'CUgraph' in found_types}} if objType == CUgraph: return sizeof(ccuda.CUgraph){{endif}} {{if 'CUgraphNode' in found_types}} if objType == CUgraphNode: return sizeof(ccuda.CUgraphNode){{endif}} {{if 'CUgraphExec' in found_types}} if objType == CUgraphExec: return sizeof(ccuda.CUgraphExec){{endif}} {{if 'CUmemoryPool' in found_types}} if objType == CUmemoryPool: return sizeof(ccuda.CUmemoryPool){{endif}} {{if 'CUuserObject' in found_types}} if objType == CUuserObject: return sizeof(ccuda.CUuserObject){{endif}} {{if 'CUgraphConditionalHandle' in found_types}} if objType == CUgraphConditionalHandle: return sizeof(ccuda.CUgraphConditionalHandle){{endif}} {{if 'CUgraphDeviceNode' in found_types}} if objType == CUgraphDeviceNode: return sizeof(ccuda.CUgraphDeviceNode){{endif}} {{if 'CUasyncCallbackHandle' in found_types}} if objType == CUasyncCallbackHandle: return sizeof(ccuda.CUasyncCallbackHandle){{endif}} {{if 'CUgreenCtx' in found_types}} if objType == CUgreenCtx: return sizeof(ccuda.CUgreenCtx){{endif}} {{if 'struct CUuuid_st' in found_types}} if objType == CUuuid_st: return sizeof(ccuda.CUuuid_st){{endif}} {{if 'CUuuid' in found_types}} if objType == CUuuid: return sizeof(ccuda.CUuuid){{endif}} {{if 'struct CUmemFabricHandle_st' in found_types}} if objType == CUmemFabricHandle_st: return sizeof(ccuda.CUmemFabricHandle_st){{endif}} {{if 'CUmemFabricHandle_v1' in found_types}} if objType == CUmemFabricHandle_v1: return sizeof(ccuda.CUmemFabricHandle_v1){{endif}} {{if 'CUmemFabricHandle' in found_types}} if objType == CUmemFabricHandle: return sizeof(ccuda.CUmemFabricHandle){{endif}} {{if 'struct CUipcEventHandle_st' in found_types}} if objType == CUipcEventHandle_st: return sizeof(ccuda.CUipcEventHandle_st){{endif}} {{if 'CUipcEventHandle_v1' in found_types}} if objType == CUipcEventHandle_v1: return sizeof(ccuda.CUipcEventHandle_v1){{endif}} {{if 'CUipcEventHandle' in found_types}} if objType == CUipcEventHandle: return sizeof(ccuda.CUipcEventHandle){{endif}} {{if 'struct CUipcMemHandle_st' in found_types}} if objType == CUipcMemHandle_st: return sizeof(ccuda.CUipcMemHandle_st){{endif}} {{if 'CUipcMemHandle_v1' in found_types}} if objType == CUipcMemHandle_v1: return sizeof(ccuda.CUipcMemHandle_v1){{endif}} {{if 'CUipcMemHandle' in found_types}} if objType == CUipcMemHandle: return sizeof(ccuda.CUipcMemHandle){{endif}} {{if 'union CUstreamBatchMemOpParams_union' in found_types}} if objType == CUstreamBatchMemOpParams_union: return sizeof(ccuda.CUstreamBatchMemOpParams_union){{endif}} {{if 'CUstreamBatchMemOpParams_v1' in found_types}} if objType == CUstreamBatchMemOpParams_v1: return sizeof(ccuda.CUstreamBatchMemOpParams_v1){{endif}} {{if 'CUstreamBatchMemOpParams' in found_types}} if objType == CUstreamBatchMemOpParams: return sizeof(ccuda.CUstreamBatchMemOpParams){{endif}} {{if 'struct CUDA_BATCH_MEM_OP_NODE_PARAMS_v1_st' in found_types}} if objType == CUDA_BATCH_MEM_OP_NODE_PARAMS_v1_st: return sizeof(ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS_v1_st){{endif}} {{if 'CUDA_BATCH_MEM_OP_NODE_PARAMS_v1' in found_types}} if objType == CUDA_BATCH_MEM_OP_NODE_PARAMS_v1: return sizeof(ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS_v1){{endif}} {{if 'CUDA_BATCH_MEM_OP_NODE_PARAMS' in found_types}} if objType == CUDA_BATCH_MEM_OP_NODE_PARAMS: return sizeof(ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS){{endif}} {{if 'struct CUDA_BATCH_MEM_OP_NODE_PARAMS_v2_st' in found_types}} if objType == CUDA_BATCH_MEM_OP_NODE_PARAMS_v2_st: return sizeof(ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS_v2_st){{endif}} {{if 'CUDA_BATCH_MEM_OP_NODE_PARAMS_v2' in found_types}} if objType == CUDA_BATCH_MEM_OP_NODE_PARAMS_v2: return sizeof(ccuda.CUDA_BATCH_MEM_OP_NODE_PARAMS_v2){{endif}} {{if 'struct CUasyncNotificationInfo_st' in found_types}} if objType == CUasyncNotificationInfo_st: return sizeof(ccuda.CUasyncNotificationInfo_st){{endif}} {{if 'CUasyncNotificationInfo' in found_types}} if objType == CUasyncNotificationInfo: return sizeof(ccuda.CUasyncNotificationInfo){{endif}} {{if 'CUasyncCallback' in found_types}} if objType == CUasyncCallback: return sizeof(ccuda.CUasyncCallback){{endif}} {{if 'struct CUdevprop_st' in found_types}} if objType == CUdevprop_st: return sizeof(ccuda.CUdevprop_st){{endif}} {{if 'CUdevprop_v1' in found_types}} if objType == CUdevprop_v1: return sizeof(ccuda.CUdevprop_v1){{endif}} {{if 'CUdevprop' in found_types}} if objType == CUdevprop: return sizeof(ccuda.CUdevprop){{endif}} {{if 'CUlinkState' in found_types}} if objType == CUlinkState: return sizeof(ccuda.CUlinkState){{endif}} {{if 'CUhostFn' in found_types}} if objType == CUhostFn: return sizeof(ccuda.CUhostFn){{endif}} {{if 'struct CUaccessPolicyWindow_st' in found_types}} if objType == CUaccessPolicyWindow_st: return sizeof(ccuda.CUaccessPolicyWindow_st){{endif}} {{if 'CUaccessPolicyWindow_v1' in found_types}} if objType == CUaccessPolicyWindow_v1: return sizeof(ccuda.CUaccessPolicyWindow_v1){{endif}} {{if 'CUaccessPolicyWindow' in found_types}} if objType == CUaccessPolicyWindow: return sizeof(ccuda.CUaccessPolicyWindow){{endif}} {{if 'struct CUDA_KERNEL_NODE_PARAMS_st' in found_types}} if objType == CUDA_KERNEL_NODE_PARAMS_st: return sizeof(ccuda.CUDA_KERNEL_NODE_PARAMS_st){{endif}} {{if 'CUDA_KERNEL_NODE_PARAMS_v1' in found_types}} if objType == CUDA_KERNEL_NODE_PARAMS_v1: return sizeof(ccuda.CUDA_KERNEL_NODE_PARAMS_v1){{endif}} {{if 'struct CUDA_KERNEL_NODE_PARAMS_v2_st' in found_types}} if objType == CUDA_KERNEL_NODE_PARAMS_v2_st: return sizeof(ccuda.CUDA_KERNEL_NODE_PARAMS_v2_st){{endif}} {{if 'CUDA_KERNEL_NODE_PARAMS_v2' in found_types}} if objType == CUDA_KERNEL_NODE_PARAMS_v2: return sizeof(ccuda.CUDA_KERNEL_NODE_PARAMS_v2){{endif}} {{if 'CUDA_KERNEL_NODE_PARAMS' in found_types}} if objType == CUDA_KERNEL_NODE_PARAMS: return sizeof(ccuda.CUDA_KERNEL_NODE_PARAMS){{endif}} {{if 'struct CUDA_KERNEL_NODE_PARAMS_v3_st' in found_types}} if objType == CUDA_KERNEL_NODE_PARAMS_v3_st: return sizeof(ccuda.CUDA_KERNEL_NODE_PARAMS_v3_st){{endif}} {{if 'CUDA_KERNEL_NODE_PARAMS_v3' in found_types}} if objType == CUDA_KERNEL_NODE_PARAMS_v3: return sizeof(ccuda.CUDA_KERNEL_NODE_PARAMS_v3){{endif}} {{if 'struct CUDA_MEMSET_NODE_PARAMS_st' in found_types}} if objType == CUDA_MEMSET_NODE_PARAMS_st: return sizeof(ccuda.CUDA_MEMSET_NODE_PARAMS_st){{endif}} {{if 'CUDA_MEMSET_NODE_PARAMS_v1' in found_types}} if objType == CUDA_MEMSET_NODE_PARAMS_v1: return sizeof(ccuda.CUDA_MEMSET_NODE_PARAMS_v1){{endif}} {{if 'CUDA_MEMSET_NODE_PARAMS' in found_types}} if objType == CUDA_MEMSET_NODE_PARAMS: return sizeof(ccuda.CUDA_MEMSET_NODE_PARAMS){{endif}} {{if 'struct CUDA_MEMSET_NODE_PARAMS_v2_st' in found_types}} if objType == CUDA_MEMSET_NODE_PARAMS_v2_st: return sizeof(ccuda.CUDA_MEMSET_NODE_PARAMS_v2_st){{endif}} {{if 'CUDA_MEMSET_NODE_PARAMS_v2' in found_types}} if objType == CUDA_MEMSET_NODE_PARAMS_v2: return sizeof(ccuda.CUDA_MEMSET_NODE_PARAMS_v2){{endif}} {{if 'struct CUDA_HOST_NODE_PARAMS_st' in found_types}} if objType == CUDA_HOST_NODE_PARAMS_st: return sizeof(ccuda.CUDA_HOST_NODE_PARAMS_st){{endif}} {{if 'CUDA_HOST_NODE_PARAMS_v1' in found_types}} if objType == CUDA_HOST_NODE_PARAMS_v1: return sizeof(ccuda.CUDA_HOST_NODE_PARAMS_v1){{endif}} {{if 'CUDA_HOST_NODE_PARAMS' in found_types}} if objType == CUDA_HOST_NODE_PARAMS: return sizeof(ccuda.CUDA_HOST_NODE_PARAMS){{endif}} {{if 'struct CUDA_HOST_NODE_PARAMS_v2_st' in found_types}} if objType == CUDA_HOST_NODE_PARAMS_v2_st: return sizeof(ccuda.CUDA_HOST_NODE_PARAMS_v2_st){{endif}} {{if 'CUDA_HOST_NODE_PARAMS_v2' in found_types}} if objType == CUDA_HOST_NODE_PARAMS_v2: return sizeof(ccuda.CUDA_HOST_NODE_PARAMS_v2){{endif}} {{if 'struct CUDA_CONDITIONAL_NODE_PARAMS' in found_types}} if objType == CUDA_CONDITIONAL_NODE_PARAMS: return sizeof(ccuda.CUDA_CONDITIONAL_NODE_PARAMS){{endif}} {{if 'struct CUgraphEdgeData_st' in found_types}} if objType == CUgraphEdgeData_st: return sizeof(ccuda.CUgraphEdgeData_st){{endif}} {{if 'CUgraphEdgeData' in found_types}} if objType == CUgraphEdgeData: return sizeof(ccuda.CUgraphEdgeData){{endif}} {{if 'struct CUDA_GRAPH_INSTANTIATE_PARAMS_st' in found_types}} if objType == CUDA_GRAPH_INSTANTIATE_PARAMS_st: return sizeof(ccuda.CUDA_GRAPH_INSTANTIATE_PARAMS_st){{endif}} {{if 'CUDA_GRAPH_INSTANTIATE_PARAMS' in found_types}} if objType == CUDA_GRAPH_INSTANTIATE_PARAMS: return sizeof(ccuda.CUDA_GRAPH_INSTANTIATE_PARAMS){{endif}} {{if 'struct CUlaunchMemSyncDomainMap_st' in found_types}} if objType == CUlaunchMemSyncDomainMap_st: return sizeof(ccuda.CUlaunchMemSyncDomainMap_st){{endif}} {{if 'CUlaunchMemSyncDomainMap' in found_types}} if objType == CUlaunchMemSyncDomainMap: return sizeof(ccuda.CUlaunchMemSyncDomainMap){{endif}} {{if 'union CUlaunchAttributeValue_union' in found_types}} if objType == CUlaunchAttributeValue_union: return sizeof(ccuda.CUlaunchAttributeValue_union){{endif}} {{if 'CUlaunchAttributeValue' in found_types}} if objType == CUlaunchAttributeValue: return sizeof(ccuda.CUlaunchAttributeValue){{endif}} {{if 'struct CUlaunchAttribute_st' in found_types}} if objType == CUlaunchAttribute_st: return sizeof(ccuda.CUlaunchAttribute_st){{endif}} {{if 'CUlaunchAttribute' in found_types}} if objType == CUlaunchAttribute: return sizeof(ccuda.CUlaunchAttribute){{endif}} {{if 'struct CUlaunchConfig_st' in found_types}} if objType == CUlaunchConfig_st: return sizeof(ccuda.CUlaunchConfig_st){{endif}} {{if 'CUlaunchConfig' in found_types}} if objType == CUlaunchConfig: return sizeof(ccuda.CUlaunchConfig){{endif}} {{if 'CUkernelNodeAttrValue_v1' in found_types}} if objType == CUkernelNodeAttrValue_v1: return sizeof(ccuda.CUkernelNodeAttrValue_v1){{endif}} {{if 'CUkernelNodeAttrValue' in found_types}} if objType == CUkernelNodeAttrValue: return sizeof(ccuda.CUkernelNodeAttrValue){{endif}} {{if 'CUstreamAttrValue_v1' in found_types}} if objType == CUstreamAttrValue_v1: return sizeof(ccuda.CUstreamAttrValue_v1){{endif}} {{if 'CUstreamAttrValue' in found_types}} if objType == CUstreamAttrValue: return sizeof(ccuda.CUstreamAttrValue){{endif}} {{if 'struct CUexecAffinitySmCount_st' in found_types}} if objType == CUexecAffinitySmCount_st: return sizeof(ccuda.CUexecAffinitySmCount_st){{endif}} {{if 'CUexecAffinitySmCount_v1' in found_types}} if objType == CUexecAffinitySmCount_v1: return sizeof(ccuda.CUexecAffinitySmCount_v1){{endif}} {{if 'CUexecAffinitySmCount' in found_types}} if objType == CUexecAffinitySmCount: return sizeof(ccuda.CUexecAffinitySmCount){{endif}} {{if 'struct CUexecAffinityParam_st' in found_types}} if objType == CUexecAffinityParam_st: return sizeof(ccuda.CUexecAffinityParam_st){{endif}} {{if 'CUexecAffinityParam_v1' in found_types}} if objType == CUexecAffinityParam_v1: return sizeof(ccuda.CUexecAffinityParam_v1){{endif}} {{if 'CUexecAffinityParam' in found_types}} if objType == CUexecAffinityParam: return sizeof(ccuda.CUexecAffinityParam){{endif}} {{if 'struct CUctxCigParam_st' in found_types}} if objType == CUctxCigParam_st: return sizeof(ccuda.CUctxCigParam_st){{endif}} {{if 'CUctxCigParam' in found_types}} if objType == CUctxCigParam: return sizeof(ccuda.CUctxCigParam){{endif}} {{if 'struct CUctxCreateParams_st' in found_types}} if objType == CUctxCreateParams_st: return sizeof(ccuda.CUctxCreateParams_st){{endif}} {{if 'CUctxCreateParams' in found_types}} if objType == CUctxCreateParams: return sizeof(ccuda.CUctxCreateParams){{endif}} {{if 'struct CUlibraryHostUniversalFunctionAndDataTable_st' in found_types}} if objType == CUlibraryHostUniversalFunctionAndDataTable_st: return sizeof(ccuda.CUlibraryHostUniversalFunctionAndDataTable_st){{endif}} {{if 'CUlibraryHostUniversalFunctionAndDataTable' in found_types}} if objType == CUlibraryHostUniversalFunctionAndDataTable: return sizeof(ccuda.CUlibraryHostUniversalFunctionAndDataTable){{endif}} {{if 'CUstreamCallback' in found_types}} if objType == CUstreamCallback: return sizeof(ccuda.CUstreamCallback){{endif}} {{if 'CUoccupancyB2DSize' in found_types}} if objType == CUoccupancyB2DSize: return sizeof(ccuda.CUoccupancyB2DSize){{endif}} {{if 'struct CUDA_MEMCPY2D_st' in found_types}} if objType == CUDA_MEMCPY2D_st: return sizeof(ccuda.CUDA_MEMCPY2D_st){{endif}} {{if 'CUDA_MEMCPY2D_v2' in found_types}} if objType == CUDA_MEMCPY2D_v2: return sizeof(ccuda.CUDA_MEMCPY2D_v2){{endif}} {{if 'CUDA_MEMCPY2D' in found_types}} if objType == CUDA_MEMCPY2D: return sizeof(ccuda.CUDA_MEMCPY2D){{endif}} {{if 'struct CUDA_MEMCPY3D_st' in found_types}} if objType == CUDA_MEMCPY3D_st: return sizeof(ccuda.CUDA_MEMCPY3D_st){{endif}} {{if 'CUDA_MEMCPY3D_v2' in found_types}} if objType == CUDA_MEMCPY3D_v2: return sizeof(ccuda.CUDA_MEMCPY3D_v2){{endif}} {{if 'CUDA_MEMCPY3D' in found_types}} if objType == CUDA_MEMCPY3D: return sizeof(ccuda.CUDA_MEMCPY3D){{endif}} {{if 'struct CUDA_MEMCPY3D_PEER_st' in found_types}} if objType == CUDA_MEMCPY3D_PEER_st: return sizeof(ccuda.CUDA_MEMCPY3D_PEER_st){{endif}} {{if 'CUDA_MEMCPY3D_PEER_v1' in found_types}} if objType == CUDA_MEMCPY3D_PEER_v1: return sizeof(ccuda.CUDA_MEMCPY3D_PEER_v1){{endif}} {{if 'CUDA_MEMCPY3D_PEER' in found_types}} if objType == CUDA_MEMCPY3D_PEER: return sizeof(ccuda.CUDA_MEMCPY3D_PEER){{endif}} {{if 'struct CUDA_MEMCPY_NODE_PARAMS_st' in found_types}} if objType == CUDA_MEMCPY_NODE_PARAMS_st: return sizeof(ccuda.CUDA_MEMCPY_NODE_PARAMS_st){{endif}} {{if 'CUDA_MEMCPY_NODE_PARAMS' in found_types}} if objType == CUDA_MEMCPY_NODE_PARAMS: return sizeof(ccuda.CUDA_MEMCPY_NODE_PARAMS){{endif}} {{if 'struct CUDA_ARRAY_DESCRIPTOR_st' in found_types}} if objType == CUDA_ARRAY_DESCRIPTOR_st: return sizeof(ccuda.CUDA_ARRAY_DESCRIPTOR_st){{endif}} {{if 'CUDA_ARRAY_DESCRIPTOR_v2' in found_types}} if objType == CUDA_ARRAY_DESCRIPTOR_v2: return sizeof(ccuda.CUDA_ARRAY_DESCRIPTOR_v2){{endif}} {{if 'CUDA_ARRAY_DESCRIPTOR' in found_types}} if objType == CUDA_ARRAY_DESCRIPTOR: return sizeof(ccuda.CUDA_ARRAY_DESCRIPTOR){{endif}} {{if 'struct CUDA_ARRAY3D_DESCRIPTOR_st' in found_types}} if objType == CUDA_ARRAY3D_DESCRIPTOR_st: return sizeof(ccuda.CUDA_ARRAY3D_DESCRIPTOR_st){{endif}} {{if 'CUDA_ARRAY3D_DESCRIPTOR_v2' in found_types}} if objType == CUDA_ARRAY3D_DESCRIPTOR_v2: return sizeof(ccuda.CUDA_ARRAY3D_DESCRIPTOR_v2){{endif}} {{if 'CUDA_ARRAY3D_DESCRIPTOR' in found_types}} if objType == CUDA_ARRAY3D_DESCRIPTOR: return sizeof(ccuda.CUDA_ARRAY3D_DESCRIPTOR){{endif}} {{if 'struct CUDA_ARRAY_SPARSE_PROPERTIES_st' in found_types}} if objType == CUDA_ARRAY_SPARSE_PROPERTIES_st: return sizeof(ccuda.CUDA_ARRAY_SPARSE_PROPERTIES_st){{endif}} {{if 'CUDA_ARRAY_SPARSE_PROPERTIES_v1' in found_types}} if objType == CUDA_ARRAY_SPARSE_PROPERTIES_v1: return sizeof(ccuda.CUDA_ARRAY_SPARSE_PROPERTIES_v1){{endif}} {{if 'CUDA_ARRAY_SPARSE_PROPERTIES' in found_types}} if objType == CUDA_ARRAY_SPARSE_PROPERTIES: return sizeof(ccuda.CUDA_ARRAY_SPARSE_PROPERTIES){{endif}} {{if 'struct CUDA_ARRAY_MEMORY_REQUIREMENTS_st' in found_types}} if objType == CUDA_ARRAY_MEMORY_REQUIREMENTS_st: return sizeof(ccuda.CUDA_ARRAY_MEMORY_REQUIREMENTS_st){{endif}} {{if 'CUDA_ARRAY_MEMORY_REQUIREMENTS_v1' in found_types}} if objType == CUDA_ARRAY_MEMORY_REQUIREMENTS_v1: return sizeof(ccuda.CUDA_ARRAY_MEMORY_REQUIREMENTS_v1){{endif}} {{if 'CUDA_ARRAY_MEMORY_REQUIREMENTS' in found_types}} if objType == CUDA_ARRAY_MEMORY_REQUIREMENTS: return sizeof(ccuda.CUDA_ARRAY_MEMORY_REQUIREMENTS){{endif}} {{if 'struct CUDA_RESOURCE_DESC_st' in found_types}} if objType == CUDA_RESOURCE_DESC_st: return sizeof(ccuda.CUDA_RESOURCE_DESC_st){{endif}} {{if 'CUDA_RESOURCE_DESC_v1' in found_types}} if objType == CUDA_RESOURCE_DESC_v1: return sizeof(ccuda.CUDA_RESOURCE_DESC_v1){{endif}} {{if 'CUDA_RESOURCE_DESC' in found_types}} if objType == CUDA_RESOURCE_DESC: return sizeof(ccuda.CUDA_RESOURCE_DESC){{endif}} {{if 'struct CUDA_TEXTURE_DESC_st' in found_types}} if objType == CUDA_TEXTURE_DESC_st: return sizeof(ccuda.CUDA_TEXTURE_DESC_st){{endif}} {{if 'CUDA_TEXTURE_DESC_v1' in found_types}} if objType == CUDA_TEXTURE_DESC_v1: return sizeof(ccuda.CUDA_TEXTURE_DESC_v1){{endif}} {{if 'CUDA_TEXTURE_DESC' in found_types}} if objType == CUDA_TEXTURE_DESC: return sizeof(ccuda.CUDA_TEXTURE_DESC){{endif}} {{if 'struct CUDA_RESOURCE_VIEW_DESC_st' in found_types}} if objType == CUDA_RESOURCE_VIEW_DESC_st: return sizeof(ccuda.CUDA_RESOURCE_VIEW_DESC_st){{endif}} {{if 'CUDA_RESOURCE_VIEW_DESC_v1' in found_types}} if objType == CUDA_RESOURCE_VIEW_DESC_v1: return sizeof(ccuda.CUDA_RESOURCE_VIEW_DESC_v1){{endif}} {{if 'CUDA_RESOURCE_VIEW_DESC' in found_types}} if objType == CUDA_RESOURCE_VIEW_DESC: return sizeof(ccuda.CUDA_RESOURCE_VIEW_DESC){{endif}} {{if 'struct CUtensorMap_st' in found_types}} if objType == CUtensorMap_st: return sizeof(ccuda.CUtensorMap_st){{endif}} {{if 'CUtensorMap' in found_types}} if objType == CUtensorMap: return sizeof(ccuda.CUtensorMap){{endif}} {{if 'struct CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_st' in found_types}} if objType == CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_st: return sizeof(ccuda.CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_st){{endif}} {{if 'CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_v1' in found_types}} if objType == CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_v1: return sizeof(ccuda.CUDA_POINTER_ATTRIBUTE_P2P_TOKENS_v1){{endif}} {{if 'CUDA_POINTER_ATTRIBUTE_P2P_TOKENS' in found_types}} if objType == CUDA_POINTER_ATTRIBUTE_P2P_TOKENS: return sizeof(ccuda.CUDA_POINTER_ATTRIBUTE_P2P_TOKENS){{endif}} {{if 'struct CUDA_LAUNCH_PARAMS_st' in found_types}} if objType == CUDA_LAUNCH_PARAMS_st: return sizeof(ccuda.CUDA_LAUNCH_PARAMS_st){{endif}} {{if 'CUDA_LAUNCH_PARAMS_v1' in found_types}} if objType == CUDA_LAUNCH_PARAMS_v1: return sizeof(ccuda.CUDA_LAUNCH_PARAMS_v1){{endif}} {{if 'CUDA_LAUNCH_PARAMS' in found_types}} if objType == CUDA_LAUNCH_PARAMS: return sizeof(ccuda.CUDA_LAUNCH_PARAMS){{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_HANDLE_DESC_st){{endif}} {{if 'CUDA_EXTERNAL_MEMORY_HANDLE_DESC_v1' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_HANDLE_DESC_v1: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_HANDLE_DESC_v1){{endif}} {{if 'CUDA_EXTERNAL_MEMORY_HANDLE_DESC' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_HANDLE_DESC: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_HANDLE_DESC){{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_BUFFER_DESC_st' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_BUFFER_DESC_st: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_BUFFER_DESC_st){{endif}} {{if 'CUDA_EXTERNAL_MEMORY_BUFFER_DESC_v1' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_BUFFER_DESC_v1: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_BUFFER_DESC_v1){{endif}} {{if 'CUDA_EXTERNAL_MEMORY_BUFFER_DESC' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_BUFFER_DESC: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_BUFFER_DESC){{endif}} {{if 'struct CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_st' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_st: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_st){{endif}} {{if 'CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_v1' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_v1: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC_v1){{endif}} {{if 'CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC' in found_types}} if objType == CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC: return sizeof(ccuda.CUDA_EXTERNAL_MEMORY_MIPMAPPED_ARRAY_DESC){{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_st){{endif}} {{if 'CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_v1' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_v1: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC_v1){{endif}} {{if 'CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_HANDLE_DESC){{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_st){{endif}} {{if 'CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_v1' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_v1: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS_v1){{endif}} {{if 'CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_SIGNAL_PARAMS){{endif}} {{if 'struct CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_st){{endif}} {{if 'CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_v1' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_v1: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS_v1){{endif}} {{if 'CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS' in found_types}} if objType == CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS: return sizeof(ccuda.CUDA_EXTERNAL_SEMAPHORE_WAIT_PARAMS){{endif}} {{if 'struct CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_st' in found_types}} if objType == CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_st: return sizeof(ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_st){{endif}} {{if 'CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v1' in found_types}} if objType == CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v1: return sizeof(ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v1){{endif}} {{if 'CUDA_EXT_SEM_SIGNAL_NODE_PARAMS' in found_types}} if objType == CUDA_EXT_SEM_SIGNAL_NODE_PARAMS: return sizeof(ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS){{endif}} {{if 'struct CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2_st' in found_types}} if objType == CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2_st: return sizeof(ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2_st){{endif}} {{if 'CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2' in found_types}} if objType == CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2: return sizeof(ccuda.CUDA_EXT_SEM_SIGNAL_NODE_PARAMS_v2){{endif}} {{if 'struct CUDA_EXT_SEM_WAIT_NODE_PARAMS_st' in found_types}} if objType == CUDA_EXT_SEM_WAIT_NODE_PARAMS_st: return sizeof(ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS_st){{endif}} {{if 'CUDA_EXT_SEM_WAIT_NODE_PARAMS_v1' in found_types}} if objType == CUDA_EXT_SEM_WAIT_NODE_PARAMS_v1: return sizeof(ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS_v1){{endif}} {{if 'CUDA_EXT_SEM_WAIT_NODE_PARAMS' in found_types}} if objType == CUDA_EXT_SEM_WAIT_NODE_PARAMS: return sizeof(ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS){{endif}} {{if 'struct CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2_st' in found_types}} if objType == CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2_st: return sizeof(ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2_st){{endif}} {{if 'CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2' in found_types}} if objType == CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2: return sizeof(ccuda.CUDA_EXT_SEM_WAIT_NODE_PARAMS_v2){{endif}} {{if 'CUmemGenericAllocationHandle_v1' in found_types}} if objType == CUmemGenericAllocationHandle_v1: return sizeof(ccuda.CUmemGenericAllocationHandle_v1){{endif}} {{if 'CUmemGenericAllocationHandle' in found_types}} if objType == CUmemGenericAllocationHandle: return sizeof(ccuda.CUmemGenericAllocationHandle){{endif}} {{if 'struct CUarrayMapInfo_st' in found_types}} if objType == CUarrayMapInfo_st: return sizeof(ccuda.CUarrayMapInfo_st){{endif}} {{if 'CUarrayMapInfo_v1' in found_types}} if objType == CUarrayMapInfo_v1: return sizeof(ccuda.CUarrayMapInfo_v1){{endif}} {{if 'CUarrayMapInfo' in found_types}} if objType == CUarrayMapInfo: return sizeof(ccuda.CUarrayMapInfo){{endif}} {{if 'struct CUmemLocation_st' in found_types}} if objType == CUmemLocation_st: return sizeof(ccuda.CUmemLocation_st){{endif}} {{if 'CUmemLocation_v1' in found_types}} if objType == CUmemLocation_v1: return sizeof(ccuda.CUmemLocation_v1){{endif}} {{if 'CUmemLocation' in found_types}} if objType == CUmemLocation: return sizeof(ccuda.CUmemLocation){{endif}} {{if 'struct CUmemAllocationProp_st' in found_types}} if objType == CUmemAllocationProp_st: return sizeof(ccuda.CUmemAllocationProp_st){{endif}} {{if 'CUmemAllocationProp_v1' in found_types}} if objType == CUmemAllocationProp_v1: return sizeof(ccuda.CUmemAllocationProp_v1){{endif}} {{if 'CUmemAllocationProp' in found_types}} if objType == CUmemAllocationProp: return sizeof(ccuda.CUmemAllocationProp){{endif}} {{if 'struct CUmulticastObjectProp_st' in found_types}} if objType == CUmulticastObjectProp_st: return sizeof(ccuda.CUmulticastObjectProp_st){{endif}} {{if 'CUmulticastObjectProp_v1' in found_types}} if objType == CUmulticastObjectProp_v1: return sizeof(ccuda.CUmulticastObjectProp_v1){{endif}} {{if 'CUmulticastObjectProp' in found_types}} if objType == CUmulticastObjectProp: return sizeof(ccuda.CUmulticastObjectProp){{endif}} {{if 'struct CUmemAccessDesc_st' in found_types}} if objType == CUmemAccessDesc_st: return sizeof(ccuda.CUmemAccessDesc_st){{endif}} {{if 'CUmemAccessDesc_v1' in found_types}} if objType == CUmemAccessDesc_v1: return sizeof(ccuda.CUmemAccessDesc_v1){{endif}} {{if 'CUmemAccessDesc' in found_types}} if objType == CUmemAccessDesc: return sizeof(ccuda.CUmemAccessDesc){{endif}} {{if 'struct CUgraphExecUpdateResultInfo_st' in found_types}} if objType == CUgraphExecUpdateResultInfo_st: return sizeof(ccuda.CUgraphExecUpdateResultInfo_st){{endif}} {{if 'CUgraphExecUpdateResultInfo_v1' in found_types}} if objType == CUgraphExecUpdateResultInfo_v1: return sizeof(ccuda.CUgraphExecUpdateResultInfo_v1){{endif}} {{if 'CUgraphExecUpdateResultInfo' in found_types}} if objType == CUgraphExecUpdateResultInfo: return sizeof(ccuda.CUgraphExecUpdateResultInfo){{endif}} {{if 'struct CUmemPoolProps_st' in found_types}} if objType == CUmemPoolProps_st: return sizeof(ccuda.CUmemPoolProps_st){{endif}} {{if 'CUmemPoolProps_v1' in found_types}} if objType == CUmemPoolProps_v1: return sizeof(ccuda.CUmemPoolProps_v1){{endif}} {{if 'CUmemPoolProps' in found_types}} if objType == CUmemPoolProps: return sizeof(ccuda.CUmemPoolProps){{endif}} {{if 'struct CUmemPoolPtrExportData_st' in found_types}} if objType == CUmemPoolPtrExportData_st: return sizeof(ccuda.CUmemPoolPtrExportData_st){{endif}} {{if 'CUmemPoolPtrExportData_v1' in found_types}} if objType == CUmemPoolPtrExportData_v1: return sizeof(ccuda.CUmemPoolPtrExportData_v1){{endif}} {{if 'CUmemPoolPtrExportData' in found_types}} if objType == CUmemPoolPtrExportData: return sizeof(ccuda.CUmemPoolPtrExportData){{endif}} {{if 'struct CUDA_MEM_ALLOC_NODE_PARAMS_v1_st' in found_types}} if objType == CUDA_MEM_ALLOC_NODE_PARAMS_v1_st: return sizeof(ccuda.CUDA_MEM_ALLOC_NODE_PARAMS_v1_st){{endif}} {{if 'CUDA_MEM_ALLOC_NODE_PARAMS_v1' in found_types}} if objType == CUDA_MEM_ALLOC_NODE_PARAMS_v1: return sizeof(ccuda.CUDA_MEM_ALLOC_NODE_PARAMS_v1){{endif}} {{if 'CUDA_MEM_ALLOC_NODE_PARAMS' in found_types}} if objType == CUDA_MEM_ALLOC_NODE_PARAMS: return sizeof(ccuda.CUDA_MEM_ALLOC_NODE_PARAMS){{endif}} {{if 'struct CUDA_MEM_ALLOC_NODE_PARAMS_v2_st' in found_types}} if objType == CUDA_MEM_ALLOC_NODE_PARAMS_v2_st: return sizeof(ccuda.CUDA_MEM_ALLOC_NODE_PARAMS_v2_st){{endif}} {{if 'CUDA_MEM_ALLOC_NODE_PARAMS_v2' in found_types}} if objType == CUDA_MEM_ALLOC_NODE_PARAMS_v2: return sizeof(ccuda.CUDA_MEM_ALLOC_NODE_PARAMS_v2){{endif}} {{if 'struct CUDA_MEM_FREE_NODE_PARAMS_st' in found_types}} if objType == CUDA_MEM_FREE_NODE_PARAMS_st: return sizeof(ccuda.CUDA_MEM_FREE_NODE_PARAMS_st){{endif}} {{if 'CUDA_MEM_FREE_NODE_PARAMS' in found_types}} if objType == CUDA_MEM_FREE_NODE_PARAMS: return sizeof(ccuda.CUDA_MEM_FREE_NODE_PARAMS){{endif}} {{if 'struct CUDA_CHILD_GRAPH_NODE_PARAMS_st' in found_types}} if objType == CUDA_CHILD_GRAPH_NODE_PARAMS_st: return sizeof(ccuda.CUDA_CHILD_GRAPH_NODE_PARAMS_st){{endif}} {{if 'CUDA_CHILD_GRAPH_NODE_PARAMS' in found_types}} if objType == CUDA_CHILD_GRAPH_NODE_PARAMS: return sizeof(ccuda.CUDA_CHILD_GRAPH_NODE_PARAMS){{endif}} {{if 'struct CUDA_EVENT_RECORD_NODE_PARAMS_st' in found_types}} if objType == CUDA_EVENT_RECORD_NODE_PARAMS_st: return sizeof(ccuda.CUDA_EVENT_RECORD_NODE_PARAMS_st){{endif}} {{if 'CUDA_EVENT_RECORD_NODE_PARAMS' in found_types}} if objType == CUDA_EVENT_RECORD_NODE_PARAMS: return sizeof(ccuda.CUDA_EVENT_RECORD_NODE_PARAMS){{endif}} {{if 'struct CUDA_EVENT_WAIT_NODE_PARAMS_st' in found_types}} if objType == CUDA_EVENT_WAIT_NODE_PARAMS_st: return sizeof(ccuda.CUDA_EVENT_WAIT_NODE_PARAMS_st){{endif}} {{if 'CUDA_EVENT_WAIT_NODE_PARAMS' in found_types}} if objType == CUDA_EVENT_WAIT_NODE_PARAMS: return sizeof(ccuda.CUDA_EVENT_WAIT_NODE_PARAMS){{endif}} {{if 'struct CUgraphNodeParams_st' in found_types}} if objType == CUgraphNodeParams_st: return sizeof(ccuda.CUgraphNodeParams_st){{endif}} {{if 'CUgraphNodeParams' in found_types}} if objType == CUgraphNodeParams: return sizeof(ccuda.CUgraphNodeParams){{endif}} {{if 'CUdevResourceDesc' in found_types}} if objType == CUdevResourceDesc: return sizeof(ccuda.CUdevResourceDesc){{endif}} {{if 'struct CUdevSmResource_st' in found_types}} if objType == CUdevSmResource_st: return sizeof(ccuda.CUdevSmResource_st){{endif}} {{if 'CUdevSmResource' in found_types}} if objType == CUdevSmResource: return sizeof(ccuda.CUdevSmResource){{endif}} {{if 'struct CUdevResource_st' in found_types}} if objType == CUdevResource_st: return sizeof(ccuda.CUdevResource_st){{endif}} {{if 'struct CUdevResource_st' in found_types}} if objType == CUdevResource_v1: return sizeof(ccuda.CUdevResource_v1){{endif}} {{if 'struct CUdevResource_st' in found_types}} if objType == CUdevResource: return sizeof(ccuda.CUdevResource){{endif}} {{if True}} if objType == CUeglFrame_st: return sizeof(ccuda.CUeglFrame_st){{endif}} {{if True}} if objType == CUeglFrame_v1: return sizeof(ccuda.CUeglFrame_v1){{endif}} {{if True}} if objType == CUeglFrame: return sizeof(ccuda.CUeglFrame){{endif}} {{if True}} if objType == CUeglStreamConnection: return sizeof(ccuda.CUeglStreamConnection){{endif}} {{if True}} if objType == GLenum: return sizeof(ccuda.GLenum){{endif}} {{if True}} if objType == GLuint: return sizeof(ccuda.GLuint){{endif}} {{if True}} if objType == EGLImageKHR: return sizeof(ccuda.EGLImageKHR){{endif}} {{if True}} if objType == EGLStreamKHR: return sizeof(ccuda.EGLStreamKHR){{endif}} {{if True}} if objType == EGLint: return sizeof(ccuda.EGLint){{endif}} {{if True}} if objType == EGLSyncKHR: return sizeof(ccuda.EGLSyncKHR){{endif}} {{if True}} if objType == VdpDevice: return sizeof(ccuda.VdpDevice){{endif}} {{if True}} if objType == VdpGetProcAddress: return sizeof(ccuda.VdpGetProcAddress){{endif}} {{if True}} if objType == VdpVideoSurface: return sizeof(ccuda.VdpVideoSurface){{endif}} {{if True}} if objType == VdpOutputSurface: return sizeof(ccuda.VdpOutputSurface){{endif}} raise TypeError("Unknown type: " + str(objType))