diff options
author | Gunhan Gulsoy <gunan@google.com> | 2017-06-06 22:34:30 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-06-06 22:37:57 -0700 |
commit | 3cf88d390c73d99a45ffc19d3e0c338e3cf46715 (patch) | |
tree | 448cd3f2e615bc2aedf721d04ac9f2b7d30c49df /configure | |
parent | 8ad6a036e544f81e74065a8ff928ff925b599fb3 (diff) |
When GPU is configured, do not require --config=cuda.
Also fix indentation in configure.
PiperOrigin-RevId: 158232959
Diffstat (limited to 'configure')
-rwxr-xr-x | configure | 6 |
1 files changed, 6 insertions, 0 deletions
@@ -670,6 +670,12 @@ if is_windows; then write_action_env_to_bazelrc "CUDA_PATH" "$CUDA_PATH" write_action_env_to_bazelrc "CUDA_COMPUTE_CAPABILITIES" "$CUDA_COMPUTE_CAPABILITIES" write_action_env_to_bazelrc "NO_WHOLE_ARCHIVE_OPTION" "1" + write_to_bazelrc "build --config=win-cuda" + write_to_bazelrc "test --config=win-cuda" +else + # If CUDA is enabled, always use GPU during build and test. + write_to_bazelrc "build --config=cuda" + write_to_bazelrc "test --config=cuda" fi # end of if "$TF_NEED_CUDA" == "1" |