Home
last modified time | relevance | path

Searched refs:PetscCallThrust (Results 1 – 7 of 7) sorted by relevance

/petsc/src/sys/objects/device/impls/cupm/
H A Dcupmthrustutility.hpp75 #define PetscCallThrust(...) \ macro
H A Dcupmallocator.hpp147 PetscCallThrust(THRUST_CALL(thrust::fill, stream->get_stream(), xptr, xptr + n, canary)); in set_canary()
/petsc/src/vec/vec/impls/seq/cupm/
H A Dvecseqcupm_impl.hpp194 PetscCallThrust( in PointwiseBinary_()
241 PetscCallThrust( in PointwiseUnary_()
1341 PetscCallThrust(THRUST_CALL(thrust::fill, stream, dptr, dptr + n, alpha)); in SetAsync()
1812 PetscCallThrust( in ExecuteWNorm()
1824 PetscCallThrust( in ExecuteWNorm()
2004 PetscCallThrust( in DotNorm2()
2113 PetscCallThrust( in MinMax_()
2122 PetscCallThrust( in MinMax_()
2187 PetscCallThrust(*sum = THRUST_CALL(thrust::reduce, stream, dptr, dptr + n, PetscScalar{0.0});); in Sum()
/petsc/include/petsc/private/
H A Dmatdensecupmimpl.h420 PetscCallThrust( in DiagonalUnaryTransform()
475 PetscCallThrust(THRUST_CALL(thrust::copy, stream, diagonal.begin(), diagonal.end(), dvp)); in GetDiagonal()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu734 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_Cholesky()
747 …PetscCallThrust(thrust::transform(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::device_poi… in MatSolve_SeqAIJCUSPARSE_Cholesky()
749 …PetscCallThrust(thrust::transform(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::device_poi… in MatSolve_SeqAIJCUSPARSE_Cholesky()
762 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_Cholesky()
1358 PetscCallThrust(thrust::sequence(thrust::device, csr2csc_a.begin(), csr2csc_a.end(), 0.0)); in MatSeqAIJCUSPARSEFormExplicitTranspose()
1391 …PetscCallThrust(thrust::transform(thrust::device, matrixT->values->begin(), matrixT->values->end()… in MatSeqAIJCUSPARSEFormExplicitTranspose()
1396 PetscCallThrust( in MatSeqAIJCUSPARSEFormExplicitTranspose()
1431 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_LU()
1452 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_LU()
1502 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolveTranspose_SeqAIJCUSPARSE_LU()
[all …]
/petsc/src/mat/impls/dense/seq/cupm/
H A Dmatseqdensecupm.hpp1479 PetscCallThrust( in Conjugate()
1492 PetscCallThrust( in Conjugate()
1530 PetscCallThrust( in Scale()
1578 PetscCallThrust( in AXPY()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx1055 PetscCallThrust(thrust::sequence(thrust::device, csr2csc_a.begin(), csr2csc_a.end(), 0.0)); in MatSeqAIJHIPSPARSEFormExplicitTranspose()
1069 …PetscCallThrust(thrust::transform(thrust::device, matrixT->values->begin(), matrixT->values->end()… in MatSeqAIJHIPSPARSEFormExplicitTranspose()
1071 PetscCallThrust( in MatSeqAIJHIPSPARSEFormExplicitTranspose()
4195 PetscCallThrust(thrust::merge(thrust::device, Azb, Aze, Bzb, Bze, Czb, IJCompare4())); in MatSeqAIJHIPSPARSEMergeMats()
4199 …PetscCallThrust(thrust::partition_copy(thrust::device, cci, cce, wPerm->begin(), p1, p2, thrust::i… in MatSeqAIJHIPSPARSEMergeMats()
4202 PetscCallThrust(thrust::copy_if(thrust::device, cci, cce, wPerm->begin(), p1, pred)); in MatSeqAIJHIPSPARSEMergeMats()
4203 PetscCallThrust(thrust::remove_copy_if(thrust::device, cci, cce, wPerm->begin(), p2, pred)); in MatSeqAIJHIPSPARSEMergeMats()