-
Notifications
You must be signed in to change notification settings - Fork 6
/
workspace.bzl
32 lines (30 loc) · 1.39 KB
/
workspace.bzl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
JAX_COMMIT = "e3618681326bdf68154de0e573b8b36624512d7d"
JAX_SHA256 = ""
ENZYME_COMMIT = "7425c37fbf4f676b8205b2dd2c0fd097d32c3f38"
ENZYME_SHA256 = ""
XLA_PATCHES = [
"""
sed -i.bak0 "s/\\/\\/third_party:repo.bzl/@bazel_tools\\/\\/tools\\/build_defs\\/repo:http.bzl/g" third_party/llvm/workspace.bzl
""",
"""
sed -i.bak0 "s/patch_file/patch_args = [\\\"-p1\\\"],patches/g" third_party/llvm/workspace.bzl
""",
"""
sed -i.bak0 "/link_file/d" third_party/llvm/workspace.bzl
""",
"""
sed -i.bak0 "s/build_file.*/build_file_content = \\\"# empty\\\",/g" third_party/llvm/workspace.bzl
""",
"""
sed -i.bak0 "s/\\/\\/third_party/@xla\\/\\/third_party/g" third_party/llvm/workspace.bzl
""",
"""
sed -i.bak0 "s/tf_http_archive/http_archive/g" third_party/llvm/workspace.bzl
""",
"""
sed -i.bak0 "s/strip_prefix/patch_cmds = [\\\"find . -type f -name config.bzl -exec sed -i.bak0 's\\/HAVE_BACKTRACE=1\\/NO_HAVE_BACKTRACE=0\\/g' {} +\\\"], strip_prefix/g" third_party/llvm/workspace.bzl
""",
"find . -type f -name BUILD -exec sed -i.bak1 's/\\/\\/third_party\\/py\\/enzyme_ad\\/\\.\\.\\./public/g' {} +",
"find . -type f -name BUILD -exec sed -i.bak2 's/\\/\\/xla\\/mlir\\/memref:friends/\\/\\/visibility:public/g' {} +",
"find xla/mlir -type f -name BUILD -exec sed -i.bak3 's/\\/\\/xla:internal/\\/\\/\\/\\/visibility:public/g' {} +"
]