diff options
| author |  Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-06-07 18:22:18 -0700 |
|---|---|---|
| committer | Benoit Steiner <benoit.steiner.goog@gmail.com> | 2016-06-07 18:22:18 -0700 |
| commit | 8fd57a97f203edac3f7e8681eafe752294386a24 (patch) | |
| tree | 44c0379a44766d7078d47b42eb10260a782cb01a /Eigen/src/Core/arch | |
| parent | d6d39c7ddb127d91ebfa4ea62e93ea51036f1760 (diff) | |
Enable the vectorization of adds and mults of fp16
Diffstat (limited to 'Eigen/src/Core/arch')
| -rw-r--r-- | Eigen/src/Core/arch/CUDA/PacketMathHalf.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h index 51386506f..959dff886 100644 --- a/Eigen/src/Core/arch/CUDA/PacketMathHalf.h +++ b/Eigen/src/Core/arch/CUDA/PacketMathHalf.h @@ -28,6 +28,8 @@ template<> struct packet_traits<Eigen::half> : default_packet_traits AlignedOnScalar = 1, size=2, HasHalfPacket = 0, + HasAdd = 1, + HasMul = 1, HasDiv = 1, HasSqrt = 1, HasRsqrt = 1, |