Declare the half float type as arithmetic.

This commit is contained in:
Benoit Steiner 2016-02-22 13:59:33 -08:00
parent 5cd00068c0
commit 6270d851e3

View File

@ -64,6 +64,7 @@ __device__ half abs(const half& a) {
namespace Eigen {
namespace internal {
template<> struct is_arithmetic<half> { enum { value = true }; };
template<> struct is_arithmetic<half2> { enum { value = true }; };
template<> struct packet_traits<half> : default_packet_traits