Lines Matching refs:A2
471 Mat A2, A3; in MatMPIXAIJComputeLocalToGlobalMapping_Private() local
528 PetscCall(MatCreate(comm, &A2)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
529 PetscCall(MatSetType(A2, MATMPIAIJ)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
530 PetscCall(MatSetSizes(A2, A->rmap->n, A->cmap->n, A->rmap->N, A->cmap->N)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
531 PetscCall(MatMPIAIJSetPreallocation(A2, 0, dnz, 0, onz)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
532 PetscCall(MatSetOption(A2, MAT_NO_OFF_PROC_ENTRIES, PETSC_TRUE)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
533 for (i = 0, j = 0; i < A2->rmap->n; i++) { in MatMPIXAIJComputeLocalToGlobalMapping_Private()
534 PetscInt row = i + A2->rmap->rstart, s = dnz[i] + onz[i]; in MatMPIXAIJComputeLocalToGlobalMapping_Private()
539 PetscCall(MatSetValues(A2, 1, &row, s, idxs, vals, INSERT_VALUES)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
546 PetscCall(MatAssemblyBegin(A2, MAT_FINAL_ASSEMBLY)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
547 PetscCall(MatAssemblyEnd(A2, MAT_FINAL_ASSEMBLY)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
548 PetscCall(MatTranspose(A2, MAT_INPLACE_MATRIX, &A2)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
554 PetscCall(MatMPIAIJGetLocalMatCondensed(A2, MAT_INITIAL_MATRIX, &is, NULL, &A3)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()
556 PetscCall(MatDestroy(&A2)); in MatMPIXAIJComputeLocalToGlobalMapping_Private()