|
2 | 2 | # |
3 | 3 | # SPDX-License-Identifier: Apache-2.0 |
4 | 4 | # |
5 | | -# This code was automatically generated across versions from 12.9.0 to 13.4.1. Do not modify it directly. |
6 | | -# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=cf88d68f5d0521e0872350a3e27b7c90b33d32c0d3d1211ae1aa746da98a588d |
| 5 | +# This code was automatically generated across versions from 12.9.0 to 13.5.0. Do not modify it directly. |
| 6 | +# CYTHON-BINDINGS-GENERATED-DO-NOT-MODIFY-THIS-FILE: format=1; content-sha256=855d1d0530ab58f150ca5fb8b9ec43262cf0df681f507e2b53feeb6cf77c7c17 |
7 | 7 |
|
8 | 8 |
|
9 | 9 | # <<<< PREAMBLE CONTENT >>>> |
@@ -591,6 +591,8 @@ cdef void* __cuMemGetLocationInfo = NULL |
591 | 591 | cdef void* __cuGraphAddNode_v3 = NULL |
592 | 592 | cdef void* __cuGraphNodeSetParams_v2 = NULL |
593 | 593 | cdef void* __cuCheckpointOperationComplete = NULL |
| 594 | +cdef void* __cuFabricCpyBatchAsync = NULL |
| 595 | +cdef void* __cuTensorMapSetAllocationClass = NULL |
594 | 596 |
|
595 | 597 | cdef int _init_driver() except -1 nogil: |
596 | 598 | global _cyb___py_driver_init |
@@ -2183,6 +2185,12 @@ cdef int _init_driver() except -1 nogil: |
2183 | 2185 | global __cuCheckpointOperationComplete |
2184 | 2186 | cuGetProcAddress_v2('cuCheckpointOperationComplete', <void **>&__cuCheckpointOperationComplete, 13041, ptds_mode, NULL) |
2185 | 2187 |
|
| 2188 | + global __cuFabricCpyBatchAsync |
| 2189 | + cuGetProcAddress_v2('cuFabricCpyBatchAsync', <void **>&__cuFabricCpyBatchAsync, 13050, ptds_mode, NULL) |
| 2190 | + |
| 2191 | + global __cuTensorMapSetAllocationClass |
| 2192 | + cuGetProcAddress_v2('cuTensorMapSetAllocationClass', <void **>&__cuTensorMapSetAllocationClass, 13050, ptds_mode, NULL) |
| 2193 | + |
2186 | 2194 | _cyb_atomic_int_store(<int *>&_cyb___py_driver_init, 1) |
2187 | 2195 | return 0 |
2188 | 2196 |
|
@@ -3771,6 +3779,12 @@ cpdef dict _inspect_function_pointers(): |
3771 | 3779 |
|
3772 | 3780 | global __cuCheckpointOperationComplete |
3773 | 3781 | data["__cuCheckpointOperationComplete"] = <intptr_t>__cuCheckpointOperationComplete |
| 3782 | + |
| 3783 | + global __cuFabricCpyBatchAsync |
| 3784 | + data["__cuFabricCpyBatchAsync"] = <intptr_t>__cuFabricCpyBatchAsync |
| 3785 | + |
| 3786 | + global __cuTensorMapSetAllocationClass |
| 3787 | + data["__cuTensorMapSetAllocationClass"] = <intptr_t>__cuTensorMapSetAllocationClass |
3774 | 3788 | _cyb_func_ptrs = data |
3775 | 3789 | return data |
3776 | 3790 |
|
@@ -9031,3 +9045,23 @@ cdef CUresult _cuCheckpointOperationComplete(CUcheckpointOperationHandle handle) |
9031 | 9045 | raise FunctionNotFoundError("function cuCheckpointOperationComplete is not found") |
9032 | 9046 | return (<CUresult (*)(CUcheckpointOperationHandle) noexcept nogil>__cuCheckpointOperationComplete)( |
9033 | 9047 | handle) |
| 9048 | + |
| 9049 | + |
| 9050 | +cdef CUresult _cuFabricCpyBatchAsync(size_t numOps, CUDA_FABRIC_COPY_BATCH_OP* opList, unsigned long long flags, CUstream hStream) except ?CUDA_ERROR_NOT_FOUND nogil: |
| 9051 | + global __cuFabricCpyBatchAsync |
| 9052 | + _check_or_init_driver() |
| 9053 | + if __cuFabricCpyBatchAsync == NULL: |
| 9054 | + with gil: |
| 9055 | + raise FunctionNotFoundError("function cuFabricCpyBatchAsync is not found") |
| 9056 | + return (<CUresult (*)(size_t, CUDA_FABRIC_COPY_BATCH_OP*, unsigned long long, CUstream) noexcept nogil>__cuFabricCpyBatchAsync)( |
| 9057 | + numOps, opList, flags, hStream) |
| 9058 | + |
| 9059 | + |
| 9060 | +cdef CUresult _cuTensorMapSetAllocationClass(CUtensorMap* tensorMap, CUtensorMapAllocClass allocClass, CUtensorMapAllocClass* outAllocClass) except ?CUDA_ERROR_NOT_FOUND nogil: |
| 9061 | + global __cuTensorMapSetAllocationClass |
| 9062 | + _check_or_init_driver() |
| 9063 | + if __cuTensorMapSetAllocationClass == NULL: |
| 9064 | + with gil: |
| 9065 | + raise FunctionNotFoundError("function cuTensorMapSetAllocationClass is not found") |
| 9066 | + return (<CUresult (*)(CUtensorMap*, CUtensorMapAllocClass, CUtensorMapAllocClass*) noexcept nogil>__cuTensorMapSetAllocationClass)( |
| 9067 | + tensorMap, allocClass, outAllocClass) |
0 commit comments