| /petsc/src/snes/tutorials/ |
| H A D | ex7.c | 112 static PetscErrorCode ComputeGamma(PetscInt d, PetscInt ldx, PetscScalar f[]) in ComputeGamma() argument
114 const PetscScalar fin[4] = {f[0 * ldx], f[1 * ldx], f[2 * ldx], f[3 * ldx]}; in ComputeGamma()
119 f[0 * ldx] = PETSC_i * fin[3]; in ComputeGamma()
120 f[1 * ldx] = PETSC_i * fin[2]; in ComputeGamma()
121 f[2 * ldx] = -PETSC_i * fin[1]; in ComputeGamma()
122 f[3 * ldx] = -PETSC_i * fin[0]; in ComputeGamma()
125 f[0 * ldx] = -fin[3]; in ComputeGamma()
126 f[1 * ldx] = fin[2]; in ComputeGamma()
127 f[2 * ldx] = fin[1]; in ComputeGamma()
128 f[3 * ldx] = -fin[0]; in ComputeGamma()
[all …]
|
| /petsc/include/petsc/private/ |
| H A D | dmpleximpl.h | 474 static inline void DMPlex_Mult2D_Internal(const PetscScalar A[], PetscInt ldx, const PetscScalar x[… in DMPlex_Mult2D_Internal() argument
476 const PetscScalar z[2] = {x[0 * ldx], x[1 * ldx]}; in DMPlex_Mult2D_Internal()
477 y[0 * ldx] = A[0] * z[0] + A[1] * z[1]; in DMPlex_Mult2D_Internal()
478 y[1 * ldx] = A[2] * z[0] + A[3] * z[1]; in DMPlex_Mult2D_Internal()
481 static inline void DMPlex_Mult3D_Internal(const PetscScalar A[], PetscInt ldx, const PetscScalar x[… in DMPlex_Mult3D_Internal() argument
483 const PetscScalar z[3] = {x[0 * ldx], x[1 * ldx], x[2 * ldx]}; in DMPlex_Mult3D_Internal()
484 y[0 * ldx] = A[0] * z[0] + A[1] * z[1] + A[2] * z[2]; in DMPlex_Mult3D_Internal()
485 y[1 * ldx] = A[3] * z[0] + A[4] * z[1] + A[5] * z[2]; in DMPlex_Mult3D_Internal()
486 y[2 * ldx] = A[6] * z[0] + A[7] * z[1] + A[8] * z[2]; in DMPlex_Mult3D_Internal()
489 static inline void DMPlex_MultTranspose2D_Internal(const PetscScalar A[], PetscInt ldx, const Petsc… in DMPlex_MultTranspose2D_Internal() argument
[all …]
|
| /petsc/src/ts/tests/ |
| H A D | ex27.c | 223 static void DMPlex_MultAdd2DReal_Internal(const PetscReal A[], PetscInt ldx, const PetscScalar x[],… in DMPlex_MultAdd2DReal_Internal() argument
227 z[1] = x[ldx]; in DMPlex_MultAdd2DReal_Internal()
229 y[ldx] += A[2] * z[0] + A[3] * z[1]; in DMPlex_MultAdd2DReal_Internal()
234 static void DMPlex_MultAdd3DReal_Internal(const PetscReal A[], PetscInt ldx, const PetscScalar x[],… in DMPlex_MultAdd3DReal_Internal() argument
238 z[1] = x[ldx]; in DMPlex_MultAdd3DReal_Internal()
239 z[2] = x[ldx * 2]; in DMPlex_MultAdd3DReal_Internal()
241 y[ldx] += A[3] * z[0] + A[4] * z[1] + A[5] * z[2]; in DMPlex_MultAdd3DReal_Internal()
242 y[ldx * 2] += A[6] * z[0] + A[7] * z[1] + A[8] * z[2]; in DMPlex_MultAdd3DReal_Internal()
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 413 …static PetscErrorCode Solve(Mat A, cupmScalar_t *x, cupmBlasInt_t ldx, cupmBlasInt_t m, cupmBlasIn… in Solve()
436 handle, op, m, nrhs, da.cupmdata(), lda, fact_ipiv, x, ldx, lwork in Solve()
442 …ER(cupmSolverXgetrs(handle, op, m, nrhs, da.cupmdata(), lda, fact_ipiv, x, ldx, mcu->d_fact_work, … in Solve()
513 …static PetscErrorCode Solve(Mat A, cupmScalar_t *x, cupmBlasInt_t ldx, cupmBlasInt_t m, cupmBlasIn… in Solve()
535 handle, CUPMSOLVER_FILL_MODE_LOWER, m, nrhs, da.cupmdata(), lda, x, ldx, lwork in Solve()
541 …Xpotrs(handle, CUPMSOLVER_FILL_MODE_LOWER, m, nrhs, da.cupmdata(), lda, x, ldx, mcu->d_fact_work, … in Solve()
600 …static PetscErrorCode Solve(Mat A, cupmScalar_t *x, cupmBlasInt_t ldx, cupmBlasInt_t m, cupmBlasIn… in Solve()
622 …_MODE_UPPER, CUPMBLAS_OP_T, CUPMBLAS_DIAG_NON_UNIT, rank, nrhs, &one, da.cupmdata(), lda, x, ldx)); in Solve()
623 …IDE_LEFT, CUPMSOLVER_OP_N, m, nrhs, rank, da.cupmdata(), lda, fact_tau, x, ldx, fact_work, fact_lw… in Solve()
628 … CUPMSOLVER_SIDE_LEFT, op, m, nrhs, rank, da.cupmdata(), lda, fact_tau, x, ldx, fact_work, fact_lw… in Solve()
[all …]
|
| /petsc/src/mat/impls/dense/seq/ |
| H A D | dense.c | 402 static PetscErrorCode MatSolve_SeqDense_Internal_LU(Mat A, PetscScalar *x, PetscBLASInt ldx, PetscB… in MatSolve_SeqDense_Internal_LU() argument
416 static PetscErrorCode MatSolve_SeqDense_Internal_Cholesky(Mat A, PetscScalar *x, PetscBLASInt ldx, … in MatSolve_SeqDense_Internal_Cholesky() argument
448 static PetscErrorCode MatSolve_SeqDense_Internal_QR(Mat A, PetscScalar *x, PetscBLASInt ldx, PetscB… in MatSolve_SeqDense_Internal_QR() argument
465 …rmqr_("L", &trans, &m, &nrhs, &mat->rank, mat->v, &mat->lda, mat->tau, x, &ldx, &fwork, &lfwork, &… in MatSolve_SeqDense_Internal_QR()
473 …rmqr_("L", &trans, &m, &nrhs, &mat->rank, mat->v, &mat->lda, mat->tau, x, &ldx, mat->fwork, &mat->… in MatSolve_SeqDense_Internal_QR()
477 …("LAPACKtrtrs", LAPACKtrtrs_("U", "N", "N", &mat->rank, &nrhs, mat->v, &mat->lda, x, &ldx, &info)); in MatSolve_SeqDense_Internal_QR()
481 for (PetscInt i = mat->rank; i < k; i++) x[j * ldx + i] = 0.; in MatSolve_SeqDense_Internal_QR()
487 …MatSolveTranspose_SeqDense_Internal_QR(Mat A, PetscScalar *x, PetscBLASInt ldx, PetscBLASInt m, Pe… in MatSolveTranspose_SeqDense_Internal_QR() argument
495 …PetscCallBLAS("LAPACKtrtrs", LAPACKtrtrs_("U", "T", "N", &m, &nrhs, mat->v, &mat->lda, x, &ldx, &i… in MatSolveTranspose_SeqDense_Internal_QR()
503 …CKormqr_("L", "N", &m, &nrhs, &mat->rank, mat->v, &mat->lda, mat->tau, x, &ldx, &fwork, &lfwork, &… in MatSolveTranspose_SeqDense_Internal_QR()
[all …]
|
| /petsc/src/mat/tests/ |
| H A D | ex70.c | 206 PetscInt m, n, k, M = 10, N = 10, K = 5, ldx = 3, ldb = 5, ldr = 4; in main() local
223 PetscCall(PetscOptionsGetInt(NULL, NULL, "-ldx", &ldx, NULL)); in main()
286 PetscCall(PetscMalloc1((m + ldx) * K, &dataX)); in main()
288 for (i = 0; i < (m + ldx) * K; i++) dataX[i] = MAGIC_NUMBER; in main()
294 PetscCall(MatDenseSetLDA(X, m + ldx)); in main()
|
| /petsc/src/mat/impls/aij/seq/ |
| H A D | aijfact.c | 802 PetscInt nz, neq, ldb, ldx; in MatMatSolve_SeqAIJ_inplace() local
822 PetscCall(MatDenseGetLDA(X, &ldx)); in MatMatSolve_SeqAIJ_inplace()
849 x += ldx; in MatMatSolve_SeqAIJ_inplace()
866 PetscInt nz, neq, ldb, ldx; in MatMatSolve_SeqAIJ() local
885 PetscCall(MatDenseGetLDA(X, &ldx)); in MatMatSolve_SeqAIJ()
913 x += ldx; in MatMatSolve_SeqAIJ()
930 PetscInt nz, neq, ldb, ldx; in MatMatSolveTranspose_SeqAIJ() local
949 PetscCall(MatDenseGetLDA(X, &ldx)); in MatMatSolveTranspose_SeqAIJ()
981 x += ldx; in MatMatSolveTranspose_SeqAIJ()
|
| /petsc/src/mat/impls/sbaij/seq/ |
| H A D | sbaijfact2.c | 2074 PetscInt nz, i, j, neq, ldb, ldx; in MatMatSolve_SeqSBAIJ_1_NaturalOrdering() local
2088 PetscCall(MatDenseGetLDA(X, &ldx)); in MatMatSolve_SeqSBAIJ_1_NaturalOrdering()
2110 x += ldx; in MatMatSolve_SeqSBAIJ_1_NaturalOrdering()
|