devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Vectorize cast

Browse Source
This commit is contained in:
Charles Schlosser
2023-04-26 02:50:13 +00:00
committed by Rasmus Munk Larsen
parent 3918768be1
commit eb5ff1861a
8 changed files with 254 additions and 128 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
Eigen/src/Core/functors/UnaryFunctors.h
View File

@@ -173,22 +173,40 @@ struct functor_traits<scalar_carg_op<Scalar>> {
*
* \sa class CwiseUnaryOp, MatrixBase::cast()
*/
template<typename Scalar, typename NewType>
template <typename SrcType, typename DstType>
struct scalar_cast_op {
typedef NewType result_type;
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const NewType operator() (const Scalar& a) const { return cast<Scalar, NewType>(a); }
using result_type = DstType;
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const DstType operator()(const SrcType& a) const {
return cast<SrcType, DstType>(a);
}
using SrcPacket = typename packet_traits<SrcType>::type;
template <typename DstPacket>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const DstPacket packetOp(const SrcPacket& a) const {
return pcast<SrcPacket, DstPacket>(a);
}
template <typename DstPacket>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const DstPacket packetOp(const SrcPacket& a, const SrcPacket& b) const {
return pcast<SrcPacket, DstPacket>(a, b);
}
template <typename DstPacket>
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE const DstPacket packetOp(const SrcPacket& a, const SrcPacket& b,
const SrcPacket& c, const SrcPacket& d) const {
return pcast<SrcPacket, DstPacket>(a, b, c, d);
}
};
template <typename Scalar>
struct scalar_cast_op<Scalar, bool> {
typedef bool result_type;
EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE bool operator()(const Scalar& a) const { return a != Scalar(0); }
template <typename SrcType, typename DstType>
struct functor_traits<scalar_cast_op<SrcType, DstType>> {
enum {
Cost = is_same<SrcType, DstType>::value ? 0 : NumTraits<DstType>::AddCost,
PacketAccess = (type_casting_traits<SrcType, DstType>::VectorizedCast != 0) &&
(type_casting_traits<SrcType, DstType>::SrcCoeffRatio <= 4)
};
};
template<typename Scalar, typename NewType>
struct functor_traits<scalar_cast_op<Scalar,NewType> >
{ enum { Cost = is_same<Scalar, NewType>::value ? 0 : NumTraits<NewType>::AddCost, PacketAccess = false }; };
/** \internal
* \brief Template functor to arithmetically shift a scalar right by a number of bits
*