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).

This commit is contained in:
Rasmus Munk Larsen 2022-01-12 20:16:16 +00:00
parent 5a0a165c09
commit a30ecb7221

View File

@ -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.