#include "tensorflow/core/kernels/bounds_check.h"
namespace Eigen {
-namespace numext {
-#if GOOGLE_CUDA
-template <>
-EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE std::complex<float> exp(
- const std::complex<float>& x) {
- auto com = ::expf(x.real());
- auto res_real = com * ::cosf(x.imag());
- auto res_imag = com * ::sinf(x.imag());
- return std::complex<float>(res_real, res_imag);
-}
-template <>
-EIGEN_DEVICE_FUNC EIGEN_ALWAYS_INLINE std::complex<double> exp(
- const std::complex<double>& x) {
- auto com = ::exp(x.real());
- auto res_real = com * ::cos(x.imag());
- auto res_imag = com * ::sin(x.imag());
- return std::complex<double>(res_real, res_imag);
-}
-#endif
-} // namespace numext
-
namespace internal {
template <typename T>
tf_http_archive(
name = "eigen_archive",
urls = [
- "https://mirror.bazel.build/bitbucket.org/eigen/eigen/get/2355b229ea4c.tar.gz",
- "https://bitbucket.org/eigen/eigen/get/2355b229ea4c.tar.gz",
+ "https://mirror.bazel.build/bitbucket.org/eigen/eigen/get/6913f0cf7d06.tar.gz",
+ "https://bitbucket.org/eigen/eigen/get/6913f0cf7d06.tar.gz",
],
- sha256 = "0cadb31a35b514bf2dfd6b5d38205da94ef326ec6908fc3fd7c269948467214f",
- strip_prefix = "eigen-eigen-2355b229ea4c",
+ sha256 = "791b836cacd03e20bae5bdd25f1c4a5505a0a9975ba94a61eb4e2631fbd1d53a",
+ strip_prefix = "eigen-eigen-6913f0cf7d06",
build_file = str(Label("//third_party:eigen.BUILD")),
patch_file = str(Label("//third_party:eigen_fix_cuda_compilation.patch"))
)