Merge pull request #24942 from ROCmSoftwarePlatform:google-upstream-pr-bazel-nvcc-fix

PiperOrigin-RevId: 229481571
This commit is contained in:
TensorFlower Gardener 2019-01-15 18:55:40 -08:00
commit 7c6f3e1a09

View File

@ -1409,9 +1409,11 @@ def _create_local_cuda_repository(repository_ctx):
# bazel's header check failing.
cuda_defines["%{extra_no_canonical_prefixes_flags}"] = (
"flag: \"-fno-canonical-system-headers\"")
nvcc_path = "cuda/bin/nvcc"
if _is_windows(repository_ctx):
nvcc_path += ".exe"
nvcc_path = str(
repository_ctx.path("%s/bin/nvcc%s" % (
cuda_config.cuda_toolkit_path,
".exe" if _is_windows(repository_ctx) else "",
)))
_tpl(
repository_ctx,
"crosstool:BUILD",