[systemds] branch main updated: [SYSTEMDS-3396] LocalVarMap Concurrency in Federated Execution

[baunsgaard]

This is an automated email from the ASF dual-hosted git repository.

baunsgaard pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/systemds.git


The following commit(s) were added to refs/heads/main by this push:
     new a9943772cf [SYSTEMDS-3396] LocalVarMap Concurrency in Federated 
Execution
a9943772cf is described below

commit a9943772cf28e82604dac88d340cae3e1e779569
Author: ywcb00 <ywc...@ywcb.org>
AuthorDate: Tue Jun 28 15:21:41 2022 +0200

    [SYSTEMDS-3396] LocalVarMap Concurrency in Federated Execution
    
    Changes the local variable map back to a ConcurrentHashMap to allow
    simultaneous modification and iteration of the map
    
    Closes #1647
---
 .../org/apache/sysds/runtime/controlprogram/LocalVariableMap.java  | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git 
a/src/main/java/org/apache/sysds/runtime/controlprogram/LocalVariableMap.java 
b/src/main/java/org/apache/sysds/runtime/controlprogram/LocalVariableMap.java
index bac6759ca5..5031f3f785 100644
--- 
a/src/main/java/org/apache/sysds/runtime/controlprogram/LocalVariableMap.java
+++ 
b/src/main/java/org/apache/sysds/runtime/controlprogram/LocalVariableMap.java
@@ -20,6 +20,7 @@
 package org.apache.sysds.runtime.controlprogram;
 
 import java.util.HashMap;
+import java.util.concurrent.ConcurrentHashMap;
 import java.util.HashSet;
 import java.util.Map;
 import java.util.Map.Entry;
@@ -44,19 +45,19 @@ public class LocalVariableMap implements Cloneable
        private static final IDSequence _seq = new IDSequence();
        
        //variable map data and id
-       private final HashMap<String, Data> localMap;
+       private final ConcurrentHashMap<String, Data> localMap;
        private final long localID;
        
        //optional set of registered outputs
        private HashSet<String> outputs = null;
        
        public LocalVariableMap() {
-               localMap = new HashMap<>();
+               localMap = new ConcurrentHashMap<>();
                localID = _seq.getNextID();
        }
        
        public LocalVariableMap(LocalVariableMap vars) {
-               localMap = new HashMap<>(vars.localMap);
+               localMap = new ConcurrentHashMap<>(vars.localMap);
                localID = _seq.getNextID();
        }