File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -1037,7 +1037,7 @@ REACTANT_ABI MlirModule ConvertLLVMStrToMLIR(const char *lmod,
10371037 return wrap (res);
10381038}
10391039
1040- typedef PjRtFuture <> FutureType;
1040+ typedef xla::Future <> FutureType;
10411041REACTANT_ABI void FreeFuture (FutureType *Future) { delete Future; }
10421042
10431043REACTANT_ABI uint8_t FutureIsReady (FutureType *Future) {
@@ -1236,7 +1236,7 @@ REACTANT_ABI void XLAExecuteSharded(xla::PjRtLoadedExecutable *exec,
12361236 options.untuple_result = true ;
12371237
12381238 // Optional future to hold asynchronous execution results.
1239- std::optional<PjRtFuture <>> returned_future;
1239+ std::optional<xla::Future <>> returned_future;
12401240
12411241 auto results = MyValueOrThrow (exec->ExecuteSharded (argument_handles, device,
12421242 options, returned_future,
Original file line number Diff line number Diff line change @@ -4,7 +4,7 @@ NSYNC_COMMIT = "82b118aa7ace3132e517e2c467f8732978cf4023"
44
55NSYNC_SHA256 = ""
66
7- ENZYMEXLA_COMMIT = "be41edd949d05733c12b59b582b14f6c9ef462c1 "
7+ ENZYMEXLA_COMMIT = "6ce8eecac2746d2c7db5820ec5967c8c120834b4 "
88
99ENZYMEXLA_SHA256 = ""
1010
You can’t perform that action at this time.
0 commit comments