Lines Matching refs:stream

50     stream_type stream{};  member
68 …PETSC_NODISCARD const cupmStream_t &get(stream_tag) const noexcept { return this->stream.get_strea… in get()
115 PetscCallCUPMBLAS(cupmBlasSetStream(handle, dci->stream.get_stream())); in initialize_handle_()
142 PetscCallCUPMSOLVER(cupmSolverSetStream(handle, dci->stream.get_stream())); in initialize_handle_()
284 PetscCall(dci->stream.destroy()); in destroy()
300 PetscCall(dci->stream.destroy()); in changeStreamType()
315 PetscCall(dci->stream.change_type(dctx->streamType)); in setUp()
328 switch (auto cerr = cupmStreamQuery(impls_cast_(dctx)->stream.get_stream())) { in query()
353 PetscCallCUPM(cupmEventRecord(event, dcib->stream.get_stream())); in waitForContext()
354 PetscCallCUPM(cupmStreamWaitEvent(impls_cast_(dctxa)->stream.get_stream(), event, 0)); in waitForContext()
365 if (!idle) PetscCallCUPM(cupmStreamSynchronize(impls_cast_(dctx)->stream.get_stream())); in synchronize()
407 PetscCallCUPM(cupmEventRecord(dci->begin, dci->stream.get_stream())); in beginTimer()
424 PetscCallCUPM(cupmEventRecord(end, dci->stream.get_stream())); in endTimer()
440 PetscCallCUPM(cupmStreamSynchronize(dci->stream.get_stream())); in getPower()
477 PetscCallCUPM(cupmStreamSynchronize(dci->stream.get_stream())); in endEnergyMeter()
487 const auto &stream = impls_cast_(dctx)->stream; in memAlloc() local
493 …efault_pool_<HostAllocator<T>>().allocate(n, reinterpret_cast<char **>(dest), &stream, alignment)); in memAlloc()
495 …ault_pool_<DeviceAllocator<T>>().allocate(n, reinterpret_cast<char **>(dest), &stream, alignment)); in memAlloc()
497 if (clear) PetscCallCUPM(cupmMemsetAsync(*dest, 0, n, stream.get_stream())); in memAlloc()
504 const auto &stream = impls_cast_(dctx)->stream; in memFree() local
511 … PetscCall(default_pool_<HostAllocator<T>>().deallocate(reinterpret_cast<char **>(ptr), &stream)); in memFree()
520 PetscCallCUPM(cupmFreeAsync(*ptr, stream.get_stream())); in memFree()
524 …PetscCall(default_pool_<DeviceAllocator<T>>().deallocate(reinterpret_cast<char **>(ptr), &stream)); in memFree()
526 if (*ptr) PetscCallCUPM(cupmFreeAsync(*ptr, stream.get_stream())); in memFree()
534 const auto stream = impls_cast_(dctx)->stream.get_stream(); in memCopy() local
539 const auto cerr = cupmStreamQuery(stream); in memCopy()
553 PetscCallCUPM(cupmMemcpyAsync(dest, src, n, PetscDeviceCopyModeToCUPMMemcpyKind(mode), stream)); in memCopy()
563 …PetscCallCUPM(cupmMemsetAsync(ptr, static_cast<int>(v), n, impls_cast_(dctx)->stream.get_stream())… in memSet()
585 PetscCall(impls_cast_(dctx)->stream.record_event(*event_cast_(event))); in recordEvent()
593 PetscCall(impls_cast_(dctx)->stream.wait_for_event(*event_cast_(event))); in waitForEvent()