Skip to content

Attempt to fix #1663 #2277

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
mdboom wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Attempt to fix #1663 #2277

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
28 changes: 14 additions & 14 deletions cuda_bindings/cuda/bindings/_internal/cudla.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE

# This code was automatically generated across versions from 1.5.0 to 13.3.0, generator version 0.3.1.dev1719+g565f73f4e. Do not modify it directly.
# This code was automatically generated across versions from 1.5.0 to 13.3.0, generator version 0.3.1.dev1779+ga8cc71818.d20260626. Do not modify it directly.

from ..cycudla cimport *

Expand All @@ -10,16 +10,16 @@ from ..cycudla cimport *
# Wrapper functions
###############################################################################

cdef cudlaStatus _cudlaGetVersion(uint64_t* const version) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaDeviceGetCount(uint64_t* const pNumDevices) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaCreateDevice(const uint64_t device, cudlaDevHandle* const devHandle, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaMemRegister(const cudlaDevHandle devHandle, const uint64_t* const ptr, const size_t size, uint64_t** const devPtr, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaModuleLoadFromMemory(const cudlaDevHandle devHandle, const uint8_t* const pModule, const size_t moduleSize, cudlaModule* const hModule, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaModuleGetAttributes(const cudlaModule hModule, const cudlaModuleAttributeType attrType, cudlaModuleAttribute* const attribute) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaModuleUnload(const cudlaModule hModule, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaSubmitTask(const cudlaDevHandle devHandle, const cudlaTask* const ptrToTasks, const uint32_t numTasks, void* const stream, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaDeviceGetAttribute(const cudlaDevHandle devHandle, const cudlaDevAttributeType attrib, cudlaDevAttribute* const pAttribute) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaMemUnregister(const cudlaDevHandle devHandle, const uint64_t* const devPtr) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaGetLastError(const cudlaDevHandle devHandle) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaDestroyDevice(const cudlaDevHandle devHandle) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaSetTaskTimeoutInMs(const cudlaDevHandle devHandle, const uint32_t timeout) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaGetVersion(uint64_t* const version) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaDeviceGetCount(uint64_t* const pNumDevices) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaCreateDevice(const uint64_t device, cudlaDevHandle* const devHandle, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaMemRegister(const cudlaDevHandle devHandle, const uint64_t* const ptr, const size_t size, uint64_t** const devPtr, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaModuleLoadFromMemory(const cudlaDevHandle devHandle, const uint8_t* const pModule, const size_t moduleSize, cudlaModule* const hModule, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaModuleGetAttributes(const cudlaModule hModule, const cudlaModuleAttributeType attrType, cudlaModuleAttribute* const attribute) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaModuleUnload(const cudlaModule hModule, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaSubmitTask(const cudlaDevHandle devHandle, const cudlaTask* const ptrToTasks, const uint32_t numTasks, void* const stream, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaDeviceGetAttribute(const cudlaDevHandle devHandle, const cudlaDevAttributeType attrib, cudlaDevAttribute* const pAttribute) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaMemUnregister(const cudlaDevHandle devHandle, const uint64_t* const devPtr) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaGetLastError(const cudlaDevHandle devHandle) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaDestroyDevice(const cudlaDevHandle devHandle) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
cdef cudlaStatus _cudlaSetTaskTimeoutInMs(const cudlaDevHandle devHandle, const uint32_t timeout) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil
28 changes: 14 additions & 14 deletions cuda_bindings/cuda/bindings/_internal/cudla_linux.pyx
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
# SPDX-FileCopyrightText: Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE

# This code was automatically generated across versions from 1.5.0 to 13.3.0, generator version 0.3.1.dev1719+g565f73f4e. Do not modify it directly.
# This code was automatically generated across versions from 1.5.0 to 13.3.0, generator version 0.3.1.dev1779+ga8cc71818.d20260626. Do not modify it directly.

from libc.stdint cimport intptr_t, uintptr_t

Expand Down Expand Up @@ -255,7 +255,7 @@ cpdef _inspect_function_pointer(str name):
# Wrapper functions
###############################################################################

cdef cudlaStatus _cudlaGetVersion(uint64_t* const version) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaGetVersion(uint64_t* const version) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaGetVersion
_check_or_init_cudla()
if __cudlaGetVersion == NULL:
Expand All @@ -265,7 +265,7 @@ cdef cudlaStatus _cudlaGetVersion(uint64_t* const version) except?_CUDLASTATUS_I
version)


cdef cudlaStatus _cudlaDeviceGetCount(uint64_t* const pNumDevices) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaDeviceGetCount(uint64_t* const pNumDevices) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaDeviceGetCount
_check_or_init_cudla()
if __cudlaDeviceGetCount == NULL:
Expand All @@ -275,7 +275,7 @@ cdef cudlaStatus _cudlaDeviceGetCount(uint64_t* const pNumDevices) except?_CUDLA
pNumDevices)


cdef cudlaStatus _cudlaCreateDevice(const uint64_t device, cudlaDevHandle* const devHandle, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaCreateDevice(const uint64_t device, cudlaDevHandle* const devHandle, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaCreateDevice
_check_or_init_cudla()
if __cudlaCreateDevice == NULL:
Expand All @@ -285,7 +285,7 @@ cdef cudlaStatus _cudlaCreateDevice(const uint64_t device, cudlaDevHandle* const
device, devHandle, flags)


cdef cudlaStatus _cudlaMemRegister(const cudlaDevHandle devHandle, const uint64_t* const ptr, const size_t size, uint64_t** const devPtr, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaMemRegister(const cudlaDevHandle devHandle, const uint64_t* const ptr, const size_t size, uint64_t** const devPtr, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaMemRegister
_check_or_init_cudla()
if __cudlaMemRegister == NULL:
Expand All @@ -295,7 +295,7 @@ cdef cudlaStatus _cudlaMemRegister(const cudlaDevHandle devHandle, const uint64_
devHandle, ptr, size, devPtr, flags)


cdef cudlaStatus _cudlaModuleLoadFromMemory(const cudlaDevHandle devHandle, const uint8_t* const pModule, const size_t moduleSize, cudlaModule* const hModule, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaModuleLoadFromMemory(const cudlaDevHandle devHandle, const uint8_t* const pModule, const size_t moduleSize, cudlaModule* const hModule, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaModuleLoadFromMemory
_check_or_init_cudla()
if __cudlaModuleLoadFromMemory == NULL:
Expand All @@ -305,7 +305,7 @@ cdef cudlaStatus _cudlaModuleLoadFromMemory(const cudlaDevHandle devHandle, cons
devHandle, pModule, moduleSize, hModule, flags)


cdef cudlaStatus _cudlaModuleGetAttributes(const cudlaModule hModule, const cudlaModuleAttributeType attrType, cudlaModuleAttribute* const attribute) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaModuleGetAttributes(const cudlaModule hModule, const cudlaModuleAttributeType attrType, cudlaModuleAttribute* const attribute) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaModuleGetAttributes
_check_or_init_cudla()
if __cudlaModuleGetAttributes == NULL:
Expand All @@ -315,7 +315,7 @@ cdef cudlaStatus _cudlaModuleGetAttributes(const cudlaModule hModule, const cudl
hModule, attrType, attribute)


cdef cudlaStatus _cudlaModuleUnload(const cudlaModule hModule, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaModuleUnload(const cudlaModule hModule, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaModuleUnload
_check_or_init_cudla()
if __cudlaModuleUnload == NULL:
Expand All @@ -325,7 +325,7 @@ cdef cudlaStatus _cudlaModuleUnload(const cudlaModule hModule, const uint32_t fl
hModule, flags)


cdef cudlaStatus _cudlaSubmitTask(const cudlaDevHandle devHandle, const cudlaTask* const ptrToTasks, const uint32_t numTasks, void* const stream, const uint32_t flags) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaSubmitTask(const cudlaDevHandle devHandle, const cudlaTask* const ptrToTasks, const uint32_t numTasks, void* const stream, const uint32_t flags) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaSubmitTask
_check_or_init_cudla()
if __cudlaSubmitTask == NULL:
Expand All @@ -335,7 +335,7 @@ cdef cudlaStatus _cudlaSubmitTask(const cudlaDevHandle devHandle, const cudlaTas
devHandle, ptrToTasks, numTasks, stream, flags)


cdef cudlaStatus _cudlaDeviceGetAttribute(const cudlaDevHandle devHandle, const cudlaDevAttributeType attrib, cudlaDevAttribute* const pAttribute) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaDeviceGetAttribute(const cudlaDevHandle devHandle, const cudlaDevAttributeType attrib, cudlaDevAttribute* const pAttribute) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaDeviceGetAttribute
_check_or_init_cudla()
if __cudlaDeviceGetAttribute == NULL:
Expand All @@ -345,7 +345,7 @@ cdef cudlaStatus _cudlaDeviceGetAttribute(const cudlaDevHandle devHandle, const
devHandle, attrib, pAttribute)


cdef cudlaStatus _cudlaMemUnregister(const cudlaDevHandle devHandle, const uint64_t* const devPtr) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaMemUnregister(const cudlaDevHandle devHandle, const uint64_t* const devPtr) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaMemUnregister
_check_or_init_cudla()
if __cudlaMemUnregister == NULL:
Expand All @@ -355,7 +355,7 @@ cdef cudlaStatus _cudlaMemUnregister(const cudlaDevHandle devHandle, const uint6
devHandle, devPtr)


cdef cudlaStatus _cudlaGetLastError(const cudlaDevHandle devHandle) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaGetLastError(const cudlaDevHandle devHandle) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaGetLastError
_check_or_init_cudla()
if __cudlaGetLastError == NULL:
Expand All @@ -365,7 +365,7 @@ cdef cudlaStatus _cudlaGetLastError(const cudlaDevHandle devHandle) except?_CUDL
devHandle)


cdef cudlaStatus _cudlaDestroyDevice(const cudlaDevHandle devHandle) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaDestroyDevice(const cudlaDevHandle devHandle) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaDestroyDevice
_check_or_init_cudla()
if __cudlaDestroyDevice == NULL:
Expand All @@ -375,7 +375,7 @@ cdef cudlaStatus _cudlaDestroyDevice(const cudlaDevHandle devHandle) except?_CUD
devHandle)


cdef cudlaStatus _cudlaSetTaskTimeoutInMs(const cudlaDevHandle devHandle, const uint32_t timeout) except?_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
cdef cudlaStatus _cudlaSetTaskTimeoutInMs(const cudlaDevHandle devHandle, const uint32_t timeout) except?<cudlaStatus>_CUDLASTATUS_INTERNAL_LOADING_ERROR nogil:
global __cudlaSetTaskTimeoutInMs
_check_or_init_cudla()
if __cudlaSetTaskTimeoutInMs == NULL:
Expand Down
5 changes: 4 additions & 1 deletion cuda_bindings/cuda/bindings/_internal/driver.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
#
# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
#
# This code was automatically generated across versions from 12.9.0 to 13.3.0, generator version 0.3.1.dev1622+g48467ab08.d20260421. Do not modify it directly.
# This code was automatically generated across versions from 12.9.0 to 13.3.0, generator version 0.3.1.dev1779+ga8cc71818.d20260626. Do not modify it directly.

from ..cydriver cimport *

Expand Down Expand Up @@ -36,6 +36,8 @@ cdef CUresult _cuDevicePrimaryCtxRelease_v2(CUdevice dev) except ?CUDA_ERROR_NOT
cdef CUresult _cuDevicePrimaryCtxSetFlags_v2(CUdevice dev, unsigned int flags) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuDevicePrimaryCtxGetState(CUdevice dev, unsigned int* flags, int* active) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuDevicePrimaryCtxReset_v2(CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuCtxCreate_v2(CUcontext* pctx, unsigned int flags, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuCtxCreate_v3(CUcontext* pctx, CUexecAffinityParam* paramsArray, int numParams, unsigned int flags, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuCtxCreate_v4(CUcontext* pctx, CUctxCreateParams* ctxCreateParams, unsigned int flags, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuCtxDestroy_v2(CUcontext ctx) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuCtxPushCurrent_v2(CUcontext ctx) except ?CUDA_ERROR_NOT_FOUND nogil
Expand Down Expand Up @@ -224,6 +226,7 @@ cdef CUresult _cuStreamBeginCaptureToGraph(CUstream hStream, CUgraph hGraph, con
cdef CUresult _cuThreadExchangeStreamCaptureMode(CUstreamCaptureMode* mode) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuStreamEndCapture(CUstream hStream, CUgraph* phGraph) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuStreamIsCapturing(CUstream hStream, CUstreamCaptureStatus* captureStatus) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuStreamGetCaptureInfo_v2(CUstream hStream, CUstreamCaptureStatus* captureStatus_out, cuuint64_t* id_out, CUgraph* graph_out, const CUgraphNode** dependencies_out, size_t* numDependencies_out) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuStreamGetCaptureInfo_v3(CUstream hStream, CUstreamCaptureStatus* captureStatus_out, cuuint64_t* id_out, CUgraph* graph_out, const CUgraphNode** dependencies_out, const CUgraphEdgeData** edgeData_out, size_t* numDependencies_out) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuStreamUpdateCaptureDependencies_v2(CUstream hStream, CUgraphNode* dependencies, const CUgraphEdgeData* dependencyData, size_t numDependencies, unsigned int flags) except ?CUDA_ERROR_NOT_FOUND nogil
cdef CUresult _cuStreamAttachMemAsync(CUstream hStream, CUdeviceptr dptr, size_t length, unsigned int flags) except ?CUDA_ERROR_NOT_FOUND nogil
Expand Down
Loading
Loading