Internal change.
PiperOrigin-RevId: 234672242
This commit is contained in:
parent
56d8b1f6c6
commit
3401519642
third_party
4
third_party/gpus/rocm_configure.bzl
vendored
4
third_party/gpus/rocm_configure.bzl
vendored
@ -250,8 +250,8 @@ def _hipcc_env(repository_ctx):
|
||||
"HIPCC_COMPILE_FLAGS_APPEND",
|
||||
]:
|
||||
if name in repository_ctx.os.environ:
|
||||
hipcc_env = hipcc_env + " " + name + "=\"" + \
|
||||
repository_ctx.os.environ[name].strip() + "\";"
|
||||
hipcc_env = (hipcc_env + " " + name + "=\"" +
|
||||
repository_ctx.os.environ[name].strip() + "\";")
|
||||
return hipcc_env.strip()
|
||||
|
||||
def _crosstool_verbose(repository_ctx):
|
||||
|
4
third_party/nccl/nccl_configure.bzl
vendored
4
third_party/nccl/nccl_configure.bzl
vendored
@ -118,8 +118,8 @@ def _check_nccl_version(repository_ctx, nccl_install_path, nccl_hdr_path, nccl_v
|
||||
|
||||
def _nccl_configure_impl(repository_ctx):
|
||||
"""Implementation of the nccl_configure repository rule."""
|
||||
if not enable_cuda(repository_ctx) or \
|
||||
get_cpu_value(repository_ctx) not in ("Linux", "FreeBSD"):
|
||||
if (not enable_cuda(repository_ctx) or
|
||||
get_cpu_value(repository_ctx) not in ("Linux", "FreeBSD")):
|
||||
# Add a dummy build file to make bazel query happy.
|
||||
repository_ctx.file("BUILD", _NCCL_DUMMY_BUILD_CONTENT)
|
||||
return
|
||||
|
Loading…
Reference in New Issue
Block a user