| // 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**) |
| IREE_CU_PFN_DECL(cuLaunchHostFunc, CUstream, CUhostFn, void*) |
