Lines Matching refs:subset_size
393 PetscInt i, subset_size, max_subset_size, n_local_subs = sub_schurs->graph->n_local_subs; in PCBDDCSubSchursSetUp() local
574 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
575 max_subset_size = PetscMax(subset_size, max_subset_size); in PCBDDCSubSchursSetUp()
576 local_size += subset_size; in PCBDDCSubSchursSetUp()
602 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
606 auxnum2[i] = subset_size * subset_size; in PCBDDCSubSchursSetUp()
608 PetscCall(PetscArraycpy(all_local_idx_N + local_size + extra, idxs, subset_size)); in PCBDDCSubSchursSetUp()
610 …for (PetscInt j = 0; j < subset_size; j++) all_local_subid_N[j + local_size + extra] = local_subs[… in PCBDDCSubSchursSetUp()
612 local_size += subset_size; in PCBDDCSubSchursSetUp()
613 local_stash_size += subset_size * subset_size; in PCBDDCSubSchursSetUp()
665 …_schurs->BtoNmap, IS_GTOLM_DROP, local_size, all_local_idx_N + extra, &subset_size, all_local_idx_… in PCBDDCSubSchursSetUp()
666 …k(subset_size == local_size, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Error in sub_schurs serial (BtoNmap… in PCBDDCSubSchursSetUp()
693 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
726 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
727 for (j = 0; j < subset_size; j++) { in PCBDDCSubSchursSetUp()
731 ii[row + 1] = ii[row] + subset_size; in PCBDDCSubSchursSetUp()
737 cum += subset_size; in PCBDDCSubSchursSetUp()
818 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
819 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, subset_size, subset_size, work, &S_Ej_expl)); in PCBDDCSubSchursSetUp()
823 for (j = 0; j < subset_size; j++) dummy_idx[j] = local_size + j; in PCBDDCSubSchursSetUp()
824 …PetscCall(MatSetValues(sub_schurs->S_Ej_all, subset_size, dummy_idx, subset_size, dummy_idx, work,… in PCBDDCSubSchursSetUp()
827 local_size += subset_size; in PCBDDCSubSchursSetUp()
1088 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
1089 PetscCall(ISCreateStride(PETSC_COMM_SELF, subset_size, cum, 1, &is)); in PCBDDCSubSchursSetUp()
1104 cum += subset_size; in PCBDDCSubSchursSetUp()
1457 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
1461 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, subset_size, subset_size, work, &T)); in PCBDDCSubSchursSetUp()
1467 for (PetscInt j = 0; j < subset_size; j++) idx_work[j] = cum + j; in PCBDDCSubSchursSetUp()
1468 PetscCall(MatGetValues(S_all, subset_size, idx_work, subset_size, idx_work, work)); in PCBDDCSubSchursSetUp()
1470 for (PetscInt k = 0; k < subset_size; k++) { in PCBDDCSubSchursSetUp()
1471 for (PetscInt j = k; j < subset_size; j++) { in PCBDDCSubSchursSetUp()
1472 PetscScalar t = work[k * subset_size + j]; in PCBDDCSubSchursSetUp()
1473 work[k * subset_size + j] = work[j * subset_size + k]; in PCBDDCSubSchursSetUp()
1474 work[j * subset_size + k] = t; in PCBDDCSubSchursSetUp()
1479 for (PetscInt k = 0; k < subset_size; k++) { in PCBDDCSubSchursSetUp()
1480 for (PetscInt j = k; j < subset_size; j++) { in PCBDDCSubSchursSetUp()
1481 work[k * subset_size + j] = rS_data[cum2 + k * size_schur + j]; in PCBDDCSubSchursSetUp()
1482 work[j * subset_size + k] = PetscConj(rS_data[cum2 + k * size_schur + j]); in PCBDDCSubSchursSetUp()
1486 for (PetscInt k = 0; k < subset_size; k++) { in PCBDDCSubSchursSetUp()
1487 …for (PetscInt j = 0; j < subset_size; j++) work[k * subset_size + j] = rS_data[cum2 + k * size_sch… in PCBDDCSubSchursSetUp()
1497 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, subset_size, subset_size, work, &SEj)); in PCBDDCSubSchursSetUp()
1512 PetscCall(PetscArraycpy(SEj_arr, work, subset_size * subset_size)); in PCBDDCSubSchursSetUp()
1519 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, subset_size, subset_size, work, &M)); in PCBDDCSubSchursSetUp()
1543 PetscCall(PetscArraycpy(SEjinv_arr, vals, subset_size * subset_size)); in PCBDDCSubSchursSetUp()
1551 PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, subset_size, subset_size, work, &SEj)); in PCBDDCSubSchursSetUp()
1553 PetscCall(VecCreateSeqWithArray(PETSC_COMM_SELF, 1, subset_size, array + cum, &D)); in PCBDDCSubSchursSetUp()
1559 PetscCall(PetscArraycpy(SEj_arr, work, subset_size * subset_size)); in PCBDDCSubSchursSetUp()
1562 cum += subset_size; in PCBDDCSubSchursSetUp()
1563 cum2 += subset_size * (size_schur + 1); in PCBDDCSubSchursSetUp()
1564 SEj_arr += subset_size * subset_size; in PCBDDCSubSchursSetUp()
1565 if (SEjinv_arr) SEjinv_arr += subset_size * subset_size; in PCBDDCSubSchursSetUp()
1749 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
1750 … PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, subset_size, subset_size, SEjinv_arr, &SEj)); in PCBDDCSubSchursSetUp()
1752 PetscCall(ISCreateStride(PETSC_COMM_SELF, subset_size, cum, 1, &EE)); in PCBDDCSubSchursSetUp()
1782 cum += subset_size; in PCBDDCSubSchursSetUp()
1783 SEjinv_arr += subset_size * subset_size; in PCBDDCSubSchursSetUp()
1796 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
1802 for (PetscInt j = 0; j < subset_size; j++) idx_work[j] = cum + j; in PCBDDCSubSchursSetUp()
1803 PetscCall(MatGetValues(S_all_inv, subset_size, idx_work, subset_size, idx_work, work)); in PCBDDCSubSchursSetUp()
1805 for (PetscInt k = 0; k < subset_size; k++) { in PCBDDCSubSchursSetUp()
1806 for (PetscInt j = k; j < subset_size; j++) { in PCBDDCSubSchursSetUp()
1807 PetscScalar t = work[k * subset_size + j]; in PCBDDCSubSchursSetUp()
1808 work[k * subset_size + j] = work[j * subset_size + k]; in PCBDDCSubSchursSetUp()
1809 work[j * subset_size + k] = t; in PCBDDCSubSchursSetUp()
1817 for (k = 0; k < subset_size; k++) { in PCBDDCSubSchursSetUp()
1818 for (j = k; j < subset_size; j++) { in PCBDDCSubSchursSetUp()
1819 work[k * subset_size + j] = rS_data[cum2 + k * size_schur + j]; in PCBDDCSubSchursSetUp()
1820 work[j * subset_size + k] = work[k * subset_size + j]; in PCBDDCSubSchursSetUp()
1824 for (k = 0; k < subset_size; k++) { in PCBDDCSubSchursSetUp()
1825 … for (j = k; j < subset_size; j++) work[k * subset_size + j] = rS_data[cum2 + k * size_schur + j]; in PCBDDCSubSchursSetUp()
1830 for (k = 0; k < subset_size; k++) { in PCBDDCSubSchursSetUp()
1831 … for (j = 0; j < subset_size; j++) work[k * subset_size + j] = rS_data[cum2 + k * size_schur + j]; in PCBDDCSubSchursSetUp()
1841 …PetscCall(MatCreateSeqDense(PETSC_COMM_SELF, subset_size, subset_size, (rS_data || multi_element) … in PCBDDCSubSchursSetUp()
1856 …f (rS_data || multi_element) PetscCall(PetscArraycpy(SEjinv_arr, work, subset_size * subset_size)); in PCBDDCSubSchursSetUp()
1857 cum += subset_size; in PCBDDCSubSchursSetUp()
1858 cum2 += subset_size * (size_schur + 1); in PCBDDCSubSchursSetUp()
1859 SEjinv_arr += subset_size * subset_size; in PCBDDCSubSchursSetUp()
1999 PetscCall(ISGetLocalSize(sub_schurs->is_subs[i], &subset_size)); in PCBDDCSubSchursSetUp()
2000 PetscCall(PetscBLASIntCast(subset_size, &B_N)); in PCBDDCSubSchursSetUp()
2018 PetscCall(PetscLogFlops(1.0 * subset_size * subset_size * subset_size)); in PCBDDCSubSchursSetUp()
2020 cum += subset_size * subset_size; in PCBDDCSubSchursSetUp()