Lines Matching refs:X_loc

402     Vec X_loc, Y_loc_transpose = NULL;  in MatCreateCeed()  local
404 PetscCall(DMCreateLocalVector(dm_x, &X_loc)); in MatCreateCeed()
405 PetscCall(VecZeroEntries(X_loc)); in MatCreateCeed()
410 …PetscCall(MatCeedContextCreate(dm_x, dm_y, X_loc, Y_loc_transpose, op_mult, op_mult_transpose, MAT… in MatCreateCeed()
412 PetscCall(VecDestroy(&X_loc)); in MatCreateCeed()
514 PetscCall(VecGetType(ctx->X_loc, &vec_type)); in MatCreateCeed()
1186 PetscErrorCode MatCeedSetLocalVectors(Mat mat, Vec X_loc, Vec Y_loc_transpose) { in MatCeedSetLocalVectors() argument
1191 if (X_loc) { in MatCeedSetLocalVectors()
1194 PetscCall(VecGetSize(ctx->X_loc, &len_old)); in MatCeedSetLocalVectors()
1195 PetscCall(VecGetSize(X_loc, &len_new)); in MatCeedSetLocalVectors()
1198 PetscCall(VecReferenceCopy(X_loc, &ctx->X_loc)); in MatCeedSetLocalVectors()
1223 PetscErrorCode MatCeedGetLocalVectors(Mat mat, Vec *X_loc, Vec *Y_loc_transpose) { in MatCeedGetLocalVectors() argument
1228 if (X_loc) { in MatCeedGetLocalVectors()
1229 *X_loc = NULL; in MatCeedGetLocalVectors()
1230 PetscCall(VecReferenceCopy(ctx->X_loc, X_loc)); in MatCeedGetLocalVectors()
1250 PetscErrorCode MatCeedRestoreLocalVectors(Mat mat, Vec *X_loc, Vec *Y_loc_transpose) { in MatCeedRestoreLocalVectors() argument
1252 if (X_loc) PetscCall(VecDestroy(X_loc)); in MatCeedRestoreLocalVectors()
1412 PetscErrorCode MatCeedContextCreate(DM dm_x, DM dm_y, Vec X_loc, Vec Y_loc_transpose, CeedOperator … in MatCeedContextCreate() argument
1431 if (X_loc) PetscCall(VecReferenceCopy(X_loc, &(*ctx)->X_loc)); in MatCeedContextCreate()
1479 if (X_loc) { in MatCeedContextCreate()
1480 PetscCall(VecGetLocalSize(X_loc, &X_loc_len)); in MatCeedContextCreate()
1557 PetscCall(VecDestroy(&(*ctx)->X_loc)); in MatCeedContextDestroy()
1639 Vec X_loc = ctx->X_loc, Y_loc; in MatMult_Ceed() local
1642 if (!ctx->X_loc) PetscCall(DMGetLocalVector(ctx->dm_x, &X_loc)); in MatMult_Ceed()
1646 PetscCall(DMGlobalToLocal(ctx->dm_x, X, INSERT_VALUES, X_loc)); in MatMult_Ceed()
1649 PetscCall(VecReadPetscToCeed(X_loc, &x_mem_type, ctx->x_loc)); in MatMult_Ceed()
1664 PetscCall(VecReadCeedToPetsc(ctx->x_loc, x_mem_type, X_loc)); in MatMult_Ceed()
1672 if (!ctx->X_loc) PetscCall(DMRestoreLocalVector(ctx->dm_x, &X_loc)); in MatMult_Ceed()
1700 Vec X_loc, Y_loc = ctx->Y_loc_transpose; in MatMultTranspose_Ceed() local
1704 PetscCall(DMGetLocalVector(ctx->dm_x, &X_loc)); in MatMultTranspose_Ceed()
1711 PetscCall(VecZeroEntries(X_loc)); in MatMultTranspose_Ceed()
1712 PetscCall(VecPetscToCeed(X_loc, &x_mem_type, ctx->x_loc)); in MatMultTranspose_Ceed()
1726 PetscCall(VecCeedToPetsc(ctx->x_loc, x_mem_type, X_loc)); in MatMultTranspose_Ceed()
1730 PetscCall(DMLocalToGlobal(ctx->dm_x, X_loc, ADD_VALUES, X)); in MatMultTranspose_Ceed()
1734 PetscCall(DMRestoreLocalVector(ctx->dm_x, &X_loc)); in MatMultTranspose_Ceed()