diff options
Diffstat (limited to 'tensorflow/workspace.bzl')
-rw-r--r-- | tensorflow/workspace.bzl | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index 431676c52d..2f24e2f019 100644 --- a/tensorflow/workspace.bzl +++ b/tensorflow/workspace.bzl @@ -99,7 +99,8 @@ def _execute_and_check_ret_code(repo_ctx, cmd_and_args): # Apply a patch_file to the repository root directory # Runs 'patch -p1' def _apply_patch(repo_ctx, patch_file): - if not repo_ctx.which("patch"): + # Don't check patch on Windows, because patch is only available under bash. + if not _is_windows(repo_ctx) and not repo_ctx.which("patch"): fail("patch command is not found, please install it") cmd = [ @@ -628,11 +629,11 @@ def tf_workspace(path_prefix="", tf_repo_name=""): temp_workaround_http_archive( name = "nccl_archive", urls = [ - "http://mirror.bazel.build/github.com/nvidia/nccl/archive/ccfc4567dc3e2a37fb42cfbc64d10eb526e7da7b.tar.gz", - "https://github.com/nvidia/nccl/archive/ccfc4567dc3e2a37fb42cfbc64d10eb526e7da7b.tar.gz", + "http://mirror.bazel.build/github.com/nvidia/nccl/archive/29a1a916dc14bb2c00feed3d4820d51fa85be1e6.tar.gz", + "https://github.com/nvidia/nccl/archive/29a1a916dc14bb2c00feed3d4820d51fa85be1e6.tar.gz", ], - sha256 = "6c34a0862d9f8ed4ad5984c6a8206b351957bb14cf6ad7822720f285f4aada04", - strip_prefix = "nccl-ccfc4567dc3e2a37fb42cfbc64d10eb526e7da7b", + sha256 = "6387030e37d14762f87eefbc86ee527293ec04745c66ccd820cf7fc0fdc23f92", + strip_prefix = "nccl-29a1a916dc14bb2c00feed3d4820d51fa85be1e6", build_file = str(Label("//third_party:nccl.BUILD")), repository = tf_repo_name, ) |