Lines Matching refs:PetscCallCUPM
156 PetscCallCUPM(cupmSetDevice(static_cast<int>(devidl))); in check_current_device_()
270 PetscCallCUPM(cupmDeviceGetMemPool(&mempool, static_cast<int>(device->deviceId))); in initialize()
271 PetscCallCUPM(cupmMemPoolSetAttribute(mempool, cupmMemPoolAttrReleaseThreshold, &threshold)); in initialize()
286 if (dci->begin) PetscCallCUPM(cupmEventDestroy(dci->begin)); in destroy()
287 if (dci->end) PetscCallCUPM(cupmEventDestroy(dci->end)); in destroy()
339 PetscCallCUPM(cerr); 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()
404 PetscCallCUPM(cupmEventCreate(&dci->begin)); in beginTimer()
405 PetscCallCUPM(cupmEventCreate(&dci->end)); in beginTimer()
407 PetscCallCUPM(cupmEventRecord(dci->begin, dci->stream.get_stream())); in beginTimer()
424 PetscCallCUPM(cupmEventRecord(end, dci->stream.get_stream())); in endTimer()
425 PetscCallCUPM(cupmEventSynchronize(end)); in endTimer()
426 PetscCallCUPM(cupmEventElapsedTime(>ime, dci->begin, end)); in endTimer()
440 PetscCallCUPM(cupmStreamSynchronize(dci->stream.get_stream())); in getPower()
477 PetscCallCUPM(cupmStreamSynchronize(dci->stream.get_stream())); in endEnergyMeter()
497 if (clear) PetscCallCUPM(cupmMemsetAsync(*dest, 0, n, stream.get_stream())); in memAlloc()
518 PetscCallCUPM(cupmFreeHost(*ptr)); in memFree()
520 PetscCallCUPM(cupmFreeAsync(*ptr, stream.get_stream())); in memFree()
526 if (*ptr) PetscCallCUPM(cupmFreeAsync(*ptr, stream.get_stream())); in memFree()
550 PetscCallCUPM(cerr); 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()