Lines Matching refs:dctx
5 PetscErrorCode PetscDeviceContextCreate_CUDA(PetscDeviceContext dctx) in PetscDeviceContextCreate_CUDA() argument
10 PetscCall(cuda_context.initialize(dctx->device)); in PetscDeviceContextCreate_CUDA()
11 dctx->data = new PetscDeviceContext_(CUDA); in PetscDeviceContextCreate_CUDA()
12 *dctx->ops = cuda_context.ops; in PetscDeviceContextCreate_CUDA()
19 PetscDeviceContext dctx; in PetscCUBLASGetHandle() local
23 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in PetscCUBLASGetHandle()
24 PetscCall(PetscDeviceContextGetBLASHandle_Internal(dctx, handle)); in PetscCUBLASGetHandle()
30 PetscDeviceContext dctx; in PetscCUSOLVERDnGetHandle() local
34 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in PetscCUSOLVERDnGetHandle()
35 PetscCall(PetscDeviceContextGetSOLVERHandle_Internal(dctx, handle)); in PetscCUSOLVERDnGetHandle()
41 PetscDeviceContext dctx; in PetscGetCurrentCUDAStream() local
46 PetscCall(PetscDeviceContextGetCurrentContextAssertType_Internal(&dctx, PETSC_DEVICE_CUDA)); in PetscGetCurrentCUDAStream()
47 PetscCall(PetscDeviceContextGetStreamHandle(dctx, &handle)); in PetscGetCurrentCUDAStream()