4379cc4cf6
Change uses to if_cuda_is_configured(), which returns a select() statement. It currently returns two different select statements because select() + select() [can not](497ef110c3) have any common values in the two dicts. I will fix uses of this in a separate change.
PiperOrigin-RevId: 353049964
Change-Id: I222b7d49263ea9b06569208e00a180d6ff54d0a0
25 lines
680 B
Python
25 lines
680 B
Python
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda_is_configured")
|
|
load("@local_config_rocm//rocm:build_defs.bzl", "if_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_cupti_deps():
|
|
return ["//tensorflow/stream_executor/cuda:cupti_stub"]
|
|
|
|
def tf_additional_cudnn_plugin_deps():
|
|
return []
|
|
|
|
# Returns whether any GPU backend is configuered.
|
|
def if_gpu_is_configured(x):
|
|
return if_cuda_is_configured(x) + if_rocm_is_configured(x)
|
|
|
|
def if_cuda_or_rocm(x):
|
|
return if_gpu_is_configured(x)
|