From eb56ac48b052cbda0ed3b746f478b9ab2850fbb0 Mon Sep 17 00:00:00 2001
From: Ilana Shapiro <ishapiro00@gmail.com>
Date: Mon, 22 Dec 2025 09:47:36 -0800
Subject: [PATCH] Some changes to improve LIA performance (#8101)

* add user params

* inprocessing flag

* playing around with clause sharing with some arith constraints (complicated version commented out)

* collect shared clauses inside share units after pop to base level (might help NIA)

* dont collect clauses twice

* dont pop to base level when sharing units, manual filter

* clean up code

---------

Co-authored-by: Ilana Shapiro <ilanashapiro@Mac.localdomain>
---
 src/params/CMakeLists.txt          |  1 +
 src/params/smt_parallel_params.pyg |  6 ++++++
 src/smt/smt_parallel.cpp           | 16 ++++++++++++++--
 src/smt/smt_parallel.h             |  1 +
 4 files changed, 22 insertions(+), 2 deletions(-)
 create mode 100644 src/params/smt_parallel_params.pyg

diff --git a/src/params/CMakeLists.txt b/src/params/CMakeLists.txt
index 9aea5b918..732430fe3 100644
--- a/src/params/CMakeLists.txt
+++ b/src/params/CMakeLists.txt
@@ -28,6 +28,7 @@ z3_add_component(params
     seq_rewriter_params.pyg
     sls_params.pyg
     smt_params_helper.pyg
+    smt_parallel_params.pyg
     solver_params.pyg
     tactic_params.pyg
   EXTRA_REGISTER_MODULE_HEADERS
diff --git a/src/params/smt_parallel_params.pyg b/src/params/smt_parallel_params.pyg
new file mode 100644
index 000000000..b69e6dd32
--- /dev/null
+++ b/src/params/smt_parallel_params.pyg
@@ -0,0 +1,6 @@
+def_module_params('smt_parallel', 
+                  export=True,
+                  description='Experimental parameters for parallel solving',
+                  params=(
+                        ('inprocessing', BOOL, True, 'integrate in-processing as a heuristic simplification'),
+              ))
\ No newline at end of file
diff --git a/src/smt/smt_parallel.cpp b/src/smt/smt_parallel.cpp
index c4ece1ad7..29915ce6d 100644
--- a/src/smt/smt_parallel.cpp
+++ b/src/smt/smt_parallel.cpp
@@ -25,6 +25,7 @@ Author:
 #include "smt/smt_parallel.h"
 #include "smt/smt_lookahead.h"
 #include "solver/solver_preprocess.h"
+#include "params/smt_parallel_params.hpp"
 
 #include <cmath>
 #include <mutex>
@@ -118,6 +119,10 @@ namespace smt {
 
                 LOG_WORKER(1, " found unsat cube\n");
                 b.backtrack(m_l2g, unsat_core, node);
+
+                if (m_config.m_share_conflicts)
+                    b.collect_clause(m_l2g, id, mk_not(mk_and(unsat_core)));
+
                 break;
             }
             }
@@ -141,21 +146,28 @@ namespace smt {
         m_num_shared_units = ctx->assigned_literals().size();
         m_num_initial_atoms = ctx->get_num_bool_vars();
         ctx->get_fparams().m_preprocess = false;  // avoid preprocessing lemmas that are exchanged
+
+        smt_parallel_params pp(p.ctx.m_params);
+        m_config.m_inprocessing = pp.inprocessing();
     }
 
     void parallel::worker::share_units() {
         // Collect new units learned locally by this worker and send to batch manager
         
-        ctx->pop_to_base_lvl();
         unsigned sz = ctx->assigned_literals().size();
         for (unsigned j = m_num_shared_units; j < sz; ++j) {  // iterate only over new literals since last sync
             literal lit = ctx->assigned_literals()[j];
+
+            // filter by assign level: do not pop to base level as this destroys the current search state
+            if (ctx->get_assign_level(lit) > ctx->m_base_lvl)
+                continue;
+
             if (!ctx->is_relevant(lit.var()) && m_config.m_share_units_relevant_only)
                 continue;
 
             if (m_config.m_share_units_initial_only && lit.var() >= m_num_initial_atoms) {
                 LOG_WORKER(4, " Skipping non-initial unit: " << lit.var() << "\n");
-                continue;  // skip non-iniial atoms if configured to do so
+                continue;  // skip non-initial atoms if configured to do so
             }
 
             expr_ref e(ctx->bool_var2expr(lit.var()), ctx->m);  // turn literal into a Boolean expression
diff --git a/src/smt/smt_parallel.h b/src/smt/smt_parallel.h
index 3c47d818d..007ab090a 100644
--- a/src/smt/smt_parallel.h
+++ b/src/smt/smt_parallel.h
@@ -104,6 +104,7 @@ namespace smt {
             struct config {
                 unsigned m_threads_max_conflicts = 1000;
                 bool m_share_units = true;
+                bool m_share_conflicts = true;
                 bool m_share_units_relevant_only = true;
                 bool m_share_units_initial_only = true;
                 double m_max_conflict_mul = 1.5;