| /petsc/src/dm/impls/composite/ |
| H A D | packm.c | 46 PetscInt m, *dnz, *onz, i, j, mA; in DMCreateMatrix_Composite_AIJ() local
67 mA = com->N; in DMCreateMatrix_Composite_AIJ()
68 PetscCall(MatMPIAIJSetPreallocation(*J, mA, NULL, mA - m, NULL)); in DMCreateMatrix_Composite_AIJ()
69 PetscCall(MatSeqAIJSetPreallocation(*J, mA, NULL)); in DMCreateMatrix_Composite_AIJ()
72 PetscCall(PetscMalloc2(mA, &values, mA, &indices)); in DMCreateMatrix_Composite_AIJ()
73 PetscCall(PetscArrayzero(values, mA)); in DMCreateMatrix_Composite_AIJ()
74 for (i = 0; i < mA; i++) indices[i] = i; in DMCreateMatrix_Composite_AIJ()
75 …for (i = rstart; i < rend; i++) PetscCall(MatSetValues(*J, 1, &i, mA, indices, values, INSERT_VALU… in DMCreateMatrix_Composite_AIJ()
100 PetscCall(MatGetLocalSize(Atmp, &mA, NULL)); in DMCreateMatrix_Composite_AIJ()
103 for (i = 0; i < mA; i++) { in DMCreateMatrix_Composite_AIJ()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpimattransposematmult.c | 15 PetscCall(MatDestroy(&atb->mA)); in MatProductCtxDestroy_MPIDense_MatTransMatMult()
42 PetscCall(MatCreateMAIJ(A, B->cmap->N, &atb->mA)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
43 if (!atb->mA->assembled) { in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
44 PetscCall(MatAssemblyBegin(atb->mA, MAT_FINAL_ASSEMBLY)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
45 PetscCall(MatAssemblyEnd(atb->mA, MAT_FINAL_ASSEMBLY)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
47 PetscCall(MatCreateVecs(atb->mA, &atb->ct, &atb->bt)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
86 PetscCall(MatMultTranspose(atb->mA, bt, ct)); in MatTransposeMatMultNumeric_MPIAIJ_MPIDense()
|
| /petsc/src/mat/tests/ |
| H A D | ex33.c | 82 PetscInt mA = 2700, nX = 80, nz = 40; in main() local
104 …PetscCall(MatCreateAIJ(PETSC_COMM_WORLD, PETSC_DECIDE, PETSC_DECIDE, mA, mA, nz, NULL, nz, NULL, &… in main()
112 PetscCall(MatCreateDense(PETSC_COMM_WORLD, PETSC_DECIDE, PETSC_DECIDE, mA, nX, NULL, &X)); in main()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | mattransposematmult.c | 15 PetscCall(MatDestroy(&atb->mA)); in MatProductCtxDestroy_SeqDense_MatTransMatMult()
49 PetscCall(MatCreateMAIJ(A, dofm, &atb->mA)); in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
50 PetscCall(MatCreateVecs(atb->mA, &atb->ct, &atb->bt)); in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
94 PetscCall(MatMultTranspose(atb->mA, bt, ct)); in MatTMatTMultNumeric_SeqAIJ_SeqDense()
119 PetscCall(MatMult(atb->mA, ct, bt)); in MatTMatTMultNumeric_SeqAIJ_SeqDense()
|
| H A D | aij.h | 89 Mat mA; /* maij matrix of A */ member
|
| /petsc/src/vec/is/sf/tests/ |
| H A D | ex5.c | 85 PetscInt i, m, n, k, nl = 2, mA, mB, nldataA, nldataB; in main() local
157 PetscCall(PetscSFGetLeafRange(sfA, NULL, &mA)); in main()
|
| /petsc/src/dm/dt/dualspace/impls/lagrange/ |
| H A D | dspacelagrange.c | 1232 PetscInt m, n, mA, nA, mB, nB, Nk, i, j, l; in MatricesMerge() local
1240 PetscCall(MatGetSize(matA, &mA, &nA)); in MatricesMerge()
1244 m = mA + mB; in MatricesMerge()
1248 for (i = 0; i < mA; i++) { in MatricesMerge()
1254 PetscCall(MatGetRow(matB, i, &nnz[i + mA], NULL, NULL)); in MatricesMerge()
1255 …PetscCheck(nnz[i + mA] % Nk == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "nonzeros in matB are not in k-… in MatricesMerge()
1256 maxnnz = PetscMax(maxnnz, nnz[i + mA]); in MatricesMerge()
1264 for (i = 0; i < mA; i++) { in MatricesMerge()
1280 PetscInt row = i + mA; in MatricesMerge()
|
| /petsc/src/dm/impls/plex/hdf5/ |
| H A D | plexhdf5.c | 3085 PetscInt mA, m, bs; in DMPlexVecLoad_HDF5_Internal() local
3103 PetscCall(PetscSFGetGraph(sf, &mA, &m, &ilocal, NULL)); in DMPlexVecLoad_HDF5_Internal()
3116 PetscCallMPI(MPIU_Allreduce(&mA, &MA, 1, MPIU_INT, MPI_SUM, comm)); in DMPlexVecLoad_HDF5_Internal()
3128 PetscCall(VecSetSizes(vecA, mA, PETSC_DECIDE)); in DMPlexVecLoad_HDF5_Internal()
|