Skip to content

Commit b484d1f

Browse files
committed
Fix rebase
1 parent edd935c commit b484d1f

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

deps/ReactantExtra/API.cpp

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2699,6 +2699,7 @@ struct LinkableRuntime {
26992699
mlir::DialectRegistry registry;
27002700
xla::PjRtClient *client;
27012701
int device;
2702+
bool shouldFreeClient;
27022703
DenseMap<const char *, std::map<std::vector<std::vector<int64_t>>,
27032704
xla::PjRtLoadedExecutable *>>
27042705
executables;
@@ -2709,8 +2710,8 @@ struct LinkableRuntime {
27092710
LinkableRuntime(const std::string &backend) : registry() {
27102711
InitializeRegistry(wrap(&registry));
27112712
InitializePasses(wrap(&registry));
2712-
27132713
InitializeLogs();
2714+
shouldFreeClient = true;
27142715
const char *error = NULL;
27152716
auto mpi = getenv("OMPI_COMM_WORLD_RANK");
27162717
device = 0;
@@ -2743,6 +2744,8 @@ struct LinkableRuntime {
27432744
if (!client)
27442745
llvm::errs() << " error: " << refstr << "\n";
27452746
assert(client);
2747+
// Weird stream issue in freeing cuda client.
2748+
shouldFreeClient = false;
27462749
} else {
27472750
client = MakeCPUClient(1, 0);
27482751
assert(client);

deps/ReactantExtra/WORKSPACE

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

14-
ENZYMEXLA_COMMIT = "1a880a1977cb01821177c60c934563ffaa0a0e6d"
14+
ENZYMEXLA_COMMIT = "20aff0cfd430339ca01c9febb96675d62a4a7995"
1515

1616
ENZYMEXLA_SHA256 = ""
1717

0 commit comments

Comments
 (0)