| load("@local_config_cuda//cuda:build_defs.bzl", "cuda_is_configured") |
| load("@local_config_rocm//rocm:build_defs.bzl", "rocm_is_configured") |
| |
| def stream_executor_friends(): |
| return ["//tensorflow/..."] |
| |
| def tf_additional_cuda_platform_deps(): |
| return [] |
| |
| def tf_additional_cuda_driver_deps(): |
| return [":cuda_stub"] |
| |
| def tf_additional_cudnn_plugin_deps(): |
| return [] |
| |
| # Returns whether any GPU backend is configuered. |
| def if_gpu_is_configured(x): |
| if cuda_is_configured() or rocm_is_configured(): |
| return x |
| return [] |
| |
| def if_cuda_or_rocm(x): |
| return if_gpu_is_configured(x) |