| 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 [] |