You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@systemds.apache.org by ba...@apache.org on 2023/12/30 12:30:11 UTC
(systemds) branch main updated: [MINOR] Reduce Epochs PararmservTest
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 e6b54129f3 [MINOR] Reduce Epochs PararmservTest
e6b54129f3 is described below
commit e6b54129f35dac76d3cd69aa76e1664bdb927546
Author: Sebastian Baunsgaard <ba...@apache.org>
AuthorDate: Sat Dec 30 13:30:00 2023 +0100
[MINOR] Reduce Epochs PararmservTest
---
.../paramserv/ParamservLocalNNAveragingTest.java | 23 +++++++++++-----------
1 file changed, 12 insertions(+), 11 deletions(-)
diff --git a/src/test/java/org/apache/sysds/test/functions/paramserv/ParamservLocalNNAveragingTest.java b/src/test/java/org/apache/sysds/test/functions/paramserv/ParamservLocalNNAveragingTest.java
index b103adf7ef..ab1cf97ab9 100644
--- a/src/test/java/org/apache/sysds/test/functions/paramserv/ParamservLocalNNAveragingTest.java
+++ b/src/test/java/org/apache/sysds/test/functions/paramserv/ParamservLocalNNAveragingTest.java
@@ -39,55 +39,56 @@ public class ParamservLocalNNAveragingTest extends AutomatedTestBase {
@Test
public void testParamservBSPBatchDisjointContiguous() {
- runDMLTest(10, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
+ runDMLTest(4, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
}
@Test
public void testParamservBSPEpoch() {
- runDMLTest(10, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.EPOCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
+ runDMLTest(4, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.EPOCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
}
@Test
public void testParamservBSPBatchDisjointRoundRobin() {
- runDMLTest(10, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_ROUND_ROBIN, true);
+ runDMLTest(4, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_ROUND_ROBIN, true);
}
@Test
public void testParamservBSPBatchDisjointRandom() {
- runDMLTest(10, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_RANDOM, true);
+ runDMLTest(4, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_RANDOM, true);
}
@Test
public void testParamservBSPBatchOverlapReshuffle() {
- runDMLTest(10, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.OVERLAP_RESHUFFLE, true);
+ runDMLTest(4, 2, Statement.PSUpdateType.BSP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.OVERLAP_RESHUFFLE, true);
}
@Test
public void testParamservSBPBatchDisjointContiguous() {
- runDMLTest(10, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
+ runDMLTest(4, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
}
@Test
public void testParamservSBPEpoch() {
- runDMLTest(10, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.EPOCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
+ runDMLTest(4, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.EPOCH, 32, Statement.PSScheme.DISJOINT_CONTIGUOUS, true);
}
@Test
public void testParamservSBPBatchDisjointRoundRobin() {
- runDMLTest(10, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_ROUND_ROBIN, true);
+ runDMLTest(4, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_ROUND_ROBIN, true);
}
@Test
public void testParamservSBPBatchDisjointRandom() {
- runDMLTest(10, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_RANDOM, true);
+ runDMLTest(4, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.DISJOINT_RANDOM, true);
}
@Test
public void testParamservSBPBatchOverlapReshuffle() {
- runDMLTest(10, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.OVERLAP_RESHUFFLE, true);
+ runDMLTest(4, 3, Statement.PSUpdateType.SBP, Statement.PSFrequency.BATCH, 32, Statement.PSScheme.OVERLAP_RESHUFFLE, true);
}
- private void runDMLTest(int epochs, int workers, Statement.PSUpdateType utype, Statement.PSFrequency freq, int batchsize, Statement.PSScheme scheme, boolean modelAvg) {
+ private void runDMLTest(int epochs, int workers, Statement.PSUpdateType utype, Statement.PSFrequency freq,
+ int batchsize, Statement.PSScheme scheme, boolean modelAvg) {
TestConfiguration config = getTestConfiguration(ParamservLocalNNAveragingTest.TEST_NAME);
loadTestConfiguration(config);
programArgs = new String[] { "-stats", "-nvargs", "mode=LOCAL", "epochs=" + epochs,