Lines Matching refs:nrow

12 #define IndexSpaceGet(buf, nrow, ncol, irowm, icolm) \  argument
14 if (nrow + ncol > (PetscInt)PETSC_STATIC_ARRAY_LENGTH(buf)) { \
15 PetscCall(PetscMalloc2(nrow, &irowm, ncol, &icolm)); \
18 icolm = &buf[nrow]; \
22 #define IndexSpaceRestore(buf, nrow, ncol, irowm, icolm) \ argument
24 … if (nrow + ncol > (PetscInt)PETSC_STATIC_ARRAY_LENGTH(buf)) PetscCall(PetscFree2(irowm, icolm)); \
35 static PetscErrorCode MatSetValuesBlockedLocal_LocalRef_Block(Mat A, PetscInt nrow, const PetscInt … in MatSetValuesBlockedLocal_LocalRef_Block() argument
41 if (!nrow || !ncol) PetscFunctionReturn(PETSC_SUCCESS); in MatSetValuesBlockedLocal_LocalRef_Block()
42 IndexSpaceGet(buf, nrow, ncol, irowm, icolm); in MatSetValuesBlockedLocal_LocalRef_Block()
43 PetscCall(ISLocalToGlobalMappingApplyBlock(A->rmap->mapping, nrow, irow, irowm)); in MatSetValuesBlockedLocal_LocalRef_Block()
45 PetscCall((*lr->SetValuesBlocked)(lr->Top, nrow, irowm, ncol, icolm, y, addv)); in MatSetValuesBlockedLocal_LocalRef_Block()
46 IndexSpaceRestore(buf, nrow, ncol, irowm, icolm); in MatSetValuesBlockedLocal_LocalRef_Block()
50 static PetscErrorCode MatSetValuesBlockedLocal_LocalRef_Scalar(Mat A, PetscInt nrow, const PetscInt… in MatSetValuesBlockedLocal_LocalRef_Scalar() argument
57 IndexSpaceGet(buf, nrow * rbs, ncol * cbs, irowm, icolm); in MatSetValuesBlockedLocal_LocalRef_Scalar()
58 BlockIndicesExpand(nrow, irow, rbs, irowm); in MatSetValuesBlockedLocal_LocalRef_Scalar()
60 PetscCall(ISLocalToGlobalMappingApplyBlock(A->rmap->mapping, nrow * rbs, irowm, irowm)); in MatSetValuesBlockedLocal_LocalRef_Scalar()
62 PetscCall((*lr->SetValues)(lr->Top, nrow * rbs, irowm, ncol * cbs, icolm, y, addv)); in MatSetValuesBlockedLocal_LocalRef_Scalar()
63 IndexSpaceRestore(buf, nrow * rbs, ncol * cbs, irowm, icolm); in MatSetValuesBlockedLocal_LocalRef_Scalar()
67 static PetscErrorCode MatSetValuesLocal_LocalRef_Scalar(Mat A, PetscInt nrow, const PetscInt irow[]… in MatSetValuesLocal_LocalRef_Scalar() argument
73 IndexSpaceGet(buf, nrow, ncol, irowm, icolm); in MatSetValuesLocal_LocalRef_Scalar()
77 PetscCall(ISLocalToGlobalMappingApply(A->rmap->mapping, nrow, irow, irowm)); in MatSetValuesLocal_LocalRef_Scalar()
79 PetscCall(ISLocalToGlobalMappingApplyBlock(A->rmap->mapping, nrow, irow, irowm)); in MatSetValuesLocal_LocalRef_Scalar()
87 PetscCall((*lr->SetValues)(lr->Top, nrow, irowm, ncol, icolm, y, addv)); in MatSetValuesLocal_LocalRef_Scalar()
88 IndexSpaceRestore(buf, nrow, ncol, irowm, icolm); in MatSetValuesLocal_LocalRef_Scalar()