experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #39849 from hlopko:fix_38205

Browse Source 
PiperOrigin-RevId: 313411456
Change-Id: I1d57346eced1e20d77b9d0c941f30a79d61746c3
This commit is contained in:
TensorFlower Gardener 2020-05-27 10:12:54 -07:00
commit db76607503
2 changed files with 13 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
third_party/gpus/crosstool/cc_toolchain_config.bzl.tpl vendored
View File

@ -293,7 +293,7 @@ def _cuda_set(cuda_path, actions):
return []
def _nologo():
return flag_group(flags = ["/nologo"])
return flag_group(flags = ["/nologo"])
def _features(cpu, compiler, ctx):
if cpu in ["local", "darwin"]:
@ -497,6 +497,11 @@ def _features(cpu, compiler, ctx):
flag_set(
actions = all_link_actions(),
flag_groups = [
flag_group(flags = (
["-Wl,-no-as-needed"] if cpu == "local" else []
) + [
"-B" + ctx.attr.linker_bin_path,
]),
flag_group(
flags = ["@%{linker_param_file}"],
expand_if_available = "linker_param_file",
@ -551,27 +556,17 @@ def _features(cpu, compiler, ctx):
"-Wl,-z,relro,-z,now",
])],
),
] if cpu == "local" else []) + [
flag_set(
actions = all_link_actions(),
flag_groups = [flag_group(flags = ["-Wl,-no-as-needed"])],
with_features = [with_feature_set(features = ["alwayslink"])],
),
] if cpu == "local" else []) + ([
flag_set(
actions = all_link_actions(),
flag_groups = [
flag_group(flags = ["-B" + ctx.attr.linker_bin_path]),
flag_group(flags = ["-Wl,--gc-sections"]),
flag_group(
flags = ["-Wl,--build-id=md5", "-Wl,--hash-style=gnu"],
),
],
),
] + ([flag_set(
actions = all_link_actions(),
flag_groups = [
flag_group(flags = ["-Wl,--gc-sections"]),
flag_group(
flags = ["-Wl,--build-id=md5", "-Wl,--hash-style=gnu"],
),
],
)] if cpu == "local" else []) + ([
] if cpu == "local" else []) + ([
flag_set(
actions = all_link_actions(),
flag_groups = [flag_group(flags = ["-undefined", "dynamic_lookup"])],
@ -588,7 +583,6 @@ def _features(cpu, compiler, ctx):
),
],
),
feature(name = "alwayslink", enabled = cpu == "local"),
feature(name = "opt"),
feature(name = "fastbuild"),
feature(name = "dbg"),

1
third_party/nccl/archive.BUILD vendored
View File

@ -88,6 +88,7 @@ cc_library(
],
hdrs = ["src/nccl.h"],
include_prefix = "third_party/nccl",
linkopts = ["-lrt"],
strip_include_prefix = "src",
visibility = ["//visibility:public"],
deps = [