(systemds) branch main updated: [MINOR] Reduce Epochs PararmservTest

[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 e6b54129f3 [MINOR] Reduce Epochs PararmservTest
e6b54129f3 is described below

commit e6b54129f35dac76d3cd69aa76e1664bdb927546
Author: Sebastian Baunsgaard <baunsga...@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,