Lines Matching refs:Ai
356 const MatRowMapType *Ai = Ai_h.data(), *Bi = Bi_h.data();
379 E_RowLen[i] = (Ai[i + 1] - Ai[i]) + (Bi[i + 1] - Bi[i]);
449 PetscInt *buf = &sendCol[Ai[i] + Bi[i]];
451 PetscInt alen = Ai[i + 1] - Ai[i], blen = Bi[i + 1] - Bi[i];
456 buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global
633 const auto &Ai = A.graph.row_map;
651 PetscInt disp = Ai(i) + Bi(i);
652 PetscInt alen = Ai(i + 1) - Ai(i);
661 val = Aa(Ai(i) + j - nzleft);
751 PetscInt n1 = B->cmap->n, *Ai = aseq->i, *Aj = aseq->j, *Bi = bseq->i, *Bj = bseq->j;
778 E_RowLen[i] = (Ai[i + 1] - Ai[i]) + (Bi[i + 1] - Bi[i]);
846 PetscInt alen = Ai[i + 1] - Ai[i], blen = Bi[i + 1] - Bi[i];
851 buf[j] = Aj[Ai[i] + j - nzLeft] + cstart; // diag A, also in global
956 const auto &Ai = akok->i_dual.view_device();
980 PetscInt alen = Ai(i + 1) - Ai(i);
988 rootBuf(disp + j) = Aa(Ai(i) + j - nzleft);