Skip to content

Commit 2dbc870

Browse files
vzhurba01vzhurba01
authored
Merge pull request NVIDIA#79 from vzhurba01/69-bindings-ctk-13.0
Rebase to CTK 13.0 RC14
2 parents 1e5097c + af48e18 commit 2dbc870

53 files changed

Lines changed: 5397 additions & 6043 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎cuda_bindings/benchmarks/conftest.py‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@
44
import numpy as np
55
import pytest
66

7-
from cuda import cuda, cudart, nvrtc
7+
from cuda.bindings import driver as cuda
8+
from cuda.bindings import nvrtc
9+
from cuda.bindings import runtime as cudart
810

911

1012
def ASSERT_DRV(err):
@@ -28,7 +30,7 @@ def init_cuda():
2830
ASSERT_DRV(err)
2931
err, device = cuda.cuDeviceGet(0)
3032
ASSERT_DRV(err)
31-
err, ctx = cuda.cuCtxCreate(0, device)
33+
err, ctx = cuda.cuCtxCreate(None, 0, device)
3234
ASSERT_DRV(err)
3335

3436
# create stream

‎cuda_bindings/benchmarks/test_launch_latency.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from conftest import ASSERT_DRV
88
from kernels import kernel_string
99

10-
from cuda import cuda
10+
from cuda.bindings import driver as cuda
1111

1212

1313
def launch(kernel, stream, args=(), arg_types=()):

‎cuda_bindings/benchmarks/test_pointer_attributes.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@
66
import pytest
77
from conftest import ASSERT_DRV
88

9-
from cuda import cuda
9+
from cuda.bindings import driver as cuda
1010

1111
random.seed(0)
1212

‎cuda_bindings/cuda/bindings/_bindings/cydriver.pxd.in‎

Lines changed: 57 additions & 77 deletions
Original file line numberDiff line numberDiff line change
@@ -39,11 +39,6 @@ cdef CUresult _cuDeviceGetCount(int* count) except ?CUDA_ERROR_NOT_FOUND nogil
3939
cdef CUresult _cuDeviceGetName(char* name, int length, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
4040
{{endif}}
4141

42-
{{if 'cuDeviceGetUuid' in found_functions}}
43-
44-
cdef CUresult _cuDeviceGetUuid(CUuuid* uuid, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
45-
{{endif}}
46-
4742
{{if 'cuDeviceGetUuid_v2' in found_functions}}
4843

4944
cdef CUresult _cuDeviceGetUuid_v2(CUuuid* uuid, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -69,6 +64,11 @@ cdef CUresult _cuDeviceGetTexture1DLinearMaxWidth(size_t* maxWidthInElements, CU
6964
cdef CUresult _cuDeviceGetAttribute(int* pi, CUdevice_attribute attrib, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
7065
{{endif}}
7166

67+
{{if 'cuDeviceGetHostAtomicCapabilities' in found_functions}}
68+
69+
cdef CUresult _cuDeviceGetHostAtomicCapabilities(unsigned int* capabilities, const CUatomicOperation* operations, unsigned int count, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
70+
{{endif}}
71+
7272
{{if 'cuDeviceGetNvSciSyncAttributes' in found_functions}}
7373

7474
cdef CUresult _cuDeviceGetNvSciSyncAttributes(void* nvSciSyncAttrList, CUdevice dev, int flags) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -134,16 +134,6 @@ cdef CUresult _cuDevicePrimaryCtxGetState(CUdevice dev, unsigned int* flags, int
134134
cdef CUresult _cuDevicePrimaryCtxReset_v2(CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
135135
{{endif}}
136136

137-
{{if 'cuCtxCreate_v2' in found_functions}}
138-
139-
cdef CUresult _cuCtxCreate_v2(CUcontext* pctx, unsigned int flags, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
140-
{{endif}}
141-
142-
{{if 'cuCtxCreate_v3' in found_functions}}
143-
144-
cdef CUresult _cuCtxCreate_v3(CUcontext* pctx, CUexecAffinityParam* paramsArray, int numParams, unsigned int flags, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
145-
{{endif}}
146-
147137
{{if 'cuCtxCreate_v4' in found_functions}}
148138

149139
cdef CUresult _cuCtxCreate_v4(CUcontext* pctx, CUctxCreateParams* ctxCreateParams, unsigned int flags, CUdevice dev) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -179,6 +169,11 @@ cdef CUresult _cuCtxGetCurrent(CUcontext* pctx) except ?CUDA_ERROR_NOT_FOUND nog
179169
cdef CUresult _cuCtxGetDevice(CUdevice* device) except ?CUDA_ERROR_NOT_FOUND nogil
180170
{{endif}}
181171

172+
{{if 'cuCtxGetDevice_v2' in found_functions}}
173+
174+
cdef CUresult _cuCtxGetDevice_v2(CUdevice* device, CUcontext ctx) except ?CUDA_ERROR_NOT_FOUND nogil
175+
{{endif}}
176+
182177
{{if 'cuCtxGetFlags' in found_functions}}
183178

184179
cdef CUresult _cuCtxGetFlags(unsigned int* flags) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -199,6 +194,11 @@ cdef CUresult _cuCtxGetId(CUcontext ctx, unsigned long long* ctxId) except ?CUDA
199194
cdef CUresult _cuCtxSynchronize() except ?CUDA_ERROR_NOT_FOUND nogil
200195
{{endif}}
201196

197+
{{if 'cuCtxSynchronize_v2' in found_functions}}
198+
199+
cdef CUresult _cuCtxSynchronize_v2(CUcontext ctx) except ?CUDA_ERROR_NOT_FOUND nogil
200+
{{endif}}
201+
202202
{{if 'cuCtxSetLimit' in found_functions}}
203203

204204
cdef CUresult _cuCtxSetLimit(CUlimit limit, size_t value) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -669,14 +669,14 @@ cdef CUresult _cuMemcpy3DAsync_v2(const CUDA_MEMCPY3D* pCopy, CUstream hStream)
669669
cdef CUresult _cuMemcpy3DPeerAsync(const CUDA_MEMCPY3D_PEER* pCopy, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
670670
{{endif}}
671671

672-
{{if 'cuMemcpyBatchAsync' in found_functions}}
672+
{{if 'cuMemcpyBatchAsync_v2' in found_functions}}
673673

674-
cdef CUresult _cuMemcpyBatchAsync(CUdeviceptr* dsts, CUdeviceptr* srcs, size_t* sizes, size_t count, CUmemcpyAttributes* attrs, size_t* attrsIdxs, size_t numAttrs, size_t* failIdx, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
674+
cdef CUresult _cuMemcpyBatchAsync_v2(CUdeviceptr* dsts, CUdeviceptr* srcs, size_t* sizes, size_t count, CUmemcpyAttributes* attrs, size_t* attrsIdxs, size_t numAttrs, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
675675
{{endif}}
676676

677-
{{if 'cuMemcpy3DBatchAsync' in found_functions}}
677+
{{if 'cuMemcpy3DBatchAsync_v2' in found_functions}}
678678

679-
cdef CUresult _cuMemcpy3DBatchAsync(size_t numOps, CUDA_MEMCPY3D_BATCH_OP* opList, size_t* failIdx, unsigned long long flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
679+
cdef CUresult _cuMemcpy3DBatchAsync_v2(size_t numOps, CUDA_MEMCPY3D_BATCH_OP* opList, unsigned long long flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
680680
{{endif}}
681681

682682
{{if 'cuMemsetD8_v2' in found_functions}}
@@ -929,6 +929,21 @@ cdef CUresult _cuMemPoolCreate(CUmemoryPool* pool, const CUmemPoolProps* poolPro
929929
cdef CUresult _cuMemPoolDestroy(CUmemoryPool pool) except ?CUDA_ERROR_NOT_FOUND nogil
930930
{{endif}}
931931

932+
{{if 'cuMemGetDefaultMemPool' in found_functions}}
933+
934+
cdef CUresult _cuMemGetDefaultMemPool(CUmemoryPool* pool_out, CUmemLocation* location, CUmemAllocationType typename) except ?CUDA_ERROR_NOT_FOUND nogil
935+
{{endif}}
936+
937+
{{if 'cuMemGetMemPool' in found_functions}}
938+
939+
cdef CUresult _cuMemGetMemPool(CUmemoryPool* pool, CUmemLocation* location, CUmemAllocationType typename) except ?CUDA_ERROR_NOT_FOUND nogil
940+
{{endif}}
941+
942+
{{if 'cuMemSetMemPool' in found_functions}}
943+
944+
cdef CUresult _cuMemSetMemPool(CUmemLocation* location, CUmemAllocationType typename, CUmemoryPool pool) except ?CUDA_ERROR_NOT_FOUND nogil
945+
{{endif}}
946+
932947
{{if 'cuMemAllocFromPoolAsync' in found_functions}}
933948

934949
cdef CUresult _cuMemAllocFromPoolAsync(CUdeviceptr* dptr, size_t bytesize, CUmemoryPool pool, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -989,24 +1004,29 @@ cdef CUresult _cuMulticastGetGranularity(size_t* granularity, const CUmulticastO
9891004
cdef CUresult _cuPointerGetAttribute(void* data, CUpointer_attribute attribute, CUdeviceptr ptr) except ?CUDA_ERROR_NOT_FOUND nogil
9901005
{{endif}}
9911006

992-
{{if 'cuMemPrefetchAsync' in found_functions}}
1007+
{{if 'cuMemPrefetchAsync_v2' in found_functions}}
9931008

994-
cdef CUresult _cuMemPrefetchAsync(CUdeviceptr devPtr, size_t count, CUdevice dstDevice, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
1009+
cdef CUresult _cuMemPrefetchAsync_v2(CUdeviceptr devPtr, size_t count, CUmemLocation location, unsigned int flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
9951010
{{endif}}
9961011

997-
{{if 'cuMemPrefetchAsync_v2' in found_functions}}
1012+
{{if 'cuMemAdvise_v2' in found_functions}}
9981013

999-
cdef CUresult _cuMemPrefetchAsync_v2(CUdeviceptr devPtr, size_t count, CUmemLocation location, unsigned int flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
1014+
cdef CUresult _cuMemAdvise_v2(CUdeviceptr devPtr, size_t count, CUmem_advise advice, CUmemLocation location) except ?CUDA_ERROR_NOT_FOUND nogil
10001015
{{endif}}
10011016

1002-
{{if 'cuMemAdvise' in found_functions}}
1017+
{{if 'cuMemPrefetchBatchAsync' in found_functions}}
10031018

1004-
cdef CUresult _cuMemAdvise(CUdeviceptr devPtr, size_t count, CUmem_advise advice, CUdevice device) except ?CUDA_ERROR_NOT_FOUND nogil
1019+
cdef CUresult _cuMemPrefetchBatchAsync(CUdeviceptr* dptrs, size_t* sizes, size_t count, CUmemLocation* prefetchLocs, size_t* prefetchLocIdxs, size_t numPrefetchLocs, unsigned long long flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
10051020
{{endif}}
10061021

1007-
{{if 'cuMemAdvise_v2' in found_functions}}
1022+
{{if 'cuMemDiscardBatchAsync' in found_functions}}
10081023

1009-
cdef CUresult _cuMemAdvise_v2(CUdeviceptr devPtr, size_t count, CUmem_advise advice, CUmemLocation location) except ?CUDA_ERROR_NOT_FOUND nogil
1024+
cdef CUresult _cuMemDiscardBatchAsync(CUdeviceptr* dptrs, size_t* sizes, size_t count, unsigned long long flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
1025+
{{endif}}
1026+
1027+
{{if 'cuMemDiscardAndPrefetchBatchAsync' in found_functions}}
1028+
1029+
cdef CUresult _cuMemDiscardAndPrefetchBatchAsync(CUdeviceptr* dptrs, size_t* sizes, size_t count, CUmemLocation* prefetchLocs, size_t* prefetchLocIdxs, size_t numPrefetchLocs, unsigned long long flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil
10101030
{{endif}}
10111031

10121032
{{if 'cuMemRangeGetAttribute' in found_functions}}
@@ -1104,21 +1124,11 @@ cdef CUresult _cuStreamEndCapture(CUstream hStream, CUgraph* phGraph) except ?CU
11041124
cdef CUresult _cuStreamIsCapturing(CUstream hStream, CUstreamCaptureStatus* captureStatus) except ?CUDA_ERROR_NOT_FOUND nogil
11051125
{{endif}}
11061126

1107-
{{if 'cuStreamGetCaptureInfo_v2' in found_functions}}
1108-
1109-
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
1110-
{{endif}}
1111-
11121127
{{if 'cuStreamGetCaptureInfo_v3' in found_functions}}
11131128

11141129
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
11151130
{{endif}}
11161131

1117-
{{if 'cuStreamUpdateCaptureDependencies' in found_functions}}
1118-
1119-
cdef CUresult _cuStreamUpdateCaptureDependencies(CUstream hStream, CUgraphNode* dependencies, size_t numDependencies, unsigned int flags) except ?CUDA_ERROR_NOT_FOUND nogil
1120-
{{endif}}
1121-
11221132
{{if 'cuStreamUpdateCaptureDependencies_v2' in found_functions}}
11231133

11241134
cdef CUresult _cuStreamUpdateCaptureDependencies_v2(CUstream hStream, CUgraphNode* dependencies, const CUgraphEdgeData* dependencyData, size_t numDependencies, unsigned int flags) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -1189,11 +1199,6 @@ cdef CUresult _cuEventSynchronize(CUevent hEvent) except ?CUDA_ERROR_NOT_FOUND n
11891199
cdef CUresult _cuEventDestroy_v2(CUevent hEvent) except ?CUDA_ERROR_NOT_FOUND nogil
11901200
{{endif}}
11911201

1192-
{{if 'cuEventElapsedTime' in found_functions}}
1193-
1194-
cdef CUresult _cuEventElapsedTime(float* pMilliseconds, CUevent hStart, CUevent hEnd) except ?CUDA_ERROR_NOT_FOUND nogil
1195-
{{endif}}
1196-
11971202
{{if 'cuEventElapsedTime_v2' in found_functions}}
11981203

11991204
cdef CUresult _cuEventElapsedTime_v2(float* pMilliseconds, CUevent hStart, CUevent hEnd) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -1604,51 +1609,26 @@ cdef CUresult _cuGraphGetNodes(CUgraph hGraph, CUgraphNode* nodes, size_t* numNo
16041609
cdef CUresult _cuGraphGetRootNodes(CUgraph hGraph, CUgraphNode* rootNodes, size_t* numRootNodes) except ?CUDA_ERROR_NOT_FOUND nogil
16051610
{{endif}}
16061611

1607-
{{if 'cuGraphGetEdges' in found_functions}}
1608-
1609-
cdef CUresult _cuGraphGetEdges(CUgraph hGraph, CUgraphNode* from_, CUgraphNode* to, size_t* numEdges) except ?CUDA_ERROR_NOT_FOUND nogil
1610-
{{endif}}
1611-
16121612
{{if 'cuGraphGetEdges_v2' in found_functions}}
16131613

16141614
cdef CUresult _cuGraphGetEdges_v2(CUgraph hGraph, CUgraphNode* from_, CUgraphNode* to, CUgraphEdgeData* edgeData, size_t* numEdges) except ?CUDA_ERROR_NOT_FOUND nogil
16151615
{{endif}}
16161616

1617-
{{if 'cuGraphNodeGetDependencies' in found_functions}}
1618-
1619-
cdef CUresult _cuGraphNodeGetDependencies(CUgraphNode hNode, CUgraphNode* dependencies, size_t* numDependencies) except ?CUDA_ERROR_NOT_FOUND nogil
1620-
{{endif}}
1621-
16221617
{{if 'cuGraphNodeGetDependencies_v2' in found_functions}}
16231618

16241619
cdef CUresult _cuGraphNodeGetDependencies_v2(CUgraphNode hNode, CUgraphNode* dependencies, CUgraphEdgeData* edgeData, size_t* numDependencies) except ?CUDA_ERROR_NOT_FOUND nogil
16251620
{{endif}}
16261621

1627-
{{if 'cuGraphNodeGetDependentNodes' in found_functions}}
1628-
1629-
cdef CUresult _cuGraphNodeGetDependentNodes(CUgraphNode hNode, CUgraphNode* dependentNodes, size_t* numDependentNodes) except ?CUDA_ERROR_NOT_FOUND nogil
1630-
{{endif}}
1631-
16321622
{{if 'cuGraphNodeGetDependentNodes_v2' in found_functions}}
16331623

16341624
cdef CUresult _cuGraphNodeGetDependentNodes_v2(CUgraphNode hNode, CUgraphNode* dependentNodes, CUgraphEdgeData* edgeData, size_t* numDependentNodes) except ?CUDA_ERROR_NOT_FOUND nogil
16351625
{{endif}}
16361626

1637-
{{if 'cuGraphAddDependencies' in found_functions}}
1638-
1639-
cdef CUresult _cuGraphAddDependencies(CUgraph hGraph, const CUgraphNode* from_, const CUgraphNode* to, size_t numDependencies) except ?CUDA_ERROR_NOT_FOUND nogil
1640-
{{endif}}
1641-
16421627
{{if 'cuGraphAddDependencies_v2' in found_functions}}
16431628

16441629
cdef CUresult _cuGraphAddDependencies_v2(CUgraph hGraph, const CUgraphNode* from_, const CUgraphNode* to, const CUgraphEdgeData* edgeData, size_t numDependencies) except ?CUDA_ERROR_NOT_FOUND nogil
16451630
{{endif}}
16461631

1647-
{{if 'cuGraphRemoveDependencies' in found_functions}}
1648-
1649-
cdef CUresult _cuGraphRemoveDependencies(CUgraph hGraph, const CUgraphNode* from_, const CUgraphNode* to, size_t numDependencies) except ?CUDA_ERROR_NOT_FOUND nogil
1650-
{{endif}}
1651-
16521632
{{if 'cuGraphRemoveDependencies_v2' in found_functions}}
16531633

16541634
cdef CUresult _cuGraphRemoveDependencies_v2(CUgraph hGraph, const CUgraphNode* from_, const CUgraphNode* to, const CUgraphEdgeData* edgeData, size_t numDependencies) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -1799,11 +1779,6 @@ cdef CUresult _cuGraphRetainUserObject(CUgraph graph, CUuserObject object, unsig
17991779
cdef CUresult _cuGraphReleaseUserObject(CUgraph graph, CUuserObject object, unsigned int count) except ?CUDA_ERROR_NOT_FOUND nogil
18001780
{{endif}}
18011781

1802-
{{if 'cuGraphAddNode' in found_functions}}
1803-
1804-
cdef CUresult _cuGraphAddNode(CUgraphNode* phGraphNode, CUgraph hGraph, const CUgraphNode* dependencies, size_t numDependencies, CUgraphNodeParams* nodeParams) except ?CUDA_ERROR_NOT_FOUND nogil
1805-
{{endif}}
1806-
18071782
{{if 'cuGraphAddNode_v2' in found_functions}}
18081783

18091784
cdef CUresult _cuGraphAddNode_v2(CUgraphNode* phGraphNode, CUgraph hGraph, const CUgraphNode* dependencies, const CUgraphEdgeData* dependencyData, size_t numDependencies, CUgraphNodeParams* nodeParams) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -2084,6 +2059,11 @@ cdef CUresult _cuCtxDisablePeerAccess(CUcontext peerContext) except ?CUDA_ERROR_
20842059
cdef CUresult _cuDeviceGetP2PAttribute(int* value, CUdevice_P2PAttribute attrib, CUdevice srcDevice, CUdevice dstDevice) except ?CUDA_ERROR_NOT_FOUND nogil
20852060
{{endif}}
20862061

2062+
{{if 'cuDeviceGetP2PAtomicCapabilities' in found_functions}}
2063+
2064+
cdef CUresult _cuDeviceGetP2PAtomicCapabilities(unsigned int* capabilities, const CUatomicOperation* operations, unsigned int count, CUdevice srcDevice, CUdevice dstDevice) except ?CUDA_ERROR_NOT_FOUND nogil
2065+
{{endif}}
2066+
20872067
{{if 'cuGraphicsUnregisterResource' in found_functions}}
20882068

20892069
cdef CUresult _cuGraphicsUnregisterResource(CUgraphicsResource resource) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -2209,6 +2189,11 @@ cdef CUresult _cuStreamGetGreenCtx(CUstream hStream, CUgreenCtx* phCtx) except ?
22092189
cdef CUresult _cuGreenCtxStreamCreate(CUstream* phStream, CUgreenCtx greenCtx, unsigned int flags, int priority) except ?CUDA_ERROR_NOT_FOUND nogil
22102190
{{endif}}
22112191

2192+
{{if 'cuGreenCtxGetId' in found_functions}}
2193+
2194+
cdef CUresult _cuGreenCtxGetId(CUgreenCtx greenCtx, unsigned long long* greenCtxId) except ?CUDA_ERROR_NOT_FOUND nogil
2195+
{{endif}}
2196+
22122197
{{if 'cuLogsRegisterCallback' in found_functions}}
22132198

22142199
cdef CUresult _cuLogsRegisterCallback(CUlogsCallback callbackFunc, void* userData, CUlogsCallbackHandle* callback_out) except ?CUDA_ERROR_NOT_FOUND nogil
@@ -2254,11 +2239,6 @@ cdef CUresult _cuCheckpointProcessLock(int pid, CUcheckpointLockArgs* args) exce
22542239
cdef CUresult _cuCheckpointProcessCheckpoint(int pid, CUcheckpointCheckpointArgs* args) except ?CUDA_ERROR_NOT_FOUND nogil
22552240
{{endif}}
22562241

2257-
{{if 'cuCheckpointProcessRestore' in found_functions}}
2258-
2259-
cdef CUresult _cuCheckpointProcessRestore(int pid, CUcheckpointRestoreArgs* args) except ?CUDA_ERROR_NOT_FOUND nogil
2260-
{{endif}}
2261-
22622242
{{if 'cuCheckpointProcessUnlock' in found_functions}}
22632243

22642244
cdef CUresult _cuCheckpointProcessUnlock(int pid, CUcheckpointUnlockArgs* args) except ?CUDA_ERROR_NOT_FOUND nogil

0 commit comments

Comments
 (0)