Set other CUDA related variables.
(environ_cp)
| 1045 | |
| 1046 | |
| 1047 | def set_other_cuda_vars(environ_cp): |
| 1048 | """Set other CUDA related variables.""" |
| 1049 | # If CUDA is enabled, always use GPU during build and test. |
| 1050 | if environ_cp.get('TF_CUDA_CLANG') == '1': |
| 1051 | write_to_bazelrc('build --config=cuda_clang') |
| 1052 | else: |
| 1053 | write_to_bazelrc('build --config=cuda') |
| 1054 | |
| 1055 | |
| 1056 | def set_host_cxx_compiler(environ_cp): |
no test coverage detected