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", ] 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 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,