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

cuda_configure.bzl: Add missing env vars deps for msvc configuration

Browse Source 
PiperOrigin-RevId: 357758933
Change-Id: Iae562808db85b81cba96a72ba9c50e806dedc885
This commit is contained in:
A. Unique TensorFlower 2021-02-16 11:01:56 -08:00 committed by TensorFlower Gardener
parent b382c6fe56
commit 22081f66ca
1 changed files with 16 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
third_party/gpus/cuda_configure.bzl vendored
View File

@ -1393,6 +1393,21 @@ def _cuda_autoconf_impl(repository_ctx):
else:
_create_local_cuda_repository(repository_ctx)
# For @bazel_tools//tools/cpp:windows_cc_configure.bzl
_MSVC_ENVVARS = [
"BAZEL_VC",
"BAZEL_VC_FULL_VERSION",
"BAZEL_VS",
"BAZEL_WINSDK_FULL_VERSION",
"VS90COMNTOOLS",
"VS100COMNTOOLS",
"VS110COMNTOOLS",
"VS120COMNTOOLS",
"VS140COMNTOOLS",
"VS150COMNTOOLS",
"VS160COMNTOOLS",
]
_ENVIRONS = [
_GCC_HOST_COMPILER_PATH,
_GCC_HOST_COMPILER_PREFIX,
@ -1410,7 +1425,7 @@ _ENVIRONS = [
"TMP",
"TMPDIR",
"TF_CUDA_PATHS",
]
] + _MSVC_ENVVARS
remote_cuda_configure = repository_rule(
implementation = _create_local_cuda_repository,