| /petsc/src/mat/impls/normal/ |
| H A D | normmh.c | 43 Mat C, Aa; in MatPermute_NormalHermitian() local
51 Aa = a->A; in MatPermute_NormalHermitian()
52 …PetscCall(ISCreateStride(PetscObjectComm((PetscObject)Aa), Aa->rmap->n, Aa->rmap->rstart, 1, &row)… in MatPermute_NormalHermitian()
54 PetscCall(MatPermute(Aa, row, colp, &C)); in MatPermute_NormalHermitian()
171 Mat_NormalHermitian *Aa; in MatNormalHermitianGetMat_NormalHermitian() local
174 PetscCall(MatShellGetContext(A, &Aa)); in MatNormalHermitianGetMat_NormalHermitian()
175 *M = Aa->A; in MatNormalHermitianGetMat_NormalHermitian()
206 Mat_NormalHermitian *Aa; in MatConvert_NormalHermitian_AIJ() local
213 PetscCall(MatShellGetContext(A, &Aa)); in MatConvert_NormalHermitian_AIJ()
219 PetscCall(MatProductReplaceMats(Aa->A, Aa->A, NULL, B)); in MatConvert_NormalHermitian_AIJ()
[all …]
|
| H A D | normm.c | 60 Mat C, Aa; in MatPermute_Normal() local
68 Aa = a->A; in MatPermute_Normal()
69 …PetscCall(ISCreateStride(PetscObjectComm((PetscObject)Aa), Aa->rmap->n, Aa->rmap->rstart, 1, &row)… in MatPermute_Normal()
71 PetscCall(MatPermute(Aa, row, colp, &C)); in MatPermute_Normal()
187 Mat_Normal *Aa; in MatNormalGetMat_Normal() local
190 PetscCall(MatShellGetContext(A, &Aa)); in MatNormalGetMat_Normal()
191 *M = Aa->A; in MatNormalGetMat_Normal()
222 Mat_Normal *Aa; in MatConvert_Normal_AIJ() local
229 PetscCall(MatShellGetContext(A, &Aa)); in MatConvert_Normal_AIJ()
235 PetscCall(MatProductReplaceMats(Aa->A, Aa->A, NULL, B)); in MatConvert_Normal_AIJ()
[all …]
|
| /petsc/src/ts/impls/implicit/alpha/ |
| H A D | alpha2.c | 26 Vec A0, Aa, A1; member
108 Vec Xa = th->Xa, Va = th->Va, Aa = th->Aa; in TSAlpha_StageVecs() local
131 PetscCall(VecWAXPY(Aa, -1.0, A0, A1)); in TSAlpha_StageVecs()
132 PetscCall(VecAYPX(Aa, Alpha_m, A0)); in TSAlpha_StageVecs()
386 Vec Xa = th->Xa, Va = th->Va, Aa = th->Aa; in SNESTSFormFunction_Alpha() local
391 PetscCall(TSComputeI2Function(ts, ta, Xa, Va, Aa, F)); in SNESTSFormFunction_Alpha()
400 Vec Xa = th->Xa, Va = th->Va, Aa = th->Aa; in SNESTSFormJacobian_Alpha() local
405 PetscCall(TSComputeI2Jacobian(ts, ta, Xa, Va, Aa, dVdX, dAdX, J, P)); in SNESTSFormJacobian_Alpha()
421 PetscCall(VecDestroy(&th->Aa)); in TSReset_Alpha()
455 PetscCall(VecDuplicate(ts->vec_sol, &th->Aa)); in TSSetUp_Alpha()
|
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 127 …nt Annz, const PetscCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, … in MatAddLocalCOOValues() argument
135 Aa[i] = (imode == INSERT_VALUES ? 0.0 : Aa[i]) + sum; in MatAddLocalCOOValues()
144 …imap2[], const PetscCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2,… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
162 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
198 PetscCall(MatSeqAIJHIPSPARSEGetArrayWrite(A, &Aa)); /* write matrix values */ in MatSetValuesCOO_MPIAIJHIPSPARSE()
201 PetscCall(MatSeqAIJHIPSPARSEGetArray(A, &Aa)); /* read & write matrix values */ in MatSetValuesCOO_MPIAIJHIPSPARSE()
216 …56), dim3(256), 0, PetscDefaultHipStream, v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm… in MatSetValuesCOO_MPIAIJHIPSPARSE()
223 …), dim3(256), 0, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjma… in MatSetValuesCOO_MPIAIJHIPSPARSE()
229 PetscCall(MatSeqAIJHIPSPARSERestoreArrayWrite(A, &Aa)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 127 …nt Annz, const PetscCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, … in MatAddLocalCOOValues() argument
135 Aa[i] = (imode == INSERT_VALUES ? 0.0 : Aa[i]) + sum; in MatAddLocalCOOValues()
144 …imap2[], const PetscCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2,… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
162 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJCUSPARSE() local
198 PetscCall(MatSeqAIJCUSPARSEGetArrayWrite(A, &Aa)); /* write matrix values */ in MatSetValuesCOO_MPIAIJCUSPARSE()
201 PetscCall(MatSeqAIJCUSPARSEGetArray(A, &Aa)); /* read & write matrix values */ in MatSetValuesCOO_MPIAIJCUSPARSE()
216 …<(int)((Annz + Bnnz + 255) / 256), 256>>>(v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm… in MatSetValuesCOO_MPIAIJCUSPARSE()
223 …t)((Annz2 + Bnnz2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjma… in MatSetValuesCOO_MPIAIJCUSPARSE()
229 PetscCall(MatSeqAIJCUSPARSERestoreArrayWrite(A, &Aa)); in MatSetValuesCOO_MPIAIJCUSPARSE()
[all …]
|
| /petsc/src/mat/impls/baij/seq/ |
| H A D | aijbaij.c | 61 Mat_SeqAIJ *Aa = (Mat_SeqAIJ *)A->data; in MatConvert_SeqAIJ_SeqBAIJ_Preallocate() local
63 const PetscInt *ai = Aa->i, *aj = Aa->j; in MatConvert_SeqAIJ_SeqBAIJ_Preallocate()
|
| H A D | baij.c | 1899 static PetscErrorCode MatView_SeqBAIJ_Draw_Zoom(PetscDraw draw, void *Aa) in MatView_SeqBAIJ_Draw_Zoom() argument
1901 Mat A = (Mat)Aa; in MatView_SeqBAIJ_Draw_Zoom()
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 321 const auto &Aa = akok->a_dual.view_device(); in MatSeqAIJKokkosGenerateTranspose_Private() local
332 …PetscGetKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { Ta(i) = Aa(perm(i)); })); in MatSeqAIJKokkosGenerateTranspose_Private()
339 …etKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { T.values(i) = Aa(perm(i)); })); in MatSeqAIJKokkosGenerateTranspose_Private()
359 const auto &Aa = akok->a_dual.view_device(); in MatSeqAIJKokkosGenerateHermitian_Private() local
370 …kosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { Ta(i) = PetscConj(Aa(perm(i))); })); in MatSeqAIJKokkosGenerateHermitian_Private()
377 …cutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt i) { T.values(i) = PetscConj(Aa(perm(i))); })); in MatSeqAIJKokkosGenerateHermitian_Private()
1046 const auto &Aa = aijkok->a_dual.view_device(); in MatShift_SeqAIJKokkos() local
1048 …<>(PetscGetKokkosExecutionSpace(), 0, n), KOKKOS_LAMBDA(const PetscInt i) { Aa(Adiag(i)) += a; })); in MatShift_SeqAIJKokkos()
1075 const auto &Aa = aijkok->a_dual.view_device(); in MatDiagonalSet_SeqAIJKokkos() local
1079 if (is == INSERT_VALUES) Aa(Adiag(i)) = dv(i); in MatDiagonalSet_SeqAIJKokkos()
[all …]
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | mpiaijsbaij.c | 55 Mat_SeqBAIJ *Aa = (Mat_SeqBAIJ *)mpimat->A->data, *Ba = (Mat_SeqBAIJ *)mpimat->B->data; in MatConvert_MPIBAIJ_MPISBAIJ() local
73 d_nnz[i] = Aa->i[i + 1] - adiag[i]; in MatConvert_MPIBAIJ_MPISBAIJ()
|
| H A D | mpisbaij.c | 1187 Mat_SeqSBAIJ *Aa = (Mat_SeqSBAIJ *)mat->A->data; in MatGetRow_MPISBAIJ() local
1191 tmp = Aa->i[i + 1] - Aa->i[i] + Ba->i[i + 1] - Ba->i[i]; /* row length */ in MatGetRow_MPISBAIJ()
|
| /petsc/src/mat/tests/ |
| H A D | ex237.c | 93 PetscScalar *val, *Aa; in main() local
106 PetscCall(MatSeqAIJGetArray(A, &Aa)); in main()
112 for (k = 0; k < bs[j] * bs[j]; ++k) val[i * bs[j] * bs[j] + k] = Aa[i] * ptr[k]; in main()
116 PetscCall(MatSeqAIJRestoreArray(A, &Aa)); in main()
|
| H A D | ex70.c | 15 const PetscScalar *Aa; in CheckLocal() local
16 PetscCall(MatDenseGetArrayRead(A, &Aa)); in CheckLocal()
17 wA = (PetscBool)(a != Aa); in CheckLocal()
22 if (Aa[j * lda + i] != MAGIC_NUMBER) wAv = PETSC_TRUE; in CheckLocal()
25 PetscCall(MatDenseRestoreArrayRead(A, &Aa)); in CheckLocal()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 631 auto &Aa = A.values; in MatMPIAIJKokkosReduceBegin() local
661 val = Aa(Ai(i) + j - nzleft); in MatMPIAIJKokkosReduceBegin()
954 const auto &Aa = akok->a_dual.view_device(); in MatMPIAIJKokkosBcastBegin() local
988 rootBuf(disp + j) = Aa(Ai(i) + j - nzleft); in MatMPIAIJKokkosBcastBegin()
1581 MatScalarKokkosView Aa, Ba; in MatSetValuesCOO_MPIAIJKokkos() local
1619 PetscCall(MatSeqAIJGetKokkosViewWrite(A, &Aa)); /* write matrix values */ in MatSetValuesCOO_MPIAIJKokkos()
1622 PetscCall(MatSeqAIJGetKokkosView(A, &Aa)); /* read & write matrix values */ in MatSetValuesCOO_MPIAIJKokkos()
1638 Aa(i) = (imode == INSERT_VALUES ? 0.0 : Aa(i)) + sum; in MatSetValuesCOO_MPIAIJKokkos()
1651 for (PetscCount k = Ajmap2(i); k < Ajmap2(i + 1); k++) Aa(Aimap2(i)) += v2(Aperm2(k)); in MatSetValuesCOO_MPIAIJKokkos()
1660 PetscCall(MatSeqAIJRestoreKokkosViewWrite(A, &Aa)); /* Increase A & B's state etc. */ in MatSetValuesCOO_MPIAIJKokkos()
[all …]
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | aijsbaij.c | 124 Mat_SeqAIJ *Aa = (Mat_SeqAIJ *)A->data; in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate() local
126 const PetscInt *ai = Aa->i, *aj = Aa->j; in MatConvert_SeqAIJ_SeqSBAIJ_Preallocate()
|
| H A D | sbaij.c | 383 static PetscErrorCode MatView_SeqSBAIJ_Draw_Zoom(PetscDraw draw, void *Aa) in MatView_SeqSBAIJ_Draw_Zoom() argument
385 Mat A = (Mat)Aa; in MatView_SeqSBAIJ_Draw_Zoom()
|
| /petsc/src/mat/matfd/ |
| H A D | fdmatrix.c | 22 static PetscErrorCode MatFDColoringView_Draw_Zoom(PetscDraw draw, void *Aa) in MatFDColoringView_Draw_Zoom() argument
24 MatFDColoring fd = (MatFDColoring)Aa; in MatFDColoringView_Draw_Zoom()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 967 static PetscErrorCode MatView_SeqAIJ_Draw_Zoom(PetscDraw draw, void *Aa) in MatView_SeqAIJ_Draw_Zoom() argument
969 Mat A = (Mat)Aa; in MatView_SeqAIJ_Draw_Zoom()
1688 PetscScalar *Aa; in MatShift_SeqAIJ() local
1690 PetscCall(MatSeqAIJGetArray(A, &Aa)); in MatShift_SeqAIJ()
1691 for (PetscInt i = 0; i < A->rmap->n; i++) Aa[diag[i]] += v; in MatShift_SeqAIJ()
1692 PetscCall(MatSeqAIJRestoreArray(A, &Aa)); in MatShift_SeqAIJ()
1697 const PetscScalar *Aa; in MatShift_SeqAIJ() local
1706 PetscCall(MatSeqAIJGetArrayRead(A, &Aa)); // sync the host in MatShift_SeqAIJ()
1707 PetscCall(MatSeqAIJRestoreArrayRead(A, &Aa)); in MatShift_SeqAIJ()
4561 PetscScalar *Aa; in MatSetPreallocationCOO_SeqAIJ() local
[all …]
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 206 const MatScalar *Aa = a->a; in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU() local
281 … PetscCall(PetscArraycpy(Ma + Mi[i], Aa + Ai[i], llen)); // entries of L in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
282 …Ma[Mi[i] + llen] = (MatScalar)1.0 / Aa[adiag[i]]; // recover the d… in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
283 …PetscCall(PetscArraycpy(Ma + Mi[i] + llen + 1, Aa + adiag[i + 1] + 1, ulen - 1)); // entries of U … in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
624 const MatScalar *Aa = a->a; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky() local
690 D[i] = Aa[adiag[i]]; // actually Aa[adiag[i]] is the inverse of the diagonal in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
692 for (PetscInt k = 0; k < Ai[i + 1] - Ai[i] - 1; k++) Ma[Ai[i] + 1 + k] = -Aa[Ai[i] + k]; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
4431 PetscScalar *Aa; in MatSetValuesCOO_SeqAIJCUSPARSE() local
4447 if (imode == INSERT_VALUES) PetscCall(MatSeqAIJCUSPARSEGetArrayWrite(A, &Aa)); in MatSetValuesCOO_SeqAIJCUSPARSE()
4448 else PetscCall(MatSeqAIJCUSPARSEGetArray(A, &Aa)); in MatSetValuesCOO_SeqAIJCUSPARSE()
[all …]
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mpisell.c | 1680 Mat_SeqAIJ *Aa = (Mat_SeqAIJ *)a->A->data, *Ba = (Mat_SeqAIJ *)a->B->data; in MatConvert_MPIAIJ_MPISELL() local
1685 d_nnz[i] = Aa->i[i + 1] - Aa->i[i]; in MatConvert_MPIAIJ_MPISELL()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 1728 Mat_SeqAIJ *Aa = (Mat_SeqAIJ *)mat->A->data, *Ba = (Mat_SeqAIJ *)mat->B->data; in MatGetRow_MPIAIJ() local
1731 tmp = Aa->i[i + 1] - Aa->i[i] + Ba->i[i + 1] - Ba->i[i]; in MatGetRow_MPIAIJ()
6625 PetscScalar *Aa, *Ba; in MatSetPreallocationCOO_MPIAIJ() local
6629 PetscCall(PetscCalloc1(Annz, &Aa)); /* Zero matrix on device */ in MatSetPreallocationCOO_MPIAIJ()
6640 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, m, n, Ai, Aj, Aa, &mpiaij->A)); in MatSetPreallocationCOO_MPIAIJ()
6709 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJ() local
6736 PetscCall(MatSeqAIJGetArray(A, &Aa)); /* Might read and write matrix values */ in MatSetValuesCOO_MPIAIJ()
6748 Aa[i] = (imode == INSERT_VALUES ? 0.0 : Aa[i]) + sum; in MatSetValuesCOO_MPIAIJ()
6759 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += recvbuf[Aperm2[k]]; in MatSetValuesCOO_MPIAIJ()
6764 PetscCall(MatSeqAIJRestoreArray(A, &Aa)); in MatSetValuesCOO_MPIAIJ()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3788 PetscScalar *Aa; in MatSetValuesCOO_SeqAIJHIPSPARSE() local
3804 if (imode == INSERT_VALUES) PetscCall(MatSeqAIJHIPSPARSEGetArrayWrite(A, &Aa)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3805 else PetscCall(MatSeqAIJHIPSPARSEGetArray(A, &Aa)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3809 …nnz + 255) / 256), dim3(256), 0, PetscDefaultHipStream, v1, Annz, coo->jmap, coo->perm, imode, Aa); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3814 if (imode == INSERT_VALUES) PetscCall(MatSeqAIJHIPSPARSERestoreArrayWrite(A, &Aa)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
3815 else PetscCall(MatSeqAIJHIPSPARSERestoreArray(A, &Aa)); in MatSetValuesCOO_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/sell/seq/ |
| H A D | sell.c | 1304 static PetscErrorCode MatView_SeqSELL_Draw_Zoom(PetscDraw draw, void *Aa) in MatView_SeqSELL_Draw_Zoom() argument
1306 Mat A = (Mat)Aa; in MatView_SeqSELL_Draw_Zoom()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 1316 Mat_SeqBAIJ *Aa = (Mat_SeqBAIJ *)mat->A->data, *Ba = (Mat_SeqBAIJ *)mat->B->data; in MatGetRow_MPIBAIJ() local
1319 tmp = Aa->i[i + 1] - Aa->i[i] + Ba->i[i + 1] - Ba->i[i]; in MatGetRow_MPIBAIJ()
|
| /petsc/src/mat/impls/dense/seq/ |
| H A D | dense.c | 1554 static PetscErrorCode MatView_SeqDense_Draw_Zoom(PetscDraw draw, void *Aa) in MatView_SeqDense_Draw_Zoom() argument
1556 Mat A = (Mat)Aa; in MatView_SeqDense_Draw_Zoom()
|