Home
last modified time | relevance | path

Searched refs:ldr (Results 1 – 4 of 4) sorted by relevance

/petsc/src/vec/is/sf/tests/
H A Dex5.c20 PetscInt ldr = PETSC_DECIDE; in TestVector() local
24 if (r == 1) ldr = mr; in TestVector()
25 if (r == 2) ldr = mr + 7; in TestVector()
29 PetscCall(PetscSFCreateStridedSF(sf, bs, ldr, ldl, &vsf)); in TestVector()
30 if (ldr == PETSC_DECIDE) ldr = mr; in TestVector()
33 … PetscCall(PetscCalloc4(bs * ldr, &rdata, bs * ldl, &ldata, bs * ldr, &rdatav, bs * ldl, &ldatav)); in TestVector()
34 for (PetscInt i = 0; i < bs * ldr; i++) rdata[i] = i + 1; in TestVector()
37 …PetscCall(PetscSFBcastBegin(sf, MPIU_INT, PetscSafePointerPlusOffset(rdata, i * ldr), PetscSafePoi… in TestVector()
38 …PetscCall(PetscSFBcastEnd(sf, MPIU_INT, PetscSafePointerPlusOffset(rdata, i * ldr), PetscSafePoint… in TestVector()
46 …ck size %" PetscInt_FMT ", ldr %" PetscInt_FMT ", ldl %" PetscInt_FMT "\n", sfname, bs, ldr, ldl)); in TestVector()
[all …]
/petsc/src/tao/leastsquares/impls/pounders/
H A Dgqt.c4 static PetscErrorCode estsv(PetscInt n, PetscReal *r, PetscInt ldr, PetscReal *svmin, PetscReal *z) in estsv() argument
12 PetscCall(PetscBLASIntCast(ldr, &blasldr)); in estsv()
25 if (PetscAbs(e - z[i]) > PetscAbs(r[i + ldr * i])) { in estsv()
26 temp = PetscMin(0.01, PetscAbs(r[i + ldr * i])) / PetscAbs(e - z[i]); in estsv()
32 if (r[i + ldr * i] == 0.0) { in estsv()
35 w = (e - z[i]) / r[i + ldr * i]; in estsv()
36 wm = -(e + z[i]) / r[i + ldr * i]; in estsv()
42 for (j = i + 1; j < n; j++) sm += PetscAbs(z[j] + wm * r[i + ldr * j]); in estsv()
45 …PetscCallBLAS("BLASaxpy", BLASaxpy_(&blasnmi, &w, &r[i + ldr * (i + 1)], &blasldr, &z[i + 1], &bla… in estsv()
51 …if (i < n - 1) PetscCallBLAS("BLASaxpy", BLASaxpy_(&blasnmi, &temp, &r[i + ldr * (i + 1)], &blasld… in estsv()
[all …]
/petsc/src/vec/is/sf/utils/
H A Dsfutils.c927 PetscErrorCode PetscSFCreateStridedSF(PetscSF sf, PetscInt bs, PetscInt ldr, PetscInt ldl, PetscSF … in PetscSFCreateStridedSF() argument
955 if (ldr == PETSC_DECIDE) ldr = nr; in PetscSFCreateStridedSF()
957 ldr /= PetscMax(1, sf->vscat.bs); in PetscSFCreateStridedSF()
958 …(ldr >= nr, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Invalid leading dimension %" PetscInt_FMT " mus… in PetscSFCreateStridedSF()
969 PetscCall(PetscSFBcastBegin(rankssf, MPIU_INT, &ldr, ldrs, MPI_REPLACE)); in PetscSFCreateStridedSF()
970 PetscCall(PetscSFBcastEnd(rankssf, MPIU_INT, &ldr, ldrs, MPI_REPLACE)); in PetscSFCreateStridedSF()
976 if (r == rank) lda = ldr; in PetscSFCreateStridedSF()
/petsc/src/mat/tests/
H A Dex70.c206 PetscInt m, n, k, M = 10, N = 10, K = 5, ldx = 3, ldb = 5, ldr = 4; in main() local
225 PetscCall(PetscOptionsGetInt(NULL, NULL, "-ldr", &ldr, NULL)); in main()
301 PetscCall(PetscMalloc1((k + ldr) * N, &dataBt)); in main()
302 for (i = 0; i < (k + ldr) * N; i++) dataBt[i] = MAGIC_NUMBER; in main()
305 if (local) PetscCall(MatDenseSetLDA(Bt, k + ldr)); in main()
456 PetscCall(PetscMalloc1((k + ldr) * M, &dataR)); in main()
458 PetscCall(MatDenseSetLDA(R, k + ldr)); in main()