| /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
158 user->problem.c_start = c_start = (i_start / bs) * (bs - 1) + (i_start % 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 PETSC_INTERN PetscErrorCode MatMultColumnRange(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_… in MatMultColumnRange() argument
551 …ethod(A, "MatMultColumnRange_C", (Mat, Vec, Vec, PetscInt, PetscInt), (A, xx, yy, c_start, c_end)); in MatMultColumnRange()
556 PETSC_INTERN PetscErrorCode MatMultAddColumnRange(Mat A, Vec xx, Vec zz, Vec yy, PetscInt c_start, … 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/seq/ |
| H A D | dense.c | 1045 …orCode MatMultColumnRangeKernel_SeqDense(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end, … in MatMultColumnRangeKernel_SeqDense() argument
1054 PetscCall(PetscBLASIntCast(c_end - c_start, &n)); in MatMultColumnRangeKernel_SeqDense()
1065 …BLASgemv", BLASgemv_("C", &m, &n, &_DOne, v + c_start * mat->lda, &mat->lda, x, &_One, &_DZero, y … in MatMultColumnRangeKernel_SeqDense()
1066 …BLASgemv", BLASgemv_("T", &m, &n, &_DOne, v + c_start * mat->lda, &mat->lda, x, &_One, &_DZero, y … in MatMultColumnRangeKernel_SeqDense()
1068 …allBLAS("BLASgemv", BLASgemv_("N", &m, &n, &_DOne, v + c_start * mat->lda, &mat->lda, x + c_start,… 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 …ultAddColumnRangeKernel_SeqDense(Mat A, Vec xx, Vec zz, Vec yy, PetscInt c_start, PetscInt c_end, … in MatMultAddColumnRangeKernel_SeqDense() argument
1114 PetscCall(PetscBLASIntCast(c_end - c_start, &n)); in MatMultAddColumnRangeKernel_SeqDense()
1120 …BLASgemv", BLASgemv_("C", &m, &n, &_DOne, v + c_start * mat->lda, &mat->lda, x, &_One, &_DOne, y +… in MatMultAddColumnRangeKernel_SeqDense()
[all …]
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 805 …:MatMultAddColumnRange_Dispatch_(Mat A, Vec xx, Vec yy, Vec zz, PetscInt c_start, PetscInt c_end) … in MatMultAddColumnRange_Dispatch_() argument
808 const auto n = static_cast<cupmBlasInt_t>(c_end - c_start); in MatMultAddColumnRange_Dispatch_()
862 …a.cupmdata() + c_start * lda, lda, dxx.cupmdata() + (transpose ? 0 : c_start), 1, y ? &one : &zero… in MatMultAddColumnRange_Dispatch_()
877 …eq_CUPM<T>::MatMultColumnRange_Dispatch_(Mat A, Vec xx, Vec yy, PetscInt c_start, PetscInt c_end) … in MatMultColumnRange_Dispatch_() argument
880 …PetscCall(MatMultAddColumnRange_Dispatch_<transpose, hermitian>(A, xx, nullptr, yy, c_start, c_end… in MatMultColumnRange_Dispatch_()
|
| /petsc/src/mat/impls/dense/mpi/ |
| H A D | mpidense.c | 473 static PetscErrorCode MatMultColumnRange_MPIDense(Mat mat, Vec xx, Vec yy, PetscInt c_start, PetscI… 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
562 c_start_local = PetscMax(c_start, r_start); 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()
|