1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
"""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,
)
|
