From bc2d9d1f337e149fb3ba997fe6cc8436891bfaaf Mon Sep 17 00:00:00 2001
From: williamzhu17 <williammzhu@gmail.com>
Date: Tue, 1 Apr 2025 11:10:50 -0700
Subject: [PATCH] added deeper cases for gates

---
 tests/silimate/extract_reduce.ys | 237 ++++++++++++++++++++++++++++++-
 1 file changed, 236 insertions(+), 1 deletion(-)

diff --git a/tests/silimate/extract_reduce.ys b/tests/silimate/extract_reduce.ys
index 50bb09078..8b642e70c 100644
--- a/tests/silimate/extract_reduce.ys
+++ b/tests/silimate/extract_reduce.ys
@@ -95,7 +95,7 @@ log -pop
 
 
 
-log -header "No off-chain"
+log -header "No off-chain for AND"
 log -push
 design -reset
 read_verilog <<EOF
@@ -176,6 +176,85 @@ select -assert-count 1 t:$reduce_and r:A_WIDTH=3 %i
 design -reset
 log -pop
 
+
+
+log -header "No off-chain with branches for AND"
+log -push
+design -reset
+read_verilog <<EOF
+module top (
+  input wire [10:0] a,
+  output wire x,
+  output wire y
+);
+  wire w0;
+
+  assign w0 = a[0] & a[1] & a[2] & (a[3] & a[4]) & a[5];
+  assign x  = w0 & a[6] & a[7] & (a[8] & a[9]) & a[10];
+
+  // Off-chain use of w0
+  assign y = w0;
+endmodule
+EOF
+check -assert
+
+# Check equivalence after extract_reduce
+equiv_opt -assert extract_reduce
+
+# Load design and run opt_clean to remove unnecessary wires
+design -load postopt
+opt_clean
+
+# Check final design has correct number of gates
+select -assert-count 0 t:$and
+select -assert-count 2 t:$reduce_and
+
+# Check that both gates are 6 bits wide
+select -assert-none t:$reduce_and r:A_WIDTH!=6 %i
+
+design -reset
+log -pop
+
+
+
+log -header "Allow off-chain with branches for AND"
+log -push
+design -reset
+read_verilog <<EOF
+module top (
+  input wire [10:0] a,
+  output wire x,
+  output wire y
+);
+  wire w0;
+
+  assign w0 = a[0] & a[1] & a[2] & (a[3] & a[4]) & a[5];
+  assign x  = w0 & a[6] & a[7] & (a[8] & a[9]) & a[10];
+
+  // Off-chain use of w0
+  assign y = w0;
+endmodule
+EOF
+check -assert
+
+# Check equivalence after extract_reduce
+equiv_opt -assert extract_reduce -allow-off-chain
+
+# Load design and run opt_clean to remove unnecessary wires
+design -load postopt
+opt_clean
+
+# Check final design has correct number of gates
+select -assert-count 0 t:$and
+select -assert-count 2 t:$reduce_and
+
+# Check that only one gate has a width of 11 and one gate has a width of 6
+select -assert-count 1 t:$reduce_and r:A_WIDTH=11 %i
+select -assert-count 1 t:$reduce_and r:A_WIDTH=6 %i
+
+design -reset
+log -pop
+
 ###################################################################
 # Extract Reduce OR Gates Tests
 ###################################################################
@@ -354,6 +433,85 @@ select -assert-count 1 t:$reduce_or r:A_WIDTH=3 %i
 design -reset
 log -pop
 
+
+
+log -header "No off-chain with branches for OR"
+log -push
+design -reset
+read_verilog <<EOF
+module top (
+  input wire [10:0] a,
+  output wire x,
+  output wire y
+);
+  wire w0;
+
+  assign w0 = a[0] | a[1] | a[2] | (a[3] | a[4]) | a[5];
+  assign x  = w0 | a[6] | a[7] | (a[8] | a[9]) | a[10];
+
+  // Off-chain use of w0
+  assign y = w0;
+endmodule
+EOF
+check -assert
+
+# Check equivalence after extract_reduce
+equiv_opt -assert extract_reduce
+
+# Load design and run opt_clean to remove unnecessary wires
+design -load postopt
+opt_clean
+
+# Check final design has correct number of gates
+select -assert-count 0 t:$and
+select -assert-count 2 t:$reduce_or
+
+# Check that both gates are 6 bits wide
+select -assert-none t:$reduce_or r:A_WIDTH!=6 %i
+
+design -reset
+log -pop
+
+
+
+log -header "Allow off-chain with branches for OR"
+log -push
+design -reset
+read_verilog <<EOF
+module top (
+  input wire [10:0] a,
+  output wire x,
+  output wire y
+);
+  wire w0;
+
+  assign w0 = a[0] | a[1] | a[2] | (a[3] | a[4]) | a[5];
+  assign x  = w0 | a[6] | a[7] | (a[8] | a[9]) | a[10];
+
+  // Off-chain use of w0
+  assign y = w0;
+endmodule
+EOF
+check -assert
+
+# Check equivalence after extract_reduce
+equiv_opt -assert extract_reduce -allow-off-chain
+
+# Load design and run opt_clean to remove unnecessary wires
+design -load postopt
+opt_clean
+
+# Check final design has correct number of gates
+select -assert-count 0 t:$and
+select -assert-count 2 t:$reduce_or
+
+# Check that only one gate has a width of 11 and one gate has a width of 6
+select -assert-count 1 t:$reduce_or r:A_WIDTH=11 %i
+select -assert-count 1 t:$reduce_or r:A_WIDTH=6 %i
+
+design -reset
+log -pop
+
 ###################################################################
 # Extract Reduce XOR Gates Tests
 ###################################################################
@@ -535,6 +693,83 @@ log -pop
 
 
 
+log -header "No off-chain with branches for XOR"
+log -push
+design -reset
+read_verilog <<EOF
+module top (
+  input wire [10:0] a,
+  output wire x,
+  output wire y
+);
+  wire w0;
+
+  assign w0 = a[0] ^ a[1] ^ a[2] ^ (a[3] ^ a[4]) ^ a[5];
+  assign x  = w0 ^ a[6] ^ a[7] ^ (a[8] ^ a[9]) ^ a[10];
+
+  // Off-chain use of w0
+  assign y = w0;
+endmodule
+EOF
+check -assert
+
+# Check equivalence after extract_reduce
+equiv_opt -assert extract_reduce
+
+# Load design and run opt_clean to remove unnecessary wires
+design -load postopt
+opt_clean
+
+# Check final design has correct number of gates
+select -assert-count 0 t:$xor
+select -assert-count 2 t:$reduce_xor
+
+# Check that both gates are 6 bits wide
+select -assert-none t:$reduce_xor r:A_WIDTH!=6 %i
+
+design -reset
+log -pop
+
+
+
+log -header "Allow off-chain with branches for XOR"
+log -push
+design -reset
+read_verilog <<EOF
+module top (
+  input wire [10:0] a,
+  output wire x,
+  output wire y
+);
+  wire w0;
+
+  assign w0 = a[0] ^ a[1] ^ a[2] ^ (a[3] ^ a[4]) ^ a[5];
+  assign x  = w0 ^ a[6] ^ a[7] ^ (a[8] ^ a[9]) ^ a[10];
+
+  // Off-chain use of w0
+  assign y = w0;
+endmodule
+EOF
+check -assert
+
+# Check equivalence after extract_reduce
+equiv_opt -assert extract_reduce -allow-off-chain
+
+# Load design and run opt_clean to remove unnecessary wires
+design -load postopt
+opt_clean
+
+# Check final design has correct number of gates
+select -assert-count 0 t:$xor
+select -assert-count 2 t:$reduce_xor
+
+# Check that only one gate has a width of 11 and one gate has a width of 6
+select -assert-count 1 t:$reduce_xor r:A_WIDTH=11 %i
+select -assert-count 1 t:$reduce_xor r:A_WIDTH=6 %i
+
+design -reset
+log -pop
+
 ###################################################################
 # Extract PMUX from MUX Chains Tests
 ###################################################################