diff options
author | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-05-23 20:32:30 -0700 |
---|---|---|
committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-05-23 20:32:30 -0700 |
commit | 6bc684ab6a9c300d9d19a6371f253a4321c37e4c (patch) | |
tree | c019d9daa9c710c726bca68523894f50fc9928f0 /Eigen/src/Core | |
parent | 283e33dea490b82c8dff78ed8827e4a4f74cfd3e (diff) |
Added missing alignment in the fp16 packet traits
Diffstat (limited to 'Eigen/src/Core')
-rw-r--r-- | Eigen/src/Core/arch/CUDA/PacketMathHalf.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h index bb3b6692f..840c40466 100644 --- a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h +++ b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h @@ -355,7 +355,7 @@ struct packet_traits<half> : default_packet_traits { }; -template<> struct unpacket_traits<Packet8h> { typedef Eigen::half type; enum {size=8}; typedef Packet8h half; }; +template<> struct unpacket_traits<Packet8h> { typedef Eigen::half type; enum {size=8, alignment=Aligned16}; typedef Packet8h half; }; template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet8h pset1<Packet8h>(const half& from) { Packet8h result; @@ -586,7 +586,7 @@ struct packet_traits<half> : default_packet_traits { }; -template<> struct unpacket_traits<Packet4h> { typedef Eigen::half type; enum {size=4}; typedef Packet4h half; }; +template<> struct unpacket_traits<Packet4h> { typedef Eigen::half type; enum {size=4, alignment=Aligned16}; typedef Packet4h half; }; template<> EIGEN_DEVICE_FUNC EIGEN_STRONG_INLINE Packet4h pset1<Packet4h>(const half& from) { Packet4h result; |