Lines Matching refs:bs2
101 PetscInt i, bs = A->rmap->bs, mbs = a->mbs, ipvt[5], bs2 = bs * bs, *v_pivots; in MatInvertBlockDiagonal_SeqBAIJ() local
115 if (!a->idiag) PetscCall(PetscMalloc1(bs2 * mbs, &a->idiag)); in MatInvertBlockDiagonal_SeqBAIJ()
205 odiag = v + bs2 * adiag[i]; in MatInvertBlockDiagonal_SeqBAIJ()
206 PetscCall(PetscArraycpy(diag, odiag, bs2)); in MatInvertBlockDiagonal_SeqBAIJ()
209 diag += bs2; in MatInvertBlockDiagonal_SeqBAIJ()
224 PetscInt m = a->mbs, i, i2, nz, bs = A->rmap->bs, bs2 = bs * bs, k, j, idx, it; in MatSOR_SeqBAIJ() local
522 idiag += bs2; in MatSOR_SeqBAIJ()
524 v = aa + bs2 * ai[i]; in MatSOR_SeqBAIJ()
539 idiag += bs2; in MatSOR_SeqBAIJ()
545 PetscCall(PetscLogFlops(1.0 * bs2 * a->nz)); in MatSOR_SeqBAIJ()
549 idiag = a->idiag + bs2 * (a->mbs - 1); in MatSOR_SeqBAIJ()
819 idiag -= bs2; in MatSOR_SeqBAIJ()
821 v = aa + bs2 * (diag[i] + 1); in MatSOR_SeqBAIJ()
835 idiag -= bs2; in MatSOR_SeqBAIJ()
840 PetscCall(PetscLogFlops(1.0 * bs2 * (a->nz))); in MatSOR_SeqBAIJ()
1038 v = aa + bs2 * ai[i]; in MatSOR_SeqBAIJ()
1052 idiag += bs2; in MatSOR_SeqBAIJ()
1057 PetscCall(PetscLogFlops(2.0 * bs2 * a->nz)); in MatSOR_SeqBAIJ()
1060 idiag = a->idiag + bs2 * (a->mbs - 1); in MatSOR_SeqBAIJ()
1252 v = aa + bs2 * ai[i]; in MatSOR_SeqBAIJ()
1266 idiag -= bs2; in MatSOR_SeqBAIJ()
1271 PetscCall(PetscLogFlops(2.0 * bs2 * (a->nz))); in MatSOR_SeqBAIJ()
1586 PetscInt itmp, i, j, k, M, bn, bp, *idx_i, bs, bs2; in MatGetRow_SeqBAIJ_private() local
1592 bs2 = bs * bs; in MatGetRow_SeqBAIJ_private()
1606 aa_i = aa + bs2 * (ai[bn] + i); in MatGetRow_SeqBAIJ_private()
1607 for (j = bp, k = 0; j < bs2; j += bs, k++) v_i[k] = aa_i[j]; in MatGetRow_SeqBAIJ_private()
1648 PetscInt bs2 = a->bs2, *ati, *atj, anzj, kr; in MatTranspose_SeqBAIJ() local
1683 for (k = 0; k < bs; k++) ata[bs2 * atfill[*aj] + k * bs + kr] = *aa++; in MatTranspose_SeqBAIJ()
1824 PetscInt i, j, bs = A->rmap->bs, k, l, bs2 = a->bs2; in MatView_SeqBAIJ_ASCII() local
1853 …if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) > 0.0 && PetscRealPart(a->a[bs2 * k + l * bs + … in MatView_SeqBAIJ_ASCII()
1854 …, bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), (double)PetscImaginaryPart(… in MatView_SeqBAIJ_ASCII()
1855 …} else if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) < 0.0 && PetscRealPart(a->a[bs2 * k + l … in MatView_SeqBAIJ_ASCII()
1856 … bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), -(double)PetscImaginaryPart(… in MatView_SeqBAIJ_ASCII()
1857 } else if (PetscRealPart(a->a[bs2 * k + l * bs + j]) != 0.0) { in MatView_SeqBAIJ_ASCII()
1858 … " (%" PetscInt_FMT ", %g) ", bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]))… in MatView_SeqBAIJ_ASCII()
1861 …if (a->a[bs2 * k + l * bs + j] != 0.0) PetscCall(PetscViewerASCIIPrintf(viewer, " (%" PetscInt_FMT… in MatView_SeqBAIJ_ASCII()
1877 if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) > 0.0) { in MatView_SeqBAIJ_ASCII()
1878 …, bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), (double)PetscImaginaryPart(… in MatView_SeqBAIJ_ASCII()
1879 } else if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) < 0.0) { in MatView_SeqBAIJ_ASCII()
1880 … bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), -(double)PetscImaginaryPart(… in MatView_SeqBAIJ_ASCII()
1882 … " (%" PetscInt_FMT ", %g) ", bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]))… in MatView_SeqBAIJ_ASCII()
1885 …Printf(viewer, " (%" PetscInt_FMT ", %g) ", bs * a->j[k] + l, (double)a->a[bs2 * k + l * bs + j])); in MatView_SeqBAIJ_ASCII()
1903 PetscInt row, i, j, k, l, mbs = a->mbs, bs = A->rmap->bs, bs2 = a->bs2; in MatView_SeqBAIJ_Draw_Zoom() local
1927 aa = a->a + j * bs2; in MatView_SeqBAIJ_Draw_Zoom()
1943 aa = a->a + j * bs2; in MatView_SeqBAIJ_Draw_Zoom()
1959 aa = a->a + j * bs2; in MatView_SeqBAIJ_Draw_Zoom()
1975 for (i = 0; i < a->nz * a->bs2; i++) { in MatView_SeqBAIJ_Draw_Zoom()
1989 aa = a->a + j * bs2; in MatView_SeqBAIJ_Draw_Zoom()
2059 PetscInt brow, bcol, ridx, cidx, bs = A->rmap->bs, bs2 = a->bs2; in MatGetValues_SeqBAIJ() local
2072 ap = PetscSafePointerPlusOffset(aa, bs2 * ai[brow]); in MatGetValues_SeqBAIJ()
2094 *v++ = ap[bs2 * i + bs * cidx + ridx]; in MatGetValues_SeqBAIJ()
2110 PetscInt *aj = a->j, nonew = a->nonew, bs2 = a->bs2, bs = A->rmap->bs, stepval; in MatSetValuesBlocked_SeqBAIJ() local
2126 if (!A->structure_only) ap = aa + bs2 * ai[row]; in MatSetValuesBlocked_SeqBAIJ()
2154 bap = ap + bs2 * i; in MatSetValuesBlocked_SeqBAIJ()
2158 for (jj = ii; jj < bs2; jj += bs) bap[jj] += *value++; in MatSetValuesBlocked_SeqBAIJ()
2162 for (jj = ii; jj < bs2; jj += bs) bap[jj] = *value++; in MatSetValuesBlocked_SeqBAIJ()
2184 …MatSeqXAIJReallocateAIJ_structure_only(A, a->mbs, bs2, nrow, row, col, rmax, ai, aj, rp, imax, non… in MatSetValuesBlocked_SeqBAIJ()
2186 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, row, col, rmax, aa, ai, aj, rp, ap, imax, nonew, Mat… in MatSetValuesBlocked_SeqBAIJ()
2194 PetscCall(PetscArraymove(ap + bs2 * (i + 1), ap + bs2 * i, bs2 * (N - i + 1))); in MatSetValuesBlocked_SeqBAIJ()
2195 bap = ap + bs2 * i; in MatSetValuesBlocked_SeqBAIJ()
2198 for (jj = ii; jj < bs2; jj += bs) bap[jj] = *value++; in MatSetValuesBlocked_SeqBAIJ()
2219 PetscInt mbs = a->mbs, bs2 = a->bs2, rmax = 0; in MatAssemblyEnd_SeqBAIJ() local
2233 ap = aa + bs2 * ai[i]; in MatAssemblyEnd_SeqBAIJ()
2236 if (!A->structure_only) PetscCall(PetscArraymove(ap - bs2 * fshift, ap, bs2 * N)); in MatAssemblyEnd_SeqBAIJ()
2259 …ock size %" PetscInt_FMT ", %" PetscInt_FMT " unneeded", m, A->cmap->n, A->rmap->bs, fshift * bs2); in MatAssemblyEnd_SeqBAIJ()
2260 …FMT " unneeded, %" PetscInt_FMT " used\n", m, A->cmap->n, A->rmap->bs, fshift * bs2, a->nz * bs2)); in MatAssemblyEnd_SeqBAIJ()
2266 A->info.nz_unneeded = (PetscReal)fshift * bs2; in MatAssemblyEnd_SeqBAIJ()
2317 PetscInt bs = A->rmap->bs, bs2 = baij->bs2, *sizes, row, bs_max; in MatZeroRows_SeqBAIJ() local
2353 aa = baij->a + baij->i[row / bs] * bs2 + (row % bs); in MatZeroRows_SeqBAIJ()
2386 PetscInt bs = A->rmap->bs, bs2 = baij->bs2, row, col; in MatZeroRowsColumns_SeqBAIJ() local
2414 aa = baij->a + j * bs2 + (i % bs) + bs * k; in MatZeroRowsColumns_SeqBAIJ()
2432 aa = baij->a + baij->i[row / bs] * bs2 + (row % bs); in MatZeroRowsColumns_SeqBAIJ()
2449 PetscInt ridx, cidx, bs2 = a->bs2; in MatSetValues_SeqBAIJ() local
2460 if (!A->structure_only) ap = PetscSafePointerPlusOffset(aa, bs2 * ai[brow]); in MatSetValues_SeqBAIJ()
2490 bap = PetscSafePointerPlusOffset(ap, bs2 * i + bs * cidx + ridx); in MatSetValues_SeqBAIJ()
2501 …MatSeqXAIJReallocateAIJ_structure_only(A, a->mbs, bs2, nrow, brow, bcol, rmax, ai, aj, rp, imax, n… in MatSetValues_SeqBAIJ()
2503 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, brow, bcol, rmax, aa, ai, aj, rp, ap, imax, nonew, M… in MatSetValues_SeqBAIJ()
2511 PetscCall(PetscArraymove(ap + bs2 * (i + 1), ap + bs2 * i, bs2 * (N - i + 1))); in MatSetValues_SeqBAIJ()
2512 PetscCall(PetscArrayzero(ap + bs2 * i, bs2)); in MatSetValues_SeqBAIJ()
2513 ap[bs2 * i + bs * cidx + ridx] = value; in MatSetValues_SeqBAIJ()
2689 … PetscInt ambs = a->mbs, bmbs = b->mbs, abs = A->rmap->bs, bbs = B->rmap->bs, bs2 = abs * abs; in MatCopy_SeqBAIJ() local
2693 PetscCall(PetscArraycpy(b->a, a->a, bs2 * a->i[ambs])); in MatCopy_SeqBAIJ()
2732 PetscInt bs = Y->rmap->bs, bs2 = bs * bs; in MatAXPY_SeqBAIJ() local
2750 PetscCall(PetscBLASIntCast(x->nz * bs2, &bnz)); in MatAXPY_SeqBAIJ()
2777 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatConjugate_SeqBAIJ()
2789 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatRealPart_SeqBAIJ()
2805 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatImaginaryPart_SeqBAIJ()
2926 PetscInt mbs = a->mbs, bs2 = a->bs2, rmax = 0; in MatEliminateZeros_SeqBAIJ() local
2936 ap = aa + bs2 * k; in MatEliminateZeros_SeqBAIJ()
2937 for (j = 0; j < bs2 && zero; j++) { in MatEliminateZeros_SeqBAIJ()
2944 PetscCall(PetscArraymove(ap - bs2 * fshift, ap, bs2)); in MatEliminateZeros_SeqBAIJ()
3116 PetscInt nz = aij->i[aij->mbs] * aij->bs2; in MatStoreValues_SeqBAIJ()
3132 PetscInt nz = aij->i[aij->mbs] * aij->bs2; in MatRetrieveValues_SeqBAIJ()
3149 PetscInt i, mbs, nbs, bs2; in MatSeqBAIJSetPreallocation_SeqBAIJ() local
3178 bs2 = bs * bs; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3346 PetscCall(PetscIntMultError(nz, bs2, &nzbs2)); in MatSeqBAIJSetPreallocation_SeqBAIJ()
3361 b->bs2 = bs2; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3365 B->info.nz_unneeded = (PetscReal)b->maxnz * bs2; in MatSeqBAIJSetPreallocation_SeqBAIJ()
3507 B->info.nz_unneeded = (PetscReal)b->maxnz * b->bs2; in MatCreate_SeqBAIJ()
3531 PetscInt i, mbs = a->mbs, nz = a->nz, bs2 = a->bs2; in MatDuplicateNoCreate_SeqBAIJ() local
3553 PetscCall(PetscShmgetAllocateArray(bs2 * nz, sizeof(PetscScalar), (void **)&c->a)); in MatDuplicateNoCreate_SeqBAIJ()
3554 PetscCall(PetscArrayzero(c->a, bs2 * nz)); in MatDuplicateNoCreate_SeqBAIJ()
3567 PetscCall(PetscShmgetAllocateArray(bs2 * nz, sizeof(PetscScalar), (void **)&c->a)); in MatDuplicateNoCreate_SeqBAIJ()
3577 PetscCall(PetscArraycpy(c->a, a->a, bs2 * nz)); in MatDuplicateNoCreate_SeqBAIJ()
3579 PetscCall(PetscArrayzero(c->a, bs2 * nz)); in MatDuplicateNoCreate_SeqBAIJ()
3593 c->bs2 = a->bs2; in MatDuplicateNoCreate_SeqBAIJ()