From a30ecb7221a46824b85cad5f9016efe6e5871d69 Mon Sep 17 00:00:00 2001
From: Rasmus Munk Larsen <rmlarsen@google.com>
Date: Wed, 12 Jan 2022 20:16:16 +0000
Subject: [PATCH] Don't use the fast implementation if EIGEN_GPU_CC, since
 integer_packet is not defined for float4 used by the GPU compiler (even on
 host).

---
 Eigen/src/Core/functors/UnaryFunctors.h | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Eigen/src/Core/functors/UnaryFunctors.h b/Eigen/src/Core/functors/UnaryFunctors.h
index 1092fee50..d56aae5c9 100644
--- a/Eigen/src/Core/functors/UnaryFunctors.h
+++ b/Eigen/src/Core/functors/UnaryFunctors.h
@@ -1033,7 +1033,9 @@ struct scalar_logistic_op {
   }
 };
 
-#ifndef EIGEN_GPU_COMPILE_PHASE
+// TODO(rmlarsen): Enable the following on host when integer_packet is defined 
+// for the relevant packet types.
+#ifdef EIGEN_GPU_CC
 
 /** \internal
   * \brief Template specialization of the logistic function for float.