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
50 lines
1.7 KiB
Smarty
50 lines
1.7 KiB
Smarty
# Macros for building ROCm code.
|
|
def if_rocm(if_true, if_false = []):
|
|
"""Shorthand for select()'ing on whether we're building with ROCm.
|
|
|
|
Returns a select statement which evaluates to if_true if we're building
|
|
with ROCm enabled. Otherwise, the select statement evaluates to if_false.
|
|
|
|
"""
|
|
return select({
|
|
"@local_config_rocm//rocm:using_hipcc": if_true,
|
|
"//conditions:default": if_false
|
|
})
|
|
|
|
|
|
def rocm_default_copts():
|
|
"""Default options for all ROCm compilations."""
|
|
return if_rocm(["-x", "rocm"] + %{rocm_extra_copts})
|
|
|
|
def rocm_copts(opts = []):
|
|
"""Gets the appropriate set of copts for (maybe) ROCm compilation.
|
|
|
|
If we're doing ROCm compilation, returns copts for our particular ROCm
|
|
compiler. If we're not doing ROCm compilation, returns an empty list.
|
|
|
|
"""
|
|
return rocm_default_copts() + select({
|
|
"//conditions:default": [],
|
|
"@local_config_rocm//rocm:using_hipcc": ([
|
|
"",
|
|
]),
|
|
}) + if_rocm_is_configured(opts)
|
|
|
|
def rocm_gpu_architectures():
|
|
"""Returns a list of supported GPU architectures."""
|
|
return %{rocm_gpu_architectures}
|
|
|
|
def if_rocm_is_configured(x):
|
|
"""Tests if the ROCm was enabled during the configure process.
|
|
|
|
Unlike if_rocm(), this does not require that we are building with
|
|
--config=rocm. Used to allow non-ROCm code to depend on ROCm libraries.
|
|
"""
|
|
if %{rocm_is_configured}:
|
|
return select({"//conditions:default": x})
|
|
return select({"//conditions:default": []})
|
|
|
|
def rocm_library(copts = [], **kwargs):
|
|
"""Wrapper over cc_library which adds default ROCm options."""
|
|
native.cc_library(copts = rocm_default_copts() + copts, **kwargs)
|