#include "util/scoped_ptr_vector.h"
#include "ast/ast_util.h"
#include "ast/ast_pp.h"
#include "ast/ast_ll_pp.h"
#include "ast/ast_translation.h"
#include "ast/simplifiers/then_simplifier.h"
#include "smt/smt_parallel.h"
#include "smt/smt_lookahead.h"
#include "solver/solver_preprocess.h"
#include "solver/parallel_params.hpp"
#include <cmath>
#include <mutex>
#include <condition_variable>
class bounded_pp_exprs {
expr_ref_vector const &es;
public:
bounded_pp_exprs(expr_ref_vector const &es) : es(es) {}
std::ostream &display(std::ostream &out) const {
for (auto e : es)
out << mk_bounded_pp(e, es.get_manager()) << "\n";
return out;
}
};
inline std::ostream &operator<<(std::ostream &out, bounded_pp_exprs const &pp) {
return pp.display(out);
}
#ifdef SINGLE_THREAD
namespace smt {
lbool parallel::operator()(expr_ref_vector const &asms) {
return l_undef;
}
}
#else
#include <thread>
#define LOG_WORKER(lvl, s) IF_VERBOSE(lvl, verbose_stream() << "Worker " << id << s)
#define LOG_BB_WORKER(lvl, s) IF_VERBOSE(lvl, verbose_stream() << "Backbones Worker " << id << s)
namespace smt {
static bool is_cancellation_exception(char const *msg) {
return msg && (strstr(msg, "canceled") != nullptr || strstr(msg, "cancelled") != nullptr);
}
void parallel::sls_worker::run() {
ptr_vector<expr> assertions;
p.ctx.get_assertions(assertions);
for (expr *e : assertions)
m_sls->assert_expr(m_g2l(e));
lbool res = l_undef;
try {
if (!m.inc())
return;
res = m_sls->check();
} catch (z3_exception &ex) {
if (m.limit().is_canceled()) {
IF_VERBOSE(1, verbose_stream() << "SLS worker canceled\n");
return;
}
if (strstr(ex.what(), "unsupported for sls") != nullptr) {
IF_VERBOSE(1, verbose_stream() << "SLS opted out: " << ex.what() << "\n");
return;
}
IF_VERBOSE(1, verbose_stream() << "SLS threw exception: " << ex.what() << "\n");
b.set_exception(ex.what());
return;
}
if (res == l_true) {
IF_VERBOSE(2, verbose_stream() << "SLS worker found SAT\n");
model_ref mdl = m_sls->get_model();
b.set_sat(m_l2g, *mdl);
}
}
void parallel::backbones_worker::run() {
if (m_use_failed_literal_test)
run_failed_literal_mode();
else
run_batch_mode();
}
void parallel::backbones_worker::run_failed_literal_mode() {
ctx->get_fparams().m_max_conflicts = 10;
auto is_unit = [&](unsigned v) {
return ctx->get_assignment(v) != l_undef && ctx->get_assign_level(v) == ctx->m_base_lvl;
};
auto probe_var = [&](unsigned v, expr* preferred, bool is_retry) -> lbool {
expr_ref e(ctx->bool_var2expr(v), m);
if (!e)
return l_undef;
if (m.is_or(e) || m.is_ite(e) || m.is_and(e) || m.is_iff(e))
return l_undef;
if (is_unit(v)) {
bool is_true = ctx->get_assignment(v) == l_true;
IF_VERBOSE(2, verbose_stream() << "backbone on trail " << mk_bounded_pp(e.get(), m) << "\n");
if (!is_true)
e = m.mk_not(e);
if (b.collect_global_backbone(m_l2g, e)) {
m_stats.m_internal_backbones_found++;
if (is_retry)
m_stats.m_retry_backbones_found++;
}
return l_undef;
}
expr_ref first(e, m), second(mk_not(e), m);
if (preferred) {
expr* atom = preferred;
bool is_negated = m.is_not(preferred, atom);
first = is_negated ? mk_not(e) : e;
second = is_negated ? e : mk_not(e);
}
lbool r = probe_literal(v, first.get(), is_retry);
if (r != l_undef || is_unit(v))
return r;
return probe_literal(v, second.get(), is_retry);
};
bb_candidates bb_candidates;
while (m.inc()) {
if (!b.wait_for_backbone_job(id, m_g2l, bb_candidates, m.limit()))
return;
if (bb_candidates.empty())
continue;
collect_shared_clauses();
unsigned local_cancel_epoch = b.get_cancel_epoch();
auto canceled = [&] { return local_cancel_epoch != b.get_cancel_epoch(); };
bool is_retry = false;
unsigned bb_candidate_epoch = b.get_bb_candidate_epoch();
expr_ref_vector bb_candidate_lits(m);
for (auto const& c : bb_candidates)
bb_candidate_lits.push_back(c.lit);
while (m.inc() && !canceled()) {
lbool terminal_result = l_undef;
uint_set seen_vars; for (expr* lit : bb_candidate_lits) {
if (is_retry && b.has_new_backbone_candidates(bb_candidate_epoch))
break;
if (!m.inc() || canceled())
break;
expr* atom = lit;
m.is_not(lit, atom);
if (!ctx->b_internalized(atom))
continue;
sat::bool_var v = ctx->get_bool_var(atom);
if (v == sat::null_bool_var || seen_vars.contains(v))
continue;
seen_vars.insert(v);
terminal_result = probe_var(v, lit, is_retry);
if (terminal_result != l_undef)
break;
}
if (terminal_result != l_undef)
break;
if (b.has_new_backbone_candidates(bb_candidate_epoch) || canceled() || !m.inc())
break;
is_retry = true;
expr_ref_vector bb_snapshot = b.get_global_backbones_snapshot(m_g2l);
expr_mark bb_mark;
for (expr* e : bb_snapshot) {
bb_mark.mark(e);
bb_mark.mark(mk_not(m, e));
}
bb_candidate_lits.reset();
for (auto const& c : bb_candidates)
if (!bb_mark.is_marked(c.lit.get()))
bb_candidate_lits.push_back(c.lit);
}
if (!m.inc())
return;
if (!canceled())
b.cancel_current_backbone_batch();
bb_candidates.reset();
}
}
lbool parallel::backbones_worker::probe_literal(bool_var v, expr *e, bool is_retry) {
asms.push_back(e);
auto terminal_result = b.check(asms, *ctx);
asms.pop_back();
if (terminal_result == l_false) {
if (!ctx->unsat_core().contains(e)) {
b.set_unsat(m_l2g, ctx->unsat_core());
return l_false;
}
IF_VERBOSE(2, verbose_stream() << "failed literal " << mk_bounded_pp(e, m) << "\n");
expr_ref not_e(mk_not(m, e), m);
m_stats.m_backbones_detected++;
if (b.collect_global_backbone(m_l2g, not_e)) {
m_stats.m_internal_backbones_found++;
if (is_retry)
m_stats.m_retry_backbones_found++;
}
ctx->assert_expr(not_e);
terminal_result = l_undef;
}
if (terminal_result == l_true) {
model_ref mdl;
ctx->get_model(mdl);
b.set_sat(m_l2g, *mdl);
}
return terminal_result;
}
void parallel::backbones_worker::run_batch_mode() {
bb_candidates bb_curr_batch_candidates;
while (m.inc()) {
if (!b.wait_for_backbone_job(id, m_g2l, bb_curr_batch_candidates, m.limit()))
return;
if (bb_curr_batch_candidates.empty())
continue;
LOG_BB_WORKER(1, " received batch of " << bb_curr_batch_candidates.size() << " candidates\n");
collect_shared_clauses();
unsigned local_cancel_epoch = b.get_cancel_epoch();
auto canceled = [&] { return local_cancel_epoch != b.get_cancel_epoch(); };
unsigned bb_candidate_epoch = b.get_bb_candidate_epoch();
auto fallback_failed_literal_probe = [&](expr_ref_vector const& chunk_lits, expr_ref_vector& bb_candidate_lits, bool is_retry = false) {
unsigned old_max_conflicts = ctx->get_fparams().m_max_conflicts;
ctx->get_fparams().m_max_conflicts = 10;
if (is_retry)
++m_stats.m_bb_retries;
else
++m_stats.m_fallback_singleton_checks;
for (expr* lit : chunk_lits) {
if (is_retry && b.has_new_backbone_candidates(bb_candidate_epoch)) {
ctx->get_fparams().m_max_conflicts = old_max_conflicts;
return;
}
if (!m.inc() || canceled()) {
ctx->get_fparams().m_max_conflicts = old_max_conflicts;
return;
}
if (!bb_candidate_lits.contains(lit)) continue;
expr_ref bb_ref(lit, m);
if (m_mode == bb_mode::bb_positive)
bb_ref = mk_not(m, bb_ref);
if (!b.is_global_backbone_or_negation(m_l2g, bb_ref)) {
expr_ref backbone(m);
if (try_get_unit_backbone(bb_ref.get(), backbone)) {
m_stats.m_backbones_detected++;
LOG_BB_WORKER(1, " fallback found unit backbone: " << mk_bounded_pp(backbone.get(), m, 3) << "\n");
if (b.collect_global_backbone(m_l2g, backbone))
m_stats.m_internal_backbones_found++;
} else {
expr* atom = bb_ref.get();
m.is_not(bb_ref.get(), atom);
if (ctx->b_internalized(atom)) {
sat::bool_var v = ctx->get_bool_var(atom);
if (v != sat::null_bool_var) {
lbool terminal_result = probe_literal(v, mk_not(m, bb_ref), is_retry); LOG_BB_WORKER(1, " RESULT: " << terminal_result << " FOR CANDIDATE: " << mk_bounded_pp(bb_ref.get(), m, 3) << "\n");
}
}
}
}
bb_candidate_lits.erase(lit);
}
ctx->get_fparams().m_max_conflicts = old_max_conflicts;
};
m_stats.m_batches_total++;
m_stats.m_candidates_total += bb_curr_batch_candidates.size();
expr_ref_vector bb_candidate_lits(m);
for (auto const& c : bb_curr_batch_candidates)
bb_candidate_lits.push_back(c.lit);
unsigned chunk_delta = 1;
while (!bb_candidate_lits.empty() && !canceled() && m.inc()) {
{
unsigned j = 0;
for (auto lit : bb_candidate_lits) {
if (!b.is_global_backbone_or_negation(m_l2g, lit))
bb_candidate_lits[j++] = lit;
}
bb_candidate_lits.shrink(j);
}
{
unsigned j = 0;
for (expr* lit : bb_candidate_lits) {
expr_ref backbone(m);
if (try_get_unit_backbone(lit, backbone)) {
IF_VERBOSE(2, verbose_stream() << "backbone on trail " << mk_bounded_pp(backbone.get(), m) << "\n");
if (b.collect_global_backbone(m_l2g, backbone))
m_stats.m_internal_backbones_found++;
m_stats.m_backbones_detected++;
continue;
}
bb_candidate_lits[j++] = lit;
}
bb_candidate_lits.shrink(j);
}
unsigned chunk_size = std::min(m_bb_chunk_size * chunk_delta, bb_candidate_lits.size());
expr_ref_vector chunk_lits(m);
expr_ref_vector negated_chunk_lits(m);
expr_mark chunk_atoms;
for (unsigned i = 0; i < bb_candidate_lits.size() && chunk_lits.size() < chunk_size; ++i) {
expr* lit = bb_candidate_lits.get(i);
expr* atom = lit;
m.is_not(lit, atom);
if (chunk_atoms.is_marked(atom))
continue;
chunk_atoms.mark(atom);
chunk_lits.push_back(lit);
negated_chunk_lits.push_back(mk_not(m, lit));
}
expr_ref_vector bb_asms(m);
if (m_mode == bb_mode::bb_negated)
bb_asms.append(negated_chunk_lits); else
bb_asms.append(chunk_lits);
collect_shared_clauses();
while (true) {
if (!m.inc())
return;
if (canceled())
break;
m_stats.m_core_refinement_rounds++;
unsigned base_asms_sz = asms.size();
for (expr* a : bb_asms)
asms.push_back(a);
lbool r = b.check(asms, *ctx);
asms.shrink(base_asms_sz);
if (!m.inc() || canceled())
break;
if (r == l_undef) {
LOG_BB_WORKER(1, " UNDEF at chunk_size=" << chunk_size << "\n");
if (chunk_size < bb_candidate_lits.size()) {
chunk_delta++; m_stats.m_num_chunk_increases++;
break;
}
LOG_BB_WORKER(1, " UNDEF and max chunk → fallback\n");
fallback_failed_literal_probe(chunk_lits, bb_candidate_lits);
m_stats.m_fallback_reason_undef++;
chunk_delta = 1;
break;
}
if (r == l_true) {
LOG_BB_WORKER(1, " batch check returned SAT, thus entire formula is SAT\n");
model_ref mdl;
ctx->get_model(mdl);
b.set_sat(m_l2g, *mdl);
bb_curr_batch_candidates.reset();
return;
}
expr_ref_vector bb_asms_in_core(m);
auto const& unsat_core = ctx->unsat_core();
for (expr* a : unsat_core)
if (bb_asms.contains(a))
bb_asms_in_core.push_back(a);
if (bb_asms_in_core.empty()) {
b.set_unsat(m_l2g, unsat_core);
return;
}
if (bb_asms_in_core.size() == 1) {
expr* a = bb_asms_in_core.back();
expr_ref backbone_lit(mk_not(m, a), m);
m_stats.m_singleton_backbones++;
m_stats.m_backbones_detected++;
if (b.collect_global_backbone(m_l2g, backbone_lit)) {
m_stats.m_internal_backbones_found++;
ctx->assert_expr(backbone_lit.get()); }
expr* candidate_to_remove =
(m_mode == bb_mode::bb_negated)
? backbone_lit.get() : a;
bb_candidate_lits.erase(candidate_to_remove);
}
unsigned sz_before = bb_asms.size();
for (expr* a : bb_asms_in_core)
bb_asms.erase(a);
m_stats.m_lits_removed_by_core += sz_before - bb_asms.size();
chunk_delta = 1;
if (bb_asms.empty()) {
LOG_BB_WORKER(1, " no more negated chunk literals, fallback to individual checks\n");
fallback_failed_literal_probe(chunk_lits, bb_candidate_lits);
m_stats.m_fallback_reason_chunk_exhausted++;
break;
}
}
}
while (!b.has_new_backbone_candidates(bb_candidate_epoch) && !canceled() && m.inc()) {
collect_shared_clauses();
unsigned found_before = m_stats.m_internal_backbones_found;
expr_ref_vector bb_snapshot = b.get_global_backbones_snapshot(m_g2l);
expr_mark bb_mark;
for (expr* e : bb_snapshot) {
bb_mark.mark(e);
bb_mark.mark(mk_not(m, e));
}
bb_candidate_lits.reset();
for (auto const& c : bb_curr_batch_candidates)
if (!bb_mark.is_marked(c.lit.get()))
bb_candidate_lits.push_back(c.lit);
if (bb_candidate_lits.empty())
break;
fallback_failed_literal_probe(bb_candidate_lits, bb_candidate_lits, true);
if (m_stats.m_internal_backbones_found == found_before)
break;
}
if (!canceled())
b.cancel_current_backbone_batch();
bb_curr_batch_candidates.reset();
}
}
void parallel::backbones_worker::cancel() {
LOG_BB_WORKER(1, " BACKBONES WORKER cancelling\n");
m.limit().cancel();
}
bool parallel::batch_manager::collect_global_backbone(ast_translation &l2g, expr_ref const &backbone, unsigned source_worker_id) {
IF_VERBOSE(1, verbose_stream() << "collect-global-backbone\n");
std::scoped_lock lock(mux);
SASSERT(&m == &l2g.to());
if (is_global_backbone_unlocked(l2g, backbone))
return false;
expr_ref g_bb_ref(l2g(backbone.get()), m);
m_global_backbones.insert(g_bb_ref.get());
++m_stats.m_backbones_found;
IF_VERBOSE(1, verbose_stream() << " Found and sharing new global backbone: " << mk_bounded_pp(g_bb_ref, m, 3) << "\n");
collect_clause_unlocked(l2g, source_worker_id, backbone.get());
expr_ref neg_g_bb_ref(mk_not(g_bb_ref), m);
vector<cube_config::literal> g_core;
g_core.push_back(neg_g_bb_ref);
vector<node_lease> targets;
collect_matching_targets_unlocked(nullptr, neg_g_bb_ref, g_core, targets);
if (!targets.empty()) {
IF_VERBOSE(1, verbose_stream() << " Closing negation of the new global backbone: " << mk_bounded_pp(g_bb_ref, m, 3) << "\n");
if (m_ablate_backtracking) {
for (auto const& target : targets) {
if (m_search_tree.is_lease_canceled(target.leased_node))
continue;
expr_ref_vector full_cube(l2g.from());
node* n = target.leased_node;
while (n) {
if (!cube_config::literal_is_null(n->get_literal())) {
expr* lit = n->get_literal().get();
full_cube.push_back(expr_ref(lit, l2g.from()));
}
n = n->parent();
}
vector<node_lease> single_target = { target };
backtrack_unlocked(l2g, UINT_MAX, full_cube, nullptr, &single_target);
}
} else {
expr_ref_vector l_core(l2g.from());
l_core.push_back(mk_not(backbone));
backtrack_unlocked(l2g, UINT_MAX, l_core, nullptr, &targets);
}
}
return true;
}
void parallel::backbones_worker::collect_statistics(::statistics& st) const {
st.update("bb-batches-total", m_stats.m_batches_total);
st.update("bb-candidates-total", m_stats.m_candidates_total);
st.update("bb-backbones-detected", m_stats.m_backbones_detected);
st.update("bb-internal-backbones-found", m_stats.m_internal_backbones_found);
st.update("bb-retry-backbones-found", m_stats.m_retry_backbones_found);
st.update("bb-retries", m_stats.m_bb_retries);
st.update("bb-core-refinement-rounds", m_stats.m_core_refinement_rounds);
st.update("bb-singleton-backbones", m_stats.m_singleton_backbones);
st.update("bb-fallback-singleton-checks", m_stats.m_fallback_singleton_checks);
st.update("bb-fallback-chunk-exhausted", m_stats.m_fallback_reason_chunk_exhausted);
st.update("bb-fallback-undef", m_stats.m_fallback_reason_undef);
st.update("bb-literals-removed-by-core", m_stats.m_lits_removed_by_core);
st.update("bb-num-chunk-increases", m_stats.m_num_chunk_increases);
auto safe_ratio = [](double num, double den) -> double {
return den > 0 ? num / den : 0.0;
};
st.update("bb-backbone-yield-pct",
100.0 * safe_ratio(m_stats.m_internal_backbones_found, m_stats.m_candidates_total));
st.update("bb-avg-backbones-per-batch",
safe_ratio(m_stats.m_internal_backbones_found, m_stats.m_batches_total));
st.update("bb-core-refinement-rounds-per-batch",
safe_ratio(m_stats.m_core_refinement_rounds, m_stats.m_batches_total));
st.update("bb-core-effectiveness-lit-removed-per-round",
safe_ratio(m_stats.m_lits_removed_by_core, m_stats.m_core_refinement_rounds));
}
void parallel::sls_worker::cancel() {
IF_VERBOSE(1, verbose_stream() << " SLS WORKER cancelling\n");
m.limit().cancel();
}
void parallel::sls_worker::collect_statistics(::statistics &st) const {
m_sls->collect_statistics(st);
}
parallel::core_minimizer_worker::core_minimizer_worker(parallel& p, expr_ref_vector const& _asms)
: b(p.m_batch_manager), asms(m), m_smt_params(p.ctx.get_fparams()), m_g2l(p.ctx.m, m), m_l2g(m, p.ctx.m) {
for (expr* e : _asms)
asms.push_back(m_g2l(e));
IF_VERBOSE(1, verbose_stream() << "Initialized core minimizer thread\n");
ctx = alloc(context, m, m_smt_params, p.ctx.get_params());
ctx->set_logic(p.ctx.m_setup.get_logic());
context::copy(p.ctx, *ctx, true);
ctx->pop_to_base_lvl();
ctx->get_fparams().m_preprocess = false; }
void parallel::core_minimizer_worker::cancel() {
IF_VERBOSE(1, verbose_stream() << "Core minimizer cancelling\n");
m.limit().cancel();
}
void parallel::core_minimizer_worker::collect_statistics(::statistics& st) const {
ctx->collect_statistics(st);
st.update("parallel-core-minimize-calls", m_num_core_minimize_calls);
st.update("parallel-core-minimize-undef", m_num_core_minimize_undef);
st.update("parallel-core-minimize-refined", m_num_core_minimize_refined);
st.update("parallel-core-minimize-lits-removed", m_num_core_minimize_lits_removed);
st.update("parallel-core-minimize-found-sat", m_num_core_minimize_found_sat);
}
void parallel::core_minimizer_worker::minimize_unsat_core(expr_ref_vector& core) {
expr_ref_vector unknown(core), mus(m), trial(m);
unsigned original_size = core.size();
++m_num_core_minimize_calls;
while (!unknown.empty()) {
if (!m.inc()) {
core.reset();
core.append(mus);
core.append(unknown);
return;
}
expr* lit = unknown.back();
unknown.pop_back();
expr_ref not_lit(mk_not(m, lit), m);
trial.reset();
trial.append(mus);
trial.append(unknown);
trial.push_back(not_lit);
lbool r = l_undef;
try {
flet<unsigned> _max_conflicts(ctx->get_fparams().m_max_conflicts, m_core_minimize_conflict_budget);
r = ctx->check(trial.size(), trial.data());
}
catch (...) {
r = l_undef;
}
switch (r) {
case l_undef: ++m_num_core_minimize_undef;
mus.push_back(lit);
break;
case l_true: { if (!asms.empty()) {
mus.push_back(lit);
break;
}
++m_num_core_minimize_found_sat;
model_ref mdl;
ctx->get_model(mdl);
b.set_sat(m_l2g, *mdl);
return;
}
case l_false: {
auto const& unsat_core = ctx->unsat_core();
if (!unsat_core.contains(not_lit)) {
++m_num_core_minimize_refined;
unknown.reset();
expr_ref_vector new_mus(m);
for (expr* c : unsat_core) {
if (mus.contains(c))
new_mus.push_back(c);
else
unknown.push_back(c);
}
mus.reset();
mus.append(new_mus);
}
break;
}
default:
UNREACHABLE();
}
}
core.reset();
core.append(mus);
core.append(unknown); if (core.size() < original_size)
m_num_core_minimize_lits_removed += original_size - core.size();
return;
}
void parallel::core_minimizer_worker::run() {
while (m.inc()) {
node* source = nullptr;
expr_ref_vector core(m);
if (!b.wait_for_core_min_job(m_g2l, source, core, m.limit()))
return;
unsigned original_size = core.size();
if (original_size <= 1)
continue;
collect_shared_clauses();
expr_ref_vector minimized(m);
minimized.append(core);
minimize_unsat_core(minimized);
if (minimized.size() < original_size)
b.publish_minimized_core(m_l2g, asms, source, original_size, minimized);
}
}
void parallel::worker::run() {
bool is_first_run = true;
node_lease lease;
expr_ref_vector cube(m);
while (true) {
if (!b.get_cube(m_g2l, id, cube, is_first_run, lease)) {
LOG_WORKER(1, " no more cubes\n");
return;
}
is_first_run = false;
collect_shared_clauses();
check_cube_start:
LOG_WORKER(1, " CUBE SIZE IN MAIN LOOP: " << cube.size() << "\n");
if (m_config.m_global_backbones) {
bb_candidates local_candidates = find_backbone_candidates();
b.collect_backbone_candidates(m_l2g, local_candidates);
bool lease_canceled = false;
if (!b.checkpoint_worker(id, lease, lease_canceled))
return;
if (lease_canceled) {
LOG_WORKER(1, " abandoning canceled lease\n");
continue;
}
}
lbool r = check_cube(cube);
bool lease_canceled = false;
if (!b.checkpoint_worker(id, lease, lease_canceled))
return;
if (lease_canceled) {
LOG_WORKER(1, " abandoning canceled lease\n");
continue;
}
switch (r) {
case l_undef: {
LOG_WORKER(1, " found undef cube\n");
std::string reason = ctx->last_failure_as_string();
if (reason != "max-conflicts-reached") {
LOG_WORKER(1, " undef cube not conflict-limited (" << reason << "); reporting unknown\n");
b.set_unknown(reason);
return;
}
update_max_thread_conflicts();
if (m_config.m_max_cube_depth <= cube.size())
goto check_cube_start;
auto atom = get_split_atom();
if (!atom)
goto check_cube_start;
b.try_split(m_l2g, id, lease, atom, m_config.m_threads_max_conflicts);
simplify();
break;
}
case l_true: {
LOG_WORKER(1, " found sat cube\n");
model_ref mdl;
ctx->get_model(mdl);
b.set_sat(m_l2g, *mdl);
return;
}
case l_false: {
expr_ref_vector const &unsat_core = ctx->unsat_core();
LOG_WORKER(2, " unsat core:\n";
for (auto c : unsat_core) verbose_stream() << mk_bounded_pp(c, m, 3) << "\n");
if (all_of(unsat_core, [&](expr *e) { return asms.contains(e); })) {
LOG_WORKER(1, " determined formula unsat\n");
b.set_unsat(m_l2g, unsat_core);
return;
}
LOG_WORKER(1, " found unsat cube\n");
node* source = lease.leased_node;
expr_ref_vector const& core_to_use = m_config.m_ablate_backtracking ? cube : unsat_core;
if (m_config.m_ablate_backtracking) {
LOG_WORKER(1, " ablating backtracking: using full cube path of size " << core_to_use.size() << "\n");
}
b.backtrack(m_l2g, id, core_to_use, lease);
if (m_config.m_core_minimize)
b.enqueue_core_minimization(m_l2g, source, unsat_core);
if (m_config.m_share_conflicts)
b.collect_clause(m_l2g, id, mk_not(mk_and(unsat_core)));
break;
}
}
if (m_config.m_share_units)
share_units();
}
}
parallel::worker::worker(unsigned id, parallel &p, expr_ref_vector const &_asms)
: id(id), p(p), b(p.m_batch_manager), asms(m), m_smt_params(p.ctx.get_fparams()), m_g2l(p.ctx.m, m),
m_l2g(m, p.ctx.m) {
for (auto e : _asms)
asms.push_back(m_g2l(e));
LOG_WORKER(1, " created with " << asms.size() << " assumptions\n");
m_smt_params.m_random_seed += id; ctx = alloc(context, m, m_smt_params, p.ctx.get_params());
ctx->set_logic(p.ctx.m_setup.get_logic());
context::copy(p.ctx, *ctx, true);
ctx->pop_to_base_lvl();
m_shared_units_prefix = ctx->assigned_literals().size();
m_num_initial_atoms = ctx->get_num_bool_vars();
ctx->get_fparams().m_preprocess = false;
parallel_params pp(p.ctx.m_params);
m_config.m_inprocessing = false;
m_config.m_global_backbones = pp.num_bb_threads() > 0;
m_config.m_local_backbones = false;
m_config.m_core_minimize = pp.core_minimize();
m_config.m_ablate_backtracking = pp.ablate_backtracking();
if (m_config.m_ablate_backtracking) {
m_config.m_core_minimize = false;
}
m_config.m_threads_max_conflicts = m_smt_params.m_threads_max_conflicts;
}
parallel::sls_worker::sls_worker(parallel& p)
: p(p), b(p.m_batch_manager), m(), m_g2l(p.ctx.m, m), m_l2g(m, p.ctx.m) {
IF_VERBOSE(1, verbose_stream() << "Initialized SLS portfolio thread\n");
m_params.append(p.ctx.m_params);
m_sls = alloc(sls::smt_solver, m, m_params);
}
parallel::backbones_worker::backbones_worker(unsigned id, parallel &p, expr_ref_vector const &_asms)
: id(id), b(p.m_batch_manager), m(), asms(m), m_smt_params(p.ctx.get_fparams()), m_g2l(p.ctx.m, m), m_l2g(m, p.ctx.m) {
for (auto e : _asms)
asms.push_back(m_g2l(e));
IF_VERBOSE(1, verbose_stream() << "Initialized backbones thread " << id << "\n");
m_mode = id == 0 ? bb_mode::bb_negated : bb_mode::bb_positive;
ctx = alloc(context, m, m_smt_params, p.ctx.get_params());
ctx->set_logic(p.ctx.m_setup.get_logic());
ctx->get_fparams().m_max_conflicts = m_bb_conflicts_per_chunk;
context::copy(p.ctx, *ctx, true);
ctx->pop_to_base_lvl();
m_shared_units_prefix = ctx->assigned_literals().size();
m_num_initial_atoms = ctx->get_num_bool_vars();
ctx->get_fparams().m_preprocess = false;
m_use_failed_literal_test = false;
}
parallel::bb_candidates parallel::worker::find_backbone_candidates(unsigned k) {
bb_candidates backbone_candidates;
expr_ref candidate(m);
unsigned curr_time = ctx->m_stats.m_num_assignments;
for (bool_var v = 0; v < ctx->get_num_bool_vars(); ++v) {
if (ctx->get_assignment(v) != l_undef && ctx->get_assign_level(v) == ctx->m_base_lvl)
continue;
candidate = ctx->bool_var2expr(v);
if (!candidate)
continue;
auto birth = ctx->m_birthdate[v];
auto age = curr_time - birth;
auto const& d = ctx->get_bdata(v);
if (d.m_phase_available && !d.m_phase)
candidate = m.mk_not(candidate);
if (b.is_global_backbone_or_negation(m_l2g, candidate))
continue;
bb_candidate bb_cand(m, candidate, age, 1);
backbone_candidates.push_back(bb_cand);
}
std::stable_sort(
backbone_candidates.begin(),
backbone_candidates.end(),
[](bb_candidate const& a, bb_candidate const& b) {
return a.age > b.age;
}
);
if (backbone_candidates.size() > k)
backbone_candidates.shrink(k);
return backbone_candidates;
}
bool parallel::backbones_worker::try_get_unit_backbone(expr* candidate, expr_ref& backbone) {
expr* atom = candidate;
m.is_not(candidate, atom);
if (!ctx->b_internalized(atom))
return false;
sat::bool_var v = ctx->get_bool_var(atom);
if (v == sat::null_bool_var || ctx->get_assignment(v) == l_undef || ctx->get_assign_level(v) != ctx->m_base_lvl)
return false;
bool is_true = ctx->get_assignment(v) == l_true;
backbone = expr_ref(atom, m);
if (!is_true)
backbone = mk_not(backbone);
return true;
}
void parallel::worker::share_units() {
unsigned sz = ctx->assigned_literals().size();
unsigned prefix_sz = m_shared_units_prefix;
bool at_prefix = true;
for (unsigned j = m_shared_units_prefix; j < sz; ++j) { literal lit = ctx->assigned_literals()[j];
if (ctx->get_assign_level(lit) > ctx->m_base_lvl) {
at_prefix = false;
continue;
}
if (at_prefix)
++prefix_sz;
if (m_known_units.contains(lit.var()))
continue;
m_known_units.insert(lit.var());
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; }
expr_ref e(ctx->bool_var2expr(lit.var()), ctx->m); if (m.is_and(e) || m.is_or(e) || m.is_ite(e) || m.is_iff(e))
continue;
if (lit.sign())
e = mk_not(e);
b.collect_global_backbone(m_l2g, e, id);
}
m_shared_units_prefix = prefix_sz;
}
void parallel::worker::simplify() {
if (!m.inc())
return;
if (!m_config.m_inprocessing)
return;
if (m_config.m_inprocessing_delay > 0) {
--m_config.m_inprocessing_delay;
return;
}
ctx->pop_to_base_lvl();
if (ctx->m_base_lvl > 0)
return; m_config.m_inprocessing = false; dependent_expr_simplifier *s = ctx->m_simplifier.get();
if (!s) {
ctx->m_fmls = alloc(base_dependent_expr_state, m);
auto then_s = alloc(then_simplifier, m, ctx->get_params(), *ctx->m_fmls);
s = then_s;
ctx->m_simplifier = s;
init_preprocess(m, ctx->get_params(), *then_s, *ctx->m_fmls);
}
dependent_expr_state &fmls = *ctx->m_fmls.get();
ptr_vector<expr> assertions;
expr_ref_vector units(m);
ctx->get_assertions(assertions);
ctx->get_units(units);
for (expr *e : assertions)
fmls.add(dependent_expr(m, e, nullptr, nullptr));
for (expr *e : units)
fmls.add(dependent_expr(m, e, nullptr, nullptr));
s->reduce();
scoped_ptr<context> new_ctx = alloc(context, m, m_smt_params, p.ctx.get_params());
for (unsigned i = 0; i < fmls.qtail(); ++i) {
auto const &de = fmls[i];
new_ctx->assert_expr(de.fml());
}
asserted_formulas &src_af = ctx->m_asserted_formulas;
asserted_formulas &dst_af = new_ctx->m_asserted_formulas;
src_af.get_macro_manager().copy_to(dst_af.get_macro_manager());
new_ctx->copy_user_propagator(*ctx, true);
ctx = new_ctx.detach();
ctx->setup_context(true);
ctx->internalize_assertions();
auto old_atoms = m_num_initial_atoms;
m_shared_units_prefix = ctx->assigned_literals().size();
m_num_initial_atoms = ctx->get_num_bool_vars();
LOG_WORKER(1, " inprocess " << old_atoms << " -> " << m_num_initial_atoms << "\n");
}
void parallel::worker::collect_statistics(::statistics &st) const {
ctx->collect_statistics(st);
}
void parallel::worker::cancel() {
LOG_WORKER(1, " canceling\n");
m.limit().cancel();
}
void parallel::worker::cancel_lease() {
LOG_WORKER(1, " canceling lease\n");
m.limit().inc_cancel();
}
lbool parallel::batch_manager::check(expr_ref_vector const &asms, context &ctx) {
lbool r = l_undef;
auto &m = asms.m();
try {
r = ctx.check(asms.size(), asms.data());
} catch (z3_error &err) {
if (!m.limit().is_canceled())
set_exception(err.error_code());
} catch (z3_exception &ex) {
if (!m.limit().is_canceled() && !is_cancellation_exception(ex.what()))
set_exception(ex.what());
} catch (...) {
if (!m.limit().is_canceled())
set_exception("unknown exception");
}
return r;
}
void parallel::batch_manager::set_canceled_unlocked() {
if (m_state != state::is_running)
return;
cancel_background_threads();
}
void parallel::batch_manager::set_canceled() {
std::scoped_lock lock(mux);
set_canceled_unlocked();
}
void parallel::batch_manager::release_worker_lease_unlocked(unsigned worker_id, node_lease& lease) {
if (worker_id >= m_worker_leases.size()) {
lease = {};
return;
}
auto& stored_lease = m_worker_leases[worker_id];
if (!stored_lease.leased_node || stored_lease.leased_node != lease.leased_node) {
lease = {};
return;
}
bool cancel_signaled = stored_lease.cancel_signaled;
m_search_tree.dec_active_workers(stored_lease.leased_node);
stored_lease = {};
lease = {};
if (cancel_signaled)
p.m_workers[worker_id]->limit().dec_cancel();
}
bool parallel::batch_manager::attempt_release_canceled_lease_unlocked(unsigned worker_id, node_lease& lease) {
if (m_state != state::is_running || !lease.leased_node || worker_id >= m_worker_leases.size())
return false;
auto& stored_lease = m_worker_leases[worker_id];
if (stored_lease.leased_node != lease.leased_node)
return false;
if (!m_search_tree.is_lease_canceled(stored_lease.leased_node))
return false;
release_worker_lease_unlocked(worker_id, lease);
return true;
}
void parallel::batch_manager::cancel_closed_leases_unlocked(unsigned source_worker_id) {
unsigned n = std::min(m_worker_leases.size(), p.m_workers.size());
for (unsigned worker_id = 0; worker_id < n; ++worker_id) {
if (worker_id == source_worker_id)
continue;
auto const& lease = m_worker_leases[worker_id];
if (lease.leased_node && !lease.cancel_signaled && m_search_tree.is_lease_canceled(lease.leased_node)) {
p.m_workers[worker_id]->cancel_lease();
m_worker_leases[worker_id].cancel_signaled = true;
}
}
}
void parallel::batch_manager::backtrack(ast_translation &l2g, unsigned worker_id, expr_ref_vector const &core,
node_lease& lease) {
std::scoped_lock lock(mux);
vector<cube_config::literal> g_core;
for (auto c : core)
g_core.push_back(expr_ref(l2g(c), m));
vector<node_lease> targets;
collect_matching_targets_unlocked(lease.leased_node, lease.leased_node->get_literal().get(), g_core, targets);
backtrack_unlocked(l2g, worker_id, core, &lease, targets.empty() ? nullptr : &targets);
}
void parallel::batch_manager::enqueue_core_minimization(ast_translation& l2g, node* source,
expr_ref_vector const& core) {
std::scoped_lock lock(mux);
if (m_state != state::is_running || !p.m_core_minimizer_worker || !source || core.empty())
return;
if (core.size() <= 1) {
++m_stats.m_core_min_jobs_skipped;
return;
}
source = find_core_source_unlocked(l2g, source, core);
if (!source) {
++m_stats.m_core_min_jobs_skipped;
return;
}
scoped_ptr<core_min_job> job = alloc(core_min_job, m, source);
for (expr* c : core)
job->core.push_back(l2g(c));
m_core_min_jobs.push_back(job.detach());
++m_stats.m_core_min_jobs_enqueued;
m_core_min_cv.notify_one();
}
bool parallel::batch_manager::wait_for_core_min_job(ast_translation& g2l, node*& source,
expr_ref_vector& core, reslimit& lim) {
std::unique_lock lock(mux);
m_core_min_cv.wait(lock, [&]() {
return lim.is_canceled() || m_state != state::is_running || !m_core_min_jobs.empty();
});
if (lim.is_canceled() || m_state != state::is_running)
return false;
unsigned best_idx = select_best_core_min_job_unlocked();
m_core_min_jobs.swap(best_idx, m_core_min_jobs.size() - 1);
core_min_job* job = m_core_min_jobs.detach_back();
m_core_min_jobs.pop_back();
SASSERT(job);
source = job->source;
core.reset();
for (expr* c : job->core)
core.push_back(g2l(c));
dealloc(job);
return source != nullptr;
}
parallel::node* parallel::batch_manager::find_core_source_unlocked(
ast_translation& l2g, node* source, expr_ref_vector const& core) {
if (!source)
return nullptr;
vector<cube_config::literal> g_core;
for (expr* c : core)
g_core.push_back(expr_ref(l2g(c), m));
for (node* cur = source; cur; cur = cur->parent()) {
if (cube_config::literal_is_null(cur->get_literal()))
continue;
if (any_of(g_core, [&](cube_config::literal const& lit) { return lit == cur->get_literal(); }))
return cur;
}
return nullptr;
}
unsigned parallel::batch_manager::select_best_core_min_job_unlocked() const {
SASSERT(!m_core_min_jobs.empty());
unsigned best_idx = 0;
node* best_source = m_core_min_jobs[0]->source;
unsigned best_depth = best_source ? best_source->depth() : 0;
unsigned best_core_size = m_core_min_jobs[0]->core.size();
for (unsigned i = 1; i < m_core_min_jobs.size(); ++i) {
core_min_job* job = m_core_min_jobs[i];
node* job_source = job->source;
unsigned job_depth = job_source ? job_source->depth() : 0;
unsigned job_core_size = job->core.size();
if (job_depth > best_depth || (job_depth == best_depth && job_core_size > best_core_size)) {
best_idx = i;
best_depth = job_depth;
best_core_size = job_core_size;
}
}
return best_idx;
}
void parallel::batch_manager::publish_minimized_core(ast_translation& l2g, expr_ref_vector const& asms, node* source,
unsigned original_core_size, expr_ref_vector const& minimized_core) {
std::scoped_lock lock(mux);
if (m_state != state::is_running || !source || minimized_core.size() >= original_core_size) {
++m_stats.m_core_min_jobs_skipped;
return;
}
vector<cube_config::literal> g_core;
for (expr* c : minimized_core)
g_core.push_back(expr_ref(l2g(c), m));
if (source->get_core().size() <= g_core.size()) {
++m_stats.m_core_min_jobs_skipped;
return;
}
IF_VERBOSE(1, verbose_stream() << "Batch manager publishing minimized core "
<< original_core_size << " -> " << g_core.size() << "\n");
if (all_of(g_core, [&](cube_config::literal const& lit) { return asms.contains(lit.get()); })) {
IF_VERBOSE(1, verbose_stream() << "Minimized core removed all path literals, setting UNSAT\n");
m_state = state::is_unsat;
SASSERT(p.ctx.m_unsat_core.empty());
for (expr* e : minimized_core)
p.ctx.m_unsat_core.push_back(l2g(e));
++m_stats.m_core_min_jobs_published;
++m_stats.m_core_min_global_unsat;
cancel_background_threads();
return;
}
m_search_tree.backtrack(source, g_core);
vector<node_lease> targets;
if (!g_core.empty()) {
collect_matching_targets_unlocked(source, g_core[0].get(), g_core, targets);
for (auto const& target : targets) {
if (!m_search_tree.is_lease_canceled(target.leased_node))
m_search_tree.backtrack(target.leased_node, g_core);
}
}
++m_stats.m_core_min_jobs_published;
cancel_closed_leases_unlocked(UINT_MAX);
IF_VERBOSE(2, m_search_tree.display(verbose_stream() << bounded_pp_exprs(minimized_core) << "\n"););
if (m_search_tree.is_closed()) {
IF_VERBOSE(1, verbose_stream() << "Search tree closed by minimized core, setting UNSAT\n");
m_state = state::is_unsat;
SASSERT(p.ctx.m_unsat_core.empty());
for (auto e : m_search_tree.get_core_from_root())
p.ctx.m_unsat_core.push_back(e);
cancel_background_threads();
}
}
void parallel::batch_manager::collect_matching_targets_unlocked(node* source, expr* lit, vector<cube_config::literal> const& core,
vector<node_lease>& targets) {
targets.reset();
if (!lit)
return;
auto is_ancestor_of = [&](node* ancestor, node* cur) {
if (!ancestor)
return false;
for (node* p = cur; p; p = p->parent()) {
if (p == ancestor)
return true;
}
return false;
};
auto path_contains = [&](node* cur, cube_config::literal const& lit) {
for (node* p = cur; p; p = p->parent()) {
if (p->get_literal() == lit)
return true;
}
return false;
};
auto path_contains_core = [&](node* cur) {
return all_of(core, [&](cube_config::literal const& c) {
return path_contains(cur, c);
});
};
ptr_vector<node> matches;
m_search_tree.find_nonclosed_nodes_with_literal(expr_ref(lit, m), matches);
for (node* t : matches) {
if (!t || t == source)
continue;
if (m_search_tree.is_lease_canceled(t))
continue;
if (source && (is_ancestor_of(source, t) || is_ancestor_of(t, source)))
continue;
if (!path_contains_core(t))
continue;
bool is_highest_ancestor = true;
for (node* p = t->parent(); p; p = p->parent()) {
if (any_of(targets, [&](node_lease const& target) { return target.leased_node == p; })) {
is_highest_ancestor = false;
break;
}
}
if (!is_highest_ancestor)
continue;
targets.push_back({t});
}
}
void parallel::batch_manager::backtrack_unlocked(ast_translation& l2g, unsigned worker_id, expr_ref_vector const& core,
node_lease* lease, vector<node_lease> const* targets) {
if (m_state != state::is_running)
return;
vector<cube_config::literal> g_core;
for (auto c : core)
g_core.push_back(expr_ref(l2g(c), m));
SASSERT(lease != nullptr || targets != nullptr);
bool did_backtrack = false;
if (lease) {
if (!m_search_tree.is_lease_canceled(lease->leased_node)) {
did_backtrack = true;
IF_VERBOSE(1, verbose_stream() << "Batch manager backtracking.\n");
node* leased_node = lease->leased_node;
release_worker_lease_unlocked(worker_id, *lease);
m_search_tree.backtrack(leased_node, g_core);
}
else {
attempt_release_canceled_lease_unlocked(worker_id, *lease);
}
}
if (targets) {
for (auto const& target : *targets) {
if (m_search_tree.is_lease_canceled(target.leased_node))
continue;
did_backtrack = true;
IF_VERBOSE(1, verbose_stream() << "Batch manager backtracking external targets.\n");
m_search_tree.backtrack(target.leased_node, g_core);
}
}
if (!did_backtrack)
return;
cancel_closed_leases_unlocked(worker_id);
IF_VERBOSE(2, m_search_tree.display(verbose_stream() << bounded_pp_exprs(core) << "\n"););
if (m_search_tree.is_closed()) {
IF_VERBOSE(1, verbose_stream() << "Search tree closed, setting UNSAT\n");
m_state = state::is_unsat;
SASSERT(p.ctx.m_unsat_core.empty());
for (auto e : m_search_tree.get_core_from_root())
p.ctx.m_unsat_core.push_back(e);
cancel_background_threads();
}
}
void parallel::batch_manager::try_split(ast_translation &l2g, unsigned worker_id,
node_lease& lease, expr *atom, unsigned effort) {
std::scoped_lock lock(mux);
if (m_state != state::is_running)
return;
if (m_search_tree.is_lease_canceled(lease.leased_node)) {
attempt_release_canceled_lease_unlocked(worker_id, lease);
return;
}
expr_ref lit(m), nlit(m);
lit = l2g(atom);
nlit = mk_not(m, lit);
node* leased_node = lease.leased_node;
VERIFY(!leased_node->path_contains_atom(lit));
VERIFY(!leased_node->path_contains_atom(nlit));
bool did_split = m_search_tree.try_split(leased_node, lit, nlit, effort);
release_worker_lease_unlocked(worker_id, lease);
if (did_split) {
++m_stats.m_num_cubes;
m_stats.m_max_cube_depth = std::max(m_stats.m_max_cube_depth, leased_node->depth() + 1);
IF_VERBOSE(1, verbose_stream() << "Batch manager splitting on literal: " << mk_bounded_pp(lit, m, 3) << "\n");
}
}
bool parallel::batch_manager::checkpoint_worker(unsigned worker_id, node_lease& lease, bool& lease_canceled) {
std::scoped_lock lock(mux);
lease_canceled = false;
SASSERT(worker_id < p.m_workers.size());
if (attempt_release_canceled_lease_unlocked(worker_id, lease)) {
lease_canceled = true;
return true;
}
if (p.m_workers[worker_id]->limit().inc())
return true;
if (attempt_release_canceled_lease_unlocked(worker_id, lease)) {
lease_canceled = true;
return true;
}
set_canceled_unlocked();
return false;
}
bool parallel::batch_manager::lease_canceled(node_lease const &lease) {
std::scoped_lock lock(mux);
return m_state == state::is_running && m_search_tree.is_lease_canceled(lease.leased_node);
}
void parallel::batch_manager::collect_clause(ast_translation &l2g, unsigned source_worker_id, expr *clause) {
std::scoped_lock lock(mux);
collect_clause_unlocked(l2g, source_worker_id, clause);
}
void parallel::batch_manager::collect_clause_unlocked(ast_translation &l2g, unsigned source_worker_id, expr *clause) {
expr *g_clause = l2g(clause);
if (!shared_clause_set.contains(g_clause)) {
shared_clause_set.insert(g_clause);
shared_clause sc{source_worker_id, expr_ref(g_clause, m)};
shared_clause_trail.push_back(sc);
}
}
void parallel::worker::collect_shared_clauses() {
expr_ref_vector new_clauses = b.return_shared_clauses(m_g2l, m_shared_clause_limit, id);
for (expr *e : new_clauses) {
ctx->assert_expr(e);
LOG_WORKER(4, " asserting shared clause: " << mk_bounded_pp(e, m, 3) << "\n");
}
}
void parallel::backbones_worker::collect_shared_clauses() {
expr_ref_vector new_clauses = b.return_shared_clauses(m_g2l, m_shared_clause_limit, UINT_MAX);
for (expr *e : new_clauses) {
ctx->assert_expr(e);
LOG_BB_WORKER(4, " asserting shared clause: " << mk_bounded_pp(e, m, 3) << "\n");
}
}
void parallel::core_minimizer_worker::collect_shared_clauses() {
expr_ref_vector new_clauses = b.return_shared_clauses(m_g2l, m_shared_clause_limit, UINT_MAX);
for (expr *e : new_clauses) {
ctx->assert_expr(e);
IF_VERBOSE(4, verbose_stream() << "Core minimizer asserting shared clause: "
<< mk_bounded_pp(e, m, 3) << "\n";);
}
}
void parallel::batch_manager::collect_backbone_candidates(ast_translation& l2g, bb_candidates& bb_candidates) {
std::scoped_lock lock(mux);
bool changed = false;
auto find_existing_candidate_idx = [&](expr* e) -> int {
for (unsigned i = 0; i < m_bb_candidates.size(); ++i) {
if (m_bb_candidates[i].lit.get() == e)
return i;
}
return -1;
};
auto rank_of = [&](bb_candidate const& c) {
return c.age * std::log2(2.0 + c.hits);
};
for (auto const& c : bb_candidates) {
expr_ref g_lit(l2g(c.lit.get()), m);
if (is_global_backbone_or_negation_unlocked(l2g, c.lit))
continue;
double age = c.age;
int idx = find_existing_candidate_idx(g_lit.get());
if (idx >= 0) {
auto& existing = m_bb_candidates[idx];
existing.age = (existing.age * existing.hits + age) / (existing.hits + 1);
existing.hits++;
continue;
}
if (m_bb_candidates.size() < m_max_global_bb_candidates) {
m_bb_candidates.push_back(bb_candidate(m, g_lit.get(), age, 1));
changed = true;
continue;
}
bb_candidate new_bb_candidate = bb_candidate(m, g_lit.get(), age, 1);
auto worst_it = std::min_element(
m_bb_candidates.begin(),
m_bb_candidates.end(),
[&](bb_candidate const& a, bb_candidate const& b) {
return rank_of(a) < rank_of(b);
}
);
if (worst_it != m_bb_candidates.end() && rank_of(new_bb_candidate) > rank_of(*worst_it)) {
*worst_it = new_bb_candidate; changed = true;
}
}
if (changed && !m_bb_candidates.empty()) {
m_bb_candidate_epoch.fetch_add(1, std::memory_order_release);
std::sort(
m_bb_candidates.begin(),
m_bb_candidates.end(),
[&](bb_candidate const& a, bb_candidate const& b) {
return rank_of(a) < rank_of(b); }
);
m_bb_cv.notify_all();
}
}
bool parallel::batch_manager::wait_for_backbone_job(unsigned bb_thread_id, ast_translation& g2l, bb_candidates& out, reslimit& lim) {
out.reset();
std::unique_lock<std::mutex> lock(mux);
m_bb_cv.wait(lock, [&]() {
return lim.is_canceled() ||
m_state != state::is_running ||
m_bb_last_batch_processed[bb_thread_id] < m_bb_batch_id ||
!m_bb_candidates.empty();
});
if (lim.is_canceled())
return false;
if (m_state != state::is_running)
return false;
if (m_bb_last_batch_processed[bb_thread_id] == m_bb_batch_id) {
unsigned n = std::min<unsigned>(m_bb_batch_size, m_bb_candidates.size());
m_bb_current_batch.reset();
for (unsigned i = 0; i < n; ++i) {
m_bb_current_batch.push_back(m_bb_candidates.back());
m_bb_candidates.pop_back();
}
m_bb_batch_id++;
m_bb_cv.notify_all();
}
for (auto const& gc : m_bb_current_batch) {
expr_ref l_lit(g2l(gc.lit.get()), g2l.to());
out.push_back(bb_candidate(g2l.to(), l_lit, gc.age, gc.hits));
}
m_bb_last_batch_processed[bb_thread_id] = m_bb_batch_id;
return true;
}
expr_ref_vector parallel::batch_manager::return_shared_clauses(ast_translation &g2l, unsigned &worker_limit,
unsigned worker_id) {
std::scoped_lock lock(mux);
expr_ref_vector result(g2l.to());
for (unsigned i = worker_limit; i < shared_clause_trail.size(); ++i) {
if (shared_clause_trail[i].source_worker_id != worker_id)
result.push_back(g2l(shared_clause_trail[i].clause.get()));
}
worker_limit = shared_clause_trail.size(); return result;
}
lbool parallel::worker::check_cube(expr_ref_vector const &cube) {
for (auto &atom : cube)
asms.push_back(atom);
ctx->get_fparams().m_max_conflicts = std::min(m_config.m_threads_max_conflicts, m_config.m_max_conflicts);
IF_VERBOSE(1, verbose_stream() << " Checking cube\n"
<< bounded_pp_exprs(cube)
<< "with max_conflicts: " << ctx->get_fparams().m_max_conflicts << "\n";);
lbool r = b.check(asms, *ctx);
asms.shrink(asms.size() - cube.size());
LOG_WORKER(1, " DONE checking cube " << r << "\n";);
return r;
}
expr_ref parallel::worker::get_split_atom() {
expr_ref result(m);
double score = 0;
unsigned n = 0;
ctx->pop_to_search_lvl();
for (bool_var v = 0; v < ctx->get_num_bool_vars(); ++v) {
if (ctx->get_assignment(v) != l_undef)
continue;
expr *e = ctx->bool_var2expr(v);
if (!e)
continue;
if (m_config.m_global_backbones) {
if (b.is_global_backbone_or_negation(m_l2g, e))
continue;
}
double new_score = ctx->get_activity(v);
if (new_score > score || !result || (new_score == score && m_rand(++n) == 0)) {
score = new_score;
result = e;
}
}
return result;
}
void parallel::batch_manager::set_sat(ast_translation &l2g, model &m) {
std::scoped_lock lock(mux);
IF_VERBOSE(1, verbose_stream() << "Batch manager setting SAT.\n");
if (m_state != state::is_running && m_state != state::is_unknown)
return;
m_state = state::is_sat;
p.ctx.set_model(m.translate(l2g));
cancel_background_threads();
}
void parallel::batch_manager::set_unsat(ast_translation &l2g, expr_ref_vector const &unsat_core) {
std::scoped_lock lock(mux);
IF_VERBOSE(1, verbose_stream() << "Batch manager setting UNSAT.\n");
if (m_state != state::is_running && m_state != state::is_unknown)
return;
m_state = state::is_unsat;
SASSERT(p.ctx.m_unsat_core.empty());
for (expr *e : unsat_core)
p.ctx.m_unsat_core.push_back(l2g(e));
cancel_background_threads();
}
void parallel::batch_manager::set_unknown(std::string const &reason) {
std::scoped_lock lock(mux);
IF_VERBOSE(1, verbose_stream() << "Batch manager setting UNKNOWN: " << reason << ".\n");
if (m_state != state::is_running)
return; m_state = state::is_unknown;
m_reason_unknown = reason;
cancel_background_threads();
}
void parallel::batch_manager::set_exception(unsigned error_code) {
std::scoped_lock lock(mux);
IF_VERBOSE(1, verbose_stream() << "Batch manager setting exception code: " << error_code << ".\n");
if (m_state != state::is_running)
return;
m_state = state::is_exception_code;
m_exception_code = error_code;
cancel_background_threads();
}
void parallel::batch_manager::set_exception(std::string const &msg) {
std::scoped_lock lock(mux);
IF_VERBOSE(1, verbose_stream() << "Batch manager setting exception msg: " << msg << ".\n");
if (m_state != state::is_running)
return;
m_state = state::is_exception_msg;
m_exception_msg = msg;
cancel_background_threads();
}
lbool parallel::batch_manager::get_result() const {
if (m.limit().is_canceled())
return l_undef; switch (m_state) {
case state::is_running: throw default_exception("inconsistent end state");
case state::is_unsat:
return l_false;
case state::is_sat:
return l_true;
case state::is_unknown:
return l_undef;
case state::is_exception_msg:
throw default_exception(m_exception_msg.c_str());
case state::is_exception_code:
throw z3_error(m_exception_code);
default:
UNREACHABLE();
return l_undef;
}
}
bool parallel::batch_manager::get_cube(ast_translation &g2l, unsigned id, expr_ref_vector &cube, bool is_first_run, node_lease &lease) {
std::scoped_lock lock(mux);
cube.reset();
if (m_search_tree.is_closed()) {
IF_VERBOSE(1, verbose_stream() << "all done\n";);
return false;
}
if (m_state != state::is_running) {
IF_VERBOSE(1, verbose_stream() << "aborting get_cube\n";);
return false;
}
node *t = is_first_run ? m_search_tree.activate_root() : m_search_tree.activate_best_node();
if (!t)
return false;
IF_VERBOSE(2, m_search_tree.display(verbose_stream()); verbose_stream() << "\n";);
lease.leased_node = t;
if (id >= m_worker_leases.size())
m_worker_leases.resize(id + 1);
m_worker_leases[id] = lease;
while (t) {
if (cube_config::literal_is_null(t->get_literal()))
break;
expr_ref lit(g2l.to());
lit = g2l(t->get_literal().get());
cube.push_back(std::move(lit));
t = t->parent();
}
return true;
}
void parallel::batch_manager::initialize(unsigned num_global_bb_threads, unsigned initial_max_thread_conflicts) {
m_state = state::is_running;
m_num_global_bb_threads = num_global_bb_threads;
m_bb_last_batch_processed.reset();
m_bb_last_batch_processed.resize(m_num_global_bb_threads);
m_bb_candidates.reset();
m_global_backbones.reset();
m_bb_candidate_epoch.store(0, std::memory_order_release);
m_core_min_jobs.reset();
m_search_tree.reset();
m_search_tree.set_effort_unit(initial_max_thread_conflicts);
m_worker_leases.reset();
m_worker_leases.resize(p.m_workers.size());
parallel_params pp(p.ctx.m_params);
m_ablate_backtracking = pp.ablate_backtracking();
m_canceled = false;
}
void parallel::batch_manager::collect_statistics(::statistics &st) const {
st.update("parallel-num_cubes", m_stats.m_num_cubes);
st.update("parallel-max-cube-size", m_stats.m_max_cube_depth);
st.update("bb-backbones-found", m_stats.m_backbones_found);
st.update("parallel-core-min-jobs-enqueued", m_stats.m_core_min_jobs_enqueued);
st.update("parallel-core-min-jobs-published", m_stats.m_core_min_jobs_published);
st.update("parallel-core-min-jobs-skipped", m_stats.m_core_min_jobs_skipped);
st.update("parallel-core-min-global-unsat", m_stats.m_core_min_global_unsat);
}
lbool parallel::operator()(expr_ref_vector const &asms) {
parallel_params pp(ctx.m_params);
unsigned num_global_bb_threads = pp.num_bb_threads();
if (num_global_bb_threads > 2)
throw default_exception("parallel.num_bb_threads must be 0, 1, or 2");
unsigned total_threads = std::min((unsigned)std::thread::hardware_concurrency(), ctx.get_fparams().m_threads);
unsigned num_workers = total_threads;
unsigned num_sls_threads = 0;
unsigned num_core_min_threads = (pp.core_minimize() ? 1 : 0);
if (num_workers > 2 + num_core_min_threads)
num_workers -= num_core_min_threads;
else
num_core_min_threads = 0;
if (num_workers > 2 + num_global_bb_threads)
num_workers -= num_global_bb_threads;
else
num_global_bb_threads = 0;
if (num_workers > 2 + num_sls_threads)
num_workers -= num_sls_threads;
else
num_sls_threads = 0;
IF_VERBOSE(1, verbose_stream() << "Parallel SMT with " << total_threads << " threads\n";);
ast_manager &m = ctx.m;
if (m.has_trace_stream())
throw default_exception("trace streams have to be off in parallel mode");
struct scoped_clear {
parallel &p;
scoped_clear(parallel &p) : p(p) {}
~scoped_clear() {
p.m_workers.reset();
p.m_sls_worker = nullptr;
p.m_core_minimizer_worker = nullptr;
p.m_global_backbones_workers.reset();
}
};
scoped_clear clear(*this);
m_workers.reset();
m_core_minimizer_worker = nullptr;
scoped_limits sl(m.limit());
flet<unsigned> _nt(ctx.m_fparams.m_threads, 1);
SASSERT(num_workers > 1);
for (unsigned i = 0; i < num_workers; ++i)
m_workers.push_back(alloc(worker, i, *this, asms));
for (auto w : m_workers)
sl.push_child(&(w->limit()));
if (num_sls_threads == 1) {
m_sls_worker = alloc(sls_worker, *this);
sl.push_child(&(m_sls_worker->limit()));
}
if (num_core_min_threads == 1) {
m_core_minimizer_worker = alloc(core_minimizer_worker, *this, asms);
sl.push_child(&(m_core_minimizer_worker->limit()));
}
for (unsigned i = 0; i < num_global_bb_threads; ++i) {
auto *w = alloc(backbones_worker, i, *this, asms);
m_global_backbones_workers.push_back(w);
sl.push_child(&(w->limit()));
}
IF_VERBOSE(1, verbose_stream() << "Launched " << m_workers.size() << " CDCL threads, "
<< (m_sls_worker ? 1 : 0) << " SLS threads, "
<< (m_core_minimizer_worker ? 1 : 0) << " core minimizer threads, "
<< m_global_backbones_workers.size() << " global backbone threads.\n";);
m_batch_manager.initialize(num_global_bb_threads);
auto safe_run = [&](auto&& run_fn, reslimit& lim) {
try {
run_fn();
if (lim.is_canceled())
m_batch_manager.set_canceled();
} catch (z3_error &err) {
IF_VERBOSE(0, verbose_stream() << "Exception in parallel solver: " << err.what() << "\n");
if (!lim.is_canceled())
m_batch_manager.set_exception(err.error_code());
else
m_batch_manager.set_canceled();
} catch (z3_exception &ex) {
IF_VERBOSE(0, verbose_stream() << "Exception in parallel solver: " << ex.what() << "\n");
if (!lim.is_canceled() && !is_cancellation_exception(ex.what()))
m_batch_manager.set_exception(ex.what());
else
m_batch_manager.set_canceled();
} catch (...) {
IF_VERBOSE(0, verbose_stream() << "Unknown exception in parallel solver\n");
if (!lim.is_canceled())
m_batch_manager.set_exception("unknown exception");
else
m_batch_manager.set_canceled();
}
};
vector<std::thread> threads(total_threads);
unsigned thread_idx = 0;
for (auto* w : m_workers)
threads[thread_idx++] = std::thread([w, &safe_run]() {
safe_run([w]() { w->run(); }, w->limit());
});
if (m_sls_worker)
threads[thread_idx++] = std::thread([this, &safe_run]() {
safe_run([this]() { m_sls_worker->run(); }, m_sls_worker->limit());
});
if (m_core_minimizer_worker)
threads[thread_idx++] = std::thread([this, &safe_run]() {
safe_run([this]() { m_core_minimizer_worker->run(); }, m_core_minimizer_worker->limit());
});
for (auto* w : m_global_backbones_workers)
threads[thread_idx++] = std::thread([w, &safe_run]() {
safe_run([w]() { w->run(); }, w->limit());
});
for (auto &th : threads)
th.join();
for (auto w : m_workers)
w->collect_statistics(ctx.m_aux_stats);
m_batch_manager.collect_statistics(ctx.m_aux_stats);
if (m_sls_worker)
m_sls_worker->collect_statistics(ctx.m_aux_stats);
if (m_core_minimizer_worker)
m_core_minimizer_worker->collect_statistics(ctx.m_aux_stats);
for (auto* bb_w : m_global_backbones_workers)
bb_w->collect_statistics(ctx.m_aux_stats);
lbool result = m_batch_manager.get_result();
if (result == l_undef && !m_batch_manager.get_reason_unknown().empty())
ctx.set_reason_unknown(m_batch_manager.get_reason_unknown().c_str());
return result;
}
} #endif