mirrors/yosys
3
0
Fork 0
mirror of https://github.com/YosysHQ/yosys synced 2025-06-23 06:13:41 +00:00
Code Activity

opt_balance_tree allow_off_chain

Browse source
This commit is contained in:
Alain Dargelas 2024-12-12 09:50:53 -08:00
parent 713f80e850
commit 41c6b71bcb
1 changed files with 53 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

68
passes/opt/opt_balance_tree.cc
View file

@ -28,15 +28,13 @@ PRIVATE_NAMESPACE_BEGIN
struct OptBalanceTreeWorker { struct OptBalanceTreeWorker {
// Module and signal map // Module and signal map
Design *design;
Module *module; Module *module;
SigMap sigmap; SigMap sigmap;
bool allow_off_chain;
// Counts of each cell type that are getting balanced // Counts of each cell type that are getting balanced
dict<IdString, int> cell_count; dict<IdString, int> cell_count;
// Cells to remove
pool<Cell*> remove_cells;
// Signal chain data structures // Signal chain data structures
dict<SigSpec, Cell*> sig_chain_next; dict<SigSpec, Cell*> sig_chain_next;
dict<SigSpec, Cell*> sig_chain_prev; dict<SigSpec, Cell*> sig_chain_prev;
@ -263,15 +261,10 @@ struct OptBalanceTreeWorker {
} }
void cleanup() { void cleanup() {
// Remove cells
for (auto cell : remove_cells)
module->remove(cell);
// Fix ports // Fix ports
module->fixup_ports(); module->fixup_ports();
// Clear data structures // Clear data structures
remove_cells.clear();
sig_chain_next.clear(); sig_chain_next.clear();
sig_chain_prev.clear(); sig_chain_prev.clear();
sigbit_with_non_chain_users.clear(); sigbit_with_non_chain_users.clear();
@ -279,8 +272,43 @@ struct OptBalanceTreeWorker {
candidate_cells.clear(); candidate_cells.clear();
} }
OptBalanceTreeWorker(Module *module, const vector<IdString> cell_types) : module(module), sigmap(module) { OptBalanceTreeWorker(Design* design, Module *module, const vector<IdString> cell_types, bool allow_off_chain) :
// Do for each cell type design(design), module(module), sigmap(module), allow_off_chain(allow_off_chain) {
if (allow_off_chain) {
// Deselect all cells
Pass::call(design, "select -none");
// Do for each cell type
bool has_cell_to_split = false;
for (auto cell_type : cell_types) {
// Find chains of ops
make_sig_chain_next_prev(cell_type);
find_chain_start_cells();
// For each chain, if len >= 3, select all the elements
for (auto c : chain_start_cells) {
vector<Cell *> chain = create_chain(c);
if (GetSize(chain) < 3)
continue;
for (auto cell : chain) {
has_cell_to_split = true;
design->select(module, cell);
}
}
// Clean up
cleanup();
}
// Splitfanout of selected cells
if (has_cell_to_split)
Pass::call(design, "splitfanout");
// Reset selection for other passes
Pass::call(design, "select -clear");
// Recreate sigmap
sigmap.set(module);
}
// Do for each cell type
for (auto cell_type : cell_types) { for (auto cell_type : cell_types) {
// Find chains of ops // Find chains of ops
make_sig_chain_next_prev(cell_type); make_sig_chain_next_prev(cell_type);
@ -309,14 +337,24 @@ struct OptBalanceTreePass : public Pass {
log("This pass converts cascaded chains of $and/$or/$xor/$xnor/$add/$mul cells into\n"); log("This pass converts cascaded chains of $and/$or/$xor/$xnor/$add/$mul cells into\n");
log("trees of cells to improve timing.\n"); log("trees of cells to improve timing.\n");
log("\n"); log("\n");
log(" -allow-off-chain\n");
log(" Allows matching of cells that have loads outside the chain. These cells\n");
log(" will be replicated and balanced into a tree, but the original\n");
log(" cell will remain, driving its original loads.\n");
log("\n");
} }
void execute(std::vector<std::string> args, RTLIL::Design *design) override { void execute(std::vector<std::string> args, RTLIL::Design *design) override {
log_header(design, "Executing OPT_BALANCE_TREE pass (cell cascades to trees).\n"); log_header(design, "Executing OPT_BALANCE_TREE pass (cell cascades to trees).\n");
// Handle arguments bool allow_off_chain = false;
size_t argidx; size_t argidx;
for (argidx = 1; argidx < args.size(); argidx++) { for (argidx = 1; argidx < args.size(); argidx++)
// No arguments yet {
if (args[argidx] == "-allow-off-chain")
{
allow_off_chain = true;
continue;
}
break; break;
} }
extra_args(args, argidx, design); extra_args(args, argidx, design);
@ -325,7 +363,7 @@ struct OptBalanceTreePass : public Pass {
dict<IdString, int> cell_count; dict<IdString, int> cell_count;
const vector<IdString> cell_types = {ID($and), ID($or), ID($xor), ID($xnor), ID($add), ID($mul)}; const vector<IdString> cell_types = {ID($and), ID($or), ID($xor), ID($xnor), ID($add), ID($mul)};
for (auto module : design->selected_modules()) { for (auto module : design->selected_modules()) {
OptBalanceTreeWorker worker(module, cell_types); OptBalanceTreeWorker worker(design, module, cell_types, allow_off_chain);
for (auto cell : worker.cell_count) { for (auto cell : worker.cell_count) {
cell_count[cell.first] += cell.second; cell_count[cell.first] += cell.second;
} }