| // Copyright 2023 The IREE Authors | |
| // | |
| // Licensed under the Apache License v2.0 with LLVM Exceptions. | |
| // See https://llvm.org/LICENSE.txt for license information. | |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception | |
| // cuGetErrorName, cuGetErrorString should be loaded first in case there are | |
| // errors loading the other symbols. | |
| IREE_CU_PFN_DECL(cuGetErrorName, CUresult, const char**) | |
| IREE_CU_PFN_DECL(cuGetErrorString, CUresult, const char**) | |
| IREE_CU_PFN_DECL(cuDriverGetVersion, int*) | |
| IREE_CU_PFN_DECL(cuCtxCreate, CUcontext*, unsigned int, CUdevice) | |
| IREE_CU_PFN_DECL(cuCtxDestroy, CUcontext) | |
| IREE_CU_PFN_DECL(cuDevicePrimaryCtxRetain, CUcontext*, CUdevice) | |
| IREE_CU_PFN_DECL(cuDevicePrimaryCtxRelease, CUdevice) | |
| IREE_CU_PFN_DECL(cuCtxSetCurrent, CUcontext) | |
| IREE_CU_PFN_DECL(cuCtxPushCurrent, CUcontext) | |
| IREE_CU_PFN_DECL(cuCtxPopCurrent, CUcontext*) | |
| IREE_CU_PFN_DECL(cuDeviceGet, CUdevice*, int) | |
| IREE_CU_PFN_DECL(cuDeviceGetCount, int*) | |
| IREE_CU_PFN_DECL(cuDeviceGetName, char*, int, CUdevice) | |
| IREE_CU_PFN_DECL(cuDeviceGetAttribute, int*, CUdevice_attribute, CUdevice) | |
| IREE_CU_PFN_DECL(cuDeviceGetUuid, CUuuid*, CUdevice) | |
| IREE_CU_PFN_DECL(cuEventCreate, CUevent*, unsigned int) | |
| IREE_CU_PFN_DECL(cuEventDestroy, CUevent) | |
| IREE_CU_PFN_DECL(cuEventElapsedTime, float*, CUevent, CUevent) | |
| IREE_CU_PFN_DECL(cuEventQuery, CUevent) | |
| IREE_CU_PFN_DECL(cuEventRecord, CUevent, CUstream) | |
| IREE_CU_PFN_DECL(cuEventSynchronize, CUevent) | |
| IREE_CU_PFN_DECL(cuGetProcAddress, const char*, void**, int, cuuint64_t) | |
| IREE_CU_PFN_DECL(cuGraphAddMemcpyNode, CUgraphNode*, CUgraph, | |
| const CUgraphNode*, size_t, const CUDA_MEMCPY3D*, CUcontext) | |
| IREE_CU_PFN_DECL(cuGraphAddMemsetNode, CUgraphNode*, CUgraph, | |
| const CUgraphNode*, size_t, const CUDA_MEMSET_NODE_PARAMS*, | |
| CUcontext) | |
| IREE_CU_PFN_DECL(cuGraphAddKernelNode, CUgraphNode*, CUgraph, | |
| const CUgraphNode*, size_t, const CUDA_KERNEL_NODE_PARAMS*) | |
| IREE_CU_PFN_DECL(cuGraphCreate, CUgraph*, unsigned int) | |
| IREE_CU_PFN_DECL(cuGraphDestroy, CUgraph) | |
| IREE_CU_PFN_DECL(cuGraphExecDestroy, CUgraphExec) | |
| IREE_CU_PFN_DECL(cuGraphGetNodes, CUgraph, CUgraphNode*, size_t*) | |
| IREE_CU_PFN_DECL(cuGraphInstantiate, CUgraphExec*, CUgraph, CUgraphNode*, char*, | |
| size_t) | |
| IREE_CU_PFN_DECL(cuGraphLaunch, CUgraphExec, CUstream) | |
| IREE_CU_PFN_DECL(cuInit, unsigned int) | |
| IREE_CU_PFN_DECL(cuMemAllocManaged, CUdeviceptr*, size_t, unsigned int) | |
| IREE_CU_PFN_DECL(cuMemPrefetchAsync, CUdeviceptr, size_t, CUdevice, CUstream) | |
| IREE_CU_PFN_DECL(cuMemAlloc, CUdeviceptr*, size_t) | |
| IREE_CU_PFN_DECL(cuMemFree, CUdeviceptr) | |
| IREE_CU_PFN_DECL(cuMemFreeHost, void*) | |
| IREE_CU_PFN_DECL(cuMemHostAlloc, void**, size_t, unsigned int) | |
| IREE_CU_PFN_DECL(cuMemHostRegister, void*, size_t, unsigned int) | |
| IREE_CU_PFN_DECL(cuMemHostUnregister, void*) | |
| IREE_CU_PFN_DECL(cuMemHostGetDevicePointer, CUdeviceptr*, void*, unsigned int) | |
| IREE_CU_PFN_DECL(cuMemPoolCreate, CUmemoryPool*, const CUmemPoolProps*) | |
| IREE_CU_PFN_DECL(cuMemPoolDestroy, CUmemoryPool) | |
| IREE_CU_PFN_DECL(cuMemPoolSetAccess, CUmemoryPool, const CUmemAccessDesc*, | |
| size_t) | |
| IREE_CU_PFN_DECL(cuMemPoolGetAttribute, CUmemoryPool, CUmemPool_attribute, | |
| void*) | |
| IREE_CU_PFN_DECL(cuMemPoolSetAttribute, CUmemoryPool, CUmemPool_attribute, | |
| void*) | |
| IREE_CU_PFN_DECL(cuMemPoolTrimTo, CUmemoryPool, size_t) | |
| IREE_CU_PFN_DECL(cuMemAllocFromPoolAsync, CUdeviceptr*, size_t, CUmemoryPool, | |
| CUstream) | |
| IREE_CU_PFN_DECL(cuMemFreeAsync, CUdeviceptr dptr, CUstream hStream) | |
| IREE_CU_PFN_DECL(cuModuleGetFunction, CUfunction*, CUmodule, const char*) | |
| IREE_CU_PFN_DECL(cuModuleLoadDataEx, CUmodule*, const void*, unsigned int, | |
| CUjit_option*, void**) | |
| IREE_CU_PFN_DECL(cuModuleUnload, CUmodule) | |
| IREE_CU_PFN_DECL(cuStreamCreate, CUstream*, unsigned int) | |
| IREE_CU_PFN_DECL(cuStreamDestroy, CUstream) | |
| IREE_CU_PFN_DECL(cuStreamSynchronize, CUstream) | |
| IREE_CU_PFN_DECL(cuStreamWaitEvent, CUstream, CUevent, unsigned int) | |
| IREE_CU_PFN_DECL(cuMemsetD32Async, unsigned long long, unsigned int, size_t, | |
| CUstream) | |
| IREE_CU_PFN_DECL(cuMemsetD16Async, unsigned long long, unsigned short, size_t, | |
| CUstream) | |
| IREE_CU_PFN_DECL(cuMemsetD8Async, unsigned long long, unsigned char, size_t, | |
| CUstream) | |
| IREE_CU_PFN_DECL(cuMemcpyAsync, CUdeviceptr, CUdeviceptr, size_t, CUstream) | |
| IREE_CU_PFN_DECL(cuMemcpyHtoDAsync, CUdeviceptr, const void*, size_t, CUstream) | |
| IREE_CU_PFN_DECL(cuFuncSetAttribute, CUfunction, CUfunction_attribute, int) | |
| IREE_CU_PFN_DECL(cuLaunchKernel, CUfunction, unsigned int, unsigned int, | |
| unsigned int, unsigned int, unsigned int, unsigned int, | |
| unsigned int, CUstream, void**, void**) |