Wrap A and B in sigmap

diff --git a/passes/opt/wreduce.cc b/passes/opt/wreduce.cc
index 23e14f7f53863a184d78cf3db22ce4f7748a0e2f..294f0d57e15c35de14e204b87b442e32704ba315 100644 (file)
--- a/passes/opt/wreduce.cc
+++ b/passes/opt/wreduce.cc
@@ -366,8 +366,8 @@ struct WreduceWorker
                }
 
                if (cell->type.in("$add", "$sub")) {
-                       SigSpec A = cell->getPort("\\A");
-                       SigSpec B = cell->getPort("\\B");
+                       SigSpec A = mi.sigmap(cell->getPort("\\A"));
+                       SigSpec B = mi.sigmap(cell->getPort("\\B"));
                        bool sub = cell->type == "$sub";
 
                        int i;