From 116c5235ac5600672982e5a12a0ab89d19d2d5f0 Mon Sep 17 00:00:00 2001
From: Sheng Yang <yang.sheng@intel.com>
Date: Wed, 1 Jul 2020 18:33:42 +0000
Subject: BF16 for scalar_cmp_with_cast_op

---
 Eigen/src/Core/functors/BinaryFunctors.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'Eigen/src/Core/functors')

diff --git a/Eigen/src/Core/functors/BinaryFunctors.h b/Eigen/src/Core/functors/BinaryFunctors.h
index 54bcddd34..d8b7b1eba 100644
--- a/Eigen/src/Core/functors/BinaryFunctors.h
+++ b/Eigen/src/Core/functors/BinaryFunctors.h
@@ -260,7 +260,7 @@ template<typename LhsScalar, typename RhsScalar, ComparisonName cmp>
 struct functor_traits<scalar_cmp_with_cast_op<LhsScalar,RhsScalar, cmp> > {
   enum {
     Cost = (NumTraits<LhsScalar>::AddCost+NumTraits<RhsScalar>::AddCost)/2,
-    PacketAccess = internal::is_same<LhsScalar, RhsScalar>::value && internal::is_same<LhsScalar, float>::value
+    PacketAccess = internal::is_same<LhsScalar, RhsScalar>::value && packet_traits<LhsScalar>::HasCmp
   };
 };
 
-- 
cgit v1.2.3