attempting simple fix

JuliaTrustworthyAI · Nov 18, 2024 · 376ec2f · 376ec2f
1 parent 156d666
commit 376ec2f
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 1 deletion.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -4,6 +4,12 @@ All notable changes to this project will be documented in this file.
 
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/), and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## Version [1.1.2] - 2024-11-18
+
+### Changed
+
+- Updated MPI functionality to avoid issue related to deserialization. [#16]
+
 ## Version [1.1.1] - 2024-11-13
 
 ### Changed

diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "TaijaParallel"
 uuid = "bf1c2c22-5e42-4e78-8b6b-92e6c673eeb0"
 authors = ["Patrick Altmeyer <[email protected]>"]
-version = "1.1.1"
+version = "1.1.2"
 
 [deps]
 CounterfactualExplanations = "2f13d31b-18db-44c1-bc43-ebaf2cff0be0"

diff --git a/ext/MPIExt/evaluate.jl b/ext/MPIExt/evaluate.jl
@@ -84,6 +84,9 @@ function TaijaBase.parallelize(
         MPI.Barrier(parallelizer.comm)
     end
 
+    # Collect all chunks in rank 0:
+    MPI.Barrier(parallelizer.comm)
+
     # Load output from rank 0:
     if parallelizer.rank == 0
         outputs = []

diff --git a/ext/MPIExt/generate_counterfactual.jl b/ext/MPIExt/generate_counterfactual.jl
@@ -84,6 +84,9 @@ function TaijaBase.parallelize(
         MPI.Barrier(parallelizer.comm)
     end
 
+    # Collect all chunks in rank 0:
+    MPI.Barrier(parallelizer.comm)
+
     # Load output from rank 0:
     if parallelizer.rank == 0
         outputs = []