Merge pull request #17 from JuliaTrustworthyAI/16-serializationdeserialization-causes-issue-with-mpijl

attempting simple fix

Author: pat-alt

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

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"

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 = []

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 = []