Lines Matching refs:bs
10 const PetscInt n = a->mbs, bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_N_inplace() local
16 PetscCheck(bs > 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Expected bs %" PetscInt_FMT " > 0", bs); in MatSolveTranspose_SeqBAIJ_N_inplace()
28 for (j = 0; j < bs; j++) t[i * bs + j] = b[c[i] * bs + j]; in MatSolveTranspose_SeqBAIJ_N_inplace()
34 PetscCall(PetscArraycpy(ls, t + i * bs, bs)); in MatSolveTranspose_SeqBAIJ_N_inplace()
35 PetscKernel_w_gets_transA_times_v(bs, ls, aa + bs2 * a->diag[i], t + i * bs); in MatSolveTranspose_SeqBAIJ_N_inplace()
40 PetscKernel_v_gets_v_minus_transA_times_w(bs, t + bs * (*vi++), v, t + i * bs); in MatSolveTranspose_SeqBAIJ_N_inplace()
51 PetscKernel_v_gets_v_minus_transA_times_w(bs, t + bs * (*vi++), v, t + i * bs); in MatSolveTranspose_SeqBAIJ_N_inplace()
58 for (j = 0; j < bs; j++) x[bs * r[i] + j] = t[bs * i + j]; in MatSolveTranspose_SeqBAIJ_N_inplace()
65 PetscCall(PetscLogFlops(2.0 * (a->bs2) * (a->nz) - A->rmap->bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_N_inplace()
76 const PetscInt bs = A->rmap->bs, bs2 = a->bs2; in MatSolveTranspose_SeqBAIJ_N() local
82 PetscCheck(bs > 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Expected bs %" PetscInt_FMT " > 0", bs); in MatSolveTranspose_SeqBAIJ_N()
94 for (j = 0; j < bs; j++) t[i * bs + j] = b[c[i] * bs + j]; in MatSolveTranspose_SeqBAIJ_N()
100 PetscCall(PetscArraycpy(ls, t + i * bs, bs)); in MatSolveTranspose_SeqBAIJ_N()
101 PetscKernel_w_gets_transA_times_v(bs, ls, aa + bs2 * diag[i], t + i * bs); in MatSolveTranspose_SeqBAIJ_N()
106 PetscKernel_v_gets_v_minus_transA_times_w(bs, t + bs * (vi[j]), v, t + i * bs); in MatSolveTranspose_SeqBAIJ_N()
117 PetscKernel_v_gets_v_minus_transA_times_w(bs, t + bs * (vi[j]), v, t + i * bs); in MatSolveTranspose_SeqBAIJ_N()
124 for (j = 0; j < bs; j++) x[bs * r[i] + j] = t[bs * i + j]; in MatSolveTranspose_SeqBAIJ_N()
131 PetscCall(PetscLogFlops(2.0 * (a->bs2) * (a->nz) - A->rmap->bs * A->cmap->n)); in MatSolveTranspose_SeqBAIJ_N()