Skip to content

Commit cb49c07

Browse files
giordanowsmoses
andauthored
[ReactantExtra] Update XLA and adapt API.cpp (#995)
* [ReactantExtra] Update XLA and adapt `API.cpp` [skip tests] * [ReactantExtra] Change the function name, not the return type Co-authored-by: Billy Moses <[email protected]> * Test XLA PR which integrates newer LLVM [skip tests] * Use commits on main for Enzyme-JAX and XLA * Update WORKSPACE * Update deps/ReactantExtra/WORKSPACE --------- Co-authored-by: Billy Moses <[email protected]>
1 parent f823b86 commit cb49c07

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

deps/ReactantExtra/API.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -855,7 +855,7 @@ ClientCompile(PjRtClient *client, MlirModule cmod, int64_t device_id,
855855
}
856856
}
857857

858-
auto exec_err = client->Compile(cmod_op, options);
858+
auto exec_err = client->CompileAndLoad(cmod_op, options);
859859

860860
if (!exec_err.ok()) {
861861
std::string err_str;

deps/ReactantExtra/WORKSPACE

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ http_archive(
99
urls = ["https://github.com/wsmoses/nsync/archive/{commit}.tar.gz".format(commit = NSYNC_COMMIT)],
1010
)
1111

12-
ENZYMEXLA_COMMIT = "e28431f92da1a29cd993823a7a4b63fee678e2af"
12+
ENZYMEXLA_COMMIT = "2db74a9afd45c7d99d5515a370ffbaac707b114f"
1313
ENZYMEXLA_SHA256 = ""
1414

1515
http_archive(
@@ -130,6 +130,8 @@ http_archive(
130130
)
131131

132132
load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256")
133+
# XLA_COMMIT = "5e01d79fae2d7884ea9d47a3251f4cec8dcc742e"
134+
# XLA_SHA256 = ""
133135

134136
http_archive(
135137
name = "xla",

0 commit comments

Comments
 (0)