Searched refs:CUstream (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/openmp/libomptarget/plugins/cuda/dynamic_cuda/ |
| H A D | cuda.h | 24 typedef struct CUstream_st *CUstream; typedef
221 unsigned, unsigned, unsigned, CUstream, void **,
228 CUresult cuMemcpyDtoDAsync(CUdeviceptr, CUdeviceptr, size_t, CUstream);
230 CUresult cuMemcpyDtoHAsync(void *, CUdeviceptr, size_t, CUstream);
232 CUresult cuMemcpyHtoDAsync(CUdeviceptr, const void *, size_t, CUstream);
241 CUresult cuStreamCreate(CUstream *, unsigned);
242 CUresult cuStreamDestroy(CUstream);
243 CUresult cuStreamSynchronize(CUstream);
255 size_t, CUstream);
261 CUresult cuEventRecord(CUevent, CUstream);
[all …]
|
| /llvm-project-15.0.7/mlir/lib/ExecutionEngine/ |
| H A D | CudaRuntimeWrappers.cpp | 88 intptr_t blockZ, int32_t smem, CUstream stream, void **params, in mgpuLaunchKernel()
96 extern "C" MLIR_CUDA_WRAPPERS_EXPORT CUstream mgpuStreamCreate() { in mgpuStreamCreate()
98 CUstream stream = nullptr; in mgpuStreamCreate()
103 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuStreamDestroy(CUstream stream) { in mgpuStreamDestroy()
108 mgpuStreamSynchronize(CUstream stream) { in mgpuStreamSynchronize()
112 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuStreamWaitEvent(CUstream stream, in mgpuStreamWaitEvent()
133 CUstream stream) {
137 extern "C" void *mgpuMemAlloc(uint64_t sizeBytes, CUstream /*stream*/) { in mgpuMemAlloc() argument
144 extern "C" void mgpuMemFree(void *ptr, CUstream /*stream*/) { in mgpuMemFree() argument
149 CUstream stream) { in mgpuMemcpy()
[all …]
|
| /llvm-project-15.0.7/openmp/libomptarget/plugins/cuda/src/ |
| H A D | rtl.cpp | 108 CUstream Stream) { in memcpyDtoD()
124 CUstream Stream = reinterpret_cast<CUstream>(AsyncInfo->Queue); in recordEvent()
186 int create(CUstream &Stream) noexcept override { in create()
195 int destroy(CUstream Stream) noexcept override { in destroy()
487 CUstream S; in getStream()
1002 CUstream Stream = getStream(DeviceId, AsyncInfo); in dataSubmit()
1019 CUstream Stream = getStream(DeviceId, AsyncInfo); in dataRetrieve()
1037 CUstream Stream = getStream(SrcDevId, AsyncInfo); in dataExchange()
1233 CUstream Stream = getStream(DeviceId, AsyncInfo); in runTargetTeamRegion()
1248 CUstream Stream = reinterpret_cast<CUstream>(AsyncInfo->Queue); in synchronize()
[all …]
|
| /llvm-project-15.0.7/polly/tools/GPURuntime/ |
| H A D | GPUJIT.c | 951 unsigned int BlockDimZ, unsigned int SharedMemBytes, CUstream HStream,
1407 CUstream Stream = 0; in launchKernelCUDA()
|