File tree Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Expand file tree Collapse file tree 2 files changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -2699,6 +2699,7 @@ struct LinkableRuntime {
2699
2699
mlir::DialectRegistry registry;
2700
2700
xla::PjRtClient *client;
2701
2701
int device;
2702
+ bool shouldFreeClient;
2702
2703
DenseMap<const char *, std::map<std::vector<std::vector<int64_t >>,
2703
2704
xla::PjRtLoadedExecutable *>>
2704
2705
executables;
@@ -2709,8 +2710,8 @@ struct LinkableRuntime {
2709
2710
LinkableRuntime (const std::string &backend) : registry() {
2710
2711
InitializeRegistry (wrap (®istry));
2711
2712
InitializePasses (wrap (®istry));
2712
-
2713
2713
InitializeLogs ();
2714
+ shouldFreeClient = true ;
2714
2715
const char *error = NULL ;
2715
2716
auto mpi = getenv (" OMPI_COMM_WORLD_RANK" );
2716
2717
device = 0 ;
@@ -2743,6 +2744,8 @@ struct LinkableRuntime {
2743
2744
if (!client)
2744
2745
llvm::errs () << " error: " << refstr << " \n " ;
2745
2746
assert (client);
2747
+ // Weird stream issue in freeing cuda client.
2748
+ shouldFreeClient = false ;
2746
2749
} else {
2747
2750
client = MakeCPUClient (1 , 0 );
2748
2751
assert (client);
Original file line number Diff line number Diff line change @@ -11,7 +11,7 @@ http_archive(
11
11
urls = ["https://github.com/wsmoses/nsync/archive/{commit}.tar.gz" .format (commit = NSYNC_COMMIT )],
12
12
)
13
13
14
- ENZYMEXLA_COMMIT = "1a880a1977cb01821177c60c934563ffaa0a0e6d "
14
+ ENZYMEXLA_COMMIT = "20aff0cfd430339ca01c9febb96675d62a4a7995 "
15
15
16
16
ENZYMEXLA_SHA256 = ""
17
17
You can’t perform that action at this time.
0 commit comments