Skip to content

Commit 818ab17

Browse files
committed
increment on mmchain
1 parent 884113c commit 818ab17

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/main/java/org/apache/sysds/runtime/controlprogram/federated/FederatedWorkloadAnalyzer.java

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,8 @@
3030
import org.apache.sysds.runtime.instructions.Instruction;
3131
import org.apache.sysds.runtime.instructions.cp.AggregateBinaryCPInstruction;
3232
import org.apache.sysds.runtime.instructions.cp.ComputationCPInstruction;
33+
import org.apache.sysds.runtime.instructions.cp.MMChainCPInstruction;
34+
import org.apache.sysds.runtime.matrix.data.MatrixBlock;
3335

3436
public class FederatedWorkloadAnalyzer {
3537
protected static final Log LOG = LogFactory.getLog(FederatedWorkloadAnalyzer.class.getName());
@@ -89,6 +91,11 @@ public void incrementWorkload(ExecutionContext ec, ConcurrentHashMap<Long, Instr
8991
counter++;
9092
}
9193

94+
}else if (cpIns instanceof MMChainCPInstruction){
95+
final String n1 = cpIns.input1.getName();
96+
getOrMakeCounter(mm, Long.parseLong(n1)).incRMM(1);
97+
getOrMakeCounter(mm, Long.parseLong(n1)).incLMM(1);
98+
9299
}
93100
}
94101

0 commit comments

Comments
 (0)