提交 a607538a 编写于 作者: B Benoit Steiner 提交者: TensorFlower Gardener

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
上级 9ca002e0
package(default_visibility = ["//visibility:public"])
archive_dir = "eigen-eigen-62a2305d5734"
archive_dir = "eigen-eigen-5f86b31739cd"
cc_library(
name = "eigen",
......
......@@ -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)
......
......@@ -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",
)
......
#include "eigen-eigen-62a2305d5734/Eigen/Cholesky"
#include "eigen-eigen-5f86b31739cd/Eigen/Cholesky"
#include "eigen-eigen-62a2305d5734/Eigen/Core"
#include "eigen-eigen-5f86b31739cd/Eigen/Core"
#include "eigen-eigen-62a2305d5734/Eigen/Eigenvalues"
#include "eigen-eigen-5f86b31739cd/Eigen/Eigenvalues"
#include "eigen-eigen-62a2305d5734/Eigen/LU"
#include "eigen-eigen-5f86b31739cd/Eigen/LU"
#include "eigen-eigen-62a2305d5734/Eigen/QR"
#include "eigen-eigen-5f86b31739cd/Eigen/QR"
#include "eigen-eigen-62a2305d5734/unsupported/Eigen/CXX11/Tensor"
#include "eigen-eigen-5f86b31739cd/unsupported/Eigen/CXX11/Tensor"
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册