mirror of
				https://github.com/YosysHQ/yosys
				synced 2025-10-27 01:39:23 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			653 lines
		
	
	
	
		
			21 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
			
		
		
	
	
			653 lines
		
	
	
	
		
			21 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
| /*
 | |
|  *  yosys -- Yosys Open SYnthesis Suite
 | |
|  *
 | |
|  *  Copyright (C) 2012  Claire Xenia Wolf <claire@yosyshq.com>
 | |
|  *
 | |
|  *  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/sigtools.h"
 | |
| #include "kernel/modtools.h"
 | |
| #include "kernel/ffinit.h"
 | |
| #include "kernel/utils.h"
 | |
| 
 | |
| USING_YOSYS_NAMESPACE
 | |
| 
 | |
| PRIVATE_NAMESPACE_BEGIN
 | |
| 
 | |
| struct WreduceConfig
 | |
| {
 | |
| 	pool<IdString> supported_cell_types;
 | |
| 	bool keepdc = false;
 | |
| 	bool mux_undef = false;
 | |
| 
 | |
| 	WreduceConfig()
 | |
| 	{
 | |
| 		supported_cell_types = pool<IdString>({
 | |
| 			ID($not), ID($pos), ID($neg),
 | |
| 			ID($and), ID($or), ID($xor), ID($xnor),
 | |
| 			ID($shl), ID($shr), ID($sshl), ID($sshr), ID($shift), ID($shiftx),
 | |
| 			ID($lt), ID($le), ID($eq), ID($ne), ID($eqx), ID($nex), ID($ge), ID($gt),
 | |
| 			ID($add), ID($sub), ID($mul), // ID($div), ID($mod), ID($divfloor), ID($modfloor), ID($pow),
 | |
| 			ID($mux), ID($pmux),
 | |
| 			ID($dff), ID($dffe), ID($adff), ID($adffe), ID($sdff), ID($sdffe), ID($sdffce),
 | |
| 			ID($dlatch), ID($adlatch),
 | |
| 		});
 | |
| 	}
 | |
| };
 | |
| 
 | |
| struct WreduceWorker
 | |
| {
 | |
| 	WreduceConfig *config;
 | |
| 	Module *module;
 | |
| 	ModIndex mi;
 | |
| 
 | |
| 	std::set<Cell*, IdString::compare_ptr_by_name<Cell>> work_queue_cells;
 | |
| 	std::set<SigBit> work_queue_bits;
 | |
| 	pool<SigBit> keep_bits;
 | |
| 	FfInitVals initvals;
 | |
| 
 | |
| 	WreduceWorker(WreduceConfig *config, Module *module) :
 | |
| 			config(config), module(module), mi(module) { }
 | |
| 
 | |
| 	void run_cell_mux(Cell *cell)
 | |
| 	{
 | |
| 		// Reduce size of MUX if inputs agree on a value for a bit or a output bit is unused
 | |
| 
 | |
| 		SigSpec sig_a = mi.sigmap(cell->getPort(ID::A));
 | |
| 		SigSpec sig_b = mi.sigmap(cell->getPort(ID::B));
 | |
| 		SigSpec sig_s = mi.sigmap(cell->getPort(ID::S));
 | |
| 		SigSpec sig_y = mi.sigmap(cell->getPort(ID::Y));
 | |
| 		std::vector<SigBit> bits_removed;
 | |
| 
 | |
| 		if (sig_y.has_const())
 | |
| 			return;
 | |
| 
 | |
| 		for (int i = GetSize(sig_y)-1; i >= 0; i--)
 | |
| 		{
 | |
| 			auto info = mi.query(sig_y[i]);
 | |
| 			if (!info->is_output && GetSize(info->ports) <= 1 && !keep_bits.count(mi.sigmap(sig_y[i]))) {
 | |
| 				bits_removed.push_back(State::Sx);
 | |
| 				continue;
 | |
| 			}
 | |
| 
 | |
| 			SigBit ref = sig_a[i];
 | |
| 			for (int k = 0; k < GetSize(sig_s); k++) {
 | |
| 				if ((config->keepdc || !config->mux_undef || (ref != State::Sx && sig_b[k*GetSize(sig_a) + i] != State::Sx)) && ref != sig_b[k*GetSize(sig_a) + i])
 | |
| 					goto no_match_ab;
 | |
| 				if (sig_b[k*GetSize(sig_a) + i] != State::Sx)
 | |
| 					ref = sig_b[k*GetSize(sig_a) + i];
 | |
| 			}
 | |
| 			if (0)
 | |
| 		no_match_ab:
 | |
| 				break;
 | |
| 			bits_removed.push_back(ref);
 | |
| 		}
 | |
| 
 | |
| 		if (bits_removed.empty())
 | |
| 			return;
 | |
| 
 | |
| 		SigSpec sig_removed;
 | |
| 		for (int i = GetSize(bits_removed)-1; i >= 0; i--)
 | |
| 			sig_removed.append(bits_removed[i]);
 | |
| 
 | |
| 		if (GetSize(bits_removed) == GetSize(sig_y)) {
 | |
| 			log("Removed cell %s.%s (%s).\n", log_id(module), log_id(cell), log_id(cell->type));
 | |
| 			module->connect(sig_y, sig_removed);
 | |
| 			module->remove(cell);
 | |
| 			return;
 | |
| 		}
 | |
| 
 | |
| 		log("Removed top %d bits (of %d) from mux cell %s.%s (%s).\n",
 | |
| 				GetSize(sig_removed), GetSize(sig_y), log_id(module), log_id(cell), log_id(cell->type));
 | |
| 
 | |
| 		int n_removed = GetSize(sig_removed);
 | |
| 		int n_kept = GetSize(sig_y) - GetSize(sig_removed);
 | |
| 
 | |
| 		SigSpec new_work_queue_bits;
 | |
| 		new_work_queue_bits.append(sig_a.extract(n_kept, n_removed));
 | |
| 		new_work_queue_bits.append(sig_y.extract(n_kept, n_removed));
 | |
| 
 | |
| 		SigSpec new_sig_a = sig_a.extract(0, n_kept);
 | |
| 		SigSpec new_sig_y = sig_y.extract(0, n_kept);
 | |
| 		SigSpec new_sig_b;
 | |
| 
 | |
| 		for (int k = 0; k < GetSize(sig_s); k++) {
 | |
| 			new_sig_b.append(sig_b.extract(k*GetSize(sig_a), n_kept));
 | |
| 			new_work_queue_bits.append(sig_b.extract(k*GetSize(sig_a) + n_kept, n_removed));
 | |
| 		}
 | |
| 
 | |
| 		for (auto bit : new_work_queue_bits)
 | |
| 			work_queue_bits.insert(bit);
 | |
| 
 | |
| 		cell->setPort(ID::A, new_sig_a);
 | |
| 		cell->setPort(ID::B, new_sig_b);
 | |
| 		cell->setPort(ID::Y, new_sig_y);
 | |
| 		cell->fixup_parameters();
 | |
| 
 | |
| 		module->connect(sig_y.extract(n_kept, n_removed), sig_removed);
 | |
| 	}
 | |
| 
 | |
| 	void run_cell_dff(Cell *cell)
 | |
| 	{
 | |
| 		// Reduce size of FF if inputs are just sign/zero extended or output bit is not used
 | |
| 
 | |
| 		SigSpec sig_d = mi.sigmap(cell->getPort(ID::D));
 | |
| 		SigSpec sig_q = mi.sigmap(cell->getPort(ID::Q));
 | |
| 		bool has_reset = false;
 | |
| 		Const initval = initvals(sig_q), rst_value;
 | |
| 
 | |
| 		int width_before = GetSize(sig_q);
 | |
| 
 | |
| 		if (width_before == 0)
 | |
| 			return;
 | |
| 
 | |
| 		if (cell->parameters.count(ID::ARST_VALUE)) {
 | |
| 			rst_value = cell->parameters[ID::ARST_VALUE];
 | |
| 			has_reset = true;
 | |
| 		} else if (cell->parameters.count(ID::SRST_VALUE)) {
 | |
| 			rst_value = cell->parameters[ID::SRST_VALUE];
 | |
| 			has_reset = true;
 | |
| 		}
 | |
| 
 | |
| 		bool zero_ext = sig_d[GetSize(sig_d)-1] == State::S0;
 | |
| 		bool sign_ext = !zero_ext;
 | |
| 
 | |
| 		for (int i = GetSize(sig_q)-1; i >= 0; i--)
 | |
| 		{
 | |
| 			if (zero_ext && sig_d[i] == State::S0 && (initval[i] == State::S0 || (!config->keepdc && initval[i] == State::Sx)) &&
 | |
| 					(!has_reset || i >= GetSize(rst_value) || rst_value[i] == State::S0 || (!config->keepdc && rst_value[i] == State::Sx))) {
 | |
| 				module->connect(sig_q[i], State::S0);
 | |
| 				initvals.remove_init(sig_q[i]);
 | |
| 				sig_d.remove(i);
 | |
| 				sig_q.remove(i);
 | |
| 				continue;
 | |
| 			}
 | |
| 
 | |
| 			if (sign_ext && i > 0 && sig_d[i] == sig_d[i-1] && initval[i] == initval[i-1] && (!config->keepdc || initval[i] != State::Sx) &&
 | |
| 					(!has_reset || i >= GetSize(rst_value) || (rst_value[i] == rst_value[i-1] && (!config->keepdc || rst_value[i] != State::Sx)))) {
 | |
| 				module->connect(sig_q[i], sig_q[i-1]);
 | |
| 				initvals.remove_init(sig_q[i]);
 | |
| 				sig_d.remove(i);
 | |
| 				sig_q.remove(i);
 | |
| 				continue;
 | |
| 			}
 | |
| 
 | |
| 			auto info = mi.query(sig_q[i]);
 | |
| 			if (info == nullptr)
 | |
| 				return;
 | |
| 			if (!info->is_output && GetSize(info->ports) == 1 && !keep_bits.count(mi.sigmap(sig_q[i]))) {
 | |
| 				initvals.remove_init(sig_q[i]);
 | |
| 				sig_d.remove(i);
 | |
| 				sig_q.remove(i);
 | |
| 				zero_ext = false;
 | |
| 				sign_ext = false;
 | |
| 				continue;
 | |
| 			}
 | |
| 
 | |
| 			break;
 | |
| 		}
 | |
| 
 | |
| 		if (width_before == GetSize(sig_q))
 | |
| 			return;
 | |
| 
 | |
| 		if (GetSize(sig_q) == 0) {
 | |
| 			log("Removed cell %s.%s (%s).\n", log_id(module), log_id(cell), log_id(cell->type));
 | |
| 			module->remove(cell);
 | |
| 			return;
 | |
| 		}
 | |
| 
 | |
| 		log("Removed top %d bits (of %d) from FF cell %s.%s (%s).\n", width_before - GetSize(sig_q), width_before,
 | |
| 				log_id(module), log_id(cell), log_id(cell->type));
 | |
| 
 | |
| 		for (auto bit : sig_d)
 | |
| 			work_queue_bits.insert(bit);
 | |
| 
 | |
| 		for (auto bit : sig_q)
 | |
| 			work_queue_bits.insert(bit);
 | |
| 
 | |
| 		// Narrow ARST_VALUE parameter to new size.
 | |
| 		if (cell->parameters.count(ID::ARST_VALUE)) {
 | |
| 			rst_value.resize(GetSize(sig_q), State::S0);
 | |
| 			cell->setParam(ID::ARST_VALUE, rst_value);
 | |
| 		} else if (cell->parameters.count(ID::SRST_VALUE)) {
 | |
| 			rst_value.resize(GetSize(sig_q), State::S0);
 | |
| 			cell->setParam(ID::SRST_VALUE, rst_value);
 | |
| 		}
 | |
| 
 | |
| 		cell->setPort(ID::D, sig_d);
 | |
| 		cell->setPort(ID::Q, sig_q);
 | |
| 		cell->fixup_parameters();
 | |
| 	}
 | |
| 
 | |
| 	void run_reduce_inport(Cell *cell, char port, int max_port_size, bool &port_signed, bool &did_something)
 | |
| 	{
 | |
| 		port_signed = cell->getParam(stringf("\\%c_SIGNED", port)).as_bool();
 | |
| 		SigSpec sig = mi.sigmap(cell->getPort(stringf("\\%c", port)));
 | |
| 
 | |
| 		if (port == 'B' && cell->type.in(ID($shl), ID($shr), ID($sshl), ID($sshr)))
 | |
| 			port_signed = false;
 | |
| 
 | |
| 		int bits_removed = 0;
 | |
| 		if (GetSize(sig) > max_port_size) {
 | |
| 			bits_removed = GetSize(sig) - max_port_size;
 | |
| 			for (auto bit : sig.extract(max_port_size, bits_removed))
 | |
| 				work_queue_bits.insert(bit);
 | |
| 			sig = sig.extract(0, max_port_size);
 | |
| 		}
 | |
| 
 | |
| 		if (port_signed) {
 | |
| 			while (GetSize(sig) > 1 && sig[GetSize(sig)-1] == sig[GetSize(sig)-2])
 | |
| 				work_queue_bits.insert(sig[GetSize(sig)-1]), sig.remove(GetSize(sig)-1), bits_removed++;
 | |
| 		} else {
 | |
| 			while (GetSize(sig) > 1 && sig[GetSize(sig)-1] == State::S0)
 | |
| 				work_queue_bits.insert(sig[GetSize(sig)-1]), sig.remove(GetSize(sig)-1), bits_removed++;
 | |
| 		}
 | |
| 
 | |
| 		if (bits_removed) {
 | |
| 			log("Removed top %d bits (of %d) from port %c of cell %s.%s (%s).\n",
 | |
| 					bits_removed, GetSize(sig) + bits_removed, port, log_id(module), log_id(cell), log_id(cell->type));
 | |
| 			cell->setPort(stringf("\\%c", port), sig);
 | |
| 			did_something = true;
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	int reduced_opsize(const SigSpec &inp, bool signed_)
 | |
| 	{
 | |
| 		int size = GetSize(inp);
 | |
| 		if (signed_) {
 | |
| 			while (size >= 2 && inp[size - 1] == inp[size - 2])
 | |
| 				size--;
 | |
| 		} else {
 | |
| 			while (size >= 1 && inp[size - 1] == State::S0)
 | |
| 				size--;
 | |
| 		}
 | |
| 		return size;
 | |
| 	}
 | |
| 
 | |
| 	void run_cell(Cell *cell)
 | |
| 	{
 | |
| 		bool did_something = false;
 | |
| 
 | |
| 		if (!config->supported_cell_types.count(cell->type))
 | |
| 			return;
 | |
| 
 | |
| 		if (cell->type.in(ID($mux), ID($pmux)))
 | |
| 			return run_cell_mux(cell);
 | |
| 
 | |
| 		if (cell->type.in(ID($dff), ID($dffe), ID($adff), ID($adffe), ID($sdff), ID($sdffe), ID($sdffce), ID($dlatch), ID($adlatch)))
 | |
| 			return run_cell_dff(cell);
 | |
| 
 | |
| 		SigSpec sig = mi.sigmap(cell->getPort(ID::Y));
 | |
| 
 | |
| 		if (sig.has_const())
 | |
| 			return;
 | |
| 
 | |
| 
 | |
| 		// Reduce size of ports A and B based on constant input bits and size of output port
 | |
| 
 | |
| 		int max_port_a_size = cell->hasPort(ID::A) ? GetSize(cell->getPort(ID::A)) : -1;
 | |
| 		int max_port_b_size = cell->hasPort(ID::B) ? GetSize(cell->getPort(ID::B)) : -1;
 | |
| 
 | |
| 		if (cell->type.in(ID($not), ID($pos), ID($neg), ID($and), ID($or), ID($xor), ID($add), ID($sub))) {
 | |
| 			max_port_a_size = min(max_port_a_size, GetSize(sig));
 | |
| 			max_port_b_size = min(max_port_b_size, GetSize(sig));
 | |
| 		}
 | |
| 
 | |
| 		bool port_a_signed = false;
 | |
| 		bool port_b_signed = false;
 | |
| 
 | |
| 		// For some operations if the output is no wider than either of the inputs
 | |
| 		// we are free to choose the signedness of the operands
 | |
| 		if (cell->type.in(ID($mul), ID($add), ID($sub)) &&
 | |
| 				max_port_a_size == GetSize(sig) &&
 | |
| 				max_port_b_size == GetSize(sig)) {
 | |
| 			SigSpec sig_a = mi.sigmap(cell->getPort(ID::A)), sig_b = mi.sigmap(cell->getPort(ID::B));
 | |
| 
 | |
| 			// Remove top bits from sig_a and sig_b which are not visible on the output
 | |
| 			sig_a.extend_u0(max_port_a_size);
 | |
| 			sig_b.extend_u0(max_port_b_size);
 | |
| 
 | |
| 			int signed_cost, unsigned_cost;
 | |
| 			if (cell->type == ID($mul)) {
 | |
| 				signed_cost = reduced_opsize(sig_a, true) * reduced_opsize(sig_b, true);
 | |
| 				unsigned_cost = reduced_opsize(sig_a, false) * reduced_opsize(sig_b, false);
 | |
| 			} else {
 | |
| 				signed_cost = max(reduced_opsize(sig_a, true), reduced_opsize(sig_b, true));
 | |
| 				unsigned_cost = max(reduced_opsize(sig_a, false), reduced_opsize(sig_b, false));
 | |
| 			}
 | |
| 
 | |
| 			if (!port_a_signed && !port_b_signed && signed_cost < unsigned_cost) {
 | |
| 				log("Converting cell %s.%s (%s) from unsigned to signed.\n",
 | |
| 						log_id(module), log_id(cell), log_id(cell->type));
 | |
| 				cell->setParam(ID::A_SIGNED, 1);
 | |
| 				cell->setParam(ID::B_SIGNED, 1);
 | |
| 				port_a_signed = true;
 | |
| 				port_b_signed = true;
 | |
| 				did_something = true;
 | |
| 			} else if (port_a_signed && port_b_signed && unsigned_cost < signed_cost) {
 | |
| 				log("Converting cell %s.%s (%s) from signed to unsigned.\n",
 | |
| 						log_id(module), log_id(cell), log_id(cell->type));
 | |
| 				cell->setParam(ID::A_SIGNED, 0);
 | |
| 				cell->setParam(ID::B_SIGNED, 0);
 | |
| 				port_a_signed = false;
 | |
| 				port_b_signed = false;
 | |
| 				did_something = true;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		if (max_port_a_size >= 0 && cell->type != ID($shiftx))
 | |
| 			run_reduce_inport(cell, 'A', max_port_a_size, port_a_signed, did_something);
 | |
| 
 | |
| 		if (max_port_b_size >= 0)
 | |
| 			run_reduce_inport(cell, 'B', max_port_b_size, port_b_signed, did_something);
 | |
| 
 | |
| 		if (cell->hasPort(ID::A) && cell->hasPort(ID::B) && port_a_signed && port_b_signed) {
 | |
| 			SigSpec sig_a = mi.sigmap(cell->getPort(ID::A)), sig_b = mi.sigmap(cell->getPort(ID::B));
 | |
| 			if (GetSize(sig_a) > 0 && sig_a[GetSize(sig_a)-1] == State::S0 &&
 | |
| 					GetSize(sig_b) > 0 && sig_b[GetSize(sig_b)-1] == State::S0) {
 | |
| 				log("Converting cell %s.%s (%s) from signed to unsigned.\n",
 | |
| 						log_id(module), log_id(cell), log_id(cell->type));
 | |
| 				cell->setParam(ID::A_SIGNED, 0);
 | |
| 				cell->setParam(ID::B_SIGNED, 0);
 | |
| 				port_a_signed = false;
 | |
| 				port_b_signed = false;
 | |
| 				did_something = true;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		if (cell->hasPort(ID::A) && !cell->hasPort(ID::B) && port_a_signed) {
 | |
| 			SigSpec sig_a = mi.sigmap(cell->getPort(ID::A));
 | |
| 			if (GetSize(sig_a) > 0 && sig_a[GetSize(sig_a)-1] == State::S0) {
 | |
| 				log("Converting cell %s.%s (%s) from signed to unsigned.\n",
 | |
| 						log_id(module), log_id(cell), log_id(cell->type));
 | |
| 				cell->setParam(ID::A_SIGNED, 0);
 | |
| 				port_a_signed = false;
 | |
| 				did_something = true;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 
 | |
| 		// Reduce size of port Y based on sizes for A and B and unused bits in Y
 | |
| 
 | |
| 		int bits_removed = 0;
 | |
| 		if (port_a_signed && cell->type == ID($shr)) {
 | |
| 			// do not reduce size of output on $shr cells with signed A inputs
 | |
| 		} else {
 | |
| 			while (GetSize(sig) > 0)
 | |
| 			{
 | |
| 				auto bit = sig[GetSize(sig)-1];
 | |
| 				if (keep_bits.count(bit))
 | |
| 					break;
 | |
| 
 | |
| 				auto info = mi.query(bit);
 | |
| 				if (info->is_output || GetSize(info->ports) > 1)
 | |
| 					break;
 | |
| 
 | |
| 				sig.remove(GetSize(sig)-1);
 | |
| 				bits_removed++;
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		if (cell->type.in(ID($pos), ID($add), ID($mul), ID($and), ID($or), ID($xor), ID($sub)))
 | |
| 		{
 | |
| 			bool is_signed = cell->getParam(ID::A_SIGNED).as_bool() || cell->type == ID($sub);
 | |
| 
 | |
| 			int a_size = 0, b_size = 0;
 | |
| 			if (cell->hasPort(ID::A)) a_size = GetSize(cell->getPort(ID::A));
 | |
| 			if (cell->hasPort(ID::B)) b_size = GetSize(cell->getPort(ID::B));
 | |
| 
 | |
| 			int max_y_size = max(a_size, b_size);
 | |
| 
 | |
| 			if (cell->type.in(ID($add), ID($sub)))
 | |
| 				max_y_size++;
 | |
| 
 | |
| 			if (cell->type == ID($mul))
 | |
| 				max_y_size = a_size + b_size;
 | |
| 
 | |
| 			max_y_size = std::max(max_y_size, 1);
 | |
| 
 | |
| 			if (GetSize(sig) > max_y_size) {
 | |
| 				SigSpec extra_bits = sig.extract(max_y_size, GetSize(sig) - max_y_size);
 | |
| 
 | |
| 				bits_removed += GetSize(extra_bits);
 | |
| 				sig.remove(max_y_size, GetSize(extra_bits));
 | |
| 
 | |
| 				SigBit padbit = is_signed ? sig[GetSize(sig)-1] : State::S0;
 | |
| 				module->connect(extra_bits, SigSpec(padbit, GetSize(extra_bits)));
 | |
| 			}
 | |
| 		}
 | |
| 
 | |
| 		if (GetSize(sig) == 0) {
 | |
| 			log("Removed cell %s.%s (%s).\n", log_id(module), log_id(cell), log_id(cell->type));
 | |
| 			module->remove(cell);
 | |
| 			return;
 | |
| 		}
 | |
| 
 | |
| 		if (bits_removed) {
 | |
| 			log("Removed top %d bits (of %d) from port Y of cell %s.%s (%s).\n",
 | |
| 					bits_removed, GetSize(sig) + bits_removed, log_id(module), log_id(cell), log_id(cell->type));
 | |
| 			cell->setPort(ID::Y, sig);
 | |
| 			did_something = true;
 | |
| 		}
 | |
| 
 | |
| 		if (did_something) {
 | |
| 			cell->fixup_parameters();
 | |
| 			run_cell(cell);
 | |
| 		}
 | |
| 	}
 | |
| 
 | |
| 	static int count_nontrivial_wire_attrs(RTLIL::Wire *w)
 | |
| 	{
 | |
| 		int count = w->attributes.size();
 | |
| 		count -= w->attributes.count(ID::src);
 | |
| 		count -= w->attributes.count(ID::unused_bits);
 | |
| 		return count;
 | |
| 	}
 | |
| 
 | |
| 	void run()
 | |
| 	{
 | |
| 		// create a copy as mi.sigmap will be updated as we process the module
 | |
| 		SigMap init_attr_sigmap = mi.sigmap;
 | |
| 		initvals.set(&init_attr_sigmap, module);
 | |
| 
 | |
| 		for (auto w : module->wires()) {
 | |
| 			if (w->get_bool_attribute(ID::keep))
 | |
| 				for (auto bit : mi.sigmap(w))
 | |
| 					keep_bits.insert(bit);
 | |
| 		}
 | |
| 
 | |
| 		for (auto c : module->selected_cells())
 | |
| 			work_queue_cells.insert(c);
 | |
| 
 | |
| 		while (!work_queue_cells.empty())
 | |
| 		{
 | |
| 			work_queue_bits.clear();
 | |
| 			for (auto c : work_queue_cells)
 | |
| 				run_cell(c);
 | |
| 
 | |
| 			work_queue_cells.clear();
 | |
| 			for (auto bit : work_queue_bits)
 | |
| 			for (auto port : mi.query_ports(bit))
 | |
| 				if (module->selected(port.cell))
 | |
| 					work_queue_cells.insert(port.cell);
 | |
| 		}
 | |
| 
 | |
| 		pool<SigSpec> complete_wires;
 | |
| 		for (auto w : module->wires())
 | |
| 			complete_wires.insert(mi.sigmap(w));
 | |
| 
 | |
| 		for (auto w : module->selected_wires())
 | |
| 		{
 | |
| 			int unused_top_bits = 0;
 | |
| 
 | |
| 			if (w->port_id > 0 || count_nontrivial_wire_attrs(w) > 0)
 | |
| 				continue;
 | |
| 
 | |
| 			for (int i = GetSize(w)-1; i >= 0; i--) {
 | |
| 				SigBit bit(w, i);
 | |
| 				auto info = mi.query(bit);
 | |
| 				if (info && (info->is_input || info->is_output || GetSize(info->ports) > 0))
 | |
| 					break;
 | |
| 				unused_top_bits++;
 | |
| 			}
 | |
| 
 | |
| 			if (unused_top_bits == 0 || unused_top_bits == GetSize(w))
 | |
| 				continue;
 | |
| 
 | |
| 			if (complete_wires[mi.sigmap(w).extract(0, GetSize(w) - unused_top_bits)])
 | |
| 				continue;
 | |
| 
 | |
| 			log("Removed top %d bits (of %d) from wire %s.%s.\n", unused_top_bits, GetSize(w), log_id(module), log_id(w));
 | |
| 			Wire *nw = module->addWire(NEW_ID, GetSize(w) - unused_top_bits);
 | |
| 			module->connect(nw, SigSpec(w).extract(0, GetSize(nw)));
 | |
| 			module->swap_names(w, nw);
 | |
| 		}
 | |
| 	}
 | |
| };
 | |
| 
 | |
| struct WreducePass : public Pass {
 | |
| 	WreducePass() : Pass("wreduce", "reduce the word size of operations if possible") { }
 | |
| 	void help() override
 | |
| 	{
 | |
| 		//   |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
 | |
| 		log("\n");
 | |
| 		log("    wreduce [options] [selection]\n");
 | |
| 		log("\n");
 | |
| 		log("This command reduces the word size of operations. For example it will replace\n");
 | |
| 		log("the 32 bit adders in the following code with adders of more appropriate widths:\n");
 | |
| 		log("\n");
 | |
| 		log("    module test(input [3:0] a, b, c, output [7:0] y);\n");
 | |
| 		log("        assign y = a + b + c + 1;\n");
 | |
| 		log("    endmodule\n");
 | |
| 		log("\n");
 | |
| 		log("Options:\n");
 | |
| 		log("\n");
 | |
| 		log("    -memx\n");
 | |
| 		log("        Do not change the width of memory address ports. Use this options in\n");
 | |
| 		log("        flows that use the 'memory_memx' pass.\n");
 | |
| 		log("\n");
 | |
| 		log("    -mux_undef\n");
 | |
| 		log("        remove 'undef' inputs from $mux, $pmux and $_MUX_ cells\n");
 | |
| 		log("\n");
 | |
| 		log("    -keepdc\n");
 | |
| 		log("        Do not optimize explicit don't-care values.\n");
 | |
| 		log("\n");
 | |
| 	}
 | |
| 	void execute(std::vector<std::string> args, Design *design) override
 | |
| 	{
 | |
| 		WreduceConfig config;
 | |
| 		bool opt_memx = false;
 | |
| 
 | |
| 		log_header(design, "Executing WREDUCE pass (reducing word size of cells).\n");
 | |
| 
 | |
| 		size_t argidx;
 | |
| 		for (argidx = 1; argidx < args.size(); argidx++) {
 | |
| 			if (args[argidx] == "-memx") {
 | |
| 				opt_memx = true;
 | |
| 				continue;
 | |
| 			}
 | |
| 			if (args[argidx] == "-keepdc") {
 | |
| 				config.keepdc = true;
 | |
| 				continue;
 | |
| 			}
 | |
| 			if (args[argidx] == "-mux_undef") {
 | |
| 				config.mux_undef = true;
 | |
| 				continue;
 | |
| 			}
 | |
| 			break;
 | |
| 		}
 | |
| 		extra_args(args, argidx, design);
 | |
| 
 | |
| 		for (auto module : design->selected_modules())
 | |
| 		{
 | |
| 			if (module->has_processes_warn())
 | |
| 				continue;
 | |
| 
 | |
| 			for (auto c : module->selected_cells())
 | |
| 			{
 | |
| 				if (c->type.in(ID($reduce_and), ID($reduce_or), ID($reduce_xor), ID($reduce_xnor), ID($reduce_bool),
 | |
| 						ID($lt), ID($le), ID($eq), ID($ne), ID($eqx), ID($nex), ID($ge), ID($gt),
 | |
| 						ID($logic_not), ID($logic_and), ID($logic_or)) && GetSize(c->getPort(ID::Y)) > 1) {
 | |
| 					SigSpec sig = c->getPort(ID::Y);
 | |
| 					if (!sig.has_const()) {
 | |
| 						c->setPort(ID::Y, sig[0]);
 | |
| 						c->setParam(ID::Y_WIDTH, 1);
 | |
| 						sig.remove(0);
 | |
| 						module->connect(sig, Const(0, GetSize(sig)));
 | |
| 					}
 | |
| 				}
 | |
| 
 | |
| 				if (c->type.in(ID($div), ID($mod), ID($divfloor), ID($modfloor), ID($pow)))
 | |
| 				{
 | |
| 					SigSpec A = c->getPort(ID::A);
 | |
| 					int original_a_width = GetSize(A);
 | |
| 					if (c->getParam(ID::A_SIGNED).as_bool()) {
 | |
| 						while (GetSize(A) > 1 && A[GetSize(A)-1] == State::S0 && A[GetSize(A)-2] == State::S0)
 | |
| 							A.remove(GetSize(A)-1, 1);
 | |
| 					} else {
 | |
| 						while (GetSize(A) > 0 && A[GetSize(A)-1] == State::S0)
 | |
| 							A.remove(GetSize(A)-1, 1);
 | |
| 					}
 | |
| 					if (original_a_width != GetSize(A)) {
 | |
| 						log("Removed top %d bits (of %d) from port A of cell %s.%s (%s).\n",
 | |
| 								original_a_width-GetSize(A), original_a_width, log_id(module), log_id(c), log_id(c->type));
 | |
| 						c->setPort(ID::A, A);
 | |
| 						c->setParam(ID::A_WIDTH, GetSize(A));
 | |
| 					}
 | |
| 
 | |
| 					SigSpec B = c->getPort(ID::B);
 | |
| 					int original_b_width = GetSize(B);
 | |
| 					if (c->getParam(ID::B_SIGNED).as_bool()) {
 | |
| 						while (GetSize(B) > 1 && B[GetSize(B)-1] == State::S0 && B[GetSize(B)-2] == State::S0)
 | |
| 							B.remove(GetSize(B)-1, 1);
 | |
| 					} else {
 | |
| 						while (GetSize(B) > 0 && B[GetSize(B)-1] == State::S0)
 | |
| 							B.remove(GetSize(B)-1, 1);
 | |
| 					}
 | |
| 					if (original_b_width != GetSize(B)) {
 | |
| 						log("Removed top %d bits (of %d) from port B of cell %s.%s (%s).\n",
 | |
| 								original_b_width-GetSize(B), original_b_width, log_id(module), log_id(c), log_id(c->type));
 | |
| 						c->setPort(ID::B, B);
 | |
| 						c->setParam(ID::B_WIDTH, GetSize(B));
 | |
| 					}
 | |
| 				}
 | |
| 
 | |
| 				if (!opt_memx && c->type.in(ID($memrd), ID($memrd_v2), ID($memwr), ID($memwr_v2), ID($meminit), ID($meminit_v2))) {
 | |
| 					IdString memid = c->getParam(ID::MEMID).decode_string();
 | |
| 					RTLIL::Memory *mem = module->memories.at(memid);
 | |
| 					if (mem->start_offset >= 0) {
 | |
| 						int cur_addrbits = c->getParam(ID::ABITS).as_int();
 | |
| 						int max_addrbits = ceil_log2(mem->start_offset + mem->size);
 | |
| 						if (cur_addrbits > max_addrbits) {
 | |
| 							log("Removed top %d address bits (of %d) from memory %s port %s.%s (%s).\n",
 | |
| 									cur_addrbits-max_addrbits, cur_addrbits,
 | |
| 									c->type == ID($memrd) ? "read" : c->type == ID($memwr) ? "write" : "init",
 | |
| 									log_id(module), log_id(c), log_id(memid));
 | |
| 							c->setParam(ID::ABITS, max_addrbits);
 | |
| 							c->setPort(ID::ADDR, c->getPort(ID::ADDR).extract(0, max_addrbits));
 | |
| 						}
 | |
| 					}
 | |
| 				}
 | |
| 			}
 | |
| 
 | |
| 			WreduceWorker worker(&config, module);
 | |
| 			worker.run();
 | |
| 		}
 | |
| 	}
 | |
| } WreducePass;
 | |
| 
 | |
| PRIVATE_NAMESPACE_END
 | |
| 
 |