Enable the vectorization of adds and mult on fp16s. This improves the

performance of the toy mnist training by 1 order of magnitude Change: 124374286
sachinravi14 · Jun 8, 2016 · a607538 · a607538
1 parent 9ca002e
commit a607538
Show file tree

Hide file tree

Showing 9 changed files with 11 additions and 11 deletions.
diff --git a/eigen.BUILD b/eigen.BUILD
@@ -1,6 +1,6 @@
 package(default_visibility = ["//visibility:public"])
 
-archive_dir = "eigen-eigen-62a2305d5734"
+archive_dir = "eigen-eigen-5f86b31739cd"
 
 cc_library(
     name = "eigen",

diff --git a/tensorflow/contrib/cmake/external/eigen.cmake b/tensorflow/contrib/cmake/external/eigen.cmake
@@ -7,7 +7,7 @@
 
 include (ExternalProject)
 
-set(eigen_archive_hash "62a2305d5734")
+set(eigen_archive_hash "5f86b31739cd")
 
 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=d5da5c60f7225bc2f104f3494323b929e68e3a188ccf01dcee61df32ff536888)
+set(eigen_HASH SHA256=e1101422f198a5d8c07e008eb801aeac385bf3022d88a0f041acc9f77e09d26e)
 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
@@ -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/62a2305d5734.tar.gz",
-    sha256 = "d5da5c60f7225bc2f104f3494323b929e68e3a188ccf01dcee61df32ff536888",
+    url = "https://bitbucket.org/eigen/eigen/get/5f86b31739cd.tar.gz",
+    sha256 = "e1101422f198a5d8c07e008eb801aeac385bf3022d88a0f041acc9f77e09d26e",
     build_file = path_prefix + "eigen.BUILD",
   )
 

diff --git a/third_party/eigen3/Eigen/Cholesky b/third_party/eigen3/Eigen/Cholesky
@@ -1 +1 @@
-#include "eigen-eigen-62a2305d5734/Eigen/Cholesky"
+#include "eigen-eigen-5f86b31739cd/Eigen/Cholesky"
diff --git a/third_party/eigen3/Eigen/Core b/third_party/eigen3/Eigen/Core
@@ -1 +1 @@
-#include "eigen-eigen-62a2305d5734/Eigen/Core"
+#include "eigen-eigen-5f86b31739cd/Eigen/Core"
diff --git a/third_party/eigen3/Eigen/Eigenvalues b/third_party/eigen3/Eigen/Eigenvalues
@@ -1 +1 @@
-#include "eigen-eigen-62a2305d5734/Eigen/Eigenvalues"
+#include "eigen-eigen-5f86b31739cd/Eigen/Eigenvalues"
diff --git a/third_party/eigen3/Eigen/LU b/third_party/eigen3/Eigen/LU
@@ -1 +1 @@
-#include "eigen-eigen-62a2305d5734/Eigen/LU"
+#include "eigen-eigen-5f86b31739cd/Eigen/LU"
diff --git a/third_party/eigen3/Eigen/QR b/third_party/eigen3/Eigen/QR
@@ -1 +1 @@
-#include "eigen-eigen-62a2305d5734/Eigen/QR"
+#include "eigen-eigen-5f86b31739cd/Eigen/QR"
diff --git a/third_party/eigen3/unsupported/Eigen/CXX11/Tensor b/third_party/eigen3/unsupported/Eigen/CXX11/Tensor
@@ -1 +1 @@
-#include "eigen-eigen-62a2305d5734/unsupported/Eigen/CXX11/Tensor"
+#include "eigen-eigen-5f86b31739cd/unsupported/Eigen/CXX11/Tensor"
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		#include "eigen-eigen-62a2305d5734/Eigen/Cholesky"
		#include "eigen-eigen-5f86b31739cd/Eigen/Cholesky"
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		#include "eigen-eigen-62a2305d5734/unsupported/Eigen/CXX11/Tensor"
		#include "eigen-eigen-5f86b31739cd/unsupported/Eigen/CXX11/Tensor"