Lines Matching refs:bs2
287 PetscInt i, j, bs = A->rmap->bs, k, l, bs2 = a->bs2; in MatView_SeqSBAIJ_ASCII() local
360 if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) > 0.0) { in MatView_SeqSBAIJ_ASCII()
361 …, bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), (double)PetscImaginaryPart(… in MatView_SeqSBAIJ_ASCII()
362 } else if (PetscImaginaryPart(a->a[bs2 * k + l * bs + j]) < 0.0) { in MatView_SeqSBAIJ_ASCII()
363 … bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]), -(double)PetscImaginaryPart(… in MatView_SeqSBAIJ_ASCII()
365 … " (%" PetscInt_FMT ", %g) ", bs * a->j[k] + l, (double)PetscRealPart(a->a[bs2 * k + l * bs + j]))… in MatView_SeqSBAIJ_ASCII()
368 …Printf(viewer, " (%" PetscInt_FMT ", %g) ", bs * a->j[k] + l, (double)a->a[bs2 * k + l * bs + j])); in MatView_SeqSBAIJ_ASCII()
387 PetscInt row, i, j, k, l, mbs = a->mbs, bs = A->rmap->bs, bs2 = a->bs2; in MatView_SeqSBAIJ_Draw_Zoom() local
409 aa = a->a + j * bs2; in MatView_SeqSBAIJ_Draw_Zoom()
425 aa = a->a + j * bs2; in MatView_SeqSBAIJ_Draw_Zoom()
441 aa = a->a + j * bs2; in MatView_SeqSBAIJ_Draw_Zoom()
515 PetscInt brow, bcol, ridx, cidx, bs = A->rmap->bs, bs2 = a->bs2; in MatGetValues_SeqSBAIJ() local
528 ap = aa + bs2 * ai[brow]; in MatGetValues_SeqSBAIJ()
550 *v++ = ap[bs2 * i + bs * cidx + ridx]; in MatGetValues_SeqSBAIJ()
580 PetscInt *aj = a->j, nonew = a->nonew, bs2 = a->bs2, bs = A->rmap->bs, stepval; in MatSetValuesBlocked_SeqSBAIJ() local
593 ap = aa + bs2 * ai[row]; in MatSetValuesBlocked_SeqSBAIJ()
621 bap = ap + bs2 * i; in MatSetValuesBlocked_SeqSBAIJ()
625 for (jj = ii; jj < bs2; jj += bs) bap[jj] += *value++; in MatSetValuesBlocked_SeqSBAIJ()
629 for (jj = ii; jj < bs2; jj += bs) bap[jj] = *value++; in MatSetValuesBlocked_SeqSBAIJ()
648 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, row, col, rmax, aa, ai, aj, rp, ap, imax, nonew, Mat… in MatSetValuesBlocked_SeqSBAIJ()
653 PetscCall(PetscArraymove(ap + bs2 * (i + 1), ap + bs2 * i, bs2 * (N - i + 1))); in MatSetValuesBlocked_SeqSBAIJ()
654 PetscCall(PetscArrayzero(ap + bs2 * i, bs2)); in MatSetValuesBlocked_SeqSBAIJ()
656 bap = ap + bs2 * i; in MatSetValuesBlocked_SeqSBAIJ()
659 for (jj = ii; jj < bs2; jj += bs) bap[jj] = *value++; in MatSetValuesBlocked_SeqSBAIJ()
679 PetscInt mbs = a->mbs, bs2 = a->bs2, rmax = 0; in MatAssemblyEnd_SeqSBAIJ() local
692 ap = aa + bs2 * ai[i]; in MatAssemblyEnd_SeqSBAIJ()
695 PetscCall(PetscArraymove(ap - bs2 * fshift, ap, bs2 * N)); in MatAssemblyEnd_SeqSBAIJ()
707 …ock size %" PetscInt_FMT ", %" PetscInt_FMT " unneeded", m, A->cmap->n, A->rmap->bs, fshift * bs2); in MatAssemblyEnd_SeqSBAIJ()
709 …FMT " unneeded, %" PetscInt_FMT " used\n", m, A->rmap->N, A->rmap->bs, fshift * bs2, a->nz * bs2)); in MatAssemblyEnd_SeqSBAIJ()
715 A->info.nz_unneeded = (PetscReal)fshift * bs2; in MatAssemblyEnd_SeqSBAIJ()
743 PetscInt ridx, cidx, bs2 = a->bs2; in MatSetValues_SeqSBAIJ() local
753 ap = aa + bs2 * ai[brow]; /*ptr to beginning of element value of the row block*/ in MatSetValues_SeqSBAIJ()
789 bap = ap + bs2 * i + bs * cidx + ridx; in MatSetValues_SeqSBAIJ()
794 bap = ap + bs2 * i + bs * ridx + cidx; in MatSetValues_SeqSBAIJ()
804 …MatSeqXAIJReallocateAIJ(A, a->mbs, bs2, nrow, brow, bcol, rmax, aa, ai, aj, rp, ap, imax, nonew, M… in MatSetValues_SeqSBAIJ()
810 PetscCall(PetscArraymove(ap + bs2 * (i + 1), ap + bs2 * i, bs2 * (N - i + 1))); in MatSetValues_SeqSBAIJ()
811 PetscCall(PetscArrayzero(ap + bs2 * i, bs2)); in MatSetValues_SeqSBAIJ()
813 ap[bs2 * i + bs * cidx + ridx] = value; in MatSetValues_SeqSBAIJ()
815 if (brow == bcol && ridx < cidx) ap[bs2 * i + bs * ridx + cidx] = value; in MatSetValues_SeqSBAIJ()
922 PetscCheck(a->bs2 == b->bs2, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Different block size"); in MatCopy_SeqSBAIJ()
923 PetscCall(PetscArraycpy(b->a, a->a, a->bs2 * a->i[a->mbs])); in MatCopy_SeqSBAIJ()
964 PetscInt bs = Y->rmap->bs, bs2 = bs * bs; in MatAXPY_SeqSBAIJ() local
982 PetscCall(PetscBLASIntCast(x->nz * bs2, &bnz)); in MatAXPY_SeqSBAIJ()
1024 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatConjugate_SeqSBAIJ()
1035 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatRealPart_SeqSBAIJ()
1046 PetscInt i, nz = a->bs2 * a->i[a->mbs]; in MatImaginaryPart_SeqSBAIJ()
1058 PetscInt bs = A->rmap->bs, bs2 = baij->bs2, row, col; in MatZeroRowsColumns_SeqSBAIJ() local
1086 aa = baij->a + j * bs2 + (i % bs) + bs * k; in MatZeroRowsColumns_SeqSBAIJ()
1102 aa = baij->a + j * bs2 + (i % bs) + bs * k; in MatZeroRowsColumns_SeqSBAIJ()
1119 aa = baij->a + baij->i[row / bs] * bs2 + (row % bs); in MatZeroRowsColumns_SeqSBAIJ()
1145 PetscInt mbs = a->mbs, bs2 = a->bs2, rmax = 0; in MatEliminateZeros_SeqSBAIJ() local
1155 ap = aa + bs2 * k; in MatEliminateZeros_SeqSBAIJ()
1156 for (j = 0; j < bs2 && zero; j++) { in MatEliminateZeros_SeqSBAIJ()
1163 PetscCall(PetscArraymove(ap - bs2 * fshift, ap, bs2)); in MatEliminateZeros_SeqSBAIJ()
1335 PetscInt nz = aij->i[mat->rmap->N] * mat->rmap->bs * aij->bs2; in MatStoreValues_SeqSBAIJ()
1351 PetscInt nz = aij->i[mat->rmap->N] * mat->rmap->bs * aij->bs2; in MatRetrieveValues_SeqSBAIJ()
1365 PetscInt i, mbs, nbs, bs2; in MatSeqSBAIJSetPreallocation_SeqSBAIJ() local
1391 bs2 = bs * bs; in MatSeqSBAIJSetPreallocation_SeqSBAIJ()
1490 PetscCall(PetscShmgetAllocateArray(bs2 * nz, sizeof(PetscScalar), (void **)&b->a)); in MatSeqSBAIJSetPreallocation_SeqSBAIJ()
1495 PetscCall(PetscArrayzero(b->a, nz * bs2)); in MatSeqSBAIJSetPreallocation_SeqSBAIJ()
1509 b->bs2 = bs2; in MatSeqSBAIJSetPreallocation_SeqSBAIJ()
1787 B->info.nz_unneeded = (PetscReal)b->maxnz * b->bs2; in MatCreate_SeqSBAIJ()
1983 PetscInt i, mbs = a->mbs, nz = a->nz, bs2 = a->bs2; in MatDuplicate_SeqSBAIJ() local
2006 c->bs2 = a->bs2; in MatDuplicate_SeqSBAIJ()
2024 PetscCall(PetscShmgetAllocateArray(bs2 * nz, sizeof(PetscScalar), (void **)&c->a)); in MatDuplicate_SeqSBAIJ()
2027 PetscCall(PetscArrayzero(c->a, bs2 * nz)); in MatDuplicate_SeqSBAIJ()
2044 PetscCall(PetscArraycpy(c->a, a->a, bs2 * nz)); in MatDuplicate_SeqSBAIJ()
2046 PetscCall(PetscArrayzero(c->a, bs2 * nz)); in MatDuplicate_SeqSBAIJ()