Lines Matching refs:X_loc

350     Vec X_loc, Y_loc_transpose = NULL;  in MatCreateCeed()  local
352 PetscCall(DMCreateLocalVector(dm_x, &X_loc)); in MatCreateCeed()
353 PetscCall(VecZeroEntries(X_loc)); in MatCreateCeed()
358 …PetscCall(MatCeedContextCreate(dm_x, dm_y, X_loc, Y_loc_transpose, op_mult, op_mult_transpose, MAT… in MatCreateCeed()
360 PetscCall(VecDestroy(&X_loc)); in MatCreateCeed()
462 PetscCall(VecGetType(ctx->X_loc, &vec_type)); in MatCreateCeed()
1110 PetscErrorCode MatCeedSetLocalVectors(Mat mat, Vec X_loc, Vec Y_loc_transpose) { in MatCeedSetLocalVectors() argument
1115 if (X_loc) { in MatCeedSetLocalVectors()
1118 PetscCall(VecGetSize(ctx->X_loc, &len_old)); in MatCeedSetLocalVectors()
1119 PetscCall(VecGetSize(X_loc, &len_new)); in MatCeedSetLocalVectors()
1122 PetscCall(VecReferenceCopy(X_loc, &ctx->X_loc)); in MatCeedSetLocalVectors()
1147 PetscErrorCode MatCeedGetLocalVectors(Mat mat, Vec *X_loc, Vec *Y_loc_transpose) { in MatCeedGetLocalVectors() argument
1152 if (X_loc) { in MatCeedGetLocalVectors()
1153 *X_loc = NULL; in MatCeedGetLocalVectors()
1154 PetscCall(VecReferenceCopy(ctx->X_loc, X_loc)); in MatCeedGetLocalVectors()
1174 PetscErrorCode MatCeedRestoreLocalVectors(Mat mat, Vec *X_loc, Vec *Y_loc_transpose) { in MatCeedRestoreLocalVectors() argument
1176 if (X_loc) PetscCall(VecDestroy(X_loc)); in MatCeedRestoreLocalVectors()
1336 PetscErrorCode MatCeedContextCreate(DM dm_x, DM dm_y, Vec X_loc, Vec Y_loc_transpose, CeedOperator … in MatCeedContextCreate() argument
1356 if (X_loc) PetscCall(VecReferenceCopy(X_loc, &(*ctx)->X_loc)); in MatCeedContextCreate()
1402 if (X_loc) { in MatCeedContextCreate()
1403 PetscCall(VecGetLocalSize(X_loc, &X_loc_len)); in MatCeedContextCreate()
1480 PetscCall(VecDestroy(&(*ctx)->X_loc)); in MatCeedContextDestroy()
1562 Vec X_loc = ctx->X_loc, Y_loc; in MatMult_Ceed() local
1565 if (!ctx->X_loc) PetscCall(DMGetLocalVector(ctx->dm_x, &X_loc)); in MatMult_Ceed()
1569 PetscCall(DMGlobalToLocal(ctx->dm_x, X, INSERT_VALUES, X_loc)); in MatMult_Ceed()
1572 PetscCall(VecReadPetscToCeed(X_loc, &x_mem_type, ctx->x_loc)); in MatMult_Ceed()
1584 PetscCall(VecReadCeedToPetsc(ctx->x_loc, x_mem_type, X_loc)); in MatMult_Ceed()
1592 if (!ctx->X_loc) PetscCall(DMRestoreLocalVector(ctx->dm_x, &X_loc)); in MatMult_Ceed()
1623 Vec X_loc, Y_loc = ctx->Y_loc_transpose; in MatMultTranspose_Ceed() local
1627 PetscCall(DMGetLocalVector(ctx->dm_x, &X_loc)); in MatMultTranspose_Ceed()
1634 PetscCall(VecZeroEntries(X_loc)); in MatMultTranspose_Ceed()
1635 PetscCall(VecPetscToCeed(X_loc, &x_mem_type, ctx->x_loc)); in MatMultTranspose_Ceed()
1646 PetscCall(VecCeedToPetsc(ctx->x_loc, x_mem_type, X_loc)); in MatMultTranspose_Ceed()
1650 PetscCall(DMLocalToGlobal(ctx->dm_x, X_loc, ADD_VALUES, X)); in MatMultTranspose_Ceed()
1654 PetscCall(DMRestoreLocalVector(ctx->dm_x, &X_loc)); in MatMultTranspose_Ceed()