Lines Matching refs:Ai
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()
641 …PetscCall(PetscArraycpy(Mj + Ai[i] + 1, Aj + Ai[i], ulen - 1)); // entries of U on the right of th… in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
648 PetscCallCUDA(cudaMemcpy(fs->csrRowPtr, Ai, sizeof(*Ai) * (m + 1), cudaMemcpyHostToDevice)); in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
688 Mnz = Ai[m]; in MatSeqAIJCUSPARSEBuildFactoredMatrix_Cholesky()
691 …Ma[Ai[i]] = (MatScalar)1.0; // set the unit diagonal, which is cosmetic since cusparse does not re… 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()
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()
1806 …PetscCallCUDA(cudaMemcpyAsync(fs->csrRowPtr32, Ai, sizeof(*Ai) * (m + 1), cudaMemcpyDeviceToDevice… in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0()
1892 PetscInt *Ai, nzRow, nzLeft; in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0() local
1897 Ai = Aseq->i; in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0()
1899 …if (Ai[i] < adiag[i] && adiag[i] < Ai[i + 1]) { /* There are nonzeros left to the diagonal of row … in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0()
1900 nzRow = Ai[i + 1] - Ai[i]; in MatILUFactorSymbolic_SeqAIJCUSPARSE_ILU0()
1901 nzLeft = adiag[i] - Ai[i]; 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()
2055 …PetscCallCUDA(cudaMemcpyAsync(fs->csrRowPtr32, Ai, sizeof(*Ai) * (m + 1), cudaMemcpyDeviceToDevice… in MatICCFactorSymbolic_SeqAIJCUSPARSE_ICC0()
2129 PetscInt *Ai, nzRow, nzLeft; in MatICCFactorSymbolic_SeqAIJCUSPARSE_ICC0() local
2132 Ai = Aseq->i; in MatICCFactorSymbolic_SeqAIJCUSPARSE_ICC0()
2134 nzRow = Ai[i + 1] - Ai[i]; in MatICCFactorSymbolic_SeqAIJCUSPARSE_ICC0()