Lines Matching refs:dctx
26 static inline PetscErrorCode PetscSetDefaultCUPMStreamFromDeviceContext(PetscDeviceContext dctx, Pe… in PetscSetDefaultCUPMStreamFromDeviceContext() argument
33 PetscCall(PetscDeviceContextGetStreamHandle_Internal(dctx, &handle)); in PetscSetDefaultCUPMStreamFromDeviceContext()
41 PetscCall(PetscDeviceContextGetStreamHandle_Internal(dctx, &handle)); in PetscSetDefaultCUPMStreamFromDeviceContext()
46 (void)dctx, (void)dtype; in PetscSetDefaultCUPMStreamFromDeviceContext()
107 PetscErrorCode PetscDeviceContextGetCurrentContext(PetscDeviceContext *dctx) in PetscDeviceContextGetCurrentContext() argument
110 PetscAssertPointer(dctx, 1); in PetscDeviceContextGetCurrentContext()
115 *dctx = globalContext; in PetscDeviceContextGetCurrentContext()
142 PetscErrorCode PetscDeviceContextSetCurrentContext(PetscDeviceContext dctx) in PetscDeviceContextSetCurrentContext() argument
147 PetscCall(PetscDeviceContextGetOptionalNullContext_Internal(&dctx)); in PetscDeviceContextSetCurrentContext()
148 …dctx->setup, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONGSTATE, "PetscDeviceContext %" PetscInt64_FMT " mu… in PetscDeviceContextSetCurrentContext()
149 PetscCall(PetscDeviceContextGetDeviceType(dctx, &dtype)); in PetscDeviceContextSetCurrentContext()
151 globalContext = dctx; in PetscDeviceContextSetCurrentContext()
152 …PetscCall(PetscInfo(dctx, "Set global PetscDeviceContext id %" PetscInt64_FMT "\n", PetscObjectCas… in PetscDeviceContextSetCurrentContext()