diff options
Diffstat (limited to 'tensorflow/workspace.bzl')
-rw-r--r-- | tensorflow/workspace.bzl | 25 |
1 files changed, 14 insertions, 11 deletions
diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index 25a2b7e84f..71e4b8f5bb 100644 --- a/tensorflow/workspace.bzl +++ b/tensorflow/workspace.bzl @@ -2,6 +2,7 @@ load("//third_party/gpus:cuda_configure.bzl", "cuda_configure") load("//third_party/sycl:sycl_configure.bzl", "sycl_configure") +load("//third_party/mkl:build_defs.bzl", "mkl_repository") load("@io_bazel_rules_closure//closure/private:java_import_external.bzl", "java_import_external") load("@io_bazel_rules_closure//closure:defs.bzl", "filegroup_external") load("//third_party/py:python_configure.bzl", "python_configure") @@ -140,6 +141,19 @@ def tf_workspace(path_prefix="", tf_repo_name=""): cuda_configure(name="local_config_cuda") sycl_configure(name="local_config_sycl") python_configure(name="local_config_python") + + mkl_repository( + name = "mkl", + urls = [ + "http://mirror.bazel.build/github.com/01org/mkl-dnn/releases/download/v0.7/mklml_lnx_2018.0.20170425.tgz", + "https://github.com/01org/mkl-dnn/releases/download/v0.7/mklml_lnx_2018.0.20170425.tgz", + ], + sha256 = "3cc2501fb209e1fd0960a5f61c919438f9619c68a644dcebf0fdf69b07460c57", + strip_prefix = "mklml_lnx_2018.0.20170425", + build_file = str(Label("//third_party/mkl:mkl.BUILD")), + repository = tf_repo_name, + ) + if path_prefix: print("path_prefix was specified to tf_workspace but is no longer used " + "and will be removed in the future.") @@ -172,17 +186,6 @@ def tf_workspace(path_prefix="", tf_repo_name=""): ) native.new_http_archive( - name = "mkl", - urls = [ - "http://mirror.bazel.build/github.com/01org/mkl-dnn/releases/download/v0.7/mklml_lnx_2018.0.20170425.tgz", - "https://github.com/01org/mkl-dnn/releases/download/v0.7/mklml_lnx_2018.0.20170425.tgz", - ], - sha256 = "3cc2501fb209e1fd0960a5f61c919438f9619c68a644dcebf0fdf69b07460c57", - strip_prefix = "mklml_lnx_2018.0.20170425", - build_file = str(Label("//third_party/mkl:mkl.BUILD")), - ) - - native.new_http_archive( name = "ortools_archive", urls = [ "http://mirror.bazel.build/github.com/google/or-tools/archive/253f7955c6a1fd805408fba2e42ac6d45b312d15.tar.gz", |