Lines Matching refs:dctx
180 static PetscErrorCode PetscDeviceCheckCapable_Private(PetscDeviceContext dctx, bool cond, const cha… in PetscDeviceCheckCapable_Private() argument
183 …ndle %s host memory", PetscObjectCast(dctx)->id, PetscObjectCast(dctx)->name, dctx->device ? Petsc… in PetscDeviceCheckCapable_Private()
325 PetscErrorCode PetscDeviceAllocate_Private(PetscDeviceContext dctx, PetscBool clear, PetscMemType m… in PetscDeviceAllocate_Private() argument
340 PetscCall(PetscDeviceContextGetOptionalNullContext_Internal(&dctx)); in PetscDeviceAllocate_Private()
343 if (dctx->ops->memalloc) { in PetscDeviceAllocate_Private()
344 PetscUseTypeMethod(dctx, memalloc, clear, mtype, n, alignment, ptr); in PetscDeviceAllocate_Private()
346 PetscCall(PetscDeviceCheckCapable_Private(dctx, PetscMemTypeHost(mtype), "allocating")); in PetscDeviceAllocate_Private()
352 …PetscCall(PetscDeviceContextMarkIntentFromID(dctx, id, PETSC_MEMORY_ACCESS_WRITE, "memory allocati… in PetscDeviceAllocate_Private()
391 PetscErrorCode PetscDeviceDeallocate_Private(PetscDeviceContext dctx, void *PETSC_RESTRICT ptr) in PetscDeviceDeallocate_Private() argument
414 PetscCall(PetscDeviceContextGetOptionalNullContext_Internal(&dctx)); in PetscDeviceDeallocate_Private()
417 …PetscCall(PetscDeviceContextMarkIntentFromID(dctx, attr.id, PETSC_MEMORY_ACCESS_WRITE, "memory dea… in PetscDeviceDeallocate_Private()
419 if (dctx->ops->memfree) { in PetscDeviceDeallocate_Private()
420 PetscUseTypeMethod(dctx, memfree, attr.mtype, (void **)&ptr); in PetscDeviceDeallocate_Private()
422 PetscCall(PetscDeviceCheckCapable_Private(dctx, PetscMemTypeHost(attr.mtype), "freeing")); in PetscDeviceDeallocate_Private()
470 PetscErrorCode PetscDeviceMemcpy(PetscDeviceContext dctx, void *PETSC_RESTRICT dest, const void *PE… in PetscDeviceMemcpy() argument
477 PetscCall(PetscDeviceContextGetOptionalNullContext_Internal(&dctx)); in PetscDeviceMemcpy()
483 …PetscCall(PetscDeviceContextMarkIntentFromID(dctx, src_attr.id, PETSC_MEMORY_ACCESS_READ, "memory … in PetscDeviceMemcpy()
484 …PetscCall(PetscDeviceContextMarkIntentFromID(dctx, dest_attr.id, PETSC_MEMORY_ACCESS_WRITE, "memor… in PetscDeviceMemcpy()
486 if (dctx->ops->memcopy) { in PetscDeviceMemcpy()
487 PetscUseTypeMethod(dctx, memcopy, dest, src, n, mode); in PetscDeviceMemcpy()
496 PetscCall(PetscDeviceCheckCapable_Private(dctx, mode == PETSC_DEVICE_COPY_HTOH, "copying")); in PetscDeviceMemcpy()
543 PetscErrorCode PetscDeviceMemset(PetscDeviceContext dctx, void *ptr, PetscInt v, std::size_t n) in PetscDeviceMemset() argument
548 PetscCall(PetscDeviceContextGetOptionalNullContext_Internal(&dctx)); in PetscDeviceMemset()
552 …PetscCall(PetscDeviceContextMarkIntentFromID(dctx, attr.id, PETSC_MEMORY_ACCESS_WRITE, "memory set… in PetscDeviceMemset()
553 if (dctx->ops->memset) { in PetscDeviceMemset()
554 PetscUseTypeMethod(dctx, memset, attr.mtype, ptr, v, n); in PetscDeviceMemset()
558 PetscCall(PetscDeviceCheckCapable_Private(dctx, PetscMemTypeHost(attr.mtype), "memsetting")); in PetscDeviceMemset()