kse-01/tensorflow/third_party/gpus/rocm/build_defs.bzl.tpl
github-classroom[bot] 1122cdd8b0
Initial commit
2023-10-09 11:37:31 +00:00

45 lines
1.4 KiB (Stored with Git LFS)
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_is_configured():
"""Returns true if ROCm was enabled during the configure process."""
return %{rocm_is_configured}
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 x
return []