increase timeout + set workers to 16

slothy-optimizer · Dec 5, 2024 · 523fc30 · 523fc30
1 parent 2c5f3be
commit 523fc30
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/test_basic.yaml b/.github/workflows/test_basic.yaml
@@ -75,7 +75,7 @@ jobs:
         echo BASH_ENV="./venv/bin/activate" >> $GITHUB_ENV
     - name: Run examples
       run: |
-        python3 example.py --examples ntt_kyber_1_23_45_67_m55,ntt_dilithium_12_34_56_78_m55 --timeout=300
+        python3 example.py --examples ntt_kyber_1_23_45_67_m55,ntt_dilithium_12_34_56_78_m55 --timeout=1200
   examples_ntt_kyber_dilithium_neon_core:
     if: ${{ github.event.label.name == 'needs-ci' ||
             github.event.pull_request.user.login == 'hanno-becker' ||
@@ -95,7 +95,7 @@ jobs:
         echo BASH_ENV="./venv/bin/activate" >> $GITHUB_ENV
     - name: Run examples
       run: |
-        python3 example.py --examples ntt_kyber_123_4567_a55,ntt_dilithium_123_45678_a55 --timeout=300
+        python3 example.py --examples ntt_kyber_123_4567_a55,ntt_dilithium_123_45678_a55 --timeout=1200
   sqmag:
     if: ${{ github.event.label.name == 'needs-ci' ||
             github.event.pull_request.user.login == 'hanno-becker' ||

diff --git a/slothy/core/core.py b/slothy/core/core.py
@@ -3322,6 +3322,8 @@ def _init_external_model_and_solver(self):
         self._model.cp_solver = cp_model.CpSolver()
         self._model.cp_solver.random_seed = self.config.solver_random_seed
 
+        self._model.cp_solver.parameters.num_workers = 16
+
     def _NewIntVar(self, minval, maxval, name=""): # pylint:disable=invalid-name
         r = self._model.cp_model.NewIntVar(minval,maxval, name)
         self._model.variables.append(r)