Lines Matching refs:bs2
43 PetscInt i, mbs = a->mbs, *jutmp, bs = A->rmap->bs, bs2 = a->bs2; in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR() local
184 PetscCall(PetscShmgetAllocateArray((iu[mbs] + 1) * bs2, sizeof(PetscScalar), (void **)&b->a)); in MatCholeskyFactorSymbolic_SeqSBAIJ_MSR()
565 PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N() local
574 PetscCall(PetscCalloc1(bs2 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
581 PetscCall(PetscMalloc3(bs2, &dk, bs2, &uik, bs, &work)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
594 PetscCall(PetscMalloc1(bs2 * ai[mbs], &aa)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
595 PetscCall(PetscArraycpy(aa, a->a, bs2 * ai[mbs])); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
607 for (k1 = 0; k1 < bs2; k1++) { in MatCholeskyFactorNumeric_SeqSBAIJ_N()
608 dk[k1] = aa[k * bs2 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
609 aa[k * bs2 + k1] = aa[j * bs2 + k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
610 aa[j * bs2 + k1] = dk[k1]; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
615 ap = aa + j * bs2; /* ptr to the beginning of j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_N()
616 for (k = 0; k < bs2; k++) dk[k] = ap[k]; /* dk <- j-th block of aa */ in MatCholeskyFactorNumeric_SeqSBAIJ_N()
632 ap = aa + jmin * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
635 rtmp_ptr = rtmp + vj * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
636 for (i = 0; i < bs2; i++) *rtmp_ptr++ = *ap++; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
640 PetscCall(PetscArraycpy(dk, rtmp + k * bs2, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
650 diag = ba + i * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
651 u = ba + ili * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
652 PetscCall(PetscArrayzero(uik, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
657 PetscCall(PetscLogFlops(4.0 * bs * bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
660 PetscCall(PetscArraycpy(ba + ili * bs2, uik, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
668 rtmp_ptr = rtmp + bj[j] * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
669 u = ba + j * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
672 PetscCall(PetscLogFlops(2.0 * bs * bs2 * (jmax - jmin))); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
686 diag = ba + k * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
687 PetscCall(PetscArraycpy(diag, dk, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N()
697 u = ba + j * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
698 rtmp_ptr = rtmp + vj * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N()
699 for (k1 = 0; k1 < bs2; k1++) { in MatCholeskyFactorNumeric_SeqSBAIJ_N()
729 PetscCall(PetscLogFlops(1.3333 * bs * bs2 * b->mbs)); /* from inverting diagonal blocks */ in MatCholeskyFactorNumeric_SeqSBAIJ_N()
738 PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering() local
746 PetscCall(PetscCalloc1(bs2 * mbs, &rtmp)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
751 PetscCall(PetscMalloc3(bs2, &dk, bs2, &uik, bs, &work)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
764 ap = aa + jmin * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
767 rtmp_ptr = rtmp + vj * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
768 for (i = 0; i < bs2; i++) *rtmp_ptr++ = *ap++; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
772 PetscCall(PetscArraycpy(dk, rtmp + k * bs2, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
782 diag = ba + i * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
783 u = ba + ili * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
784 PetscCall(PetscArrayzero(uik, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
789 PetscCall(PetscLogFlops(2.0 * bs * bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
792 PetscCall(PetscArraycpy(ba + ili * bs2, uik, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
800 rtmp_ptr = rtmp + bj[j] * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
801 u = ba + j * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
804 PetscCall(PetscLogFlops(2.0 * bs * bs2 * (jmax - jmin))); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
818 diag = ba + k * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
819 PetscCall(PetscArraycpy(diag, dk, bs2)); in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
829 u = ba + j * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
830 rtmp_ptr = rtmp + vj * bs2; in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
831 for (k1 = 0; k1 < bs2; k1++) { in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()
857 PetscCall(PetscLogFlops(1.3333 * bs * bs2 * b->mbs)); /* from inverting diagonal blocks */ in MatCholeskyFactorNumeric_SeqSBAIJ_N_NaturalOrdering()