Lines Matching refs:mbs

22   m  = a->mbs;  in MatIncreaseOverlap_SeqBAIJ()
105 …PetscCheck(c->mbs == nrows && c->nbs == ncols && (*B)->rmap->bs == bs, PETSC_COMM_SELF, PETSC_ERR_… in MatCreateSubMatrix_SeqBAIJ_Private()
106 PetscCall(PetscArraycmp(c->ilen, lens, c->mbs, &flag)); in MatCreateSubMatrix_SeqBAIJ_Private()
108 PetscCall(PetscArrayzero(c->ilen, c->mbs)); in MatCreateSubMatrix_SeqBAIJ_Private()
242 PetscInt mbs, i, n; in MatMult_SeqBAIJ_1() local
251 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_1()
254 PetscCall(PetscArrayzero(z, a->mbs)); in MatMult_SeqBAIJ_1()
256 mbs = a->mbs; in MatMult_SeqBAIJ_1()
260 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_1()
288 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_2() local
298 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_2()
301 PetscCall(PetscArrayzero(zarray, 2 * a->mbs)); in MatMult_SeqBAIJ_2()
303 mbs = a->mbs; in MatMult_SeqBAIJ_2()
308 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_2()
340 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_3() local
354 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_3()
357 PetscCall(PetscArrayzero(zarray, 3 * a->mbs)); in MatMult_SeqBAIJ_3()
359 mbs = a->mbs; in MatMult_SeqBAIJ_3()
364 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_3()
401 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_4() local
411 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_4()
414 PetscCall(PetscArrayzero(zarray, 4 * a->mbs)); in MatMult_SeqBAIJ_4()
416 mbs = a->mbs; in MatMult_SeqBAIJ_4()
421 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_4()
463 PetscInt mbs, i, j, n; in MatMult_SeqBAIJ_5() local
473 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_5()
476 PetscCall(PetscArrayzero(zarray, 5 * a->mbs)); in MatMult_SeqBAIJ_5()
478 mbs = a->mbs; in MatMult_SeqBAIJ_5()
483 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_5()
528 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_6() local
538 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_6()
541 PetscCall(PetscArrayzero(zarray, 6 * a->mbs)); in MatMult_SeqBAIJ_6()
543 mbs = a->mbs; in MatMult_SeqBAIJ_6()
548 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_6()
599 PetscInt mbs, i, *idx, *ii, j, n, *ridx = NULL; in MatMult_SeqBAIJ_7() local
609 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_7()
612 PetscCall(PetscArrayzero(zarray, 7 * a->mbs)); in MatMult_SeqBAIJ_7()
614 mbs = a->mbs; in MatMult_SeqBAIJ_7()
619 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_7()
674 PetscInt mbs, i, bs = A->rmap->bs, j, n, bs2 = a->bs2; in MatMult_SeqBAIJ_9_AVX2() local
691 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_9_AVX2()
694 PetscCall(PetscArrayzero(zarray, bs * a->mbs)); in MatMult_SeqBAIJ_9_AVX2()
696 mbs = a->mbs; in MatMult_SeqBAIJ_9_AVX2()
707 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_9_AVX2()
827 PetscInt mbs, i, j, k, n, *ridx = NULL; in MatMult_SeqBAIJ_11() local
836 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_11()
839 PetscCall(PetscArrayzero(zarray, 11 * a->mbs)); in MatMult_SeqBAIJ_11()
841 mbs = a->mbs; in MatMult_SeqBAIJ_11()
846 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_11()
911 PetscInt mbs, i, j, k, n, *ridx = NULL; in MatMult_SeqBAIJ_12_ver1() local
920 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_12_ver1()
923 PetscCall(PetscArrayzero(zarray, 12 * a->mbs)); in MatMult_SeqBAIJ_12_ver1()
925 mbs = a->mbs; in MatMult_SeqBAIJ_12_ver1()
930 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_12_ver1()
995 PetscInt mbs = a->mbs, i, j, k, n, *ridx = NULL; in MatMultAdd_SeqBAIJ_12_ver1() local
1004 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 12 * mbs)); in MatMultAdd_SeqBAIJ_12_ver1()
1005 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_12_ver1()
1014 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_12_ver1()
1088 PetscInt mbs, i, j, n; in MatMult_SeqBAIJ_12_ver2() local
1097 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_12_ver2()
1100 PetscCall(PetscArrayzero(zarray, 12 * a->mbs)); in MatMult_SeqBAIJ_12_ver2()
1102 mbs = a->mbs; in MatMult_SeqBAIJ_12_ver2()
1107 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_12_ver2()
1200 PetscInt mbs = a->mbs, i, j, n; in MatMultAdd_SeqBAIJ_12_ver2() local
1209 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 12 * mbs)); in MatMultAdd_SeqBAIJ_12_ver2()
1210 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_12_ver2()
1219 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_12_ver2()
1328 PetscInt mbs, i, j, n; in MatMult_SeqBAIJ_12_AVX2() local
1342 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_12_AVX2()
1345 PetscCall(PetscArrayzero(zarray, bs * a->mbs)); in MatMult_SeqBAIJ_12_AVX2()
1347 mbs = a->mbs; in MatMult_SeqBAIJ_12_AVX2()
1352 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_12_AVX2()
1495 PetscInt mbs, i, j, k, n, *ridx = NULL; in MatMult_SeqBAIJ_15_ver1() local
1504 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_15_ver1()
1507 PetscCall(PetscArrayzero(zarray, 15 * a->mbs)); in MatMult_SeqBAIJ_15_ver1()
1509 mbs = a->mbs; in MatMult_SeqBAIJ_15_ver1()
1514 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_15_ver1()
1591 PetscInt mbs, i, j, n, *ridx = NULL; in MatMult_SeqBAIJ_15_ver2() local
1600 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_15_ver2()
1603 PetscCall(PetscArrayzero(zarray, 15 * a->mbs)); in MatMult_SeqBAIJ_15_ver2()
1605 mbs = a->mbs; in MatMult_SeqBAIJ_15_ver2()
1610 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_15_ver2()
1752 PetscInt mbs, i, j, n, *ridx = NULL; in MatMult_SeqBAIJ_15_ver3() local
1761 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_15_ver3()
1764 PetscCall(PetscArrayzero(zarray, 15 * a->mbs)); in MatMult_SeqBAIJ_15_ver3()
1766 mbs = a->mbs; in MatMult_SeqBAIJ_15_ver3()
1771 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_15_ver3()
1878 PetscInt mbs, i, j, n, *ridx = NULL; in MatMult_SeqBAIJ_15_ver4() local
1887 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_15_ver4()
1890 PetscCall(PetscArrayzero(zarray, 15 * a->mbs)); in MatMult_SeqBAIJ_15_ver4()
1892 mbs = a->mbs; in MatMult_SeqBAIJ_15_ver4()
1897 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_15_ver4()
1986 PetscInt mbs, i, bs = A->rmap->bs, j, n, bs2 = a->bs2; in MatMult_SeqBAIJ_N() local
1998 mbs = a->compressedrow.nrows; in MatMult_SeqBAIJ_N()
2001 PetscCall(PetscArrayzero(zarray, bs * a->mbs)); in MatMult_SeqBAIJ_N()
2003 mbs = a->mbs; in MatMult_SeqBAIJ_N()
2013 for (i = 0; i < mbs; i++) { in MatMult_SeqBAIJ_N()
2040 PetscInt mbs = a->mbs, i, n, *ridx = NULL; in MatMultAdd_SeqBAIJ_1() local
2051 if (zz != yy) PetscCall(PetscArraycpy(z, y, mbs)); in MatMultAdd_SeqBAIJ_1()
2052 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_1()
2059 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_1()
2091 PetscInt mbs = a->mbs, i, n, j; in MatMultAdd_SeqBAIJ_2() local
2102 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 2 * mbs)); in MatMultAdd_SeqBAIJ_2()
2103 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_2()
2112 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_2()
2151 PetscInt mbs = a->mbs, i, j, n; in MatMultAdd_SeqBAIJ_3() local
2162 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 3 * mbs)); in MatMultAdd_SeqBAIJ_3()
2163 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_3()
2172 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_3()
2214 PetscInt mbs = a->mbs, i, j, n; in MatMultAdd_SeqBAIJ_4() local
2225 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 4 * mbs)); in MatMultAdd_SeqBAIJ_4()
2226 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_4()
2235 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_4()
2282 PetscInt mbs = a->mbs, i, j, n; in MatMultAdd_SeqBAIJ_5() local
2293 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 5 * mbs)); in MatMultAdd_SeqBAIJ_5()
2294 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_5()
2303 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_5()
2354 PetscInt mbs = a->mbs, i, j, n; in MatMultAdd_SeqBAIJ_6() local
2365 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 6 * mbs)); in MatMultAdd_SeqBAIJ_6()
2366 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_6()
2375 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_6()
2430 PetscInt mbs = a->mbs, i, j, n; in MatMultAdd_SeqBAIJ_7() local
2441 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 7 * mbs)); in MatMultAdd_SeqBAIJ_7()
2442 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_7()
2451 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_7()
2510 PetscInt mbs, i, j, n; in MatMultAdd_SeqBAIJ_9_AVX2() local
2528 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_9_AVX2()
2532 mbs = a->mbs; in MatMultAdd_SeqBAIJ_9_AVX2()
2543 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_9_AVX2()
2662 PetscInt mbs = a->mbs, i, j, n; in MatMultAdd_SeqBAIJ_11() local
2673 if (zz != yy) PetscCall(PetscArraycpy(zarray, yarray, 7 * mbs)); in MatMultAdd_SeqBAIJ_11()
2674 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_11()
2683 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_11()
2757 PetscInt mbs, i, bs = A->rmap->bs, j, n, bs2 = a->bs2; in MatMultAdd_SeqBAIJ_N() local
2770 mbs = a->compressedrow.nrows; in MatMultAdd_SeqBAIJ_N()
2774 mbs = a->mbs; in MatMultAdd_SeqBAIJ_N()
2784 for (i = 0; i < mbs; i++) { in MatMultAdd_SeqBAIJ_N()
2831 PetscInt mbs, i, rval, bs = A->rmap->bs, j, n; in MatMultHermitianTransposeAdd_SeqBAIJ() local
2844 mbs = cprow.nrows; in MatMultHermitianTransposeAdd_SeqBAIJ()
2848 mbs = a->mbs; in MatMultHermitianTransposeAdd_SeqBAIJ()
2855 for (i = 0; i < mbs; i++) { in MatMultHermitianTransposeAdd_SeqBAIJ()
2870 for (i = 0; i < mbs; i++) { in MatMultHermitianTransposeAdd_SeqBAIJ()
2887 for (i = 0; i < mbs; i++) { in MatMultHermitianTransposeAdd_SeqBAIJ()
2906 for (i = 0; i < mbs; i++) { in MatMultHermitianTransposeAdd_SeqBAIJ()
2927 for (i = 0; i < mbs; i++) { in MatMultHermitianTransposeAdd_SeqBAIJ()
2962 for (i=0; i<mbs; i++) { in MatMultHermitianTransposeAdd_SeqBAIJ()
2992 PetscInt mbs, i, rval, bs = A->rmap->bs, j, n, bs2 = a->bs2; in MatMultTransposeAdd_SeqBAIJ() local
3005 mbs = cprow.nrows; in MatMultTransposeAdd_SeqBAIJ()
3009 mbs = a->mbs; in MatMultTransposeAdd_SeqBAIJ()
3016 for (i = 0; i < mbs; i++) { in MatMultTransposeAdd_SeqBAIJ()
3031 for (i = 0; i < mbs; i++) { in MatMultTransposeAdd_SeqBAIJ()
3048 for (i = 0; i < mbs; i++) { in MatMultTransposeAdd_SeqBAIJ()
3067 for (i = 0; i < mbs; i++) { in MatMultTransposeAdd_SeqBAIJ()
3088 for (i = 0; i < mbs; i++) { in MatMultTransposeAdd_SeqBAIJ()
3120 for (i = 0; i < mbs; i++) { in MatMultTransposeAdd_SeqBAIJ()
3201 for (j = 0; j < a->mbs; j++) { in MatNorm_SeqBAIJ()
3222 const PetscInt bs = A->rmap->bs, ambs = a->mbs, bs2 = a->bs2; in MatGetDiagonal_SeqBAIJ()
3262 PetscInt i, j, k, lm, rn, M, m, n, mbs, tmp, bs, bs2, iai; in MatDiagonalScale_SeqBAIJ() local
3272 mbs = a->mbs; in MatDiagonalScale_SeqBAIJ()
3278 for (i = 0; i < mbs; i++) { /* for each block row */ in MatDiagonalScale_SeqBAIJ()
3294 for (i = 0; i < mbs; i++) { /* for each block row */ in MatDiagonalScale_SeqBAIJ()
3342 PetscCall(PetscArrayzero(a->a, a->bs2 * a->i[a->mbs])); in MatZeroEntries_SeqBAIJ()
3369 PetscInt mbs, i, *idx, *ii, j, *jj, n, k, *ridx = NULL; in MatMatMult_SeqBAIJ_1_Private() local
3376 mbs = a->compressedrow.nrows; in MatMatMult_SeqBAIJ_1_Private()
3380 mbs = a->mbs; in MatMatMult_SeqBAIJ_1_Private()
3385 for (i = 0; i < mbs; i++) { in MatMatMult_SeqBAIJ_1_Private()
3417 PetscInt mbs, i, *idx, *ii, j, *jj, n, k, *ridx = NULL; in MatMatMult_SeqBAIJ_2_Private() local
3424 mbs = a->compressedrow.nrows; in MatMatMult_SeqBAIJ_2_Private()
3428 mbs = a->mbs; in MatMatMult_SeqBAIJ_2_Private()
3433 for (i = 0; i < mbs; i++) { in MatMatMult_SeqBAIJ_2_Private()
3469 PetscInt mbs, i, *idx, *ii, j, *jj, n, k, *ridx = NULL; in MatMatMult_SeqBAIJ_3_Private() local
3476 mbs = a->compressedrow.nrows; in MatMatMult_SeqBAIJ_3_Private()
3480 mbs = a->mbs; in MatMatMult_SeqBAIJ_3_Private()
3485 for (i = 0; i < mbs; i++) { in MatMatMult_SeqBAIJ_3_Private()
3525 PetscInt mbs, i, *idx, *ii, j, *jj, n, k, *ridx = NULL; in MatMatMult_SeqBAIJ_4_Private() local
3532 mbs = a->compressedrow.nrows; in MatMatMult_SeqBAIJ_4_Private()
3536 mbs = a->mbs; in MatMatMult_SeqBAIJ_4_Private()
3541 for (i = 0; i < mbs; i++) { in MatMatMult_SeqBAIJ_4_Private()
3585 PetscInt mbs, i, *idx, *ii, j, *jj, n, k, *ridx = NULL; in MatMatMult_SeqBAIJ_5_Private() local
3592 mbs = a->compressedrow.nrows; in MatMatMult_SeqBAIJ_5_Private()
3596 mbs = a->mbs; in MatMatMult_SeqBAIJ_5_Private()
3601 for (i = 0; i < mbs; i++) { in MatMatMult_SeqBAIJ_5_Private()
3648 PetscInt mbs, i, bs = A->rmap->bs, j, n, bs2 = a->bs2; in MatMatMultNumeric_SeqBAIJ_SeqDense() local
3689 mbs = a->compressedrow.nrows; in MatMatMultNumeric_SeqBAIJ_SeqDense()
3693 mbs = a->mbs; in MatMatMultNumeric_SeqBAIJ_SeqDense()
3697 for (i = 0; i < mbs; i++) { in MatMatMultNumeric_SeqBAIJ_SeqDense()
3721 PetscInt mbs, i, *idx, *ii, j, n, k, *ridx = NULL; in MatTransposeMatMult_SeqBAIJ_1_Private() local
3729 mbs = a->compressedrow.nrows; in MatTransposeMatMult_SeqBAIJ_1_Private()
3733 mbs = a->mbs; in MatTransposeMatMult_SeqBAIJ_1_Private()
3737 for (i = 0; i < mbs; i++) { in MatTransposeMatMult_SeqBAIJ_1_Private()
3758 PetscInt mbs, i, *idx, *ii, j, n, k, *ridx = NULL; in MatTransposeMatMult_SeqBAIJ_2_Private() local
3767 mbs = a->compressedrow.nrows; in MatTransposeMatMult_SeqBAIJ_2_Private()
3771 mbs = a->mbs; in MatTransposeMatMult_SeqBAIJ_2_Private()
3775 for (i = 0; i < mbs; i++) { in MatTransposeMatMult_SeqBAIJ_2_Private()
3801 PetscInt mbs, i, *idx, *ii, j, n, k, *ridx = NULL; in MatTransposeMatMult_SeqBAIJ_3_Private() local
3810 mbs = a->compressedrow.nrows; in MatTransposeMatMult_SeqBAIJ_3_Private()
3814 mbs = a->mbs; in MatTransposeMatMult_SeqBAIJ_3_Private()
3818 for (i = 0; i < mbs; i++) { in MatTransposeMatMult_SeqBAIJ_3_Private()
3846 PetscInt mbs, i, *idx, *ii, j, n, k, *ridx = NULL; in MatTransposeMatMult_SeqBAIJ_4_Private() local
3855 mbs = a->compressedrow.nrows; in MatTransposeMatMult_SeqBAIJ_4_Private()
3859 mbs = a->mbs; in MatTransposeMatMult_SeqBAIJ_4_Private()
3863 for (i = 0; i < mbs; i++) { in MatTransposeMatMult_SeqBAIJ_4_Private()
3893 PetscInt mbs, i, *idx, *ii, j, n, k, *ridx = NULL; in MatTransposeMatMult_SeqBAIJ_5_Private() local
3902 mbs = a->compressedrow.nrows; in MatTransposeMatMult_SeqBAIJ_5_Private()
3906 mbs = a->mbs; in MatTransposeMatMult_SeqBAIJ_5_Private()
3910 for (i = 0; i < mbs; i++) { in MatTransposeMatMult_SeqBAIJ_5_Private()
3944 PetscInt mbs, i, bs = A->rmap->bs, j, n, bs2 = a->bs2; in MatTransposeMatMultNumeric_SeqBAIJ_SeqDense() local
3984 mbs = a->compressedrow.nrows; in MatTransposeMatMultNumeric_SeqBAIJ_SeqDense()
3988 mbs = a->mbs; in MatTransposeMatMultNumeric_SeqBAIJ_SeqDense()
3991 for (i = 0; i < mbs; i++) { in MatTransposeMatMultNumeric_SeqBAIJ_SeqDense()