| /petsc/src/mat/impls/aij/mpi/pastix/ |
| H A D | pastix.c | 56 Mat A_loc, A_aij; in MatConvertToSPM() local
76 if (ismpiaij || ismpisbaij) PetscCall(MatMPIAIJGetLocalMat(A_aij, MAT_INITIAL_MATRIX, &A_loc)); in MatConvertToSPM()
77 else if (isseqaij || isseqsbaij) A_loc = A_aij; in MatConvertToSPM()
81 PetscCall(MatGetRowIJ(A_loc, 0, PETSC_FALSE, PETSC_FALSE, &n, &row, &col, &flag)); in MatConvertToSPM()
83 PetscCall(MatSeqAIJGetArrayRead(A_loc, &val)); in MatConvertToSPM()
106 PetscCall(MatRestoreRowIJ(A_loc, 0, PETSC_FALSE, PETSC_FALSE, &n, &row, &col, &flag)); in MatConvertToSPM()
108 PetscCall(MatSeqAIJRestoreArrayRead(A_loc, &val)); in MatConvertToSPM()
109 if (ismpiaij || ismpisbaij) PetscCall(MatDestroy(&A_loc)); in MatConvertToSPM()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimatmatmult.c | 1309 PetscCall(MatMPIAIJGetLocalMat(A, MAT_INITIAL_MATRIX, &ptap->A_loc)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1316 PetscCall(MatMatMultSymbolic_SeqAIJ_SeqAIJ(ptap->Ro, ptap->A_loc, fill, ptap->C_oth)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1373 PetscCall(MatMatMultSymbolic_SeqAIJ_SeqAIJ(ptap->Rd, ptap->A_loc, fill, ptap->C_loc)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1382 a_loc = (Mat_SeqAIJ *)ptap->A_loc->data; in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1386 MatRowMergeMax_SeqAIJ(a_loc, ptap->A_loc->rmap->N, ta); in MatTransposeMatMultSymbolic_MPIAIJ_MPIAIJ_nonscalable()
1526 Mat A_loc, C_loc, C_oth; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable() local
1535 …PetscCheck(ptap->A_loc, PetscObjectComm((PetscObject)C), PETSC_ERR_ARG_WRONGSTATE, "PtA cannot be … in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
1546 PetscCall(MatMPIAIJGetLocalMat(A, MAT_REUSE_MATRIX, &ptap->A_loc)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
1549 A_loc = ptap->A_loc; in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
1550 PetscCall(ptap->C_loc->ops->matmultnumeric(ptap->Rd, A_loc, ptap->C_loc)); in MatTransposeMatMultNumeric_MPIAIJ_MPIAIJ_nonscalable()
[all …]
|
| H A D | mpiaij.c | 5077 PetscErrorCode MatAIJGetLocalMat(Mat A, Mat *A_loc) in MatAIJGetLocalMat() argument
5084 PetscCall(MatMPIAIJGetLocalMat(A, MAT_INITIAL_MATRIX, A_loc)); in MatAIJGetLocalMat()
5086 *A_loc = A; in MatAIJGetLocalMat()
5087 PetscCall(PetscObjectReference((PetscObject)*A_loc)); in MatAIJGetLocalMat()
5120 PetscErrorCode MatMPIAIJGetLocalMat(Mat A, MatReuse scall, Mat *A_loc) in MatMPIAIJGetLocalMat() argument
5139 *A_loc = mpimat->A; in MatMPIAIJGetLocalMat()
5141 PetscCall(MatCopy(mpimat->A, *A_loc, SAME_NONZERO_PATTERN)); in MatMPIAIJGetLocalMat()
5187 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, am, A->cmap->N, ci, cj, ca, A_loc)); in MatMPIAIJGetLocalMat()
5190 mat = (Mat_SeqAIJ *)(*A_loc)->data; in MatMPIAIJGetLocalMat()
5195 mat = (Mat_SeqAIJ *)(*A_loc)->data; in MatMPIAIJGetLocalMat()
[all …]
|
| H A D | mpiaij.h | 22 Mat A_loc; /* used by MatTransposeMatMult(), contains api and apj */ member
|
| H A D | mpiptap.c | 52 PetscCall(MatDestroy(&ptap->A_loc)); /* used by MatTransposeMatMult() */ in MatProductCtxDestroy_MPIAIJ_PtAP()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 240 …tscErrorCode MatMPIAIJGetLocalMatMerge_MPIAIJHIPSPARSE(Mat A, MatReuse scall, IS *glob, Mat *A_loc) in MatMPIAIJGetLocalMatMerge_MPIAIJHIPSPARSE() argument
247 PetscCall(MatSeqAIJHIPSPARSEMergeMats(Ad, Ao, scall, A_loc)); in MatMPIAIJGetLocalMatMerge_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 240 …etscErrorCode MatMPIAIJGetLocalMatMerge_MPIAIJCUSPARSE(Mat A, MatReuse scall, IS *glob, Mat *A_loc) in MatMPIAIJGetLocalMatMerge_MPIAIJCUSPARSE() argument
247 PetscCall(MatSeqAIJCUSPARSEMergeMats(Ad, Ao, scall, A_loc)); in MatMPIAIJGetLocalMatMerge_MPIAIJCUSPARSE()
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mpisell.c | 1572 PetscErrorCode MatMPISELLGetLocalMatCondensed(Mat A, MatReuse scall, IS *row, IS *col, Mat *A_loc) in MatMPISELLGetLocalMatCondensed() argument
1611 aloc[0] = *A_loc; in MatMPISELLGetLocalMatCondensed()
1614 *A_loc = aloc[0]; in MatMPISELLGetLocalMatCondensed()
|
| /petsc/src/ksp/pc/impls/bddc/ |
| H A D | bddcprivate.c | 9809 Mat A_loc; in MatAIJExtractRows() local
9825 PetscCall(MatMPIAIJGetLocalMat(A, MAT_INITIAL_MATRIX, &A_loc)); in MatAIJExtractRows()
9826 da = (Mat_SeqAIJ *)A_loc->data; in MatAIJExtractRows()
9862 PetscCall(MatDestroy(&A_loc)); in MatAIJExtractRows()
|