From 44b54f5988f4e7b996b31bbb771efdc63f74d3ba Mon Sep 17 00:00:00 2001 From: cong235 Date: Thu, 26 Dec 2019 13:29:35 +0700 Subject: [PATCH 1/3] Modifications to support building w/ CUDA 10.2, NCCL for 10.2 & Intel Chip Build string: bazel build --config=mkl --config=v2 --config=opt //tensorflow/tools/pip_package:build_pip_package --- third_party/gpus/find_cuda_config.py | 1 + 1 file changed, 1 insertion(+) diff --git a/third_party/gpus/find_cuda_config.py b/third_party/gpus/find_cuda_config.py index be59515661a3a3..546f311dd4bd33 100644 --- a/third_party/gpus/find_cuda_config.py +++ b/third_party/gpus/find_cuda_config.py @@ -188,6 +188,7 @@ def _library_paths(): "lib/*-linux-gnu", "lib/x64", "extras/CUPTI/*", + "targets/x86_64-linux/include", ] From 729e80ca40155bd5e549c72b46c8b31341195446 Mon Sep 17 00:00:00 2001 From: cong235 Date: Thu, 26 Dec 2019 13:30:23 +0700 Subject: [PATCH 2/3] Modifications to support building w/ CUDA 10.2, NCCL for 10.2 & Intel Chip Build string: bazel build --config=mkl --config=v2 --config=opt //tensorflow/tools/pip_package:build_pip_package --- third_party/mpi/mpi.bzl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/third_party/mpi/mpi.bzl b/third_party/mpi/mpi.bzl index 3a483351d1f982..626a9bbccdd5a7 100644 --- a/third_party/mpi/mpi.bzl +++ b/third_party/mpi/mpi.bzl @@ -2,7 +2,7 @@ #based on the configuration options return one or the other def mpi_hdr(): - MPI_LIB_IS_OPENMPI = True + MPI_LIB_IS_OPENMPI = False hdrs = [] if MPI_LIB_IS_OPENMPI: hdrs = ["mpi.h", "mpi_portable_platform.h"] #When using OpenMPI From e4dc23ddf15f03a19cf195911d54ffe51792a923 Mon Sep 17 00:00:00 2001 From: cong235 Date: Thu, 26 Dec 2019 13:31:40 +0700 Subject: [PATCH 3/3] Modifications to support building w/ CUDA 10.2, NCCL for 10.2 & Intel Chip Build string: bazel build --config=mkl --config=v2 --config=opt //tensorflow/tools/pip_package:build_pip_package --- third_party/nccl/build_defs.bzl.tpl | 1 - 1 file changed, 1 deletion(-) diff --git a/third_party/nccl/build_defs.bzl.tpl b/third_party/nccl/build_defs.bzl.tpl index 57191398553f88..5f5c3a1008a151 100644 --- a/third_party/nccl/build_defs.bzl.tpl +++ b/third_party/nccl/build_defs.bzl.tpl @@ -113,7 +113,6 @@ def _device_link_impl(ctx): "--cmdline=--compile-only", "--link", "--compress-all", - "--bin2c-path=%s" % bin2c.dirname, "--create=%s" % tmp_fatbin.path, "--embedded-fatbin=%s" % fatbin_h.path, ] + images,