Lines Matching refs:o_nnz
1904 PetscInt *d_nnz, *g_nnz, *o_nnz; in MatTranspose_MPIAIJ() local
1908 PetscCall(PetscMalloc4(na, &d_nnz, na, &o_nnz, nb, &g_nnz, nb, &oloc)); in MatTranspose_MPIAIJ()
1919 PetscCall(PetscArrayzero(o_nnz, na)); in MatTranspose_MPIAIJ()
1920 PetscCall(PetscSFReduceBegin(sf, MPIU_INT, g_nnz, o_nnz, MPI_SUM)); in MatTranspose_MPIAIJ()
1921 PetscCall(PetscSFReduceEnd(sf, MPIU_INT, g_nnz, o_nnz, MPI_SUM)); in MatTranspose_MPIAIJ()
1928 PetscCall(MatMPIAIJSetPreallocation(B, 0, d_nnz, 0, o_nnz)); in MatTranspose_MPIAIJ()
1929 PetscCall(PetscFree4(d_nnz, o_nnz, g_nnz, oloc)); in MatTranspose_MPIAIJ()
2873 …ocation_MPIAIJ(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPIAIJSetPreallocation_MPIAIJ() argument
2914 PetscCall(MatSeqAIJSetPreallocation(b->B, o_nz, o_nnz)); in MatMPIAIJSetPreallocation_MPIAIJ()
3869 PetscInt *d_nnz, *o_nnz, nnz_max = 0, rstart, ii, irstart; in MatMPIAIJSetPreallocationCSR_MPIAIJ() local
3883 PetscCall(PetscCalloc2(m, &d_nnz, m, &o_nnz)); in MatMPIAIJSetPreallocationCSR_MPIAIJ()
3904 o_nnz[i] = nnz - d; in MatMPIAIJSetPreallocationCSR_MPIAIJ()
3906 PetscCall(MatMPIAIJSetPreallocation(B, 0, d_nnz, 0, o_nnz)); in MatMPIAIJSetPreallocationCSR_MPIAIJ()
3907 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatMPIAIJSetPreallocationCSR_MPIAIJ()
4119 …tPreallocation(Mat B, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[]) in MatMPIAIJSetPreallocation() argument
4124 …_C", (Mat, PetscInt, const PetscInt[], PetscInt, const PetscInt[]), (B, d_nz, d_nnz, o_nz, o_nnz)); in MatMPIAIJSetPreallocation()
4515 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateAIJ() argument
4525 PetscCall(MatMPIAIJSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateAIJ()
7763 PetscInt *d_nnz, *o_nnz; in MatCreateGraph_Simple_AIJ() local
7776 PetscCall(PetscMalloc2(nloc, &d_nnz, (isseqaij ? 0 : nloc), &o_nnz)); in MatCreateGraph_Simple_AIJ()
7778 PetscInt *nnz = (c == a) ? d_nnz : o_nnz; in MatCreateGraph_Simple_AIJ()
7799 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatCreateGraph_Simple_AIJ()
7806 PetscCall(MatMPIAIJSetPreallocation(Gmat, 0, d_nnz, 0, o_nnz)); in MatCreateGraph_Simple_AIJ()
7807 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatCreateGraph_Simple_AIJ()
7892 PetscInt *d_nnz, *o_nnz, *w0, *w1, *w2; in MatCreateGraph_Simple_AIJ() local
7898 PetscCall(PetscMalloc2(nloc, &d_nnz, (isseqaij ? 0 : nloc), &o_nnz)); in MatCreateGraph_Simple_AIJ()
7928 o_nnz[jj] = 0; in MatCreateGraph_Simple_AIJ()
7931 o_nnz[jj] += ncols; in MatCreateGraph_Simple_AIJ()
7934 if (o_nnz[jj] > (NN / bs - nloc)) o_nnz[jj] = NN / bs - nloc; in MatCreateGraph_Simple_AIJ()
7939 PetscCall(MatMPIAIJSetPreallocation(Gmat, 0, d_nnz, 0, o_nnz)); in MatCreateGraph_Simple_AIJ()
7940 PetscCall(PetscFree2(d_nnz, o_nnz)); in MatCreateGraph_Simple_AIJ()