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

Prevent TensorFlow from crashing when attempting to reduce an empty tensor on GPU

Change: 120505517
上级 bcf3dcb4
package(default_visibility = ["//visibility:public"])
archive_dir = "eigen-eigen-0823d98fdde7"
archive_dir = "eigen-eigen-ffa59a39ffb9"
cc_library(
name = "eigen",
......
......@@ -7,7 +7,7 @@
include (ExternalProject)
set(eigen_archive_hash "0823d98fdde7")
set(eigen_archive_hash "ffa59a39ffb9")
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=bbb8504e2fbb02937bb4efba4631f2344cd898a4fcee2cf4e7f22542de9aeb52)
set(eigen_HASH SHA256=ccca47fe6528a97bcec26c733908114bbfb701c5d818d095820da94b158ef8e1)
set(eigen_BUILD ${CMAKE_CURRENT_BINARY_DIR}/eigen/src/eigen)
set(eigen_INSTALL ${CMAKE_CURRENT_BINARY_DIR}/eigen/install)
......
......@@ -13,8 +13,8 @@ def tf_workspace(path_prefix = "", tf_repo_name = ""):
native.new_http_archive(
name = "eigen_archive",
url = "https://bitbucket.org/eigen/eigen/get/0823d98fdde7.tar.gz",
sha256 = "bbb8504e2fbb02937bb4efba4631f2344cd898a4fcee2cf4e7f22542de9aeb52",
url = "https://bitbucket.org/eigen/eigen/get/ffa59a39ffb9.tar.gz",
sha256 = "ccca47fe6528a97bcec26c733908114bbfb701c5d818d095820da94b158ef8e1",
build_file = path_prefix + "eigen.BUILD",
)
......
#include "eigen-eigen-0823d98fdde7/Eigen/Cholesky"
#include "eigen-eigen-ffa59a39ffb9/Eigen/Cholesky"
#include "eigen-eigen-0823d98fdde7/Eigen/Core"
#include "eigen-eigen-ffa59a39ffb9/Eigen/Core"
#include "eigen-eigen-0823d98fdde7/Eigen/Eigenvalues"
#include "eigen-eigen-ffa59a39ffb9/Eigen/Eigenvalues"
#include "eigen-eigen-0823d98fdde7/Eigen/LU"
#include "eigen-eigen-ffa59a39ffb9/Eigen/LU"
#include "eigen-eigen-0823d98fdde7/Eigen/QR"
#include "eigen-eigen-ffa59a39ffb9/Eigen/QR"
#include "eigen-eigen-0823d98fdde7/unsupported/Eigen/CXX11/Tensor"
#include "eigen-eigen-ffa59a39ffb9/unsupported/Eigen/CXX11/Tensor"
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册