mirror of
https://github.com/tensorflow/tensorflow.git
synced 2024-11-21 21:05:19 +00:00
24 lines
561 B
Python
24 lines
561 B
Python
"""TensorFlow workspace initialization. Consult the WORKSPACE on how to use it."""
|
|
|
|
load("@llvm-raw//utils/bazel:configure.bzl", "llvm_configure")
|
|
|
|
# The subset of LLVM targets that TensorFlow cares about.
|
|
_LLVM_TARGETS = [
|
|
"AArch64",
|
|
"AMDGPU",
|
|
"ARM",
|
|
"NVPTX",
|
|
"PowerPC",
|
|
"RISCV",
|
|
"SystemZ",
|
|
"X86",
|
|
]
|
|
|
|
def llvm_setup(name):
|
|
# Build @llvm-project from @llvm-raw using overlays.
|
|
llvm_configure(
|
|
name = name,
|
|
repo_mapping = {"@python_runtime": "@local_config_python"},
|
|
targets = _LLVM_TARGETS,
|
|
)
|