Searched refs:THRUST_CALL (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/sys/objects/device/impls/cupm/ |
| H A D | cupmthrustutility.hpp | 69 #define THRUST_CALL(...) \ macro
|
| H A D | cupmallocator.hpp | 147 PetscCallThrust(THRUST_CALL(thrust::fill, stream->get_stream(), xptr, xptr + n, canary)); in set_canary()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 197 THRUST_CALL( in PointwiseBinary_()
244 THRUST_CALL( in PointwiseUnary_()
1341 PetscCallThrust(THRUST_CALL(thrust::fill, stream, dptr, dptr + n, alpha)); in SetAsync()
1813 thrust::tie(*norm, *norma, *normr, *norm_loc, *norma_loc, *normr_loc) = THRUST_CALL( in ExecuteWNorm()
1825 thrust::tie(*norm, *norma, *normr, *norm_loc, *norma_loc, *normr_loc) = THRUST_CALL( in ExecuteWNorm()
2005 thrust::tie(*dp, *nm) = THRUST_CALL( in DotNorm2()
2096 #define THRUST_MINMAX_REDUCE(...) THRUST_CALL(thrust::transform_reduce, __VA_ARGS__) in MinMax_()
2098 …#define THRUST_MINMAX_REDUCE(s, b, e, real_part__, ...) THRUST_CALL(thrust::reduce, s, b, e, __VA_… in MinMax_()
2187 PetscCallThrust(*sum = THRUST_CALL(thrust::reduce, stream, dptr, dptr + n, PetscScalar{0.0});); in Sum()
|
| /petsc/include/petsc/private/ |
| H A D | matdensecupmimpl.h | 421 THRUST_CALL( in DiagonalUnaryTransform()
475 PetscCallThrust(THRUST_CALL(thrust::copy, stream, diagonal.begin(), diagonal.end(), dvp)); in GetDiagonal()
|
| /petsc/src/mat/impls/dense/seq/cupm/ |
| H A D | matseqdensecupm.hpp | 1482 THRUST_CALL( in Conjugate()
1495 THRUST_CALL( in Conjugate()
1533 THRUST_CALL( in Scale()
1582 THRUST_CALL( in AXPY()
|