From eb5abf9c12e86fbe8222cf5317f08bfac58ccaa2 Mon Sep 17 00:00:00 2001
From: Krystine Sherwin <93062060+KrystalDelusion@users.noreply.github.com>
Date: Mon, 31 Mar 2025 16:39:13 +1300
Subject: [PATCH] test_cell: Add $bweqx, kinda

It fails in sat, and in `ConstEval::eval()`.  But it's pseudo supported in test_cell at least.

Also fix `RTLIL::Cell::fixup_parameters()` for $bweqx.
---
 kernel/rtlil.cc           |  2 +-
 passes/tests/test_cell.cc | 15 ++++++++++-----
 2 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/kernel/rtlil.cc b/kernel/rtlil.cc
index 2b085fd4e..790480edf 100644
--- a/kernel/rtlil.cc
+++ b/kernel/rtlil.cc
@@ -4063,7 +4063,7 @@ void RTLIL::Cell::fixup_parameters(bool set_a_signed, bool set_b_signed)
 			type.begins_with("$verific$") || type.begins_with("$array:") || type.begins_with("$extern:"))
 		return;
 
-	if (type == ID($buf) || type == ID($mux) || type == ID($pmux) || type == ID($bmux) || type == ID($bwmux)) {
+	if (type == ID($buf) || type == ID($mux) || type == ID($pmux) || type == ID($bmux) || type == ID($bwmux) || type == ID($bweqx)) {
 		parameters[ID::WIDTH] = GetSize(connections_[ID::Y]);
 		if (type.in(ID($pmux), ID($bmux)))
 			parameters[ID::S_WIDTH] = GetSize(connections_[ID::S]);
diff --git a/passes/tests/test_cell.cc b/passes/tests/test_cell.cc
index 203768ea6..d1beafd7b 100644
--- a/passes/tests/test_cell.cc
+++ b/passes/tests/test_cell.cc
@@ -429,17 +429,20 @@ static RTLIL::Cell* create_gold_module(RTLIL::Design *design, RTLIL::IdString ce
 		cell->setPort(ID::Y, wire);
 	}
 
-	if (cell_type == ID($bwmux))
+	if (cell_type.in(ID($bwmux), ID($bweqx)))
 	{
 		int a_size = GetSize(cell->getPort(ID::A));
 		wire = module->addWire(ID::B);
 		wire->width = a_size;
 		wire->port_output = true;
 		cell->setPort(ID::B, wire);
-		wire = module->addWire(ID::S);
-		wire->width = a_size;
-		wire->port_output = true;
-		cell->setPort(ID::S, wire);
+		if (cell_type == ID($bwmux))
+		{
+			wire = module->addWire(ID::S);
+			wire->width = a_size;
+			wire->port_output = true;
+			cell->setPort(ID::S, wire);
+		}
 		wire = module->addWire(ID::Y);
 		wire->width = a_size;
 		wire->port_output = true;
@@ -1112,6 +1115,8 @@ struct TestCellPass : public Pass {
 			cell_types[ID($pmux)] = "*";
 		if (noeval)
 			cell_types[ID($bwmux)] = "A";
+		if (nosat && noeval)
+			cell_types[ID($bweqx)] = "A";
 
 		cell_types[ID($slice)] = "A";
 		cell_types[ID($concat)] = "AB";