Skip to content

Add missing driver & runtime bindings for functions new in CTK 13.1.0 #1337

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

Merged
rwgk merged 7 commits into from
Dec 9, 2025
Merged

Add missing driver & runtime bindings for functions new in CTK 13.1.0 #1337

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
af5c0b7
Add driver bindings for new functions in CTK 13.1
rwgk Dec 6, 2025
7d9cee6
Add runtime bindings for new functions in CTK 13.1
rwgk Dec 6, 2025
a28aa9f
Fix UnicodeDecodeError on Windows by using UTF-8 encoding in setup.py
rwgk Dec 6, 2025
cbd7d9a
Add Phase 1 unit tests for CUDA 13.1+ graph/node ID functions
rwgk Dec 7, 2025
d22dcba
Add Phase 2 unit tests for CUDA 13.1+ resource management functions
rwgk Dec 7, 2025
1520d38
Add Phase 3 unit tests for CUDA 13.1+ complex resource functions
rwgk Dec 7, 2025
056861a
Remove # Phase 1, 2, 3 comments (they will be more distracting than h…
rwgk Dec 7, 2025
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
35 changes: 35 additions & 0 deletions cuda_bindings/cuda/bindings/_bindings/cydriver.pxd.in
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1609,6 +1609,31 @@ cdef CUresult _cuGraphNodeFindInClone(CUgraphNode* phNode, CUgraphNode hOriginal
cdef CUresult _cuGraphNodeGetType(CUgraphNode hNode, CUgraphNodeType* typename) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuGraphNodeGetContainingGraph' in found_functions}}

cdef CUresult _cuGraphNodeGetContainingGraph(CUgraphNode hNode, CUgraph* phGraph) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuGraphNodeGetLocalId' in found_functions}}

cdef CUresult _cuGraphNodeGetLocalId(CUgraphNode hNode, unsigned int* nodeId) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuGraphNodeGetToolsId' in found_functions}}

cdef CUresult _cuGraphNodeGetToolsId(CUgraphNode hNode, unsigned long long* toolsNodeId) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuGraphGetId' in found_functions}}

cdef CUresult _cuGraphGetId(CUgraph hGraph, unsigned int* graphId) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuGraphExecGetId' in found_functions}}

cdef CUresult _cuGraphExecGetId(CUgraphExec hGraphExec, unsigned int* graphId) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuGraphGetNodes' in found_functions}}

cdef CUresult _cuGraphGetNodes(CUgraph hGraph, CUgraphNode* nodes, size_t* numNodes) except ?CUDA_ERROR_NOT_FOUND nogil
Expand Down Expand Up @@ -2174,6 +2199,11 @@ cdef CUresult _cuGreenCtxGetDevResource(CUgreenCtx hCtx, CUdevResource* resource
cdef CUresult _cuDevSmResourceSplitByCount(CUdevResource* result, unsigned int* nbGroups, const CUdevResource* input, CUdevResource* remainder, unsigned int flags, unsigned int minCount) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuDevSmResourceSplit' in found_functions}}

cdef CUresult _cuDevSmResourceSplit(CUdevResource* result, unsigned int nbGroups, const CUdevResource* input, CUdevResource* remainder, unsigned int flags, CU_DEV_SM_RESOURCE_GROUP_PARAMS* groupParams) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuDevResourceGenerateDesc' in found_functions}}

cdef CUresult _cuDevResourceGenerateDesc(CUdevResourceDesc* phDesc, CUdevResource* resources, unsigned int nbResources) except ?CUDA_ERROR_NOT_FOUND nogil
Expand Down Expand Up @@ -2204,6 +2234,11 @@ cdef CUresult _cuGreenCtxStreamCreate(CUstream* phStream, CUgreenCtx greenCtx, u
cdef CUresult _cuGreenCtxGetId(CUgreenCtx greenCtx, unsigned long long* greenCtxId) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuStreamGetDevResource' in found_functions}}

cdef CUresult _cuStreamGetDevResource(CUstream hStream, CUdevResource* resource, CUdevResourceType typename) except ?CUDA_ERROR_NOT_FOUND nogil
{{endif}}

{{if 'cuLogsRegisterCallback' in found_functions}}

cdef CUresult _cuLogsRegisterCallback(CUlogsCallback callbackFunc, void* userData, CUlogsCallbackHandle* callback_out) except ?CUDA_ERROR_NOT_FOUND nogil
Expand Down
236 changes: 236 additions & 0 deletions cuda_bindings/cuda/bindings/_bindings/cydriver.pyx.in
View file
Open in desktop

Large diffs are not rendered by default.

85 changes: 85 additions & 0 deletions cuda_bindings/cuda/bindings/_bindings/cyruntime.pxd.in
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -1186,6 +1186,31 @@ cdef cudaError_t _cudaGraphNodeFindInClone(cudaGraphNode_t* pNode, cudaGraphNode
cdef cudaError_t _cudaGraphNodeGetType(cudaGraphNode_t node, cudaGraphNodeType* pType) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGraphNodeGetContainingGraph' in found_functions}}

cdef cudaError_t _cudaGraphNodeGetContainingGraph(cudaGraphNode_t hNode, cudaGraph_t* phGraph) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGraphNodeGetLocalId' in found_functions}}

cdef cudaError_t _cudaGraphNodeGetLocalId(cudaGraphNode_t hNode, unsigned int* nodeId) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGraphNodeGetToolsId' in found_functions}}

cdef cudaError_t _cudaGraphNodeGetToolsId(cudaGraphNode_t hNode, unsigned long long* toolsNodeId) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGraphGetId' in found_functions}}

cdef cudaError_t _cudaGraphGetId(cudaGraph_t hGraph, unsigned int* graphID) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGraphExecGetId' in found_functions}}

cdef cudaError_t _cudaGraphExecGetId(cudaGraphExec_t hGraphExec, unsigned int* graphID) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGraphGetNodes' in found_functions}}

cdef cudaError_t _cudaGraphGetNodes(cudaGraph_t graph, cudaGraphNode_t* nodes, size_t* numNodes) except ?cudaErrorCallRequiresNewerDriver nogil
Expand Down Expand Up @@ -1381,6 +1406,11 @@ cdef cudaError_t _cudaGraphExecNodeSetParams(cudaGraphExec_t graphExec, cudaGrap
cdef cudaError_t _cudaGraphConditionalHandleCreate(cudaGraphConditionalHandle* pHandle_out, cudaGraph_t graph, unsigned int defaultLaunchValue, unsigned int flags) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGraphConditionalHandleCreate_v2' in found_functions}}

cdef cudaError_t _cudaGraphConditionalHandleCreate_v2(cudaGraphConditionalHandle* pHandle_out, cudaGraph_t graph, cudaExecutionContext_t ctx, unsigned int defaultLaunchValue, unsigned int flags) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGetDriverEntryPoint' in found_functions}}

cdef cudaError_t _cudaGetDriverEntryPoint(const char* symbol, void** funcPtr, unsigned long long flags, cudaDriverEntryPointQueryResult* driverStatus) except ?cudaErrorCallRequiresNewerDriver nogil
Expand Down Expand Up @@ -1441,16 +1471,66 @@ cdef cudaError_t _cudaLibraryEnumerateKernels(cudaKernel_t* kernels, unsigned in
cdef cudaError_t _cudaKernelSetAttributeForDevice(cudaKernel_t kernel, cudaFuncAttribute attr, int value, int device) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaDeviceGetDevResource' in found_functions}}

cdef cudaError_t _cudaDeviceGetDevResource(int device, cudaDevResource* resource, cudaDevResourceType typename) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaDevSmResourceSplitByCount' in found_functions}}

cdef cudaError_t _cudaDevSmResourceSplitByCount(cudaDevResource* result, unsigned int* nbGroups, const cudaDevResource* input, cudaDevResource* remaining, unsigned int flags, unsigned int minCount) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaDevSmResourceSplit' in found_functions}}

cdef cudaError_t _cudaDevSmResourceSplit(cudaDevResource* result, unsigned int nbGroups, const cudaDevResource* input, cudaDevResource* remainder, unsigned int flags, cudaDevSmResourceGroupParams* groupParams) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaDevResourceGenerateDesc' in found_functions}}

cdef cudaError_t _cudaDevResourceGenerateDesc(cudaDevResourceDesc_t* phDesc, cudaDevResource* resources, unsigned int nbResources) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGreenCtxCreate' in found_functions}}

cdef cudaError_t _cudaGreenCtxCreate(cudaExecutionContext_t* phCtx, cudaDevResourceDesc_t desc, int device, unsigned int flags) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaExecutionCtxDestroy' in found_functions}}

cdef cudaError_t _cudaExecutionCtxDestroy(cudaExecutionContext_t ctx) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaExecutionCtxGetDevResource' in found_functions}}

cdef cudaError_t _cudaExecutionCtxGetDevResource(cudaExecutionContext_t ctx, cudaDevResource* resource, cudaDevResourceType typename) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaExecutionCtxGetDevice' in found_functions}}

cdef cudaError_t _cudaExecutionCtxGetDevice(int* device, cudaExecutionContext_t ctx) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaExecutionCtxGetId' in found_functions}}

cdef cudaError_t _cudaExecutionCtxGetId(cudaExecutionContext_t ctx, unsigned long long* ctxId) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaExecutionCtxStreamCreate' in found_functions}}

cdef cudaError_t _cudaExecutionCtxStreamCreate(cudaStream_t* phStream, cudaExecutionContext_t ctx, unsigned int flags, int priority) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaExecutionCtxSynchronize' in found_functions}}

cdef cudaError_t _cudaExecutionCtxSynchronize(cudaExecutionContext_t ctx) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaStreamGetDevResource' in found_functions}}

cdef cudaError_t _cudaStreamGetDevResource(cudaStream_t hStream, cudaDevResource* resource, cudaDevResourceType typename) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaExecutionCtxRecordEvent' in found_functions}}

cdef cudaError_t _cudaExecutionCtxRecordEvent(cudaExecutionContext_t ctx, cudaEvent_t event) except ?cudaErrorCallRequiresNewerDriver nogil
Expand All @@ -1461,6 +1541,11 @@ cdef cudaError_t _cudaExecutionCtxRecordEvent(cudaExecutionContext_t ctx, cudaEv
cdef cudaError_t _cudaExecutionCtxWaitEvent(cudaExecutionContext_t ctx, cudaEvent_t event) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaDeviceGetExecutionCtx' in found_functions}}

cdef cudaError_t _cudaDeviceGetExecutionCtx(cudaExecutionContext_t* ctx, int device) except ?cudaErrorCallRequiresNewerDriver nogil
{{endif}}

{{if 'cudaGetExportTable' in found_functions}}

cdef cudaError_t _cudaGetExportTable(const void** ppExportTable, const cudaUUID_t* pExportTableId) except ?cudaErrorCallRequiresNewerDriver nogil
Expand Down
153 changes: 153 additions & 0 deletions cuda_bindings/cuda/bindings/_bindings/cyruntime.pyx.in
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2150,6 +2150,51 @@ cdef cudaError_t _cudaGraphNodeGetType(cudaGraphNode_t node, cudaGraphNodeType*
return cudaGraphNodeGetType(node, pType)
{{endif}}

{{if 'cudaGraphNodeGetContainingGraph' in found_functions}}

cdef cudaError_t _cudaGraphNodeGetContainingGraph(cudaGraphNode_t hNode, cudaGraph_t* phGraph) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaGraphNodeGetContainingGraph(hNode, phGraph)
return cudaGraphNodeGetContainingGraph(hNode, phGraph)
{{endif}}

{{if 'cudaGraphNodeGetLocalId' in found_functions}}

cdef cudaError_t _cudaGraphNodeGetLocalId(cudaGraphNode_t hNode, unsigned int* nodeId) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaGraphNodeGetLocalId(hNode, nodeId)
return cudaGraphNodeGetLocalId(hNode, nodeId)
{{endif}}

{{if 'cudaGraphNodeGetToolsId' in found_functions}}

cdef cudaError_t _cudaGraphNodeGetToolsId(cudaGraphNode_t hNode, unsigned long long* toolsNodeId) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaGraphNodeGetToolsId(hNode, toolsNodeId)
return cudaGraphNodeGetToolsId(hNode, toolsNodeId)
{{endif}}

{{if 'cudaGraphGetId' in found_functions}}

cdef cudaError_t _cudaGraphGetId(cudaGraph_t hGraph, unsigned int* graphID) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaGraphGetId(hGraph, graphID)
return cudaGraphGetId(hGraph, graphID)
{{endif}}

{{if 'cudaGraphExecGetId' in found_functions}}

cdef cudaError_t _cudaGraphExecGetId(cudaGraphExec_t hGraphExec, unsigned int* graphID) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaGraphExecGetId(hGraphExec, graphID)
return cudaGraphExecGetId(hGraphExec, graphID)
{{endif}}

{{if 'cudaGraphGetNodes' in found_functions}}

cdef cudaError_t _cudaGraphGetNodes(cudaGraph_t graph, cudaGraphNode_t* nodes, size_t* numNodes) except ?cudaErrorCallRequiresNewerDriver nogil:
Expand Down Expand Up @@ -2501,6 +2546,15 @@ cdef cudaError_t _cudaGraphConditionalHandleCreate(cudaGraphConditionalHandle* p
return cudaGraphConditionalHandleCreate(pHandle_out, graph, defaultLaunchValue, flags)
{{endif}}

{{if 'cudaGraphConditionalHandleCreate_v2' in found_functions}}

cdef cudaError_t _cudaGraphConditionalHandleCreate_v2(cudaGraphConditionalHandle* pHandle_out, cudaGraph_t graph, cudaExecutionContext_t ctx, unsigned int defaultLaunchValue, unsigned int flags) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaGraphConditionalHandleCreate_v2(pHandle_out, graph, ctx, defaultLaunchValue, flags)
return cudaGraphConditionalHandleCreate_v2(pHandle_out, graph, ctx, defaultLaunchValue, flags)
{{endif}}

{{if 'cudaGetDriverEntryPoint' in found_functions}}

cdef cudaError_t _cudaGetDriverEntryPoint(const char* symbol, void** funcPtr, unsigned long long flags, cudaDriverEntryPointQueryResult* driverStatus) except ?cudaErrorCallRequiresNewerDriver nogil:
Expand Down Expand Up @@ -2609,6 +2663,51 @@ cdef cudaError_t _cudaKernelSetAttributeForDevice(cudaKernel_t kernel, cudaFuncA
return cudaKernelSetAttributeForDevice(kernel, attr, value, device)
{{endif}}

{{if 'cudaDeviceGetDevResource' in found_functions}}

cdef cudaError_t _cudaDeviceGetDevResource(int device, cudaDevResource* resource, cudaDevResourceType typename) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaDeviceGetDevResource(device, resource, typename)
return cudaDeviceGetDevResource(device, resource, typename)
{{endif}}

{{if 'cudaDevSmResourceSplitByCount' in found_functions}}

cdef cudaError_t _cudaDevSmResourceSplitByCount(cudaDevResource* result, unsigned int* nbGroups, const cudaDevResource* input, cudaDevResource* remaining, unsigned int flags, unsigned int minCount) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaDevSmResourceSplitByCount(result, nbGroups, input, remaining, flags, minCount)
return cudaDevSmResourceSplitByCount(result, nbGroups, input, remaining, flags, minCount)
{{endif}}

{{if 'cudaDevSmResourceSplit' in found_functions}}

cdef cudaError_t _cudaDevSmResourceSplit(cudaDevResource* result, unsigned int nbGroups, const cudaDevResource* input, cudaDevResource* remainder, unsigned int flags, cudaDevSmResourceGroupParams* groupParams) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaDevSmResourceSplit(result, nbGroups, input, remainder, flags, groupParams)
return cudaDevSmResourceSplit(result, nbGroups, input, remainder, flags, groupParams)
{{endif}}

{{if 'cudaDevResourceGenerateDesc' in found_functions}}

cdef cudaError_t _cudaDevResourceGenerateDesc(cudaDevResourceDesc_t* phDesc, cudaDevResource* resources, unsigned int nbResources) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaDevResourceGenerateDesc(phDesc, resources, nbResources)
return cudaDevResourceGenerateDesc(phDesc, resources, nbResources)
{{endif}}

{{if 'cudaGreenCtxCreate' in found_functions}}

cdef cudaError_t _cudaGreenCtxCreate(cudaExecutionContext_t* phCtx, cudaDevResourceDesc_t desc, int device, unsigned int flags) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaGreenCtxCreate(phCtx, desc, device, flags)
return cudaGreenCtxCreate(phCtx, desc, device, flags)
{{endif}}

{{if 'cudaExecutionCtxDestroy' in found_functions}}

cdef cudaError_t _cudaExecutionCtxDestroy(cudaExecutionContext_t ctx) except ?cudaErrorCallRequiresNewerDriver nogil:
Expand All @@ -2618,6 +2717,42 @@ cdef cudaError_t _cudaExecutionCtxDestroy(cudaExecutionContext_t ctx) except ?cu
return cudaExecutionCtxDestroy(ctx)
{{endif}}

{{if 'cudaExecutionCtxGetDevResource' in found_functions}}

cdef cudaError_t _cudaExecutionCtxGetDevResource(cudaExecutionContext_t ctx, cudaDevResource* resource, cudaDevResourceType typename) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaExecutionCtxGetDevResource(ctx, resource, typename)
return cudaExecutionCtxGetDevResource(ctx, resource, typename)
{{endif}}

{{if 'cudaExecutionCtxGetDevice' in found_functions}}

cdef cudaError_t _cudaExecutionCtxGetDevice(int* device, cudaExecutionContext_t ctx) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaExecutionCtxGetDevice(device, ctx)
return cudaExecutionCtxGetDevice(device, ctx)
{{endif}}

{{if 'cudaExecutionCtxGetId' in found_functions}}

cdef cudaError_t _cudaExecutionCtxGetId(cudaExecutionContext_t ctx, unsigned long long* ctxId) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaExecutionCtxGetId(ctx, ctxId)
return cudaExecutionCtxGetId(ctx, ctxId)
{{endif}}

{{if 'cudaExecutionCtxStreamCreate' in found_functions}}

cdef cudaError_t _cudaExecutionCtxStreamCreate(cudaStream_t* phStream, cudaExecutionContext_t ctx, unsigned int flags, int priority) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaExecutionCtxStreamCreate(phStream, ctx, flags, priority)
return cudaExecutionCtxStreamCreate(phStream, ctx, flags, priority)
{{endif}}

{{if 'cudaExecutionCtxSynchronize' in found_functions}}

cdef cudaError_t _cudaExecutionCtxSynchronize(cudaExecutionContext_t ctx) except ?cudaErrorCallRequiresNewerDriver nogil:
Expand All @@ -2627,6 +2762,15 @@ cdef cudaError_t _cudaExecutionCtxSynchronize(cudaExecutionContext_t ctx) except
return cudaExecutionCtxSynchronize(ctx)
{{endif}}

{{if 'cudaStreamGetDevResource' in found_functions}}

cdef cudaError_t _cudaStreamGetDevResource(cudaStream_t hStream, cudaDevResource* resource, cudaDevResourceType typename) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaStreamGetDevResource(hStream, resource, typename)
return cudaStreamGetDevResource(hStream, resource, typename)
{{endif}}

{{if 'cudaExecutionCtxRecordEvent' in found_functions}}

cdef cudaError_t _cudaExecutionCtxRecordEvent(cudaExecutionContext_t ctx, cudaEvent_t event) except ?cudaErrorCallRequiresNewerDriver nogil:
Expand All @@ -2645,6 +2789,15 @@ cdef cudaError_t _cudaExecutionCtxWaitEvent(cudaExecutionContext_t ctx, cudaEven
return cudaExecutionCtxWaitEvent(ctx, event)
{{endif}}

{{if 'cudaDeviceGetExecutionCtx' in found_functions}}

cdef cudaError_t _cudaDeviceGetExecutionCtx(cudaExecutionContext_t* ctx, int device) except ?cudaErrorCallRequiresNewerDriver nogil:
cdef bint usePTDS = cudaPythonInit()
if usePTDS:
return ptds._cudaDeviceGetExecutionCtx(ctx, device)
return cudaDeviceGetExecutionCtx(ctx, device)
{{endif}}

{{if 'cudaGetExportTable' in found_functions}}

cdef cudaError_t _cudaGetExportTable(const void** ppExportTable, const cudaUUID_t* pExportTableId) except ?cudaErrorCallRequiresNewerDriver nogil:
Expand Down
Loading
Loading