From aa33446dace833fbf06632e586c80119b3d8ac11 Mon Sep 17 00:00:00 2001
From: Benoit Steiner <benoit.steiner.goog@gmail.com>
Date: Thu, 9 Jun 2016 08:22:27 -0700
Subject: Improved support for vectorization of 16-bit floats

---
 unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h | 5 +++++
 1 file changed, 5 insertions(+)

(limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h')

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h b/unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h
index b1645d56f..82a905a65 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorMeta.h
@@ -54,6 +54,11 @@ struct PacketType {
 
 // For CUDA packet types when using a GpuDevice
 #if defined(EIGEN_USE_GPU) && defined(__CUDACC__)
+template <>
+  struct PacketType<half, GpuDevice> {
+  typedef half2 type;
+  static const int size = 2;
+ };
 template <>
 struct PacketType<float, GpuDevice> {
   typedef float4 type;
-- 
cgit v1.2.3