Lines Matching refs:nr
20 for (i = 0; i < bA->nr; i++) { /* rows */ in MatNestGetSizes_Private()
42 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatMult_Nest() local
45 for (i = 0; i < nr; i++) PetscCall(VecGetSubVector(y, bA->isglobal.row[i], &by[i])); in MatMult_Nest()
47 for (i = 0; i < nr; i++) { in MatMult_Nest()
55 for (i = 0; i < nr; i++) PetscCall(VecRestoreSubVector(y, bA->isglobal.row[i], &by[i])); in MatMult_Nest()
64 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatMultAdd_Nest() local
67 for (i = 0; i < nr; i++) PetscCall(VecGetSubVector(z, bA->isglobal.row[i], &bz[i])); in MatMultAdd_Nest()
69 for (i = 0; i < nr; i++) { in MatMultAdd_Nest()
82 for (i = 0; i < nr; i++) PetscCall(VecRestoreSubVector(z, bA->isglobal.row[i], &bz[i])); in MatMultAdd_Nest()
100 PetscInt i, j, M, N, nr, nc, ldb, ldc; in MatProductNumeric_Nest_Dense() local
116 nr = bA->nr; in MatProductNumeric_Nest_Dense()
123 for (i = 0; i < nr; i++) { in MatProductNumeric_Nest_Dense()
173 PetscInt i, j, M, N, m, n, nr, nc, maxm = 0, ldb; in MatProductSymbolic_Nest_Dense() local
186 nr = bA->nr; in MatProductSymbolic_Nest_Dense()
208 PetscCall(PetscCalloc3(nr + 1, &contents->dm, nc + 1, &contents->dn, nr * nc, &contents->workC)); in MatProductSymbolic_Nest_Dense()
209 contents->k = nr * nc; in MatProductSymbolic_Nest_Dense()
210 for (i = 0; i < nr; i++) { in MatProductSymbolic_Nest_Dense()
228 for (i = 0; i < nr; i++) { in MatProductSymbolic_Nest_Dense()
265 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatMultTransposeKernel_Nest() local
268 for (i = 0; i < nr; i++) PetscCall(VecGetSubVector(x, bA->isglobal.row[i], &bx[i])); in MatMultTransposeKernel_Nest()
272 for (i = 0; i < nr; i++) { in MatMultTransposeKernel_Nest()
278 for (i = 0; i < nr; i++) PetscCall(VecRestoreSubVector(x, bA->isglobal.row[i], &bx[i])); in MatMultTransposeKernel_Nest()
301 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatMultTransposeAddKernel_Nest() local
304 for (i = 0; i < nr; i++) PetscCall(VecGetSubVector(x, bA->isglobal.row[i], &bx[i])); in MatMultTransposeAddKernel_Nest()
313 for (i = 0; i < nr; i++) { in MatMultTransposeAddKernel_Nest()
319 for (i = 0; i < nr; i++) PetscCall(VecRestoreSubVector(x, bA->isglobal.row[i], &bx[i])); in MatMultTransposeAddKernel_Nest()
342 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatTranspose_Nest() local
346 …PetscCheck(reuse != MAT_INPLACE_MATRIX || nr == nc, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG… in MatTranspose_Nest()
352 PetscCall(PetscCalloc1(nr * nc, &subs)); in MatTranspose_Nest()
353 PetscCall(PetscMalloc2(nr, &is_row, nc, &is_col)); in MatTranspose_Nest()
356 for (i = 0; i < nr; i++) { in MatTranspose_Nest()
357 for (j = 0; j < nc; j++) subs[i + nr * j] = bA->m[i][j]; in MatTranspose_Nest()
361 PetscCall(MatCreateNest(PetscObjectComm((PetscObject)A), nc, is_col, nr, is_row, subs, &C)); in MatTranspose_Nest()
369 for (i = 0; i < nr; i++) { in MatTranspose_Nest()
408 PetscCall(MatNestDestroyISList(vs->nr, &vs->isglobal.row)); in MatReset_Nest()
410 PetscCall(MatNestDestroyISList(vs->nr, &vs->islocal.row)); in MatReset_Nest()
421 for (i = 0; i < vs->nr; i++) { in MatReset_Nest()
429 vs->nr = 0; in MatReset_Nest()
466 for (i = 0; i < vs->nr; i++) { in MatAssemblyBegin_Nest()
496 for (i = 0; i < vs->nr; i++) { in MatAssemblyEnd_Nest()
527 sub = (col < vs->nr) ? vs->m[col][col] : (Mat)NULL; /* Prefer to find on the diagonal */ in MatNestFindNonzeroSubMatCol()
528 for (i = 0; !sub && i < vs->nr; i++) sub = vs->m[i][col]; in MatNestFindNonzeroSubMatCol()
613 PetscInt i, j, k, l, nr = rend - rbegin, nc = cend - cbegin; in MatNestGetBlock_Private() local
624 PetscCall(PetscMalloc2(nr * nc, &a, nr * nc, &b)); in MatNestGetBlock_Private()
625 for (i = 0; i < nr; i++) { in MatNestGetBlock_Private()
631 if (nc != vs->nc && nr != vs->nr) { in MatNestGetBlock_Private()
632 for (i = 0; i < nr; i++) { in MatNestGetBlock_Private()
635 for (k = 0; (k < nr && !flg); k++) { in MatNestGetBlock_Private()
651 …PetscCall(MatCreateNest(PetscObjectComm((PetscObject)A), nr, nr != vs->nr ? NULL : vs->isglobal.ro… in MatNestGetBlock_Private()
652 for (i = 0; i < nr; i++) { in MatNestGetBlock_Private()
670 PetscCall(MatNestFindISRange(A, vs->nr, is->row, isrow, &rbegin, &rend)); in MatNestFindSubMat()
741 for (i = 0; i < bA->nr; i++) { in MatGetDiagonal_Nest()
766 for (i = 0; i < bA->nr; i++) { in MatDiagonalScale_Nest()
786 for (i = 0; i < bA->nr; i++) { in MatScale_Nest()
801 for (i = 0; i < bA->nr; i++) { in MatShift_Nest()
820 for (i = 0; i < bA->nr; i++) { in MatDiagonalSet_Nest()
842 for (i = 0; i < bA->nr; i++) { in MatSetRandom_Nest()
864 for (i = 0; i < bA->nr; i++) { in MatCreateVecs_Nest()
870 …PetscCheck(i != bA->nr, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_WRONG, "Mat(Nest) contains … in MatCreateVecs_Nest()
880 PetscCall(PetscMalloc1(bA->nr, &L)); in MatCreateVecs_Nest()
882 for (i = 0; i < bA->nr; i++) { in MatCreateVecs_Nest()
892 PetscCall(VecCreateNest(comm, bA->nr, bA->isglobal.row, L, left)); in MatCreateVecs_Nest()
893 for (i = 0; i < bA->nr; i++) PetscCall(VecDestroy(&L[i])); in MatCreateVecs_Nest()
922 …(viewer, "MatNest, rows=%" PetscInt_FMT ", cols=%" PetscInt_FMT ", structure:\n", bA->nr, bA->nc)); in MatView_Nest()
923 for (i = 0; i < bA->nr; i++) { in MatView_Nest()
960 for (i = 0; i < bA->nr; i++) { in MatZeroEntries_Nest()
972 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatCopy_Nest() local
976 …nr == bB->nr && nc == bB->nc, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_INCOMP, "Cannot copy … in MatCopy_Nest()
977 for (i = 0; i < nr; i++) { in MatCopy_Nest()
1005 PetscInt i, j, nr = bY->nr, nc = bY->nc; in MatAXPY_Nest() local
1009 …nr == bX->nr && nc == bX->nc, PetscObjectComm((PetscObject)Y), PETSC_ERR_ARG_INCOMP, "Cannot AXPY … in MatAXPY_Nest()
1010 for (i = 0; i < nr; i++) { in MatAXPY_Nest()
1036 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatDuplicate_Nest() local
1039 PetscCall(PetscMalloc1(nr * nc, &b)); in MatDuplicate_Nest()
1040 for (i = 0; i < nr; i++) { in MatDuplicate_Nest()
1049 …PetscCall(MatCreateNest(PetscObjectComm((PetscObject)A), nr, bA->isglobal.row, nc, bA->isglobal.co… in MatDuplicate_Nest()
1051 for (i = 0; i < nr * nc; i++) PetscCall(MatDestroy(&b[i])); in MatDuplicate_Nest()
1065 …xm < bA->nr, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" Pet… in MatNestGetSubMat_Nest()
1106 …xm < bA->nr, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_OUTOFRANGE, "Row too large: row %" Pet… in MatNestSetSubMat_Nest()
1169 if (M) *M = bA->nr; in MatNestGetSubMats_Nest()
1214 if (M) *M = bA->nr; in MatNestGetSize_Nest()
1254 for (i = 0; i < vs->nr; i++) rows[i] = vs->isglobal.row[i]; in MatNestGetISs_Nest()
1295 for (i = 0; i < vs->nr; i++) rows[i] = vs->islocal.row[i]; in MatNestGetLocalISs_Nest()
1362 static PetscErrorCode MatNestSetSubMats_Nest(Mat A, PetscInt nr, const IS is_row[], PetscInt nc, co… in MatNestSetSubMats_Nest() argument
1372 s->nr = nr; in MatNestSetSubMats_Nest()
1376 PetscCall(PetscMalloc1(nr, &s->m)); in MatNestSetSubMats_Nest()
1377 PetscCall(PetscMalloc1(nr * nc, &s->m[0])); in MatNestSetSubMats_Nest()
1378 for (i = 0; i < nr; i++) { in MatNestSetSubMats_Nest()
1390 for (i = 0; i < nr; i++) { in MatNestSetSubMats_Nest()
1408 PetscCall(MatSetUp_NestIS_Private(A, nr, is_row, nc, is_col)); in MatNestSetSubMats_Nest()
1410 PetscCall(PetscMalloc1(nr, &s->row_len)); in MatNestSetSubMats_Nest()
1412 for (i = 0; i < nr; i++) s->row_len[i] = -1; in MatNestSetSubMats_Nest()
1415 PetscCall(PetscCalloc1(nr * nc, &s->nnzstate)); in MatNestSetSubMats_Nest()
1416 for (i = 0; i < nr; i++) { in MatNestSetSubMats_Nest()
1435 if (cong && nr != nc) cong = PETSC_FALSE; in MatNestSetSubMats_Nest()
1437 …for (i = 0; cong && i < nr; i++) PetscCall(ISEqualUnsorted(s->isglobal.row[i], s->isglobal.col[i],… in MatNestSetSubMats_Nest()
1445 PetscCall(PetscCalloc2(nr, &s->left, nc, &s->right)); in MatNestSetSubMats_Nest()
1479 PetscErrorCode MatNestSetSubMats(Mat A, PetscInt nr, const IS is_row[], PetscInt nc, const IS is_co… in MatNestSetSubMats() argument
1483 PetscValidLogicalCollectiveInt(A, nr, 2); in MatNestSetSubMats()
1484 …PetscCheck(nr >= 0, PetscObjectComm((PetscObject)A), PETSC_ERR_ARG_OUTOFRANGE, "Number of rows can… in MatNestSetSubMats()
1485 if (nr && is_row) { in MatNestSetSubMats()
1487 for (PetscInt i = 0; i < nr; i++) PetscValidHeaderSpecific(is_row[i], IS_CLASSID, 3); in MatNestSetSubMats()
1495 …ts_C", (Mat, PetscInt, const IS[], PetscInt, const IS[], const Mat[]), (A, nr, is_row, nc, is_col,… in MatNestSetSubMats()
1601 static PetscErrorCode MatSetUp_NestIS_Private(Mat A, PetscInt nr, const IS is_row[], PetscInt nc, c… in MatSetUp_NestIS_Private() argument
1608 PetscCall(PetscMalloc1(nr, &vs->isglobal.row)); in MatSetUp_NestIS_Private()
1612 for (i = 0; i < vs->nr; i++) { in MatSetUp_NestIS_Private()
1618 for (i = 0; i < vs->nr; i++) { /* Add up the local sizes to compute the aggregate offset */ in MatSetUp_NestIS_Private()
1627 for (i = 0; i < vs->nr; i++) { in MatSetUp_NestIS_Private()
1666 PetscCall(PetscMalloc1(vs->nr, &vs->islocal.row)); in MatSetUp_NestIS_Private()
1668 for (i = 0, offset = 0; i < vs->nr; i++) { in MatSetUp_NestIS_Private()
1708 …PetscCall(MatNestCreateAggregateL2G_Private(A, vs->nr, vs->islocal.row, vs->isglobal.row, PETSC_FA… in MatSetUp_NestIS_Private()
1716 for (i = 0; i < vs->nr; i++) { in MatSetUp_NestIS_Private()
1734 for (i = 0; i < vs->nr; i++) { in MatSetUp_NestIS_Private()
1774 PetscErrorCode MatCreateNest(MPI_Comm comm, PetscInt nr, const IS is_row[], PetscInt nc, const IS i… in MatCreateNest() argument
1780 PetscCall(MatNestSetSubMats(*B, nr, is_row, nc, is_col, a)); in MatCreateNest()
1792 PetscInt nr, nc, nnz, i, j; in MatConvert_Nest_SeqAIJ_fast() local
1796 PetscCall(MatGetSize(A, &nr, &nc)); in MatConvert_Nest_SeqAIJ_fast()
1802 …PetscCheck(rnr == nr, PetscObjectComm((PetscObject)A), PETSC_ERR_USER, "Cannot reuse matrix, wrong… in MatConvert_Nest_SeqAIJ_fast()
1807 …PetscCall(PetscCalloc4(nest->nr * nest->nc, &aii, nest->nr * nest->nc, &ajj, nest->nr * nest->nc, … in MatConvert_Nest_SeqAIJ_fast()
1808 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_SeqAIJ_fast()
1845 PetscCall(PetscMalloc1(nr + 1, &ii)); in MatConvert_Nest_SeqAIJ_fast()
1849 …PetscCheck(nnz == ii[nr], PetscObjectComm((PetscObject)A), PETSC_ERR_USER, "Cannot reuse matrix, w… in MatConvert_Nest_SeqAIJ_fast()
1853 PetscCall(PetscArrayzero(ii, nr + 1)); in MatConvert_Nest_SeqAIJ_fast()
1854 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_SeqAIJ_fast()
1871 for (i = 0; i < nr; i++) ii[i + 1] += ii[i]; in MatConvert_Nest_SeqAIJ_fast()
1874 PetscCall(PetscCalloc1(nr, &ci)); in MatConvert_Nest_SeqAIJ_fast()
1875 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_SeqAIJ_fast()
1911 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_SeqAIJ_fast()
1929 … PetscCall(MatCreateSeqAIJWithArrays(PetscObjectComm((PetscObject)A), nr, nc, ii, jj, vv, newmat)); in MatConvert_Nest_SeqAIJ_fast()
1933 PetscCall(MatCreateSeqAIJWithArrays(PetscObjectComm((PetscObject)A), nr, nc, ii, jj, vv, &B)); in MatConvert_Nest_SeqAIJ_fast()
1962 for (i = 0; i < nest->nr; ++i) { in MatAXPY_Dense_Nest()
2032 for (i = 0; i < nest->nr && fast; ++i) { in MatConvert_Nest_AIJ()
2060 for (i = 0, nf = 0; i < nest->nr && fast; ++i) { in MatConvert_Nest_AIJ()
2119 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_AIJ()
2229 PetscInt i, j, nr = bA->nr, nc = bA->nc; in MatHasOperation_Nest() local
2237 for (i = 0; i < nr; i++) { in MatHasOperation_Nest()
2274 s->nr = -1; in MatCreate_Nest()