Lines Matching refs:ns
10 static PetscErrorCode MatCreateColInode_Private(Mat A, PetscInt *size, PetscInt **ns) in MatCreateColInode_Private() argument
22 if (!ns) { in MatCreateColInode_Private()
46 *ns = ns_col; in MatCreateColInode_Private()
362 PetscInt *ns; in MatMult_SeqAIJ_Inode() local
367 ns = a->inode.size_csr; /* Node Size array */ in MatMult_SeqAIJ_Inode()
381 row = ns[i]; in MatMult_SeqAIJ_Inode()
382 nsz = ns[i + 1] - ns[i]; in MatMult_SeqAIJ_Inode()
571 const PetscInt *idx, *ns, *ii; in MatMultAdd_SeqAIJ_Inode() local
576 ns = a->inode.size_csr; /* Node Size array */ in MatMultAdd_SeqAIJ_Inode()
587 row = ns[i]; in MatMultAdd_SeqAIJ_Inode()
588 nsz = ns[i + 1] - ns[i]; in MatMultAdd_SeqAIJ_Inode()
769 PetscInt node_max, *ns, row, nsz, aii, i0, i1; in MatSolve_SeqAIJ_Inode_inplace() local
779 ns = a->inode.size_csr; /* Node Size array */ in MatSolve_SeqAIJ_Inode_inplace()
797 row = ns[i]; in MatSolve_SeqAIJ_Inode_inplace()
798 nsz = ns[i + 1] - ns[i]; in MatSolve_SeqAIJ_Inode_inplace()
808 …PetscPrefetchBlock(aa + ai[row + nsz], ad[ns[i + 2] - 1] - ai[row + nsz], 0, PETSC_PREFETCH_HINT_N… in MatSolve_SeqAIJ_Inode_inplace()
993 row = ns[i + 1]; in MatSolve_SeqAIJ_Inode_inplace()
994 nsz = ns[i + 1] - ns[i]; in MatSolve_SeqAIJ_Inode_inplace()
1214 const PetscInt *ns; in MatLUFactorNumeric_SeqAIJ_Inode() local
1246 ns = a->inode.size_csr; in MatLUFactorNumeric_SeqAIJ_Inode()
1247 PetscCheck(ns, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Matrix without inode information"); in MatLUFactorNumeric_SeqAIJ_Inode()
1253 nodesz = ns[i + 1] - ns[i]; in MatLUFactorNumeric_SeqAIJ_Inode()
1284 ns = tmp_vec2; in MatLUFactorNumeric_SeqAIJ_Inode()
1290 nodesz = ns[inod + 1] - ns[inod]; in MatLUFactorNumeric_SeqAIJ_Inode()
1989 const PetscInt *ai = a->i, *a_j = a->j, *ns, *vi, *ad, *aj; in MatSolve_SeqAIJ_Inode() local
1998 ns = a->inode.size_csr; /* Node Size array */ in MatSolve_SeqAIJ_Inode()
2016 row = ns[i]; in MatSolve_SeqAIJ_Inode()
2017 nsz = ns[i + 1] - ns[i]; in MatSolve_SeqAIJ_Inode()
2027 … PetscPrefetchBlock(aa + ai[row + nsz], ai[ns[i + 2]] - ai[row + nsz], 0, PETSC_PREFETCH_HINT_NTA); in MatSolve_SeqAIJ_Inode()
2190 row = ns[i + 1] - 1; in MatSolve_SeqAIJ_Inode()
2191 nsz = ns[i + 1] - ns[i]; in MatSolve_SeqAIJ_Inode()
2201 …PetscPrefetchBlock(aa + ad[row - nsz + 1] + 1, ad[ns[i - 1] + 1] - ad[row - nsz + 1], 0, PETSC_PRE… in MatSolve_SeqAIJ_Inode()
2389 PetscInt n = mat->cmap->n, m = a->inode.node_count, j, *ns = a->inode.size_csr, row; in MatColoringPatch_SeqAIJ_Inode() local
2399 …for (j = 0; j < (ns[i + 1] - ns[i]); j++) PetscCall(ISColoringValueCast(coloring[i] + j * ncolors,… in MatColoringPatch_SeqAIJ_Inode()
3919 PetscInt i, j, m, nzx, nzy, *ns, node_count, blk_size; in MatSeqAIJCheckInode() local
3933 ns = a->inode.size_csr; in MatSeqAIJCheckInode()
3934 ns[0] = 0; in MatSeqAIJCheckInode()
3951 ns[node_count + 1] = ns[node_count] + blk_size; in MatSeqAIJCheckInode()
4049 PetscInt *cols1, *cols2, *ns; in MatSeqAIJCheckInode_FactorLU() local
4058 if (a->inode.size_csr) ns = a->inode.size_csr; in MatSeqAIJCheckInode_FactorLU()
4059 else PetscCall(PetscMalloc1(m + 1, &ns)); in MatSeqAIJCheckInode_FactorLU()
4060 ns[0] = 0; in MatSeqAIJCheckInode_FactorLU()
4081 ns[node_count + 1] = ns[node_count] + blk_size; in MatSeqAIJCheckInode_FactorLU()
4088 PetscCall(PetscFree(ns)); in MatSeqAIJCheckInode_FactorLU()
4106 a->inode.size_csr = ns; in MatSeqAIJCheckInode_FactorLU()