File tree Expand file tree Collapse file tree 2 files changed +11
-7
lines changed Expand file tree Collapse file tree 2 files changed +11
-7
lines changed Original file line number Diff line number Diff line change @@ -758,7 +758,7 @@ cc_toolchain_config(
758
758
"-D__TIME__=\" redacted\" " ,
759
759
"-Wno-unused-command-line-argument" ,
760
760
"-Wno-gnu-offsetof-extensions" ,
761
- "-mxsave" ,
761
+ "-mxsave" ,
762
762
],
763
763
)
764
764
@@ -1193,7 +1193,7 @@ cc_library(
1193
1193
],
1194
1194
"@xla//xla/tsl:windows" : [
1195
1195
"@xla//xla/tsl/platform/windows:platform_port" ,
1196
- ],
1196
+ ],
1197
1197
"//conditions:default" : [
1198
1198
"@gloo//:transport_tcp" ,
1199
1199
"@xla//xla/backends/cpu/collectives:gloo_collectives" ,
Original file line number Diff line number Diff line change @@ -40,10 +40,10 @@ NEW_XLA_PATCHES = CUPTI_NEW + [
40
40
"""
41
41
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
42
42
""" ,
43
- """
43
+ """
44
44
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
45
45
""" ,
46
- """
46
+ """
47
47
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
48
48
""" ,
49
49
"""
@@ -138,10 +138,15 @@ LLVM_TARGETS = [
138
138
# )
139
139
140
140
load ("@enzyme_ad//third_party/jax:workspace.bzl" , jax_workspace = "repo" )
141
+
141
142
jax_workspace ()
142
143
143
144
load ("@enzyme_ad//third_party/xla:workspace.bzl" , xla_workspace = "repo" )
144
- xla_workspace (NEW_XLA_PATCHES , override_commit = "3db2ac3bf43c3e250ef7614eb5c8594e716ecc48" )
145
+
146
+ xla_workspace (
147
+ NEW_XLA_PATCHES ,
148
+ override_commit = "3db2ac3bf43c3e250ef7614eb5c8594e716ecc48" ,
149
+ )
145
150
146
151
#
147
152
# load("@xla//third_party/py:python_init_pip.bzl", "python_init_pip")
@@ -158,10 +163,9 @@ xla_workspace(NEW_XLA_PATCHES, override_commit = "3db2ac3bf43c3e250ef7614eb5c859
158
163
#
159
164
# pip_install_dependencies()
160
165
161
-
162
166
load ("@enzyme_ad//third_party/enzyme:workspace.bzl" , enzyme_workspace = "repo" )
163
- enzyme_workspace ()
164
167
168
+ enzyme_workspace ()
165
169
166
170
# http_archive(
167
171
# name = "upb",
You can’t perform that action at this time.
0 commit comments