| /petsc/src/tao/unconstrained/tutorials/ |
| H A D | rosenbrock4.h | 17 PetscInt c_start, c_end; member
142 PetscInt c_start, c_end; in CreateHessian() local
159 user->problem.c_end = c_end = (i_end / bs) * (bs - 1) + (i_end % bs); in CreateHessian()
160 user->n_local_comp = n_local_comp = c_end - c_start; in CreateHessian()
182 for (PetscInt c = c_start, k = 0; c < c_end; c++, k += 4) { in CreateHessian()
259 PetscInt c_start = user->problem.c_start, c_end = user->problem.c_end, bs = user->problem.bs; in CreateVectors() local
288 for (PetscInt c = c_start, k = 0; c < c_end; c++, k += 2) { in CreateVectors()
330 for (PetscInt c = r.c_start + idx, k = idx; c < r.c_end; c += num_threads, k += num_threads) { in rosenbrock_for_loop()
362 PetscInt n_comp = r.c_end - r.c_start; in RosenbrockObjective_Device()
372 PetscInt n_comp = r.c_end - r.c_start; in RosenbrockGradient_Device()
[all …]
|
| /petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/ |
| H A D | blas_cyclic.c | 547 …C_INTERN PetscErrorCode MatMultColumnRange(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultColumnRange() argument
551 …ethod(A, "MatMultColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (A, xx, yy, c_start, c_end)); in MatMultColumnRange()
556 …tscErrorCode MatMultAddColumnRange(Mat A, Vec xx, Vec zz, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultAddColumnRange() argument
560 …tMultAddColumnRange_C", (Mat, Vec, Vec, Vec, PetscInt, PetscInt), (A, xx, zz, yy, c_start, c_end)); in MatMultAddColumnRange()
565 …rCode MatMultHermitianTransposeColumnRange(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultHermitianTransposeColumnRange() argument
569 …ermitianTransposeColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (A, xx, yy, c_start, c_end)); in MatMultHermitianTransposeColumnRange()
574 …ltHermitianTransposeAddColumnRange(Mat A, Vec xx, Vec zz, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultHermitianTransposeAddColumnRange() argument
578 …sposeAddColumnRange_C", (Mat, Vec, Vec, Vec, PetscInt, PetscInt), (A, xx, zz, yy, c_start, c_end)); in MatMultHermitianTransposeAddColumnRange()
|
| /petsc/src/mat/impls/dense/mpi/ |
| H A D | mpidense.c | 473 …tscErrorCode MatMultColumnRange_MPIDense(Mat mat, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultColumnRange_MPIDense() argument
488 …tMultColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (mdn->A, mdn->lvec, yy, c_start, c_end)); in MatMultColumnRange_MPIDense()
511 …e MatMultAddColumnRange_MPIDense(Mat mat, Vec xx, Vec yy, Vec zz, PetscInt c_start, PetscInt c_end) in MatMultAddColumnRange_MPIDense() argument
526 …mnRange_C", (Mat, Vec, Vec, Vec, PetscInt, PetscInt), (mdn->A, mdn->lvec, yy, zz, c_start, c_end)); in MatMultAddColumnRange_MPIDense()
549 …MultHermitianTransposeColumnRange_MPIDense(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultHermitianTransposeColumnRange_MPIDense() argument
563 c_end_local = PetscMin(c_end, r_end); in MatMultHermitianTransposeColumnRange_MPIDense()
573 …TransposeColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (a->A, xx, a->lvec, c_start, c_end)); in MatMultHermitianTransposeColumnRange_MPIDense()
603 …anTransposeAddColumnRange_MPIDense(Mat A, Vec xx, Vec yy, Vec zz, PetscInt c_start, PetscInt c_end) in MatMultHermitianTransposeAddColumnRange_MPIDense() argument
614 …TransposeColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (a->A, xx, a->lvec, c_start, c_end)); in MatMultHermitianTransposeAddColumnRange_MPIDense()
|
| /petsc/src/mat/impls/dense/seq/ |
| H A D | dense.c | 1045 …umnRangeKernel_SeqDense(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end, PetscBool trans, … in MatMultColumnRangeKernel_SeqDense() argument
1054 PetscCall(PetscBLASIntCast(c_end - c_start, &n)); in MatMultColumnRangeKernel_SeqDense()
1077 …MultHermitianTransposeColumnRange_SeqDense(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultHermitianTransposeColumnRange_SeqDense() argument
1080 PetscCall(MatMultColumnRangeKernel_SeqDense(A, xx, yy, c_start, c_end, PETSC_TRUE, PETSC_TRUE)); in MatMultHermitianTransposeColumnRange_SeqDense()
1105 …Kernel_SeqDense(Mat A, Vec xx, Vec zz, Vec yy, PetscInt c_start, PetscInt c_end, PetscBool trans, … in MatMultAddColumnRangeKernel_SeqDense() argument
1114 PetscCall(PetscBLASIntCast(c_end - c_start, &n)); in MatMultAddColumnRangeKernel_SeqDense()
1131 PetscErrorCode MatMultColumnRange_SeqDense(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultColumnRange_SeqDense() argument
1134 PetscCall(MatMultColumnRangeKernel_SeqDense(A, xx, yy, c_start, c_end, PETSC_FALSE, PETSC_FALSE)); in MatMultColumnRange_SeqDense()
1138 …ode MatMultAddColumnRange_SeqDense(Mat A, Vec xx, Vec zz, Vec yy, PetscInt c_start, PetscInt c_end) in MatMultAddColumnRange_SeqDense() argument
1141 …PetscCall(MatMultAddColumnRangeKernel_SeqDense(A, xx, zz, yy, c_start, c_end, PETSC_FALSE, PETSC_F… in MatMultAddColumnRange_SeqDense()
[all …]
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 805 …tAddColumnRange_Dispatch_(Mat A, Vec xx, Vec yy, Vec zz, PetscInt c_start, PetscInt c_end) noexcept in MatMultAddColumnRange_Dispatch_() argument
808 const auto n = static_cast<cupmBlasInt_t>(c_end - c_start); in MatMultAddColumnRange_Dispatch_()
877 …<T>::MatMultColumnRange_Dispatch_(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) noexcept in MatMultColumnRange_Dispatch_() argument
880 …tscCall(MatMultAddColumnRange_Dispatch_<transpose, hermitian>(A, xx, nullptr, yy, c_start, c_end)); in MatMultColumnRange_Dispatch_()
|