2020-06-29 17:06:43 -05:00
|
|
|
/* -*- c++ -*-
|
2020-03-23 02:42:48 -05:00
|
|
|
* yosys -- Yosys Open SYnthesis Suite
|
|
|
|
*
|
|
|
|
* Copyright (C) 2020 Alberto Gonzalez <boqwxp@airmail.cc>
|
|
|
|
*
|
|
|
|
* Permission to use, copy, modify, and/or distribute this software for any
|
|
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
|
|
* copyright notice and this permission notice appear in all copies.
|
|
|
|
*
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
2020-03-24 20:58:41 -05:00
|
|
|
#include "kernel/yosys.h"
|
2020-04-24 23:12:02 -05:00
|
|
|
#include "kernel/consteval.h"
|
2020-06-29 17:06:43 -05:00
|
|
|
#include "qbfsat.h"
|
2020-03-24 20:58:41 -05:00
|
|
|
|
2020-03-23 02:42:48 -05:00
|
|
|
USING_YOSYS_NAMESPACE
|
|
|
|
PRIVATE_NAMESPACE_BEGIN
|
|
|
|
|
2020-05-26 18:12:15 -05:00
|
|
|
static inline unsigned int difference(unsigned int a, unsigned int b) {
|
|
|
|
if (a < b)
|
|
|
|
return b - a;
|
|
|
|
else
|
|
|
|
return a - b;
|
|
|
|
}
|
|
|
|
|
2020-06-29 17:06:43 -05:00
|
|
|
pool<std::string> validate_design_and_get_inputs(RTLIL::Module *module, bool assume_outputs) {
|
2020-04-24 23:12:02 -05:00
|
|
|
bool found_input = false;
|
|
|
|
bool found_hole = false;
|
|
|
|
bool found_1bit_output = false;
|
|
|
|
bool found_assert_assume = false;
|
|
|
|
pool<std::string> input_wires;
|
|
|
|
for (auto wire : module->wires()) {
|
|
|
|
if (wire->port_input) {
|
|
|
|
found_input = true;
|
|
|
|
input_wires.insert(wire->name.str());
|
|
|
|
}
|
|
|
|
if (wire->port_output && wire->width == 1)
|
|
|
|
found_1bit_output = true;
|
|
|
|
}
|
|
|
|
for (auto cell : module->cells()) {
|
|
|
|
if (cell->type == "$allconst")
|
|
|
|
found_input = true;
|
|
|
|
if (cell->type == "$anyconst")
|
|
|
|
found_hole = true;
|
|
|
|
if (cell->type.in("$assert", "$assume"))
|
|
|
|
found_assert_assume = true;
|
|
|
|
}
|
|
|
|
if (!found_input)
|
|
|
|
log_cmd_error("Can't perform QBF-SAT on a miter with no inputs!\n");
|
|
|
|
if (!found_hole)
|
|
|
|
log_cmd_error("Did not find any existentially-quantified variables. Use 'sat' instead.\n");
|
|
|
|
if (!found_1bit_output && !found_assert_assume)
|
|
|
|
log_cmd_error("Did not find any single-bit outputs or $assert/$assume cells. Is this a miter circuit?\n");
|
2020-06-29 17:06:43 -05:00
|
|
|
if (!found_assert_assume && !assume_outputs)
|
2020-04-24 23:12:02 -05:00
|
|
|
log_cmd_error("Did not find any $assert/$assume cells. Single-bit outputs were found, but `-assume-outputs` was not specified.\n");
|
|
|
|
|
|
|
|
return input_wires;
|
|
|
|
}
|
|
|
|
|
2020-03-25 20:02:53 -05:00
|
|
|
void specialize_from_file(RTLIL::Module *module, const std::string &file) {
|
2020-06-09 16:44:45 -05:00
|
|
|
YS_REGEX_TYPE hole_bit_assn_regex = YS_REGEX_COMPILE_WITH_SUBS("^(.+) ([0-9]+) ([^ ]+) \\[([0-9]+)] = ([01])$");
|
|
|
|
YS_REGEX_TYPE hole_assn_regex = YS_REGEX_COMPILE_WITH_SUBS("^(.+) ([0-9]+) ([^ ]+) = ([01])$"); //if no index specified
|
2020-06-09 00:27:09 -05:00
|
|
|
YS_REGEX_MATCH_TYPE bit_m, m;
|
2020-05-21 23:48:33 -05:00
|
|
|
dict<pool<std::string>, RTLIL::Cell*> anyconst_loc_to_cell;
|
|
|
|
dict<RTLIL::SigBit, RTLIL::State> hole_assignments;
|
|
|
|
|
|
|
|
for (auto cell : module->cells())
|
|
|
|
if (cell->type == "$anyconst")
|
|
|
|
anyconst_loc_to_cell[cell->get_strpool_attribute(ID::src)] = cell;
|
|
|
|
|
2020-03-25 20:02:53 -05:00
|
|
|
std::ifstream fin(file.c_str());
|
|
|
|
if (!fin)
|
|
|
|
log_cmd_error("could not read solution file.\n");
|
|
|
|
|
|
|
|
std::string buf;
|
|
|
|
while (std::getline(fin, buf)) {
|
2020-06-09 16:44:45 -05:00
|
|
|
bool bit_assn = true;
|
2020-06-09 00:27:09 -05:00
|
|
|
if (!YS_REGEX_NS::regex_search(buf, bit_m, hole_bit_assn_regex)) {
|
2020-06-09 16:44:45 -05:00
|
|
|
bit_assn = false;
|
|
|
|
if (!YS_REGEX_NS::regex_search(buf, m, hole_assn_regex))
|
2020-06-09 00:27:09 -05:00
|
|
|
log_cmd_error("solution file is not formatted correctly: \"%s\"\n", buf.c_str());
|
|
|
|
}
|
2020-04-06 22:29:54 -05:00
|
|
|
|
2020-06-09 16:44:45 -05:00
|
|
|
std::string hole_loc = bit_assn? bit_m[1].str() : m[1].str();
|
|
|
|
unsigned int hole_bit = bit_assn? atoi(bit_m[2].str().c_str()) : atoi(m[2].str().c_str());
|
|
|
|
std::string hole_name = bit_assn? bit_m[3].str() : m[3].str();
|
|
|
|
unsigned int hole_offset = bit_assn? atoi(bit_m[4].str().c_str()) : 0;
|
|
|
|
RTLIL::State hole_value = bit_assn? (atoi(bit_m[5].str().c_str()) == 1? RTLIL::State::S1 : RTLIL::State::S0)
|
|
|
|
: (atoi(m[4].str().c_str()) == 1? RTLIL::State::S1 : RTLIL::State::S0);
|
|
|
|
|
2020-05-21 23:48:33 -05:00
|
|
|
//We have two options to identify holes. First, try to match wire names. If we can't find a matching wire,
|
|
|
|
//then try to find a cell with a matching location.
|
|
|
|
RTLIL::SigBit hole_sigbit;
|
|
|
|
if (module->wire(hole_name) != nullptr) {
|
|
|
|
RTLIL::Wire *hole_wire = module->wire(hole_name);
|
|
|
|
hole_sigbit = RTLIL::SigSpec(hole_wire)[hole_offset];
|
|
|
|
} else {
|
2020-06-09 16:31:58 -05:00
|
|
|
auto locs = split_tokens(hole_loc, "|");
|
|
|
|
pool<std::string> hole_loc_pool(locs.begin(), locs.end());
|
2020-05-21 23:48:33 -05:00
|
|
|
auto hole_cell_it = anyconst_loc_to_cell.find(hole_loc_pool);
|
|
|
|
if (hole_cell_it == anyconst_loc_to_cell.end())
|
2020-06-29 20:53:21 -05:00
|
|
|
log_cmd_error("cannot find matching wire name or $anyconst cell location for hole spec \"%s\"\n", buf.c_str());
|
2020-05-21 23:48:33 -05:00
|
|
|
|
|
|
|
RTLIL::Cell *hole_cell = hole_cell_it->second;
|
|
|
|
hole_sigbit = hole_cell->getPort(ID::Y)[hole_bit];
|
2020-04-06 22:29:54 -05:00
|
|
|
}
|
2020-05-21 23:48:33 -05:00
|
|
|
hole_assignments[hole_sigbit] = hole_value;
|
|
|
|
}
|
2020-03-25 20:02:53 -05:00
|
|
|
|
2020-05-21 23:48:33 -05:00
|
|
|
for (auto &it : anyconst_loc_to_cell)
|
|
|
|
module->remove(it.second);
|
2020-03-25 20:02:53 -05:00
|
|
|
|
2020-05-21 23:48:33 -05:00
|
|
|
for (auto &it : hole_assignments) {
|
|
|
|
RTLIL::SigSpec lhs(it.first);
|
|
|
|
RTLIL::SigSpec rhs(it.second);
|
2020-06-09 00:27:09 -05:00
|
|
|
log("Specializing %s from file with %s = %d.\n", module->name.c_str(), log_signal(it.first), it.second == RTLIL::State::S1? 1 : 0);
|
2020-05-21 23:48:33 -05:00
|
|
|
module->connect(lhs, rhs);
|
2020-03-25 20:02:53 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-24 23:12:02 -05:00
|
|
|
void specialize(RTLIL::Module *module, const QbfSolutionType &sol, bool quiet = false) {
|
2020-06-29 17:06:43 -05:00
|
|
|
auto hole_loc_idx_to_sigbit = sol.get_hole_loc_idx_sigbit_map(module);
|
2020-04-11 04:41:09 -05:00
|
|
|
pool<RTLIL::Cell *> anyconsts_to_remove;
|
2020-04-06 22:29:54 -05:00
|
|
|
for (auto cell : module->cells())
|
|
|
|
if (cell->type == "$anyconst")
|
2020-05-21 23:48:33 -05:00
|
|
|
if (hole_loc_idx_to_sigbit.find(std::make_pair(cell->get_strpool_attribute(ID::src), 0)) != hole_loc_idx_to_sigbit.end())
|
2020-04-06 22:29:54 -05:00
|
|
|
anyconsts_to_remove.insert(cell);
|
|
|
|
for (auto cell : anyconsts_to_remove)
|
|
|
|
module->remove(cell);
|
2020-03-25 18:51:32 -05:00
|
|
|
for (auto &it : sol.hole_to_value) {
|
2020-05-21 23:48:33 -05:00
|
|
|
pool<std::string> hole_loc = it.first;
|
2020-03-25 18:17:50 -05:00
|
|
|
std::string hole_value = it.second;
|
|
|
|
|
2020-05-21 23:48:33 -05:00
|
|
|
for (unsigned int i = 0; i < hole_value.size(); ++i) {
|
|
|
|
int bit_idx = GetSize(hole_value) - 1 - i;
|
|
|
|
auto it = hole_loc_idx_to_sigbit.find(std::make_pair(hole_loc, i));
|
|
|
|
log_assert(it != hole_loc_idx_to_sigbit.end());
|
2020-03-25 18:17:50 -05:00
|
|
|
|
2020-05-21 23:48:33 -05:00
|
|
|
RTLIL::SigBit hole_sigbit = it->second;
|
|
|
|
log_assert(hole_sigbit.wire != nullptr);
|
|
|
|
log_assert(hole_value[bit_idx] == '0' || hole_value[bit_idx] == '1');
|
|
|
|
RTLIL::SigSpec lhs(hole_sigbit.wire, hole_sigbit.offset, 1);
|
|
|
|
RTLIL::State hole_bit_val = hole_value[bit_idx] == '1'? RTLIL::State::S1 : RTLIL::State::S0;
|
|
|
|
if (!quiet)
|
2020-06-09 00:27:09 -05:00
|
|
|
log("Specializing %s with %s = %d.\n", module->name.c_str(), log_signal(hole_sigbit), hole_bit_val == RTLIL::State::S0? 0 : 1)
|
2020-05-21 23:48:33 -05:00
|
|
|
;
|
|
|
|
module->connect(lhs, hole_bit_val);
|
|
|
|
}
|
2020-03-25 18:17:50 -05:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2020-04-11 04:41:09 -05:00
|
|
|
void allconstify_inputs(RTLIL::Module *module, const pool<std::string> &input_wires) {
|
2020-03-26 20:32:53 -05:00
|
|
|
for (auto &n : input_wires) {
|
2020-03-25 18:37:49 -05:00
|
|
|
RTLIL::Wire *input = module->wire(n);
|
|
|
|
log_assert(input != nullptr);
|
|
|
|
|
|
|
|
RTLIL::Cell *allconst = module->addCell("$allconst$" + n, "$allconst");
|
|
|
|
allconst->setParam(ID(WIDTH), input->width);
|
|
|
|
allconst->setPort(ID::Y, input);
|
|
|
|
allconst->set_src_attribute(input->get_src_attribute());
|
|
|
|
input->port_input = false;
|
|
|
|
log("Replaced input %s with $allconst cell.\n", n.c_str());
|
|
|
|
}
|
|
|
|
module->fixup_ports();
|
|
|
|
}
|
|
|
|
|
2020-06-29 17:06:43 -05:00
|
|
|
void assume_miter_outputs(RTLIL::Module *module, bool assume_neg) {
|
2020-03-26 20:32:53 -05:00
|
|
|
std::vector<RTLIL::Wire *> wires_to_assume;
|
|
|
|
for (auto w : module->wires())
|
2020-03-27 18:25:24 -05:00
|
|
|
if (w->port_output && w->width == 1)
|
|
|
|
wires_to_assume.push_back(w);
|
|
|
|
|
2020-03-26 20:32:53 -05:00
|
|
|
if (wires_to_assume.size() == 0)
|
|
|
|
return;
|
|
|
|
else {
|
2020-04-01 17:19:24 -05:00
|
|
|
log("Adding $assume cell for output(s): ");
|
2020-03-26 20:32:53 -05:00
|
|
|
for (auto w : wires_to_assume)
|
|
|
|
log("\"%s\" ", w->name.c_str());
|
|
|
|
log("\n");
|
|
|
|
}
|
|
|
|
|
2020-06-29 17:06:43 -05:00
|
|
|
if (assume_neg) {
|
2020-04-22 23:06:15 -05:00
|
|
|
for (unsigned int i = 0; i < wires_to_assume.size(); ++i) {
|
|
|
|
RTLIL::SigSpec n_wire = module->LogicNot(wires_to_assume[i]->name.str() + "__n__qbfsat", wires_to_assume[i], false, wires_to_assume[i]->get_src_attribute());
|
|
|
|
wires_to_assume[i] = n_wire.as_wire();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
for (auto i = 0; wires_to_assume.size() > 1; ++i) {
|
2020-03-26 20:32:53 -05:00
|
|
|
std::vector<RTLIL::Wire *> buf;
|
2020-03-27 18:25:24 -05:00
|
|
|
for (auto j = 0; j + 1 < GetSize(wires_to_assume); j += 2) {
|
|
|
|
std::stringstream strstr; strstr << i << "_" << j;
|
2020-03-26 20:32:53 -05:00
|
|
|
RTLIL::Wire *and_wire = module->addWire("\\_qbfsat_and_" + strstr.str(), 1);
|
2020-03-27 18:25:24 -05:00
|
|
|
module->addLogicAnd("$_qbfsat_and_" + strstr.str(), wires_to_assume[j], wires_to_assume[j+1], and_wire, false, wires_to_assume[j]->get_src_attribute());
|
2020-03-26 20:32:53 -05:00
|
|
|
buf.push_back(and_wire);
|
|
|
|
}
|
2020-03-27 18:25:24 -05:00
|
|
|
if (wires_to_assume.size() % 2 == 1)
|
|
|
|
buf.push_back(wires_to_assume[wires_to_assume.size() - 1]);
|
2020-03-26 20:32:53 -05:00
|
|
|
wires_to_assume.swap(buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
log_assert(wires_to_assume.size() == 1);
|
|
|
|
module->addAssume("$assume_qbfsat_miter_outputs", wires_to_assume[0], RTLIL::S1);
|
|
|
|
}
|
|
|
|
|
2020-04-24 23:12:02 -05:00
|
|
|
QbfSolutionType call_qbf_solver(RTLIL::Module *mod, const QbfSolveOptions &opt, const std::string &tempdir_name, const bool quiet = false, const int iter_num = 0) {
|
|
|
|
//Execute and capture stdout from `yosys-smtbmc -s z3 -t 1 -g --binary [--dump-smt2 <file>]`
|
2020-03-23 02:42:48 -05:00
|
|
|
QbfSolutionType ret;
|
2020-03-27 17:03:06 -05:00
|
|
|
const std::string yosys_smtbmc_exe = proc_self_dirname() + "yosys-smtbmc";
|
2020-06-29 18:01:56 -05:00
|
|
|
const std::string smt2_command = stringf("write_smt2 -stbv -wires %s/problem%d.smt2", tempdir_name.c_str(), iter_num);
|
2020-03-27 17:03:06 -05:00
|
|
|
const std::string smtbmc_warning = "z3: WARNING:";
|
2020-06-29 18:01:56 -05:00
|
|
|
const std::string smtbmc_cmd = stringf("%s -s %s %s -t 1 -g --binary %s %s/problem%d.smt2 2>&1",
|
|
|
|
yosys_smtbmc_exe.c_str(), opt.get_solver_name().c_str(),
|
|
|
|
(opt.timeout != 0? stringf("--timeout %d", opt.timeout) : "").c_str(),
|
|
|
|
(opt.dump_final_smt2? "--dump-smt2 " + opt.dump_final_smt2_file : "").c_str(),
|
|
|
|
tempdir_name.c_str(), iter_num);
|
2020-03-24 20:58:41 -05:00
|
|
|
|
|
|
|
Pass::call(mod->design, smt2_command);
|
|
|
|
|
2020-04-30 16:27:18 -05:00
|
|
|
auto process_line = [&ret, &smtbmc_warning, &opt, &quiet](const std::string &line) {
|
2020-04-24 23:12:02 -05:00
|
|
|
ret.stdout_lines.push_back(line.substr(0, line.size()-1)); //don't include trailing newline
|
|
|
|
auto warning_pos = line.find(smtbmc_warning);
|
|
|
|
if (warning_pos != std::string::npos)
|
|
|
|
log_warning("%s", line.substr(warning_pos + smtbmc_warning.size() + 1).c_str());
|
|
|
|
else
|
2020-04-30 16:27:18 -05:00
|
|
|
if (opt.show_smtbmc && !quiet)
|
2020-04-24 23:12:02 -05:00
|
|
|
log("smtbmc output: %s", line.c_str());
|
|
|
|
};
|
|
|
|
log_header(mod->design, "Solving QBF-SAT problem.\n");
|
|
|
|
if (!quiet) log("Launching \"%s\".\n", smtbmc_cmd.c_str());
|
2020-06-07 02:45:24 -05:00
|
|
|
int64_t begin = PerformanceTimer::query();
|
2020-05-01 18:17:35 -05:00
|
|
|
run_command(smtbmc_cmd, process_line);
|
2020-06-07 02:45:24 -05:00
|
|
|
int64_t end = PerformanceTimer::query();
|
|
|
|
ret.solver_time = (end - begin) / 1e9f;
|
|
|
|
if (!quiet) log("Solver finished in %.3f seconds.\n", ret.solver_time);
|
2020-03-23 02:42:48 -05:00
|
|
|
|
2020-06-29 17:06:43 -05:00
|
|
|
ret.recover_solution();
|
2020-03-23 02:42:48 -05:00
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
2020-04-24 23:12:02 -05:00
|
|
|
QbfSolutionType qbf_solve(RTLIL::Module *mod, const QbfSolveOptions &opt) {
|
|
|
|
QbfSolutionType ret, best_soln;
|
2020-06-29 18:01:56 -05:00
|
|
|
const std::string tempdir_name = make_temp_dir("/tmp/yosys-qbfsat-XXXXXX");
|
2020-04-24 23:12:02 -05:00
|
|
|
RTLIL::Module *module = mod;
|
|
|
|
RTLIL::Design *design = module->design;
|
|
|
|
std::string module_name = module->name.str();
|
2020-06-30 00:47:03 -05:00
|
|
|
RTLIL::IdString wire_to_optimize_name = "";
|
2020-04-24 23:12:02 -05:00
|
|
|
bool maximize = false;
|
|
|
|
log_assert(module->design != nullptr);
|
|
|
|
|
|
|
|
Pass::call(design, "design -push-copy");
|
|
|
|
|
|
|
|
//Replace input wires with wires assigned $allconst cells:
|
2020-06-29 17:06:43 -05:00
|
|
|
pool<std::string> input_wires = validate_design_and_get_inputs(module, opt.assume_outputs);
|
2020-04-24 23:12:02 -05:00
|
|
|
allconstify_inputs(module, input_wires);
|
|
|
|
if (opt.assume_outputs)
|
2020-06-29 17:06:43 -05:00
|
|
|
assume_miter_outputs(module, opt.assume_neg);
|
2020-04-24 23:12:02 -05:00
|
|
|
|
|
|
|
//Find the wire to be optimized, if any:
|
2020-06-30 00:47:03 -05:00
|
|
|
for (auto wire : module->wires()) {
|
|
|
|
if (wire->get_bool_attribute("\\maximize") || wire->get_bool_attribute("\\minimize")) {
|
|
|
|
wire_to_optimize_name = wire->name;
|
|
|
|
maximize = wire->get_bool_attribute("\\maximize");
|
|
|
|
if (opt.nooptimize) {
|
|
|
|
if (maximize)
|
|
|
|
wire->set_bool_attribute("\\maximize", false);
|
|
|
|
else
|
|
|
|
wire->set_bool_attribute("\\minimize", false);
|
|
|
|
}
|
|
|
|
}
|
2020-03-25 18:37:49 -05:00
|
|
|
}
|
2020-04-24 23:12:02 -05:00
|
|
|
|
2020-06-30 00:47:03 -05:00
|
|
|
//If -O1 or -O2 was specified, use ABC to simplify the problem:
|
|
|
|
if (opt.oflag == opt.OptimizationLevel::O1)
|
|
|
|
Pass::call(module->design, "abc -g AND,NAND,OR,NOR,XOR,XNOR,MUX,NMUX -script +print_stats;strash;print_stats;drwsat;print_stats;fraig;print_stats;refactor,-N,10,-lz;print_stats;&get,-n;&dch,-pem;&nf;&put " + mod->name.str());
|
|
|
|
else if (opt.oflag == opt.OptimizationLevel::O2)
|
|
|
|
Pass::call(module->design, "abc -g AND,NAND,OR,NOR,XOR,XNOR,MUX,NMUX -script +print_stats;strash;print_stats;drwsat;print_stats;dch,-S,1000000,-C,100000,-p;print_stats;fraig;print_stats;refactor,-N,15,-lz;print_stats;dc2,-pbl;print_stats;drwsat;print_stats;&get,-n;&dch,-pem;&nf;&put " + mod->name.str());
|
|
|
|
if (opt.oflag != opt.OptimizationLevel::O0) {
|
|
|
|
Pass::call(module->design, "techmap");
|
|
|
|
Pass::call(module->design, "opt");
|
|
|
|
}
|
|
|
|
|
|
|
|
if (opt.nobisection || opt.nooptimize || wire_to_optimize_name == "") {
|
2020-04-24 23:12:02 -05:00
|
|
|
ret = call_qbf_solver(module, opt, tempdir_name, false, 0);
|
|
|
|
} else {
|
|
|
|
//Do the iterated bisection method:
|
|
|
|
unsigned int iter_num = 1;
|
|
|
|
unsigned int success = 0;
|
|
|
|
unsigned int failure = 0;
|
|
|
|
unsigned int cur_thresh = 0;
|
|
|
|
|
2020-06-30 00:47:03 -05:00
|
|
|
log_assert(wire_to_optimize_name != "");
|
|
|
|
log_assert(module->wire(wire_to_optimize_name) != nullptr);
|
|
|
|
log("%s wire \"%s\".\n", (maximize? "Maximizing" : "Minimizing"), wire_to_optimize_name.c_str());
|
2020-04-24 23:12:02 -05:00
|
|
|
|
|
|
|
//If maximizing, grow until we get a failure. Then bisect success and failure.
|
2020-05-26 18:12:15 -05:00
|
|
|
while (failure == 0 || difference(success, failure) > 1) {
|
2020-04-24 23:12:02 -05:00
|
|
|
Pass::call(design, "design -push-copy");
|
|
|
|
log_header(design, "Preparing QBF-SAT problem.\n");
|
|
|
|
|
|
|
|
if (cur_thresh != 0) {
|
|
|
|
//Add thresholding logic (but not on the initial run when we don't have a sense of where to start):
|
|
|
|
RTLIL::SigSpec comparator = maximize? module->Ge(NEW_ID, module->wire(wire_to_optimize_name), RTLIL::Const(cur_thresh), false)
|
|
|
|
: module->Le(NEW_ID, module->wire(wire_to_optimize_name), RTLIL::Const(cur_thresh), false);
|
|
|
|
|
|
|
|
module->addAssume(wire_to_optimize_name.str() + "__threshold", comparator, RTLIL::Const(1, 1));
|
|
|
|
log("Trying to solve with %s %s %d.\n", wire_to_optimize_name.c_str(), (maximize? ">=" : "<="), cur_thresh);
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = call_qbf_solver(module, opt, tempdir_name, false, iter_num);
|
|
|
|
Pass::call(design, "design -pop");
|
|
|
|
module = design->module(module_name);
|
|
|
|
|
|
|
|
if (!ret.unknown && ret.sat) {
|
|
|
|
Pass::call(design, "design -push-copy");
|
|
|
|
specialize(module, ret, true);
|
|
|
|
|
|
|
|
RTLIL::SigSpec wire, value, undef;
|
|
|
|
RTLIL::SigSpec::parse_sel(wire, design, module, wire_to_optimize_name.str());
|
|
|
|
|
|
|
|
ConstEval ce(module);
|
|
|
|
value = wire;
|
|
|
|
if (!ce.eval(value, undef))
|
|
|
|
log_cmd_error("Failed to evaluate signal %s: Missing value for %s.\n", log_signal(wire), log_signal(undef));
|
|
|
|
log_assert(value.is_fully_const());
|
|
|
|
success = value.as_const().as_int();
|
|
|
|
best_soln = ret;
|
|
|
|
log("Problem is satisfiable with %s = %d.\n", wire_to_optimize_name.c_str(), success);
|
|
|
|
Pass::call(design, "design -pop");
|
|
|
|
module = design->module(module_name);
|
|
|
|
|
|
|
|
//sometimes this happens if we get an 'unknown' or timeout
|
|
|
|
if (!maximize && success < failure)
|
|
|
|
break;
|
2020-05-26 18:12:15 -05:00
|
|
|
else if (maximize && failure != 0 && success > failure)
|
2020-04-24 23:12:02 -05:00
|
|
|
break;
|
2020-05-26 18:12:15 -05:00
|
|
|
|
2020-04-24 23:12:02 -05:00
|
|
|
} else {
|
|
|
|
//Treat 'unknown' as UNSAT
|
|
|
|
failure = cur_thresh;
|
|
|
|
if (failure == 0) {
|
|
|
|
log("Problem is NOT satisfiable.\n");
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
log("Problem is NOT satisfiable with %s %s %d.\n", wire_to_optimize_name.c_str(), (maximize? ">=" : "<="), failure);
|
|
|
|
}
|
|
|
|
|
|
|
|
iter_num++;
|
2020-05-26 18:12:15 -05:00
|
|
|
if (maximize && failure == 0 && success == 0)
|
|
|
|
cur_thresh = 2;
|
|
|
|
else if (maximize && failure == 0)
|
|
|
|
cur_thresh = 2 * success; //growth
|
|
|
|
else //if (!maximize || failure != 0)
|
|
|
|
cur_thresh = (success + failure) / 2; //bisection
|
2020-04-24 23:12:02 -05:00
|
|
|
}
|
|
|
|
if (success != 0 || failure != 0) {
|
|
|
|
log("Wire %s is %s at %d.\n", wire_to_optimize_name.c_str(), (maximize? "maximized" : "minimized"), success);
|
|
|
|
ret = best_soln;
|
|
|
|
}
|
2020-03-25 18:37:49 -05:00
|
|
|
}
|
|
|
|
|
2020-04-24 23:12:02 -05:00
|
|
|
if(!opt.nocleanup)
|
|
|
|
remove_directory(tempdir_name);
|
|
|
|
|
|
|
|
Pass::call(design, "design -pop");
|
|
|
|
|
|
|
|
return ret;
|
2020-03-25 18:37:49 -05:00
|
|
|
}
|
|
|
|
|
|
|
|
QbfSolveOptions parse_args(const std::vector<std::string> &args) {
|
|
|
|
QbfSolveOptions opt;
|
|
|
|
for (opt.argidx = 1; opt.argidx < args.size(); opt.argidx++) {
|
2020-03-26 20:36:04 -05:00
|
|
|
if (args[opt.argidx] == "-nocleanup") {
|
2020-03-25 18:37:49 -05:00
|
|
|
opt.nocleanup = true;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
else if (args[opt.argidx] == "-specialize") {
|
|
|
|
opt.specialize = true;
|
|
|
|
continue;
|
|
|
|
}
|
2020-03-26 20:32:53 -05:00
|
|
|
else if (args[opt.argidx] == "-assume-outputs") {
|
|
|
|
opt.assume_outputs = true;
|
|
|
|
continue;
|
|
|
|
}
|
2020-04-22 23:06:15 -05:00
|
|
|
else if (args[opt.argidx] == "-assume-negative-polarity") {
|
|
|
|
opt.assume_neg = true;
|
|
|
|
continue;
|
|
|
|
}
|
2020-04-24 23:12:02 -05:00
|
|
|
else if (args[opt.argidx] == "-nooptimize") {
|
|
|
|
opt.nooptimize = true;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
else if (args[opt.argidx] == "-nobisection") {
|
|
|
|
opt.nobisection = true;
|
|
|
|
continue;
|
|
|
|
}
|
2020-04-30 16:27:18 -05:00
|
|
|
else if (args[opt.argidx] == "-solver") {
|
|
|
|
if (args.size() <= opt.argidx + 1)
|
|
|
|
log_cmd_error("solver not specified.\n");
|
|
|
|
else {
|
|
|
|
if (args[opt.argidx+1] == "z3")
|
|
|
|
opt.solver = opt.Solver::Z3;
|
|
|
|
else if (args[opt.argidx+1] == "yices")
|
|
|
|
opt.solver = opt.Solver::Yices;
|
2020-05-01 03:12:23 -05:00
|
|
|
else if (args[opt.argidx+1] == "cvc4")
|
|
|
|
opt.solver = opt.Solver::CVC4;
|
2020-04-30 16:27:18 -05:00
|
|
|
else
|
|
|
|
log_cmd_error("Unknown solver \"%s\".\n", args[opt.argidx+1].c_str());
|
|
|
|
opt.argidx++;
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
2020-05-01 18:17:35 -05:00
|
|
|
else if (args[opt.argidx] == "-timeout") {
|
|
|
|
if (args.size() <= opt.argidx + 1)
|
|
|
|
log_cmd_error("timeout not specified.\n");
|
|
|
|
else {
|
|
|
|
int timeout = atoi(args[opt.argidx+1].c_str());
|
|
|
|
if (timeout > 0)
|
|
|
|
opt.timeout = timeout;
|
|
|
|
else
|
|
|
|
log_cmd_error("timeout must be greater than 0.\n");
|
|
|
|
opt.argidx++;
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
2020-06-30 00:47:03 -05:00
|
|
|
else if (args[opt.argidx].substr(0, 2) == "-O" && args[opt.argidx].size() == 3) {
|
|
|
|
switch (args[opt.argidx][2]) {
|
|
|
|
case '0':
|
|
|
|
opt.oflag = opt.OptimizationLevel::O0;
|
|
|
|
break;
|
|
|
|
case '1':
|
|
|
|
opt.oflag = opt.OptimizationLevel::O1;
|
|
|
|
break;
|
|
|
|
case '2':
|
|
|
|
opt.oflag = opt.OptimizationLevel::O2;
|
|
|
|
break;
|
|
|
|
default:
|
|
|
|
log_cmd_error("unknown argument %s\n", args[opt.argidx].c_str());
|
|
|
|
}
|
|
|
|
continue;
|
|
|
|
}
|
2020-03-26 20:32:53 -05:00
|
|
|
else if (args[opt.argidx] == "-sat") {
|
|
|
|
opt.sat = true;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
else if (args[opt.argidx] == "-unsat") {
|
|
|
|
opt.unsat = true;
|
|
|
|
continue;
|
|
|
|
}
|
2020-04-01 14:42:45 -05:00
|
|
|
else if (args[opt.argidx] == "-show-smtbmc") {
|
|
|
|
opt.show_smtbmc = true;
|
|
|
|
continue;
|
|
|
|
}
|
2020-03-25 18:37:49 -05:00
|
|
|
else if (args[opt.argidx] == "-dump-final-smt2") {
|
|
|
|
opt.dump_final_smt2 = true;
|
|
|
|
if (args.size() <= opt.argidx + 1)
|
|
|
|
log_cmd_error("smt2 file not specified.\n");
|
|
|
|
else
|
|
|
|
opt.dump_final_smt2_file = args[++opt.argidx];
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
else if (args[opt.argidx] == "-specialize-from-file") {
|
|
|
|
opt.specialize_from_file = true;
|
|
|
|
if (args.size() <= opt.argidx + 1)
|
|
|
|
log_cmd_error("solution file not specified.\n");
|
|
|
|
else
|
|
|
|
opt.specialize_soln_file = args[++opt.argidx];
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
else if (args[opt.argidx] == "-write-solution") {
|
|
|
|
opt.write_solution = true;
|
|
|
|
if (args.size() <= opt.argidx + 1)
|
|
|
|
log_cmd_error("solution file not specified.\n");
|
|
|
|
else
|
|
|
|
opt.write_soln_soln_file = args[++opt.argidx];
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
return opt;
|
|
|
|
}
|
2020-03-25 17:37:54 -05:00
|
|
|
|
2020-03-23 02:42:48 -05:00
|
|
|
struct QbfSatPass : public Pass {
|
|
|
|
QbfSatPass() : Pass("qbfsat", "solve a 2QBF-SAT problem in the circuit") { }
|
2020-06-18 18:34:52 -05:00
|
|
|
void help() override
|
2020-03-23 02:42:48 -05:00
|
|
|
{
|
|
|
|
// |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
|
|
|
|
log("\n");
|
|
|
|
log(" qbfsat [options] [selection]\n");
|
|
|
|
log("\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log("This command solves an \"exists-forall\" 2QBF-SAT problem defined over the currently\n");
|
|
|
|
log("selected module. Existentially-quantified variables are declared by assigning a wire\n");
|
|
|
|
log("\"$anyconst\". Universally-quantified variables may be explicitly declared by assigning\n");
|
|
|
|
log("a wire \"$allconst\", but module inputs will be treated as universally-quantified\n");
|
|
|
|
log("variables by default.\n");
|
2020-03-23 02:42:48 -05:00
|
|
|
log("\n");
|
2020-03-24 20:58:41 -05:00
|
|
|
log(" -nocleanup\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" Do not delete temporary files and directories. Useful for debugging.\n");
|
2020-03-24 20:58:41 -05:00
|
|
|
log("\n");
|
|
|
|
log(" -dump-final-smt2 <file>\n");
|
|
|
|
log(" Pass the --dump-smt2 option to yosys-smtbmc.\n");
|
|
|
|
log("\n");
|
2020-03-26 20:32:53 -05:00
|
|
|
log(" -assume-outputs\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" Add an \"$assume\" cell for the conjunction of all one-bit module output wires.\n");
|
2020-03-26 20:32:53 -05:00
|
|
|
log("\n");
|
2020-04-22 23:06:15 -05:00
|
|
|
log(" -assume-negative-polarity\n");
|
|
|
|
log(" When adding $assume cells for one-bit module output wires, assume they are\n");
|
|
|
|
log(" negative polarity signals and should always be low, for example like the\n");
|
|
|
|
log(" miters created with the `miter` command.\n");
|
|
|
|
log("\n");
|
2020-04-24 23:12:02 -05:00
|
|
|
log(" -nooptimize\n");
|
|
|
|
log(" Ignore \"\\minimize\" and \"\\maximize\" attributes, do not emit \"(maximize)\" or\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" \"(minimize)\" in the SMT-LIBv2, and generally make no attempt to optimize anything.\n");
|
2020-04-24 23:12:02 -05:00
|
|
|
log("\n");
|
|
|
|
log(" -nobisection\n");
|
|
|
|
log(" If a wire is marked with the \"\\minimize\" or \"\\maximize\" attribute, do not\n");
|
|
|
|
log(" attempt to optimize that value with the default iterated solving and threshold\n");
|
|
|
|
log(" bisection approach. Instead, have yosys-smtbmc emit a \"(minimize)\" or \"(maximize)\"\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" command in the SMT-LIBv2 output and hope that the solver supports optimizing\n");
|
2020-04-24 23:12:02 -05:00
|
|
|
log(" quantified bitvector problems.\n");
|
|
|
|
log("\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" -solver <solver>\n");
|
2020-05-01 03:12:23 -05:00
|
|
|
log(" Use a particular solver. Choose one of: \"z3\", \"yices\", and \"cvc4\".\n");
|
2020-06-30 01:57:45 -05:00
|
|
|
log(" (default: yices)\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log("\n");
|
2020-05-01 18:17:35 -05:00
|
|
|
log(" -timeout <value>\n");
|
|
|
|
log(" Set the per-iteration timeout in seconds.\n");
|
2020-06-30 01:57:45 -05:00
|
|
|
log(" (default: no timeout)\n");
|
2020-05-01 18:17:35 -05:00
|
|
|
log("\n");
|
2020-06-30 00:47:03 -05:00
|
|
|
log(" -O0, -O1, -O2\n");
|
|
|
|
log(" Control the use of ABC to simplify the QBF-SAT problem before solving.\n");
|
|
|
|
log("\n");
|
2020-03-26 20:32:53 -05:00
|
|
|
log(" -sat\n");
|
|
|
|
log(" Generate an error if the solver does not return \"sat\".\n");
|
|
|
|
log("\n");
|
|
|
|
log(" -unsat\n");
|
|
|
|
log(" Generate an error if the solver does not return \"unsat\".\n");
|
|
|
|
log("\n");
|
2020-04-01 14:42:45 -05:00
|
|
|
log(" -show-smtbmc\n");
|
|
|
|
log(" Print the output from yosys-smtbmc.\n");
|
|
|
|
log("\n");
|
2020-03-23 02:42:48 -05:00
|
|
|
log(" -specialize\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" If the problem is satisfiable, replace each \"$anyconst\" cell with its\n");
|
|
|
|
log(" corresponding constant value from the model produced by the solver.\n");
|
2020-03-23 02:42:48 -05:00
|
|
|
log("\n");
|
|
|
|
log(" -specialize-from-file <solution file>\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" Do not run the solver, but instead only attempt to replace each \"$anyconst\"\n");
|
|
|
|
log(" cell in the current module with a constant value provided by the specified file.\n");
|
2020-03-23 02:42:48 -05:00
|
|
|
log("\n");
|
|
|
|
log(" -write-solution <solution file>\n");
|
2020-04-30 16:27:18 -05:00
|
|
|
log(" If the problem is satisfiable, write the corresponding constant value for each\n");
|
|
|
|
log(" \"$anyconst\" cell from the model produced by the solver to the specified file.");
|
2020-03-23 02:42:48 -05:00
|
|
|
log("\n");
|
|
|
|
log("\n");
|
|
|
|
}
|
2020-04-09 02:40:03 -05:00
|
|
|
|
2020-06-18 18:34:52 -05:00
|
|
|
void execute(std::vector<std::string> args, RTLIL::Design *design) override
|
2020-03-23 02:42:48 -05:00
|
|
|
{
|
2020-04-09 02:40:03 -05:00
|
|
|
log_header(design, "Executing QBFSAT pass (solving QBF-SAT problems in the circuit).\n");
|
2020-03-25 18:37:49 -05:00
|
|
|
QbfSolveOptions opt = parse_args(args);
|
|
|
|
extra_args(args, opt.argidx, design);
|
2020-03-23 02:42:48 -05:00
|
|
|
|
2020-04-09 02:40:03 -05:00
|
|
|
RTLIL::Module *module = nullptr;
|
2020-03-23 02:42:48 -05:00
|
|
|
for (auto mod : design->selected_modules()) {
|
|
|
|
if (module)
|
|
|
|
log_cmd_error("Only one module must be selected for the QBF-SAT pass! (selected: %s and %s)\n", log_id(module), log_id(mod));
|
|
|
|
module = mod;
|
|
|
|
}
|
2020-04-09 02:40:03 -05:00
|
|
|
if (module == nullptr)
|
2020-03-23 02:42:48 -05:00
|
|
|
log_cmd_error("Can't perform QBF-SAT on an empty selection!\n");
|
|
|
|
|
2020-04-01 14:28:07 -05:00
|
|
|
log_push();
|
2020-03-25 20:02:53 -05:00
|
|
|
if (!opt.specialize_from_file) {
|
|
|
|
//Save the design to restore after modiyfing the current module.
|
|
|
|
std::string module_name = module->name.str();
|
2020-03-23 02:42:48 -05:00
|
|
|
|
2020-03-25 20:02:53 -05:00
|
|
|
QbfSolutionType ret = qbf_solve(module, opt);
|
|
|
|
module = design->module(module_name);
|
2020-04-24 23:12:02 -05:00
|
|
|
if (ret.unknown) {
|
|
|
|
if (opt.sat || opt.unsat)
|
|
|
|
log_cmd_error("expected problem to be %s\n", opt.sat? "SAT" : "UNSAT");
|
|
|
|
}
|
|
|
|
else if (ret.sat) {
|
2020-03-25 20:02:53 -05:00
|
|
|
print_qed();
|
|
|
|
if (opt.write_solution) {
|
2020-06-29 17:06:43 -05:00
|
|
|
ret.write_solution(module, opt.write_soln_soln_file);
|
2020-03-25 20:02:53 -05:00
|
|
|
}
|
|
|
|
if (opt.specialize) {
|
|
|
|
specialize(module, ret);
|
2020-04-01 14:28:07 -05:00
|
|
|
} else {
|
2020-06-29 17:06:43 -05:00
|
|
|
ret.dump_model(module);
|
2020-03-25 20:02:53 -05:00
|
|
|
}
|
2020-03-26 20:32:53 -05:00
|
|
|
if (opt.unsat)
|
|
|
|
log_cmd_error("expected problem to be UNSAT\n");
|
2020-03-25 18:51:32 -05:00
|
|
|
}
|
2020-04-24 23:12:02 -05:00
|
|
|
else {
|
|
|
|
print_proof_failed();
|
|
|
|
if (opt.sat)
|
|
|
|
log_cmd_error("expected problem to be SAT\n");
|
|
|
|
}
|
2020-04-09 02:40:03 -05:00
|
|
|
} else
|
2020-03-25 20:02:53 -05:00
|
|
|
specialize_from_file(module, opt.specialize_soln_file);
|
2020-04-01 14:28:07 -05:00
|
|
|
log_pop();
|
2020-03-23 02:42:48 -05:00
|
|
|
}
|
|
|
|
} QbfSatPass;
|
|
|
|
|
|
|
|
PRIVATE_NAMESPACE_END
|