-
Notifications
You must be signed in to change notification settings - Fork 5
/
workspace.bzl
35 lines (32 loc) · 1.66 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
33
34
35
JAX_COMMIT = "a820387a79041d795b72188710fbf52caefc2602"
JAX_SHA256 = "" # f8bbcc40cdee9d8d83a7f6e197ce111f1c01ee00341eab83ddd9367e48519665"
ENZYME_COMMIT = "4317a8c04381eb82f9db141731a9d2774684ab7d"
ENZYME_SHA256 = "ffa70210ae42246f72b35033fdee53c8e8334863585751ee09bac8d0890ac2c1"
PYRULES_COMMIT = "fe33a4582c37499f3caeb49a07a78fc7948a8949"
PYRULES_SHA256 = "cfa6957832ae0e0c7ee2ccf455a888a291e8419ed8faf45f4420dd7414d5dd96"
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' {} +"
]