Home
last modified time | relevance | path

Searched refs:Aj (Results 1 – 9 of 9) sorted by relevance

/petsc/src/mat/tests/
H A Dex251.c15 …PetscInt Aj[] = {0, 1, 2, 7, 3, 8, 4, 9, 5, 8, 2, 6, 11, 0, 7, 1, 6, 2, 4, 10, 16, 11, 15, 12, 17,… in main() local
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 Dex250.c15 …PetscInt Aj[] = {0, 1, 2, 7, 3, 8, 4, 9, 5, 8, 2, 6, 11, 0, 7, 1, 6, 2, 4, 10, 16, 11, 15, 12, 17,… in main() local
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 Dex237.c94 const PetscInt *Ai, *Aj; in main() local
104 PetscCall(MatGetRowIJ(A, 0, PETSC_FALSE, PETSC_FALSE, &An, &Ai, &Aj, &done)); 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()
173 for (k = 0; k < Ai[An]; ++k) ja_ptr[k] = Aj[k]; in main()
180 …for (k = 0; k < Ai[An]; ++k) ja_ptr[k] = Aj[k] + 1; /* Fortran indexing to maximize cases covered … in main()
187 …for (k = 0; k < Ai[An]; ++k) ja_ptr[k] = Aj[k] + 1; /* Fortran indexing to maximize cases covered … in main()
199 …PetscCall(MatRestoreRowIJ(A, 0, PETSC_FALSE, flg ? PETSC_FALSE : PETSC_TRUE, &An, &Ai, &Aj, &done)… in main()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx266 const PetscInt *Ai = aseq->i, *Aj = aseq->j; in MatSeqAIJKokkosGenerateTransposeStructure() local
279 for (PetscInt i = 0; i < nz; i++) Ti[Aj[i]]++; in MatSeqAIJKokkosGenerateTransposeStructure()
287 PetscInt r = Aj[j]; // row r of T in MatSeqAIJKokkosGenerateTransposeStructure()
1104 const auto &Aj = aijkok->j_dual.view_device(); in MatDiagonalScale_SeqAIJKokkos() local
1124 …(PetscGetKokkosExecutionSpace(), 0, nz), KOKKOS_LAMBDA(const PetscInt k) { Aa(k) *= rv(Aj(k)); })); in MatDiagonalScale_SeqAIJKokkos()
1503 auto Aj = akok->j_dual.view_device(); in MatInvertVariableBlockDiagonal_SeqAIJKokkos() local
1528 … } else if (Aj(first + c) == rstart + c) { // this entry is right on the (rstart+c) column in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1539 if (rstart <= Aj(j) && Aj(j) < rstart + m) B(r, Aj(j) - rstart) = Aa(j); in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
1540 else if (Aj(j) >= rstart + m) break; in MatInvertVariableBlockDiagonal_SeqAIJKokkos()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu205 const PetscInt *Ai = a->i, *Aj = a->j, *adiag; in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU() local
223 … PetscCall(PetscArraycpy(Mj + Mi[i], Aj + Ai[i], llen)); // entries of L in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
225 …PetscCall(PetscArraycpy(Mj + Mi[i] + llen + 1, Aj + adiag[i + 1] + 1, ulen - 1)); // entries of U … in MatSeqAIJCUSPARSEBuildFactoredMatrix_LU()
623 const PetscInt *Ai = a->i, *Aj = a->j, *adiag; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky() local
641 …PetscCall(PetscArraycpy(Mj + Ai[i] + 1, Aj + Ai[i], ulen - 1)); // entries of U on the right of th… in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
1797 const int *Ai, *Aj; in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0() local
1805 …PetscCall(MatSeqAIJCUSPARSEGetIJ(A, PETSC_FALSE, &Ai, &Aj)); /* Do not use compressed Ai. The ret… in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0()
1807 …PetscCallCUDA(cudaMemcpyAsync(fs->csrColIdx32, Aj, sizeof(*Aj) * nz, cudaMemcpyDeviceToDevice, Pet… in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0()
2046 const int *Ai, *Aj; in MatICCFactorSymbolic_SeqAIJCUSPARSE_ICC0() local
2054 PetscCall(MatSeqAIJCUSPARSEGetIJ(A, PETSC_FALSE, &Ai, &Aj)); /* Do not use compressed Ai */ in MatICCFactorSymbolic_SeqAIJCUSPARSE_ICC0()
[all …]
/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx357 const MatColIdxType *Aj = Aj_h.data(), *Bj = Bj_h.data(); in MatMPIAIJKokkosReduceBegin() local
456 buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global in MatMPIAIJKokkosReduceBegin()
751 PetscInt n1 = B->cmap->n, *Ai = aseq->i, *Aj = aseq->j, *Bi = bseq->i, *Bj = bseq->j; in MatMPIAIJKokkosBcastBegin() local
851 buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global in MatMPIAIJKokkosBcastBegin()
/petsc/src/mat/impls/aij/seq/
H A Daij.c4560 PetscInt *Aj; in MatSetPreallocationCOO_SeqAIJ() local
4602 …PetscCall(PetscShmgetAllocateArray(coo_n - nneg, sizeof(PetscInt), (void **)&Aj)); /* We have at m… in MatSetPreallocationCOO_SeqAIJ()
4664 Aj[q] = j[p]; in MatSetPreallocationCOO_SeqAIJ()
4671 Aj[q] = j[start]; /* Log the first nonzero in this row */ in MatSetPreallocationCOO_SeqAIJ()
4680 Aj[q] = j[p]; in MatSetPreallocationCOO_SeqAIJ()
4707 PetscCall(PetscArraycpy(Aj_new, Aj, nnz)); in MatSetPreallocationCOO_SeqAIJ()
4708 PetscCall(PetscShmgetDeallocateArray((void **)&Aj)); in MatSetPreallocationCOO_SeqAIJ()
4709 Aj = Aj_new; in MatSetPreallocationCOO_SeqAIJ()
4724 PetscCall(MatSetSeqAIJWithArrays_private(PETSC_COMM_SELF, M, N, Ai, Aj, Aa, rtype, mat)); in MatSetPreallocationCOO_SeqAIJ()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx1431 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()
1441 …PetscCallHIP(hipMemcpyAsync(fs->csrColIdx, Aj, sizeof(int) * nz, hipMemcpyDeviceToDevice, PetscDef… 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()
1686 …PetscCallHIP(hipMemcpyAsync(fs->csrColIdx, Aj, sizeof(int) * nz, hipMemcpyDeviceToDevice, PetscDef… in MatICCFactorSymbolic_SeqAIJHIPSPARSE_ICC0()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c6567 PetscInt *Aj, *Bj; in MatSetPreallocationCOO_MPIAIJ() local
6571 …PetscCall(PetscMalloc1(Annz1 + Annz2, &Aj)); /* Since local and remote entries might have dups, we… in MatSetPreallocationCOO_MPIAIJ()
6580 …rnal(mat, j1, j2, rowBegin1, rowMid1, rowBegin2, rowMid2, Ajmap1, Ajmap2, Aimap1, Aimap2, Ai, Aj)); in MatSetPreallocationCOO_MPIAIJ()
6611 PetscCall(PetscArraycpy(Aj_new, Aj, Annz)); in MatSetPreallocationCOO_MPIAIJ()
6612 PetscCall(PetscFree(Aj)); in MatSetPreallocationCOO_MPIAIJ()
6613 Aj = Aj_new; in MatSetPreallocationCOO_MPIAIJ()
6633 for (k = 0; k < Annz; k++) Aj[k] -= cstart; in MatSetPreallocationCOO_MPIAIJ()
6640 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, m, n, Ai, Aj, Aa, &mpiaij->A)); in MatSetPreallocationCOO_MPIAIJ()