| /petsc/src/dm/impls/stag/tests/ |
| H A D | ex20.c | 26 Mat Ai; in main() local
31 PetscCall(DMCreateInterpolation(dmc, dmf, &Ai, NULL)); in main()
32 PetscCall(MatCreateVecs(Ai, &vc, &vf)); in main()
34 PetscCall(MatMult(Ai, vc, vf)); in main()
38 PetscCall(MatDestroy(&Ai)); in main()
|
| H A D | ex19.c | 12 Mat Ai; in main() local
20 PetscCall(DMCreateInterpolation(dmCoarse, dm, &Ai, NULL)); in main()
50 PetscCall(MatInterpolate(Ai, globalCoarse, globalFine)); in main()
79 PetscCall(MatDestroy(&Ai)); in main()
|
| H A D | ex15.c | 43 Mat Ai; in main() local
44 PetscCall(DMCreateInterpolation(dmCoarse, dm, &Ai, NULL)); in main()
45 PetscCall(MatDestroy(&Ai)); in main()
|
| /petsc/src/mat/tests/ |
| H A D | ex251.c | 14 …PetscInt Ai[] = {0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12… in main() local
19 PetscInt Annz = PETSC_STATIC_ARRAY_LENGTH(Ai); in main()
34 …for (k = 0; k < Annz; k++) PetscCall(MatSetValue(A, Ai[k], Aj[k], Ai[k] + Aj[k] + 1.0, INSERT_VALU… in main()
|
| H A D | ex250.c | 14 …PetscInt Ai[] = {0, 0, 1, 1, 2, 2, 3, 3, 4, 4, 5, 5, 5, 6, 6, 7, 7, 8, 8, 9, 9, 10, 10, 11, 11, 12… in main() local
19 PetscInt Annz = PETSC_STATIC_ARRAY_LENGTH(Ai); in main()
34 …for (k = 0; k < Annz; k++) PetscCall(MatSetValue(A, Ai[k], Aj[k], Ai[k] + Aj[k] + 1.0, INSERT_VALU… in main()
|
| H A D | ex237.c | 94 const PetscInt *Ai, *Aj; in main() local
104 PetscCall(MatGetRowIJ(A, 0, PETSC_FALSE, PETSC_FALSE, &An, &Ai, &Aj, &done)); in main()
110 PetscCall(PetscMalloc1(Ai[An] * bs[j] * bs[j], &val)); in main()
111 for (i = 0; i < Ai[An]; ++i) in main()
114 PetscCall(MatSeqBAIJSetPreallocationCSR(B, bs[j], Ai, Aj, val)); in main()
117 PetscCall(MatRestoreRowIJ(A, 0, PETSC_FALSE, PETSC_FALSE, &An, &Ai, &Aj, &done)); in main()
160 const PetscInt *Ai, *Aj; in main() local
167 … PetscCall(MatGetRowIJ(A, 0, PETSC_FALSE, flg ? PETSC_FALSE : PETSC_TRUE, &An, &Ai, &Aj, &done)); in main()
170 PetscCall(PetscMalloc1(Ai[An], &ja_ptr)); in main()
172 for (k = 0; k < An + 1; ++k) ia_ptr[k] = Ai[k]; in main()
[all …]
|
| /petsc/src/tao/constrained/tutorials/ |
| H A D | ex1.c | 42 Mat Ae, Ai, H; member
90 …PetscCall(TaoSetJacobianInequalityRoutine(tao, user.Ai, user.Ai, FormInequalityJacobian, (void *)&… in main()
122 PetscCall(FormInequalityJacobian(tao, user.x, user.Ai, user.Ai, (void *)&user)); in main()
123 PetscCall(MatCreateVecs(user.Ai, NULL, &CI)); in main()
127 PetscCall(MatView(user.Ai, PETSC_VIEWER_STDOUT_WORLD)); in main()
221 PetscCall(MatCreate(PETSC_COMM_WORLD, &user->Ai)); in InitializeProblem()
222 PetscCall(MatSetSizes(user->Ai, niloc, nloc, user->ni, user->n)); in InitializeProblem()
223 PetscCall(MatSetFromOptions(user->Ai)); in InitializeProblem()
224 PetscCall(MatSetUp(user->Ai)); in InitializeProblem()
237 PetscCall(MatDestroy(&user->Ai)); in DestroyProblem()
|
| /petsc/src/tao/constrained/impls/ipm/ |
| H A D | ipm.c | 499 PetscCall(MatDestroy(&ipmP->Ai)); in TaoDestroy_IPM()
582 PetscCall(MatMultTranspose(ipmP->Ai, ipmP->lambdai, ipmP->work)); in IPMComputeKKT()
690 if (!ipmP->Ai) { in IPMUpdateAi()
702 PetscCall(MatCreate(comm, &ipmP->Ai)); in IPMUpdateAi()
703 PetscCall(MatSetType(ipmP->Ai, MATAIJ)); in IPMUpdateAi()
707 PetscCall(MatSetSizes(ipmP->Ai, PETSC_DECIDE, nloc, ipmP->nb, PETSC_DECIDE)); in IPMUpdateAi()
708 PetscCall(MatSetFromOptions(ipmP->Ai)); in IPMUpdateAi()
709 PetscCall(MatMPIAIJSetPreallocation(ipmP->Ai, ipmP->nb, NULL, ipmP->nb, NULL)); in IPMUpdateAi()
710 PetscCall(MatSeqAIJSetPreallocation(ipmP->Ai, PETSC_DEFAULT, nonzeros)); in IPMUpdateAi()
715 PetscCall(MatGetOwnershipRange(ipmP->Ai, &astart, &aend)); in IPMUpdateAi()
[all …]
|
| H A D | ipm.h | 29 Mat Ai; /* JacI (lb) member
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 356 const MatRowMapType *Ai = Ai_h.data(), *Bi = Bi_h.data(); in MatMPIAIJKokkosReduceBegin() local
379 E_RowLen[i] = (Ai[i + 1] - Ai[i]) + (Bi[i + 1] - Bi[i]); in MatMPIAIJKokkosReduceBegin()
449 PetscInt *buf = &sendCol[Ai[i] + Bi[i]]; in MatMPIAIJKokkosReduceBegin()
451 PetscInt alen = Ai[i + 1] - Ai[i], blen = Bi[i + 1] - Bi[i]; in MatMPIAIJKokkosReduceBegin()
456 buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global in MatMPIAIJKokkosReduceBegin()
633 const auto &Ai = A.graph.row_map; in MatMPIAIJKokkosReduceBegin() local
651 PetscInt disp = Ai(i) + Bi(i); in MatMPIAIJKokkosReduceBegin()
652 PetscInt alen = Ai(i + 1) - Ai(i); in MatMPIAIJKokkosReduceBegin()
661 val = Aa(Ai(i) + j - nzleft); in MatMPIAIJKokkosReduceBegin()
751 PetscInt n1 = B->cmap->n, *Ai = aseq->i, *Aj = aseq->j, *Bi = bseq->i, *Bj = bseq->j; in MatMPIAIJKokkosBcastBegin() local
[all …]
|
| /petsc/src/tao/constrained/impls/almm/ |
| H A D | almm.h | 12 Mat Ae, Ai; /* aliased constraint Jacobians (do not destroy!) */ member
|
| H A D | almm.c | 170 auglag->Ai = tao->jacobian_inequality; in TaoSetUp_ALMM()
616 PetscCall(TaoComputeJacobianInequality(tao, auglag->Px, auglag->Ai, auglag->Ai)); in TaoALMMEvaluateIterate_Private()
625 PetscCall(MatScale(auglag->Ai, -1.0)); in TaoALMMEvaluateIterate_Private()
665 PetscCall(MatMultTransposeAdd(auglag->Ai, auglag->Ciwork, auglag->LgradX, auglag->LgradX)); in TaoALMMComputePHRLagAndGradient_Private()
705 PetscCall(MatMultTransposeAdd(auglag->Ai, auglag->Yi, auglag->LgradX, auglag->LgradX)); in TaoALMMComputeAugLagAndGradient_Private()
707 PetscCall(MatMultTranspose(auglag->Ai, auglag->Ci, auglag->Xwork)); in TaoALMMComputeAugLagAndGradient_Private()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 205 const PetscInt *Ai = a->i, *Aj = a->j, *adiag; in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU() local
215 …Mnz = (Ai[m] - Ai[0]) + (adiag[0] - adiag[m]); // Lnz (without the unit diagonal) + Unz (with the … in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
221 PetscInt llen = Ai[i + 1] - Ai[i]; in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
223 … PetscCall(PetscArraycpy(Mj + Mi[i], Aj + Ai[i], llen)); // entries of L in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
279 PetscInt llen = Ai[i + 1] - Ai[i]; in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
281 … PetscCall(PetscArraycpy(Ma + Mi[i], Aa + Ai[i], llen)); // entries of L in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
623 const PetscInt *Ai = a->i, *Aj = a->j, *adiag; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky() local
634 Mnz = Ai[m]; // Unz (with the unit diagonal) in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
639 PetscInt ulen = Ai[i + 1] - Ai[i]; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
640 Mj[Ai[i]] = i; // diagonal entry in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
[all …]
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 266 const PetscInt *Ai = aseq->i, *Aj = aseq->j; in MatSeqAIJKokkosGenerateTransposeStructure() local
286 for (PetscInt j = Ai[i]; j < Ai[i + 1]; j++) { // A's (i,j) is T's (j,i) in MatSeqAIJKokkosGenerateTransposeStructure()
1103 const auto &Ai = aijkok->i_dual.view_device(); in MatDiagonalScale_SeqAIJKokkos() local
1112 PetscInt len = Ai(i + 1) - Ai(i); in MatDiagonalScale_SeqAIJKokkos()
1114 …Kokkos::parallel_for(Kokkos::TeamThreadRange(t, len), [&](PetscInt j) { Aa(Ai(i) + j) *= lv(i); }); in MatDiagonalScale_SeqAIJKokkos()
1163 const auto &Ai = aijkok->i_dual.view_device(); in MatGetDiagonal_SeqAIJKokkos() local
1169 if (Adiag(i) < Ai(i + 1)) xv(i) = Aa(Adiag(i)); in MatGetDiagonal_SeqAIJKokkos()
1502 auto Ai = akok->i_dual.view_device(); in MatInvertVariableBlockDiagonal_SeqAIJKokkos() local
1522 if (Ai(i) <= Adiag(i) && Adiag(i) < Ai(i + 1)) { // if the diagonal exists (common case) in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1526 …if (first + c < Ai(i) || first + c >= Ai(i + 1)) { // this entry (first+c) is out of range of this… in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
[all …]
|
| /petsc/src/mat/impls/composite/ |
| H A D | mcomposite.c | 780 static PetscErrorCode MatCompositeGetMat_Composite(Mat mat, PetscInt i, Mat *Ai) in MatCompositeGetMat_Composite() argument
791 *Ai = ilink->mat; in MatCompositeGetMat_Composite()
811 PetscErrorCode MatCompositeGetMat(Mat mat, PetscInt i, Mat *Ai) in MatCompositeGetMat() argument
816 PetscAssertPointer(Ai, 3); in MatCompositeGetMat()
817 PetscUseMethod(mat, "MatCompositeGetMat_C", (Mat, PetscInt, Mat *), (mat, i, Ai)); in MatCompositeGetMat()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 1431 const int *Ai, *Aj; in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0() local
1439 PetscCall(MatSeqAIJHIPSPARSEGetIJ(A, PETSC_FALSE, &Ai, &Aj)); /* Do not use compressed Ai */ in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1440 …PetscCallHIP(hipMemcpyAsync(fs->csrRowPtr, Ai, sizeof(int) * (m + 1), hipMemcpyDeviceToDevice, Pet… in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1524 PetscInt *Ai, nzRow, nzLeft; in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0() local
1529 Ai = Aseq->i; in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1531 …if (Ai[i] < Adiag[i] && Adiag[i] < Ai[i + 1]) { /* There are nonzeros left to the diagonal of row … in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1532 nzRow = Ai[i + 1] - Ai[i]; in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1533 nzLeft = Adiag[i] - Ai[i]; in MatILUFactorSymbolic_SeqAIJHIPSPARSE_ILU0()
1676 const int *Ai, *Aj; in MatICCFactorSymbolic_SeqAIJHIPSPARSE_ICC0() local
1684 PetscCall(MatSeqAIJHIPSPARSEGetIJ(A, PETSC_FALSE, &Ai, &Aj)); /* Do not use compressed Ai */ in MatICCFactorSymbolic_SeqAIJHIPSPARSE_ICC0()
[all …]
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aij.c | 4559 …PetscInt *Ai; /* Change to PetscCount once we use it for ro… in MatSetPreallocationCOO_SeqAIJ() local
4600 PetscCall(PetscShmgetAllocateArray(M + 1, sizeof(PetscInt), (void **)&Ai)); /* CSR of A */ in MatSetPreallocationCOO_SeqAIJ()
4601 PetscCall(PetscArrayzero(Ai, M + 1)); in MatSetPreallocationCOO_SeqAIJ()
4607 Ai++; /* Inc by 1 for convenience */ in MatSetPreallocationCOO_SeqAIJ()
4667 PetscCall(PetscIntCast(end - start, Ai + row)); in MatSetPreallocationCOO_SeqAIJ()
4668 nnz += Ai[row]; // q is already advanced in MatSetPreallocationCOO_SeqAIJ()
4673 Ai[row] = 1; in MatSetPreallocationCOO_SeqAIJ()
4681 Ai[row]++; in MatSetPreallocationCOO_SeqAIJ()
4691 Ai--; /* Back to the beginning of Ai[] */ in MatSetPreallocationCOO_SeqAIJ()
4692 for (k = 0; k < M; k++) Ai[k + 1] += Ai[k]; in MatSetPreallocationCOO_SeqAIJ()
[all …]
|
| /petsc/src/dm/impls/plex/ |
| H A D | plex.c | 10074 PetscReal *ci, *fi, *Ai; in DMPlexComputeOrthogonalQuality() local
10125 PetscCall(PetscMalloc5(cEnd - cStart, &idx, cEnd - cStart, &oqVals, nc, &ci, nc, &fi, nc, &Ai)); in DMPlexComputeOrthogonalQuality()
10164 Ai[i] = fg->normal[i]; in DMPlexComputeOrthogonalQuality()
10167 normai += PetscPowReal(Ai[i], 2); in DMPlexComputeOrthogonalQuality()
10177 Ai[i] = Ai[i] / normai; in DMPlexComputeOrthogonalQuality()
10179 cArr[cellneighiter] += PetscAbs(Ai[i] * ci[i]); in DMPlexComputeOrthogonalQuality()
10180 fArr[cellneighiter] += PetscAbs(Ai[i] * fi[i]); in DMPlexComputeOrthogonalQuality()
10202 PetscCall(PetscFree5(idx, oqVals, ci, fi, Ai)); in DMPlexComputeOrthogonalQuality()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 6566 PetscInt *Ai, *Bi; in MatSetPreallocationCOO_MPIAIJ() local
6569 PetscCall(PetscMalloc1(m + 1, &Ai)); in MatSetPreallocationCOO_MPIAIJ()
6580 …rnal(mat, j1, j2, rowBegin1, rowMid1, rowBegin2, rowMid2, Ajmap1, Ajmap2, Aimap1, Aimap2, Ai, Aj)); in MatSetPreallocationCOO_MPIAIJ()
6585 PetscInt Annz = Ai[m]; in MatSetPreallocationCOO_MPIAIJ()
6640 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, m, n, Ai, Aj, Aa, &mpiaij->A)); in MatSetPreallocationCOO_MPIAIJ()
|