Lines Matching refs:nest
1786 Mat_Nest *nest = (Mat_Nest *)A->data; in MatConvert_Nest_SeqAIJ_fast() local
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()
1809 for (j = 0; j < nest->nc; ++j) { in MatConvert_Nest_SeqAIJ_fast()
1810 Mat B = nest->m[i][j]; in MatConvert_Nest_SeqAIJ_fast()
1821 PetscCall(MatTranspose(Bt, MAT_INITIAL_MATRIX, &trans[i * nest->nc + j])); in MatConvert_Nest_SeqAIJ_fast()
1822 B = trans[i * nest->nc + j]; in MatConvert_Nest_SeqAIJ_fast()
1829 PetscCall(MatHermitianTranspose(Bt, MAT_INITIAL_MATRIX, &trans[i * nest->nc + j])); in MatConvert_Nest_SeqAIJ_fast()
1830 B = trans[i * nest->nc + j]; in MatConvert_Nest_SeqAIJ_fast()
1838 aii[i * nest->nc + j] = nii; in MatConvert_Nest_SeqAIJ_fast()
1839 ajj[i * nest->nc + j] = njj; in MatConvert_Nest_SeqAIJ_fast()
1840 avv[i * nest->nc + j] = naa; in MatConvert_Nest_SeqAIJ_fast()
1854 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_SeqAIJ_fast()
1857 PetscCall(ISStrideGetInfo(nest->isglobal.row[i], &rst, NULL)); in MatConvert_Nest_SeqAIJ_fast()
1858 PetscCall(ISGetLocalSize(nest->isglobal.row[i], &ncr)); in MatConvert_Nest_SeqAIJ_fast()
1859 for (j = 0; j < nest->nc; ++j) { in MatConvert_Nest_SeqAIJ_fast()
1860 if (aii[i * nest->nc + j]) { in MatConvert_Nest_SeqAIJ_fast()
1861 PetscInt *nii = aii[i * nest->nc + j]; in MatConvert_Nest_SeqAIJ_fast()
1875 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_SeqAIJ_fast()
1878 PetscCall(ISStrideGetInfo(nest->isglobal.row[i], &rst, NULL)); in MatConvert_Nest_SeqAIJ_fast()
1879 PetscCall(ISGetLocalSize(nest->isglobal.row[i], &ncr)); in MatConvert_Nest_SeqAIJ_fast()
1880 for (j = 0; j < nest->nc; ++j) { in MatConvert_Nest_SeqAIJ_fast()
1881 if (aii[i * nest->nc + j]) { in MatConvert_Nest_SeqAIJ_fast()
1882 PetscScalar *nvv = avv[i * nest->nc + j], vscale = 1.0, vshift = 0.0; in MatConvert_Nest_SeqAIJ_fast()
1883 PetscInt *nii = aii[i * nest->nc + j]; in MatConvert_Nest_SeqAIJ_fast()
1884 PetscInt *njj = ajj[i * nest->nc + j]; in MatConvert_Nest_SeqAIJ_fast()
1887 if (trans[i * nest->nc + j]) { in MatConvert_Nest_SeqAIJ_fast()
1888 vscale = ((Mat_Shell *)nest->m[i][j]->data)->vscale; in MatConvert_Nest_SeqAIJ_fast()
1889 vshift = ((Mat_Shell *)nest->m[i][j]->data)->vshift; in MatConvert_Nest_SeqAIJ_fast()
1891 PetscCall(ISStrideGetInfo(nest->isglobal.col[j], &cst, NULL)); in MatConvert_Nest_SeqAIJ_fast()
1911 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_SeqAIJ_fast()
1912 for (j = 0; j < nest->nc; ++j) { in MatConvert_Nest_SeqAIJ_fast()
1913 Mat B = nest->m[i][j]; in MatConvert_Nest_SeqAIJ_fast()
1915 PetscInt nnr = 0, k = i * nest->nc + j; in MatConvert_Nest_SeqAIJ_fast()
1948 Mat_Nest *nest = (Mat_Nest *)X->data; in MatAXPY_Dense_Nest() local
1954 for (j = 0; j < nest->nc; ++j) { in MatAXPY_Dense_Nest()
1959 PetscCall(ISAllGather(nest->isglobal.col[j], &bNis)); in MatAXPY_Dense_Nest()
1962 for (i = 0; i < nest->nr; ++i) { in MatAXPY_Dense_Nest()
1963 Mat B = nest->m[i][j], D = NULL; in MatAXPY_Dense_Nest()
1980 PetscCall(ISGetLocalSize(nest->isglobal.row[i], &bm)); in MatAXPY_Dense_Nest()
1981 PetscCall(ISGetIndices(nest->isglobal.row[i], &bmindices)); in MatAXPY_Dense_Nest()
2007 PetscCall(ISRestoreIndices(nest->isglobal.row[i], &bmindices)); in MatAXPY_Dense_Nest()
2019 Mat_Nest *nest = (Mat_Nest *)A->data; in MatConvert_Nest_AIJ() local
2032 for (i = 0; i < nest->nr && fast; ++i) { in MatConvert_Nest_AIJ()
2033 for (j = 0; j < nest->nc && fast; ++j) { in MatConvert_Nest_AIJ()
2034 Mat B = nest->m[i][j]; in MatConvert_Nest_AIJ()
2060 for (i = 0, nf = 0; i < nest->nr && fast; ++i) { in MatConvert_Nest_AIJ()
2061 PetscCall(PetscObjectTypeCompare((PetscObject)nest->isglobal.row[i], ISSTRIDE, &fast)); in MatConvert_Nest_AIJ()
2065 PetscCall(ISStrideGetInfo(nest->isglobal.row[i], &f, &s)); in MatConvert_Nest_AIJ()
2069 PetscCall(ISGetSize(nest->isglobal.row[i], &f)); in MatConvert_Nest_AIJ()
2074 for (i = 0, nf = 0; i < nest->nc && fast; ++i) { in MatConvert_Nest_AIJ()
2075 PetscCall(PetscObjectTypeCompare((PetscObject)nest->isglobal.col[i], ISSTRIDE, &fast)); in MatConvert_Nest_AIJ()
2079 PetscCall(ISStrideGetInfo(nest->isglobal.col[i], &f, &s)); in MatConvert_Nest_AIJ()
2083 PetscCall(ISGetSize(nest->isglobal.col[i], &f)); in MatConvert_Nest_AIJ()
2110 for (j = 0; j < nest->nc; ++j) { in MatConvert_Nest_AIJ()
2116 PetscCall(ISAllGather(nest->isglobal.col[j], &bNis)); in MatConvert_Nest_AIJ()
2119 for (i = 0; i < nest->nr; ++i) { in MatConvert_Nest_AIJ()
2122 Mat B = nest->m[i][j], D = NULL; in MatConvert_Nest_AIJ()
2126 PetscCall(ISGetLocalSize(nest->isglobal.row[i], &bm)); in MatConvert_Nest_AIJ()
2127 PetscCall(ISGetIndices(nest->isglobal.row[i], &bmindices)); in MatConvert_Nest_AIJ()
2177 PetscCall(ISRestoreIndices(nest->isglobal.row[i], &bmindices)); in MatConvert_Nest_AIJ()