Switched to the latest version of Eigen that provides significant performance

improvements for fp16
Added SpecialFunctions to the list of eigen headers TensorFlow depends on
Change: 127264575

11 files changed, 14 insertions, 12 deletions

diff --git a/eigen.BUILD b/eigen.BUILD
index 14591aa42f..fa054e426a 100644
--- a/eigen.BUILD
+++ b/eigen.BUILD
@@ -1,9 +1,9 @@
 package(default_visibility = ["//visibility:public"])
 
-archive_dir = "eigen-eigen-334b1d428283"
+archive_dir = "eigen-eigen-b4fa9622b809"
 cc_library(
     name = "eigen",
-    hdrs = glob([archive_dir+"/**/*.h", archive_dir+"/unsupported/Eigen/CXX11/*", archive_dir+"/Eigen/*"]),
+    hdrs = glob([archive_dir+"/**/*.h", archive_dir+"/unsupported/Eigen/*", archive_dir+"/unsupported/Eigen/CXX11/*", archive_dir+"/Eigen/*"]),
     includes = [ archive_dir ],
     visibility = ["//visibility:public"],
 )
diff --git a/tensorflow/contrib/cmake/external/eigen.cmake b/tensorflow/contrib/cmake/external/eigen.cmake
index c633a9d1c1..156d93affb 100644
--- a/tensorflow/contrib/cmake/external/eigen.cmake
+++ b/tensorflow/contrib/cmake/external/eigen.cmake
@@ -7,7 +7,7 @@
 
 include (ExternalProject)
 
-set(eigen_archive_hash "802d984ade26")
+set(eigen_archive_hash "b4fa9622b809")
 
 set(eigen_INCLUDE_DIRS
     ${CMAKE_CURRENT_BINARY_DIR}
@@ -16,7 +16,7 @@ set(eigen_INCLUDE_DIRS
     ${tensorflow_source_dir}/third_party/eigen3
 )
 set(eigen_URL https://bitbucket.org/eigen/eigen/get/${eigen_archive_hash}.tar.gz)
-set(eigen_HASH SHA256=1499997676bd9006082950a761b88d5c48554fd550747763b2b34951da29a2e8)
+set(eigen_HASH SHA256=2862840c2de9c0473a4ef20f8678949ae89ab25965352ee53329e63ba46cec62)
 set(eigen_BUILD ${CMAKE_CURRENT_BINARY_DIR}/eigen/src/eigen)
 set(eigen_INSTALL ${CMAKE_CURRENT_BINARY_DIR}/eigen/install)
 
diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl
index 14ad56bd17..0c0018098b 100644
--- a/tensorflow/workspace.bzl
+++ b/tensorflow/workspace.bzl
@@ -6,8 +6,8 @@
 def tf_workspace(path_prefix = "", tf_repo_name = ""):
   native.new_http_archive(
     name = "eigen_archive",
-    url = "https://bitbucket.org/eigen/eigen/get/334b1d428283.tar.gz",
-    sha256 = "6d5efd02c7c11fbb9d02df4f0b64f22ecbd348e7549f8a83c13fb4d8d9e19d4b",
+    url = "https://bitbucket.org/eigen/eigen/get/b4fa9622b809.tar.gz",
+    sha256 = "2862840c2de9c0473a4ef20f8678949ae89ab25965352ee53329e63ba46cec62",
     build_file = path_prefix + "eigen.BUILD",
   )
 
diff --git a/third_party/eigen3/BUILD b/third_party/eigen3/BUILD
index e942a80b2d..74d4212d1e 100644
--- a/third_party/eigen3/BUILD
+++ b/third_party/eigen3/BUILD
@@ -8,6 +8,7 @@ cc_library(
         "Eigen/Cholesky",
         "Eigen/Eigenvalues",
         "Eigen/QR",
+        "unsupported/Eigen/SpecialFunctions",
         "unsupported/Eigen/CXX11/Tensor",
         "unsupported/Eigen/CXX11/FixedPoint",
         "unsupported/Eigen/CXX11/src/FixedPoint/*.h",
diff --git a/third_party/eigen3/Eigen/Cholesky b/third_party/eigen3/Eigen/Cholesky
index e3bc2a524c..eaa82ee9a8 100644
--- a/third_party/eigen3/Eigen/Cholesky
+++ b/third_party/eigen3/Eigen/Cholesky
@@ -1 +1 @@
-#include "eigen-eigen-334b1d428283/Eigen/Cholesky"
+#include "eigen-eigen-b4fa9622b809/Eigen/Cholesky"
diff --git a/third_party/eigen3/Eigen/Core b/third_party/eigen3/Eigen/Core
index cd000b599c..90e8342131 100644
--- a/third_party/eigen3/Eigen/Core
+++ b/third_party/eigen3/Eigen/Core
@@ -1 +1 @@
-#include "eigen-eigen-334b1d428283/Eigen/Core"
+#include "eigen-eigen-b4fa9622b809/Eigen/Core"
diff --git a/third_party/eigen3/Eigen/Eigenvalues b/third_party/eigen3/Eigen/Eigenvalues
index ac7f28d926..be5e8bacec 100644
--- a/third_party/eigen3/Eigen/Eigenvalues
+++ b/third_party/eigen3/Eigen/Eigenvalues
@@ -1 +1 @@
-#include "eigen-eigen-334b1d428283/Eigen/Eigenvalues"
+#include "eigen-eigen-b4fa9622b809/Eigen/Eigenvalues"
diff --git a/third_party/eigen3/Eigen/LU b/third_party/eigen3/Eigen/LU
index 1e9443f7a1..60d382d438 100644
--- a/third_party/eigen3/Eigen/LU
+++ b/third_party/eigen3/Eigen/LU
@@ -1 +1 @@
-#include "eigen-eigen-334b1d428283/Eigen/LU"
+#include "eigen-eigen-b4fa9622b809/Eigen/LU"
diff --git a/third_party/eigen3/Eigen/QR b/third_party/eigen3/Eigen/QR
index 2a79083ae9..1dcc8d74f3 100644
--- a/third_party/eigen3/Eigen/QR
+++ b/third_party/eigen3/Eigen/QR
@@ -1 +1 @@
-#include "eigen-eigen-334b1d428283/Eigen/QR"
+#include "eigen-eigen-b4fa9622b809/Eigen/QR"
diff --git a/third_party/eigen3/unsupported/Eigen/CXX11/Tensor b/third_party/eigen3/unsupported/Eigen/CXX11/Tensor
index 708503dec5..f1fb5ed77b 100644
--- a/third_party/eigen3/unsupported/Eigen/CXX11/Tensor
+++ b/third_party/eigen3/unsupported/Eigen/CXX11/Tensor
@@ -1 +1 @@
-#include "eigen-eigen-334b1d428283/unsupported/Eigen/CXX11/Tensor"
+#include "eigen-eigen-b4fa9622b809/unsupported/Eigen/CXX11/Tensor"
diff --git a/third_party/eigen3/unsupported/Eigen/SpecialFunctions b/third_party/eigen3/unsupported/Eigen/SpecialFunctions
new file mode 100644
index 0000000000..47036387df
--- /dev/null
+++ b/third_party/eigen3/unsupported/Eigen/SpecialFunctions
@@ -0,0 +1 @@
+#include "eigen-eigen-b4fa9622b809/unsupported/Eigen/SpecialFunctions"
\ No newline at end of file