| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 242 const auto xptr = thrust::device_pointer_cast(xinout); in PointwiseUnary_()
247 xptr, xptr + n, in PointwiseUnary_()
248 (yin && (yin != xinout)) ? thrust::device_pointer_cast(yin) : xptr, in PointwiseUnary_()
473 const auto xptr = DeviceArrayRead(dctx, xin); in AYPXAsync() local
477 …PetscCallCUPMBLAS(cupmBlasXaxpy(cupmBlasHandle, n, calpha, xptr.cupmdata(), 1, yptr.cupmdata(), 1)… in AYPXAsync()
482 … PetscCallCUPMBLAS(cupmBlasXaxpy(cupmBlasHandle, n, &one, xptr.cupmdata(), 1, yptr.cupmdata(), 1)); in AYPXAsync()
857 PETSC_KERNEL_DECL static void MAXPY_kernel(const PetscInt size, PetscScalar *PETSC_RESTRICT xptr, c… in MAXPY_kernel() argument
876 xptr[i] += sum; in MAXPY_kernel()
878 auto sum = xptr[i]; in MAXPY_kernel()
882 xptr[i] = sum; in MAXPY_kernel()
[all …]
|
| /petsc/src/sys/objects/device/impls/cupm/ |
| H A D | cupmallocator.hpp | 144 const auto xptr = thrust::device_pointer_cast(ptr); in set_canary() local
147 PetscCallThrust(THRUST_CALL(thrust::fill, stream->get_stream(), xptr, xptr + n, canary)); in set_canary()
|
| /petsc/src/tao/linesearch/impls/owarmijo/ |
| H A D | owarmijo.c | 13 const PetscReal *xptr, *gptr; in ProjWork_OWLQN() local
24 PetscCall(VecGetArrayRead(x, &xptr)); in ProjWork_OWLQN()
28 if (xptr[i] * wptr[i] < 0.0) wptr[i] = 0.0; in ProjWork_OWLQN()
29 *gdx = *gdx + gptr[i] * (wptr[i] - xptr[i]); in ProjWork_OWLQN()
32 PetscCall(VecRestoreArrayRead(x, &xptr)); in ProjWork_OWLQN()
|
| /petsc/src/tao/unconstrained/impls/owlqn/ |
| H A D | owlqn.c | 30 const PetscReal *xptr; in ComputePseudoGrad_OWLQN() local
38 PetscCall(VecGetArrayRead(x, &xptr)); in ComputePseudoGrad_OWLQN()
41 if (xptr[i] < 0.0) gptr[i] = gptr[i] - lambda; in ComputePseudoGrad_OWLQN()
42 else if (xptr[i] > 0.0) gptr[i] = gptr[i] + lambda; in ComputePseudoGrad_OWLQN()
48 PetscCall(VecRestoreArrayRead(x, &xptr)); in ComputePseudoGrad_OWLQN()
|
| /petsc/src/mat/impls/aij/mpi/strumpack/ |
| H A D | strumpack.c | 845 PetscScalar *xptr; in MatSolve_STRUMPACK() local
848 PetscCall(VecGetArray(x, &xptr)); in MatSolve_STRUMPACK()
851 …ackCallExternalVoid("STRUMPACK_solve", sp_err = STRUMPACK_solve(*S, (PetscScalar *)bptr, xptr, 0)); in MatSolve_STRUMPACK()
866 PetscCall(VecRestoreArray(x, &xptr)); in MatSolve_STRUMPACK()
878 PetscScalar *xptr; in MatMatSolve_STRUMPACK() local
887 PetscCall(MatDenseGetArray(X, &xptr)); in MatMatSolve_STRUMPACK()
890 …CallExternalVoid("STRUMPACK_matsolve", sp_err = STRUMPACK_matsolve(*S, nrhs, bptr, m, xptr, m, 0)); in MatMatSolve_STRUMPACK()
906 PetscCall(MatDenseRestoreArray(X, &xptr)); in MatMatSolve_STRUMPACK()
|
| /petsc/src/vec/vec/utils/ |
| H A D | projection.c | 757 const PetscScalar *xptr, *xlptr, *xuptr; in VecBoundGradientProjection() local
775 PetscCall(VecGetArrayRead(X, &xptr)); in VecBoundGradientProjection()
782 xval = xptr[i]; in VecBoundGradientProjection()
791 PetscCall(VecRestoreArrayRead(X, &xptr)); in VecBoundGradientProjection()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3026 PetscScalar *xarray, *zarray, *dptr, *beta, *xptr; in MatMultAddKernel_SeqAIJHIPSPARSE() local
3065 xptr = xarray; in MatMultAddKernel_SeqAIJHIPSPARSE()
3081 xptr = compressed ? hipsparsestruct->workVector->data().get() : xarray; in MatMultAddKernel_SeqAIJHIPSPARSE()
3104 …PetscCallHIPSPARSE(hipsparseCreateDnVec(&matstruct->hipSpMV[opA].vecXDescr, nx, xptr, hipsparse_sc… in MatMultAddKernel_SeqAIJHIPSPARSE()
3112 PetscCallHIPSPARSE(hipsparseDnVecSetValues(matstruct->hipSpMV[opA].vecXDescr, xptr)); in MatMultAddKernel_SeqAIJHIPSPARSE()
3121 …ata().get(), mat->row_offsets->data().get(), mat->column_indices->data().get(), xptr, beta, dptr)); in MatMultAddKernel_SeqAIJHIPSPARSE()
3126 …v(hipsparsestruct->handle, opA, matstruct->alpha_one, matstruct->descr, hybMat, xptr, beta, dptr)); in MatMultAddKernel_SeqAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3560 PetscScalar *xarray, *zarray, *dptr, *beta, *xptr; in MatMultAddKernel_SeqAIJCUSPARSE() local
3602 xptr = xarray; in MatMultAddKernel_SeqAIJCUSPARSE()
3620 xptr = compressed ? cusparsestruct->workVector->data().get() : xarray; in MatMultAddKernel_SeqAIJCUSPARSE()
3660 …PetscCallCUSPARSE(cusparseCreateDnVec(&matstruct->cuSpMV[opA].vecXDescr, nx, xptr, cusparse_scalar… in MatMultAddKernel_SeqAIJCUSPARSE()
3672 PetscCallCUSPARSE(cusparseDnVecSetValues(matstruct->cuSpMV[opA].vecXDescr, xptr)); in MatMultAddKernel_SeqAIJCUSPARSE()
3679 …ata().get(), mat->row_offsets->data().get(), mat->column_indices->data().get(), xptr, beta, dptr)); in MatMultAddKernel_SeqAIJCUSPARSE()
3687 …mv(cusparsestruct->handle, opA, matstruct->alpha_one, matstruct->descr, hybMat, xptr, beta, dptr)); in MatMultAddKernel_SeqAIJCUSPARSE()
|