set TF_CUDA_CLANG action_env. Args: environ_cp: copy of the os.environ.
(environ_cp)
| 535 | |
| 536 | |
| 537 | def set_tf_cuda_clang(environ_cp): |
| 538 | """set TF_CUDA_CLANG action_env. |
| 539 | |
| 540 | Args: |
| 541 | environ_cp: copy of the os.environ. |
| 542 | """ |
| 543 | question = 'Do you want to use clang as CUDA compiler?' |
| 544 | yes_reply = 'Clang will be used as CUDA compiler.' |
| 545 | no_reply = 'nvcc will be used as CUDA compiler.' |
| 546 | set_action_env_var( |
| 547 | environ_cp, |
| 548 | 'TF_CUDA_CLANG', |
| 549 | None, |
| 550 | False, |
| 551 | question=question, |
| 552 | yes_reply=yes_reply, |
| 553 | no_reply=no_reply, |
| 554 | bazel_config_name='cuda_clang') |
| 555 | |
| 556 | |
| 557 | def set_tf_download_clang(environ_cp): |
no test coverage detected