[MINOR] Federated Parameter server multithreaded
This commit change the parameter server to run in unlimited number of
threads for each of the federated workers instructions.
diff --git a/src/main/java/org/apache/sysds/runtime/instructions/cp/ParamservBuiltinCPInstruction.java b/src/main/java/org/apache/sysds/runtime/instructions/cp/ParamservBuiltinCPInstruction.java
index 5e8ad32..7a285f6 100644
--- a/src/main/java/org/apache/sysds/runtime/instructions/cp/ParamservBuiltinCPInstruction.java
+++ b/src/main/java/org/apache/sysds/runtime/instructions/cp/ParamservBuiltinCPInstruction.java
@@ -96,6 +96,7 @@
@Override
public void processInstruction(ExecutionContext ec) {
// check if the input is federated
+
if(ec.getMatrixObject(getParam(PS_FEATURES)).isFederated() ||
ec.getMatrixObject(getParam(PS_LABELS)).isFederated()) {
runFederated(ec);
@@ -142,7 +143,7 @@
LocalVariableMap newVarsMap = createVarsMap(ec);
// Level of par is 1 because one worker will be launched per task
// TODO: Fix recompilation
- ExecutionContext newEC = ParamservUtils.createExecutionContext(ec, newVarsMap, updFunc, aggFunc, 1, true);
+ ExecutionContext newEC = ParamservUtils.createExecutionContext(ec, newVarsMap, updFunc, aggFunc, -1, true);
// Create workers' execution context
List<ExecutionContext> federatedWorkerECs = ParamservUtils.copyExecutionContext(newEC, workerNum);
// Create the agg service's execution context