Skip to content

Commit 71160f6

Browse files
wsmoseswsmoses
authored
Refactor CUPTI patches in WORKSPACE file
1 parent 0176cd7 commit 71160f6

File tree

1 file changed

+1
-115
lines changed

1 file changed

+1
-115
lines changed

‎deps/ReactantExtra/WORKSPACE‎

Lines changed: 1 addition & 115 deletions
Original file line numberDiff line numberDiff line change
@@ -27,121 +27,7 @@ sed -i.bak0 "s/\\\\\\\\\\\\\\\\\\/\\\\\\\\\\\\\\\\\\/:patches/@enzyme_ad\\\\\\\\
2727
urls = ["https://github.com/EnzymeAD/Enzyme-JAX/archive/{commit}.tar.gz".format(commit = ENZYMEXLA_COMMIT)],
2828
)
2929

30-
CUPTI_OLD = [
31-
"""
32-
sed -i.bak0 "s/cupti_driver_cbid/cupti/g" xla/backends/profiler/gpu/cupti_tracer.cc
33-
""",
34-
"""
35-
sed -i.bak0 "/CUPTI_DRIVER_TRACE_CBID_cuGraphAddNode/d" xla/backends/profiler/gpu/cupti_tracer.cc
36-
""",
37-
"""
38-
sed -i.bak0 "/CUPTI_DRIVER_TRACE_CBID_cuGraphAddNode_v2/d" xla/backends/profiler/gpu/cupti_tracer.cc
39-
""",
40-
]
41-
42-
CUPTI_NEW = []
43-
44-
NEW_XLA_PATCHES = CUPTI_NEW + [
45-
"""
46-
echo "--- a/src/google/protobuf/stubs/port.h" >> third_party/proto.patch
47-
echo "+++ b/src/google/protobuf/stubs/port.h" >> third_party/proto.patch
48-
echo "@@ -27,7 +27,7 @@" >> third_party/proto.patch
49-
echo " #include <intrin.h>" >> third_party/proto.patch
50-
echo " #elif defined(__APPLE__)" >> third_party/proto.patch
51-
echo " #include <libkern/OSByteOrder.h>" >> third_party/proto.patch
52-
echo "-#elif defined(__linux__) || defined(__ANDROID__) || defined(__CYGWIN__)" >> third_party/proto.patch
53-
echo "+#elif !defined(__NVCC__) && (defined(__linux__) || defined(__ANDROID__) || defined(__CYGWIN__))" >> third_party/proto.patch
54-
echo " #include <byteswap.h> // IWYU pragma: export" >> third_party/proto.patch
55-
echo " #endif" >> third_party/proto.patch
56-
echo "" >> third_party/proto.patch
57-
echo "@@ -143,7 +143,7 @@" >> third_party/proto.patch
58-
echo " #define bswap_32(x) OSSwapInt32(x)" >> third_party/proto.patch
59-
echo " #define bswap_64(x) OSSwapInt64(x)" >> third_party/proto.patch
60-
echo "" >> third_party/proto.patch
61-
echo "-#elif !defined(__linux__) && !defined(__ANDROID__) && !defined(__CYGWIN__)" >> third_party/proto.patch
62-
echo "+#elif defined(__NVCC__) || (!defined(__linux__) && !defined(__ANDROID__) && !defined(__CYGWIN__))" >> third_party/proto.patch
63-
echo "" >> third_party/proto.patch
64-
echo " #ifndef bswap_16" >> third_party/proto.patch
65-
echo " static inline uint16_t bswap_16(uint16_t x) {" >> third_party/proto.patch
66-
sed -i.bak0 "s/protobuf.patch\\"/protobuf.patch\\", \\":proto.patch\\"/g" workspace2.bzl
67-
sed -i.bak0 "s/patch_file = \\[\\"\\/\\/third_party\\/protobuf:protobuf.patch\\"/patches = \\[Label(\\"\\/\\/third_party\\/protobuf:protobuf.patch\\"), Label(\\"\\/\\/third_party:proto.patc
68-
\\"\\)], patch_args = \\[\\"-p1\\"/g" third_party/py/python_init_rules.bzl
69-
sed -i.bak0 "s/tf_http_archive(/http_archive(/g" third_party/py/python_init_rules.bzl
70-
71-
""",
72-
"""
73-
sed -i.bak0 "s/def main():/def main():\\n if TMPDIR: os.environ['TMPDIR'] = TMPDIR/g" third_party/gpus/crosstool/clang/bin/crosstool_wrapper_driver_is_not_gcc.tpl
74-
""",
75-
"""
76-
sed -i.bak0 "s/__chkstk/__chkstk_ms/g" xla/service/cpu/runtime_symbol_generator.cc
77-
""",
78-
"""
79-
sed -i.bak0 "1s/^/#include \\"llvm\\/Support\\/DynamicLibrary.h\\"\\n/g" xla/service/cpu/runtime_symbol_generator.cc
80-
""",
81-
"""
82-
sed -i.bak0 "s/(__chkstk_ms)/(llvm::sys::DynamicLibrary::SearchForAddressOfSymbol(\\"__chkstk_ms\\"))/g" xla/service/cpu/runtime_symbol_generator.cc
83-
""",
84-
"""
85-
sed -i.bak0 "s/Shlwapi/shlwapi/g" xla/tsl/platform/windows/load_library.cc xla/tsl/platform/windows/windows_file_system.cc xla/tsl/platform/windows/env.cc
86-
""",
87-
"""
88-
sed -i.bak0 "1s/^/#ifdef PLATFORM_WINDOWS\\n#include <immintrin.h>\\n#include <intrin.h>\\n#endif/g" third_party/tsl/tsl/platform/cpu_info.cc
89-
""",
90-
"""
91-
sed -i.bak0 "1s/^/#define _USE_MATH_DEFINES\\n/g" xla/fp_util.h xla/hlo/builder/lib/prng.cc xla/literal_comparison.cc xla/hlo/builder/lib/math.cc xla/service/spmd/fft_handler.cc xla/service/cpu/onednn_contraction_rewriter.cc xla/hlo/evaluator/hlo_evaluator.cc
92-
""",
93-
"""
94-
sed -i.bak0 "s/Windows\\.h/windows\\.h/g" xla/tsl/platform/windows/port.cc xla/tsl/platform/windows/wide_char.cc xla/tsl/platform/windows/env.cc xla/tsl/platform/windows/windows_file_system.cc
95-
""",
96-
"""
97-
sed -i.bak0 "/D_FORTIFY_SOURCE/d" third_party/gpus/crosstool/cc_toolchain_config.bzl.tpl tools/toolchains/cross_compile/cc/BUILD tools/toolchains/clang6/CROSSTOOL.tpl
98-
""",
99-
"""
100-
sed -i.bak0 "s/i64/LL/g" xla/tsl/platform/windows/env_time.cc
101-
""",
102-
"""
103-
sed -i.bak0 "s/\\/D/-D/g" third_party/farmhash/farmhash.BUILD
104-
""",
105-
"""
106-
sed -i.bak0 "s/Node::Leaf(std::forward<decltype(pair.second)>/Node::Leaf(std::forward<T>/g" xla/tuple_tree.h
107-
""",
108-
"""
109-
sed -i.bak0 "s/kDeprecatedFlags({/kDeprecatedFlags(absl::flat_hash_set<std::string>{/g" xla/debug_options_flags.cc
110-
""",
111-
"""
112-
sed -i.bak0 "s/kStableFlags({/kStableFlags(absl::flat_hash_set<std::string>{/g" xla/debug_options_flags.cc
113-
""",
114-
"""
115-
sed -i.bak0 "s/cupti_driver_cbid/cupti/g" xla/backends/profiler/gpu/cupti_tracer.cc
116-
""",
117-
"""
118-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.bzl -exec sed -i.bak0 's\\/HAVE_LINK_H=1\\/HAVE_LINK_H=0\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
119-
""",
120-
"""
121-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.bzl -exec sed -i.bak0 's\\/LLVM_ENABLE_THREADS=1\\/LLVM_ENABLE_THREADS=0\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
122-
""",
123-
"""
124-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.bzl -exec sed -i.bak0 's\\/HAVE_MALLINFO=1\\/DONT_HAVE_ANY_MALLINFO=0\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
125-
""",
126-
"""
127-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.bzl -exec sed -i.bak0 's\\/HAVE_PTHREAD_GETNAME_NP=1\\/FAKE_HAVE_PTHREAD_GETNAME_NP=0\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
128-
""",
129-
"""
130-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.bzl -exec sed -i.bak0 's\\/HAVE_PTHREAD_SETNAME_NP=1\\/FAKE_HAVE_PTHREAD_SETNAME_NP=0\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
131-
""",
132-
"""
133-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.h -exec sed -i.bak0 's\\/ENABLE_CRASH_OVERRIDES 1\\/ENABLE_CRASH_OVERRIDES 0\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
134-
""",
135-
"""
136-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.h -exec sed -i.bak0 's\\/HAVE_PTHREAD_GETNAME_NP\\/FAKE_HAVE_PTHREAD_GETNAME_NP\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
137-
""",
138-
"""
139-
sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\[\\\"find . -type f -name config.h -exec sed -i.bak0 's\\/HAVE_PTHREAD_SETNAME_NP\\/FAKE_HAVE_PTHREAD_SETNAME_NP\\/g' {} +\\\",/g" third_party/llvm/workspace.bzl
140-
""",
141-
# """
142-
# sed -i.bak0 "s/patch_cmds = \\[/patch_cmds = \\['find . -type f -name BUILD.bazel -exec sed -i.bak0 \\\\\\'s\\/\\\"CAPIIR\\\",\\/\\\"CAPIIR\\\",alwayslink=1,\\/g\\\\\\\\' {} +',/g" third_party/llvm/workspace.bzl
143-
# """,
144-
]
30+
NEW_XLA_PATCHES = []
14531

14632
LLVM_TARGETS = [
14733
"AMDGPU",

0 commit comments

Comments
 (0)