Lines Matching refs:d2d
433 PetscScalar *darray, *d2d, v; in MatGetDiagonal_ScaLAPACK() local
451 PetscCall(PetscCalloc1(lszd, &d2d)); in MatGetDiagonal_ScaLAPACK()
461 PetscCallBLAS("SCALAPACKelset", SCALAPACKelset_(d2d, &j, &one, d2desc, &v)); in MatGetDiagonal_ScaLAPACK()
465 …PetscCallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, d2d, &one, &one, d2desc, darray, &o… in MatGetDiagonal_ScaLAPACK()
466 PetscCall(PetscFree(d2d)); in MatGetDiagonal_ScaLAPACK()
479 PetscCall(PetscCalloc1(lszd, &d2d)); in MatGetDiagonal_ScaLAPACK()
488 PetscCallBLAS("SCALAPACKelset", SCALAPACKelset_(d2d, &one, &j, d2desc, &v)); in MatGetDiagonal_ScaLAPACK()
492 …PetscCallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, d2d, &one, &one, d2desc, darray, &o… in MatGetDiagonal_ScaLAPACK()
493 PetscCall(PetscFree(d2d)); in MatGetDiagonal_ScaLAPACK()
507 PetscScalar *d2d; in MatDiagonalScale_ScaLAPACK() local
522 PetscCall(PetscCalloc1(lszd, &d2d)); in MatDiagonalScale_ScaLAPACK()
529 …PetscCallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, d, &one, &one, ddesc, d2d, &one, &o… in MatDiagonalScale_ScaLAPACK()
532 if (!a->grid->myrow) Cdgebs2d(a->grid->ictxt, "C", " ", 1, lszd, d2d, dlld); in MatDiagonalScale_ScaLAPACK()
533 else Cdgebr2d(a->grid->ictxt, "C", " ", 1, lszd, d2d, dlld, 0, a->grid->mycol); in MatDiagonalScale_ScaLAPACK()
537 for (i = 0; i < a->locr; i++) a->loc[i + j * a->lld] *= d2d[j]; in MatDiagonalScale_ScaLAPACK()
539 PetscCall(PetscFree(d2d)); in MatDiagonalScale_ScaLAPACK()
553 PetscCall(PetscCalloc1(lszd, &d2d)); in MatDiagonalScale_ScaLAPACK()
561 …PetscCallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, d, &one, &one, ddesc, d2d, &one, &o… in MatDiagonalScale_ScaLAPACK()
564 if (!a->grid->mycol) Cdgebs2d(a->grid->ictxt, "R", " ", lszd, 1, d2d, dlld); in MatDiagonalScale_ScaLAPACK()
565 else Cdgebr2d(a->grid->ictxt, "R", " ", lszd, 1, d2d, dlld, a->grid->myrow, 0); in MatDiagonalScale_ScaLAPACK()
569 for (j = 0; j < a->locc; j++) a->loc[i + j * a->lld] *= d2d[i]; in MatDiagonalScale_ScaLAPACK()
571 PetscCall(PetscFree(d2d)); in MatDiagonalScale_ScaLAPACK()