adjust for processor counter

kendryte · Dec 25, 2023 · 435f2a6 · 435f2a6
1 parent c232f96
commit 435f2a6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/Nncase.EGraph/Passes/EGraphExtractors/SatExtractor.cs b/src/Nncase.EGraph/Passes/EGraphExtractors/SatExtractor.cs
@@ -77,7 +77,7 @@ public Expr Extract(EClass root, IEGraph eGraph, out IReadOnlyDictionary<ENode,
         }
 
         var solver = new CpSolver();
-        int max_time = 60;
+        int max_time = 600;
         if (System.Environment.GetEnvironmentVariable("SOLVE_MAX_TIME") is string s_solve_max_time)
         {
             try
@@ -90,7 +90,8 @@ public Expr Extract(EClass root, IEGraph eGraph, out IReadOnlyDictionary<ENode,
             }
         }
 
-        solver.StringParameters = $"max_time_in_seconds:{max_time},num_workers:0";
+        int processorCount = Math.Max(System.Environment.ProcessorCount / 2, 1);
+        solver.StringParameters = $"max_time_in_seconds:{max_time},num_workers:{processorCount}";
 
         var enableDump = DumpScope.Current.IsEnabled(DumpFlags.EGraphCost);
         CpSolverStatus status;