File tree Expand file tree Collapse file tree 2 files changed +7
-6
lines changed Expand file tree Collapse file tree 2 files changed +7
-6
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,9 +138,11 @@ 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" )
145
+
144
146
xla_workspace (NEW_XLA_PATCHES )
145
147
146
148
#
@@ -158,10 +160,9 @@ xla_workspace(NEW_XLA_PATCHES)
158
160
#
159
161
# pip_install_dependencies()
160
162
161
-
162
163
load ("@enzyme_ad//third_party/enzyme:workspace.bzl" , enzyme_workspace = "repo" )
163
- enzyme_workspace ()
164
164
165
+ enzyme_workspace ()
165
166
166
167
# http_archive(
167
168
# name = "upb",
You can’t perform that action at this time.
0 commit comments