Added "aig" pass
authorClifford Wolf <clifford@clifford.at>
Tue, 9 Jun 2015 20:33:26 +0000 (22:33 +0200)
committerClifford Wolf <clifford@clifford.at>
Tue, 9 Jun 2015 20:33:26 +0000 (22:33 +0200)
kernel/cellaigs.cc
passes/techmap/Makefile.inc
passes/techmap/aig.cc [new file with mode: 0644]

index 735c86654487dce348ea4c6792761ba97632de54..b738b9dbaa26b111a4730cba12d9412383207a2c 100644 (file)
@@ -44,12 +44,42 @@ unsigned int AigNode::hash() const
 struct AigMaker
 {
        Aig *aig;
+       Cell *cell;
        idict<AigNode> aig_indices;
 
-       AigMaker(Aig *aig) : aig(aig) { }
+       int the_true_node;
+       int the_false_node;
 
-       int inport(IdString portname, int portbit, bool inverter = false)
+       AigMaker(Aig *aig, Cell *cell) : aig(aig), cell(cell)
+       {
+               the_true_node = -1;
+               the_false_node = -1;
+       }
+
+       int bool_node(bool value)
        {
+               AigNode node;
+               node.portbit = -1;
+               node.inverter = !value;
+               node.left_parent = -1;
+               node.right_parent = -1;
+
+               if (!aig_indices.count(node)) {
+                       aig_indices.expect(node, GetSize(aig->nodes));
+                       aig->nodes.push_back(node);
+               }
+
+               return aig_indices.at(node);
+       }
+
+       int inport(IdString portname, int portbit = 0, bool inverter = false)
+       {
+               if (portbit >= GetSize(cell->getPort(portname))) {
+                       if (cell->parameters.count(portname.str() + "_SIGNED") && cell->getParam(portname.str() + "_SIGNED").as_bool())
+                               return inport(portname, GetSize(cell->getPort(portname))-1, inverter);
+                       return bool_node(!inverter);
+               }
+
                AigNode node;
                node.portname = portname;
                node.portbit = portbit;
@@ -65,8 +95,16 @@ struct AigMaker
                return aig_indices.at(node);
        }
 
-       int gate(int left_parent, int right_parent, bool inverter = false)
+       int not_inport(IdString portname, int portbit = 0)
        {
+               return inport(portname, portbit, true);
+       }
+
+       int and_gate(int left_parent, int right_parent, bool inverter = false)
+       {
+               if (left_parent > right_parent)
+                       std::swap(left_parent, right_parent);
+
                AigNode node;
                node.portbit = -1;
                node.inverter = inverter;
@@ -81,9 +119,15 @@ struct AigMaker
                return aig_indices.at(node);
        }
 
-       void outport(int node, IdString portname, int portbit)
+       int nand_gate(int left_parent, int right_parent)
+       {
+               return and_gate(left_parent, right_parent, true);
+       }
+
+       void outport(int node, IdString portname, int portbit = 0)
        {
-               aig->nodes.at(node).outports.push_back(pair<IdString, int>(portname, portbit));
+               if (portbit < GetSize(cell->getPort(portname)))
+                       aig->nodes.at(node).outports.push_back(pair<IdString, int>(portname, portbit));
        }
 };
 
@@ -92,28 +136,110 @@ Aig::Aig(Cell *cell)
        if (cell->type[0] != '$')
                return;
 
-       AigMaker mk(this);
+       AigMaker mk(this, cell);
        name = cell->type.str();
 
        cell->parameters.sort();
        for (auto p : cell->parameters)
                name += stringf(":%d", p.second.as_int());
 
-       if (cell->type == "$_AND_" || cell->type == "$_NAND_")
+       if (cell->type.in("$and", "$_AND_", "$_NAND_"))
+       {
+               for (int i = 0; i < GetSize(cell->getPort("\\Y")); i++) {
+                       int A = mk.inport("\\A", i);
+                       int B = mk.inport("\\B", i);
+                       int Y = mk.and_gate(A, B, cell->type == "$_NAND_");
+                       mk.outport(Y, "\\Y", i);
+               }
+               return;
+       }
+
+       if (cell->type.in("$or", "$_OR_", "$_NOR_"))
+       {
+               for (int i = 0; i < GetSize(cell->getPort("\\Y")); i++) {
+                       int A = mk.not_inport("\\A", i);
+                       int B = mk.not_inport("\\B", i);
+                       int Y = mk.and_gate(A, B, cell->type != "$_NOR_");
+                       mk.outport(Y, "\\Y", i);
+               }
+               return;
+       }
+
+       if (cell->type.in("$xor", "$xnor", "$_XOR_", "$_XNOR_"))
+       {
+               for (int i = 0; i < GetSize(cell->getPort("\\Y")); i++) {
+                       int A = mk.inport("\\A", i);
+                       int B = mk.inport("\\B", i);
+                       int NA = mk.not_inport("\\A", i);
+                       int NB = mk.not_inport("\\B", i);
+                       int NOT_AB = mk.nand_gate(A, B);
+                       int NOT_NAB = mk.nand_gate(NA, NB);
+                       int Y = mk.and_gate(NOT_AB, NOT_NAB, !cell->type.in("$xor", "$_XOR_"));
+                       mk.outport(Y, "\\Y", i);
+               }
+               return;
+       }
+
+       if (cell->type.in("$mux", "$_MUX_"))
+       {
+               int S = mk.inport("\\S");
+               int NS = mk.not_inport("\\S");
+               for (int i = 0; i < GetSize(cell->getPort("\\Y")); i++) {
+                       int A = mk.inport("\\A", i);
+                       int B = mk.inport("\\B", i);
+                       int NOT_SB = mk.nand_gate(S, B);
+                       int NOT_NSA = mk.nand_gate(NS, A);
+                       int Y = mk.nand_gate(NOT_SB, NOT_NSA);
+                       mk.outport(Y, "\\Y", i);
+               }
+               return;
+       }
+
+       if (cell->type == "$_AOI3_")
+       {
+               int A = mk.inport("\\A");
+               int B = mk.inport("\\B");
+               int NC = mk.not_inport("\\C");
+               int NOT_AB = mk.nand_gate(A, B);
+               int Y = mk.and_gate(NOT_AB, NC);
+               mk.outport(Y, "\\Y");
+               return;
+       }
+
+       if (cell->type == "$_OAI3_")
+       {
+               int NA = mk.not_inport("\\A");
+               int NB = mk.not_inport("\\B");
+               int C = mk.inport("\\C");
+               int NOT_NAB = mk.nand_gate(NA, NB);
+               int Y = mk.nand_gate(NOT_NAB, C);
+               mk.outport(Y, "\\Y");
+               return;
+       }
+
+       if (cell->type == "$_AOI4_")
        {
-               int A = mk.inport("A", 0);
-               int B = mk.inport("B", 0);
-               int Y = mk.gate(A, B, cell->type == "$_NAND_");
-               mk.outport(Y, "Y", 0);
+               int A = mk.inport("\\A");
+               int B = mk.inport("\\B");
+               int C = mk.inport("\\C");
+               int D = mk.inport("\\D");
+               int NOT_AB = mk.nand_gate(A, B);
+               int NOT_CD = mk.nand_gate(C, D);
+               int Y = mk.and_gate(NOT_AB, NOT_CD);
+               mk.outport(Y, "\\Y");
                return;
        }
 
-       if (cell->type == "$_OR_" || cell->type == "$_NOR_")
+       if (cell->type == "$_OAI4_")
        {
-               int A = mk.inport("A", 0, true);
-               int B = mk.inport("B", 0, true);
-               int Y = mk.gate(A, B, cell->type == "$_OR_");
-               mk.outport(Y, "Y", 0);
+               int NA = mk.not_inport("\\A");
+               int NB = mk.not_inport("\\B");
+               int NC = mk.not_inport("\\C");
+               int ND = mk.not_inport("\\D");
+               int NOT_NAB = mk.nand_gate(NA, NB);
+               int NOT_NCD = mk.nand_gate(NC, ND);
+               int Y = mk.nand_gate(NOT_NAB, NOT_NCD);
+               mk.outport(Y, "\\Y");
                return;
        }
 
index 6ef4fbc2681cd02b54cdbf7be466f9ddad96467f..3cbee5b85a7879c58c562f4c45363e2c6c985ece 100644 (file)
@@ -18,6 +18,7 @@ OBJS += passes/techmap/dff2dffe.o
 OBJS += passes/techmap/dffinit.o
 OBJS += passes/techmap/pmuxtree.o
 OBJS += passes/techmap/muxcover.o
+OBJS += passes/techmap/aig.o
 endif
 
 GENFILES += passes/techmap/techmap.inc
diff --git a/passes/techmap/aig.cc b/passes/techmap/aig.cc
new file mode 100644 (file)
index 0000000..0c0f0b1
--- /dev/null
@@ -0,0 +1,148 @@
+/*
+ *  yosys -- Yosys Open SYnthesis Suite
+ *
+ *  Copyright (C) 2012  Clifford Wolf <clifford@clifford.at>
+ *  
+ *  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.
+ *
+ */
+
+#include "kernel/yosys.h"
+#include "kernel/cellaigs.h"
+
+USING_YOSYS_NAMESPACE
+PRIVATE_NAMESPACE_BEGIN
+
+struct AigPass : public Pass {
+       AigPass() : Pass("aig", "convert logic to and-inverter circuit") { }
+       virtual void help()
+       {
+               log("\n");
+               log("    aig [options] [selection]\n");
+               log("\n");
+               log("Replace all logic cells with circuits made of only $_AND_ and\n");
+               log("$_NOT_ cells.\n");
+               log("\n");
+               log("    -nand\n");
+               log("        Enable creation of $_NAND_ cells\n");
+               log("\n");
+       }
+       virtual void execute(std::vector<std::string> args, RTLIL::Design *design)
+       {
+               bool nand_mode = false;
+
+               log_header("Executing AIG pass (converting logic to AIG).\n");
+
+               size_t argidx;
+               for (argidx = 1; argidx < args.size(); argidx++)
+               {
+                       if (args[argidx] == "-nand") {
+                               nand_mode = true;
+                               continue;
+                       }
+                       break;
+               }
+               extra_args(args, argidx, design);
+
+               for (auto module : design->selected_modules())
+               {
+                       vector<Cell*> replaced_cells;
+                       int not_replaced_count = 0;
+                       dict<IdString, int> stat_replaced;
+                       dict<IdString, int> stat_not_replaced;
+                       int orig_num_cells = GetSize(module->cells());
+
+                       for (auto cell : module->selected_cells())
+                       {
+                               Aig aig(cell);
+
+                               if (cell->type == "$_AND_" || cell->type == "$_NOT_")
+                                       aig.name.clear();
+
+                               if (nand_mode && cell->type == "$_NAND_")
+                                       aig.name.clear();
+
+                               if (aig.name.empty()) {
+                                       not_replaced_count++;
+                                       stat_not_replaced[cell->type]++;
+                                       continue;
+                               }
+
+                               vector<SigBit> sigs;
+                               dict<pair<int, int>, SigBit> and_cache;
+
+                               for (int node_idx = 0; node_idx < GetSize(aig.nodes); node_idx++)
+                               {
+                                       SigBit bit;
+                                       auto &node = aig.nodes[node_idx];
+
+                                       if (node.portbit >= 0) {
+                                               bit = cell->getPort(node.portname)[node.portbit];
+                                       } else if (node.left_parent < 0 && node.right_parent < 0) {
+                                               bit = node.inverter ? State::S0 : State::S1;
+                                       } else {
+                                               SigBit A = sigs.at(node.left_parent);
+                                               SigBit B = sigs.at(node.right_parent);
+                                               if (nand_mode && node.inverter) {
+                                                       bit = module->NandGate(NEW_ID, A, B);
+                                                       goto nand_inverter;
+                                               } else {
+                                                       pair<int, int> key(node.left_parent, node.right_parent);
+                                                       if (and_cache.count(key))
+                                                               bit = and_cache.at(key);
+                                                       else
+                                                               bit = module->AndGate(NEW_ID, A, B);
+                                               }
+                                       }
+
+                                       if (node.inverter)
+                                               bit = module->NotGate(NEW_ID, bit);
+                               
+                               nand_inverter:
+                                       for (auto &op : node.outports)
+                                               module->connect(cell->getPort(op.first)[op.second], bit);
+
+                                       sigs.push_back(bit);
+                               }
+
+                               replaced_cells.push_back(cell);
+                               stat_replaced[cell->type]++;
+                       }
+
+                       if (not_replaced_count == 0 && replaced_cells.empty())
+                               continue;
+
+                       log("Module %s: replaced %d cells with %d new cells, skipped %d cells.\n", log_id(module),
+                                       GetSize(replaced_cells), GetSize(module->cells()) - orig_num_cells, not_replaced_count);
+
+                       if (!stat_replaced.empty()) {
+                               stat_replaced.sort();
+                               log("  replaced %d cell types:\n", GetSize(stat_replaced));
+                               for (auto &it : stat_replaced)
+                                       log("%8d %s\n", it.second, log_id(it.first));
+                       }
+
+                       if (!stat_not_replaced.empty()) {
+                               stat_not_replaced.sort();
+                               log("  not replaced %d cell types:\n", GetSize(stat_not_replaced));
+                               for (auto &it : stat_not_replaced)
+                                       log("%8d %s\n", it.second, log_id(it.first));
+                       }
+
+                       for (auto cell : replaced_cells)
+                               module->remove(cell);
+               }
+       }
+} AigPass;
+PRIVATE_NAMESPACE_END