Lines Matching refs:stream
88 intptr_t blockZ, int32_t smem, CUstream stream, void **params, in mgpuLaunchKernel() argument
92 blockY, blockZ, smem, stream, params, in mgpuLaunchKernel()
98 CUstream stream = nullptr; in mgpuStreamCreate() local
99 CUDA_REPORT_IF_ERROR(cuStreamCreate(&stream, CU_STREAM_NON_BLOCKING)); in mgpuStreamCreate()
100 return stream; in mgpuStreamCreate()
103 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuStreamDestroy(CUstream stream) { in mgpuStreamDestroy() argument
104 CUDA_REPORT_IF_ERROR(cuStreamDestroy(stream)); in mgpuStreamDestroy()
108 mgpuStreamSynchronize(CUstream stream) { in mgpuStreamSynchronize() argument
109 CUDA_REPORT_IF_ERROR(cuStreamSynchronize(stream)); in mgpuStreamSynchronize()
112 extern "C" MLIR_CUDA_WRAPPERS_EXPORT void mgpuStreamWaitEvent(CUstream stream, in mgpuStreamWaitEvent() argument
114 CUDA_REPORT_IF_ERROR(cuStreamWaitEvent(stream, event, /*flags=*/0)); in mgpuStreamWaitEvent()
133 CUstream stream) {
134 CUDA_REPORT_IF_ERROR(cuEventRecord(event, stream));
149 CUstream stream) { in mgpuMemcpy() argument
152 sizeBytes, stream)); in mgpuMemcpy()
156 CUstream stream) { in mgpuMemset32() argument
158 value, count, stream)); in mgpuMemset32()