| /petsc/src/ksp/ksp/impls/hpddm/cuda/ |
| H A D | hpddm.cu | 22 thrust::device_ptr<K> dptr[2]; in KSPSolve_HPDDM_CUDA_Private() local
26 dptr[0] = thrust::device_pointer_cast(ptr); in KSPSolve_HPDDM_CUDA_Private()
27 dptr[1] = thrust::device_pointer_cast(ptr + N); in KSPSolve_HPDDM_CUDA_Private()
28 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), db, N, dptr[0]); in KSPSolve_HPDDM_CUDA_Private()
29 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), dx, N, dptr[1]); in KSPSolve_HPDDM_CUDA_Private()
33 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), dptr[1], N, dx); in KSPSolve_HPDDM_CUDA_Private()
|
| /petsc/src/tao/matrix/ |
| H A D | adamat.c | 152 PetscScalar *dtemp, *dptr; in MatADAComputeDiagonal() local
168 PetscCall(VecGetArray(ctx->ADADiag, &dptr)); in MatADAComputeDiagonal()
169 for (i = low; i < high; i++) dptr[i - low] = dtemp[i]; in MatADAComputeDiagonal()
170 PetscCall(VecRestoreArray(ctx->ADADiag, &dptr)); in MatADAComputeDiagonal()
277 const PetscScalar *dptr; in MatConvert_ADA() local
287 PetscCall(VecGetArrayRead(X, &dptr)); in MatConvert_ADA()
288 …for (j = 0; j < high - low; j++) PetscCall(MatSetValue(*NewMat, low + j, i, dptr[j], INSERT_VALUES… in MatConvert_ADA()
289 PetscCall(VecRestoreArrayRead(X, &dptr)); in MatConvert_ADA()
296 const PetscScalar *dptr; in MatConvert_ADA() local
306 PetscCall(VecGetArrayRead(X, &dptr)); in MatConvert_ADA()
[all …]
|
| /petsc/src/tao/unconstrained/impls/owlqn/ |
| H A D | owlqn.c | 11 PetscReal *dptr; in ProjDirect_OWLQN() local
19 PetscCall(VecGetArray(d, &dptr)); in ProjDirect_OWLQN()
21 if (dptr[i] * gptr[i] <= 0.0) dptr[i] = 0.0; in ProjDirect_OWLQN()
23 PetscCall(VecRestoreArray(d, &dptr)); in ProjDirect_OWLQN()
|
| /petsc/include/petsc/private/ |
| H A D | matdensecupmimpl.h | 391 const auto dptr = thrust::device_pointer_cast(data); in MakeDiagonalIterator() local
393 return {dptr + begin, dptr + end, lda + 1}; in MakeDiagonalIterator()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 1339 const auto dptr = thrust::device_pointer_cast(xptr.data()); in SetAsync() local
1341 PetscCallThrust(THRUST_CALL(thrust::fill, stream, dptr, dptr + n, alpha)); in SetAsync()
2185 const auto dptr = thrust::device_pointer_cast(DeviceArrayRead(dctx, v).data()); in Sum() local
2187 PetscCallThrust(*sum = THRUST_CALL(thrust::reduce, stream, dptr, dptr + n, PetscScalar{0.0});); in Sum()
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1445 const auto dptr = thrust::device_pointer_cast(ptr); in make_submat_iterator() local
1447 …return {dptr + (rstart * lda) + cstart, dptr + ((rstart + nrows) * lda) + cstart, nrows, ncols, ld… in make_submat_iterator()
|
| /petsc/src/mat/impls/aij/seq/seqhipsparse/ |
| H A D | aijhipsparse.hip.cxx | 3026 PetscScalar *xarray, *zarray, *dptr, *beta, *xptr; in MatMultAddKernel_SeqAIJHIPSPARSE() local
3066 dptr = compressed ? hipsparsestruct->workVector->data().get() : zarray; in MatMultAddKernel_SeqAIJHIPSPARSE()
3082 dptr = zarray; in MatMultAddKernel_SeqAIJHIPSPARSE()
3105 …PetscCallHIPSPARSE(hipsparseCreateDnVec(&matstruct->hipSpMV[opA].vecYDescr, ny, dptr, hipsparse_sc… in MatMultAddKernel_SeqAIJHIPSPARSE()
3113 PetscCallHIPSPARSE(hipsparseDnVecSetValues(matstruct->hipSpMV[opA].vecYDescr, dptr)); 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
3603 dptr = compressed ? cusparsestruct->workVector->data().get() : zarray; in MatMultAddKernel_SeqAIJCUSPARSE()
3621 dptr = zarray; in MatMultAddKernel_SeqAIJCUSPARSE()
3661 …PetscCallCUSPARSE(cusparseCreateDnVec(&matstruct->cuSpMV[opA].vecYDescr, ny, dptr, cusparse_scalar… in MatMultAddKernel_SeqAIJCUSPARSE()
3673 PetscCallCUSPARSE(cusparseDnVecSetValues(matstruct->cuSpMV[opA].vecYDescr, dptr)); 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()
|