Lines Matching refs:imax

29 static PetscErrorCode ISAdjustForBlockSize(PetscInt bs, PetscInt imax, IS is[])  in ISAdjustForBlockSize()  argument
32 for (PetscInt i = 0; i < imax; i++) { in ISAdjustForBlockSize()
66 PetscErrorCode MatIncreaseOverlap_MPIAIJ(Mat C, PetscInt imax, IS is[], PetscInt ov) in MatIncreaseOverlap_MPIAIJ() argument
72 for (i = 0; i < ov; ++i) PetscCall(MatIncreaseOverlap_MPIAIJ_Once(C, imax, is)); in MatIncreaseOverlap_MPIAIJ()
73 …rmap->bs > 1 && C->rmap->bs == C->cmap->bs) PetscCall(ISAdjustForBlockSize(C->rmap->bs, imax, is)); in MatIncreaseOverlap_MPIAIJ()
77 PetscErrorCode MatIncreaseOverlap_MPIAIJ_Scalable(Mat C, PetscInt imax, IS is[], PetscInt ov) in MatIncreaseOverlap_MPIAIJ_Scalable() argument
83 for (i = 0; i < ov; ++i) PetscCall(MatIncreaseOverlap_MPIAIJ_Once_Scalable(C, imax, is)); in MatIncreaseOverlap_MPIAIJ_Scalable()
84 …rmap->bs > 1 && C->rmap->bs == C->cmap->bs) PetscCall(ISAdjustForBlockSize(C->rmap->bs, imax, is)); in MatIncreaseOverlap_MPIAIJ_Scalable()
492 static PetscErrorCode MatIncreaseOverlap_MPIAIJ_Once(Mat C, PetscInt imax, IS is[]) in MatIncreaseOverlap_MPIAIJ_Once() argument
523 PetscCall(PetscMalloc2(imax, (PetscInt ***)&idx, imax, &n)); in MatIncreaseOverlap_MPIAIJ_Once()
525 for (PetscInt i = 0; i < imax; i++) { in MatIncreaseOverlap_MPIAIJ_Once()
533 for (PetscInt i = 0; i < imax; i++) { in MatIncreaseOverlap_MPIAIJ_Once()
612 PetscCall(PetscIntMultError(M / PETSC_BITS_PER_BYTE + 1, imax, &M_BPB_imax)); in MatIncreaseOverlap_MPIAIJ_Once()
613 PetscCall(PetscMalloc1(imax, &table_data)); in MatIncreaseOverlap_MPIAIJ_Once()
614 for (PetscInt i = 0; i < imax; i++) PetscCall(PetscHMapICreateWithSize(n[i], table_data + i)); in MatIncreaseOverlap_MPIAIJ_Once()
615 PetscCall(PetscCalloc4(imax, &table, imax, &data, imax, &isz, M_BPB_imax, &t_p)); in MatIncreaseOverlap_MPIAIJ_Once()
616 for (PetscInt i = 0; i < imax; i++) table[i] = t_p + (M / PETSC_BITS_PER_BYTE + 1) * i; in MatIncreaseOverlap_MPIAIJ_Once()
619 PetscCall(PetscIntMultError(M, imax, &Mimax)); in MatIncreaseOverlap_MPIAIJ_Once()
620 PetscCall(PetscIntMultError(M / PETSC_BITS_PER_BYTE + 1, imax, &M_BPB_imax)); in MatIncreaseOverlap_MPIAIJ_Once()
621 PetscCall(PetscCalloc5(imax, &table, imax, &data, imax, &isz, Mimax, &d_p, M_BPB_imax, &t_p)); in MatIncreaseOverlap_MPIAIJ_Once()
622 for (PetscInt i = 0; i < imax; i++) { in MatIncreaseOverlap_MPIAIJ_Once()
634 for (PetscInt i = 0; i < imax; i++) { in MatIncreaseOverlap_MPIAIJ_Once()
682 PetscCall(PetscMalloc1(imax, &iscomms)); in MatIncreaseOverlap_MPIAIJ_Once()
683 for (PetscInt i = 0; i < imax; ++i) { in MatIncreaseOverlap_MPIAIJ_Once()
689 for (PetscInt i = 0; i < imax; ++i) PetscCall(ISDestroy(&is[i])); in MatIncreaseOverlap_MPIAIJ_Once()
693 PetscCall(MatIncreaseOverlap_MPIAIJ_Local(C, imax, table, isz, NULL, table_data)); in MatIncreaseOverlap_MPIAIJ_Once()
695 PetscCall(MatIncreaseOverlap_MPIAIJ_Local(C, imax, table, isz, data, NULL)); in MatIncreaseOverlap_MPIAIJ_Once()
780 for (PetscInt i = 0; i < imax; ++i) { in MatIncreaseOverlap_MPIAIJ_Once()
788 for (PetscInt i = 0; i < imax; ++i) { in MatIncreaseOverlap_MPIAIJ_Once()
826 for (PetscInt i = 0; i < imax; i++) PetscCall(PetscHMapIDestroy(table_data + i)); in MatIncreaseOverlap_MPIAIJ_Once()
850 static PetscErrorCode MatIncreaseOverlap_MPIAIJ_Local(Mat C, PetscInt imax, PetscBT *table, PetscIn… in MatIncreaseOverlap_MPIAIJ_Local() argument
875 for (i = 0; i < imax; i++) { in MatIncreaseOverlap_MPIAIJ_Local()