correct nets for experiment2

[soclayout.git] / experiments2 / test_partsig.py

diff --git a/experiments2/test_partsig.py b/experiments2/test_partsig.py
index 73a63339e23d16805287378608f3d678bc4aeabf..447f88c3298a1cb687054df0a183d0d322de6d21 100644 (file)
--- a/experiments2/test_partsig.py
+++ b/experiments2/test_partsig.py
@@ -17,7 +17,8 @@ class TestLS(Elaboratable):
         self.a = PartitionedSignal(partpoints, width, name="a")
         self.b = PartitionedSignal(partpoints, width, name="b")
         self.ls_output = Signal(width) # left shift
-        self.dummy_output = Signal(width) # left shift
+        self.dummy_output = Signal(width)
+        self.dummy_mask = Signal(self.partpoints.shape()) 
 
     def elaborate(self, platform):
         m = Module()
@@ -26,6 +27,8 @@ class TestLS(Elaboratable):
         self.a.set_module(m)
         self.b.set_module(m)
         # left shift
+        wid = len(self.partpoints)
+        sync += self.dummy_mask.eq(self.partpoints) 
         sync += self.dummy_output.eq(self.b.sig) # stops sigs being ignored
         sync += self.ls_output.eq(self.a << self.b)
         ppts = self.partpoints
@@ -35,7 +38,8 @@ class TestLS(Elaboratable):
     def ports(self):
         return [self.a.sig, self.b.sig,
                 self.ls_output,
-                self.dummy_output]
+                self.dummy_output,
+                self.dummy_mask]
 
 
 # XXX this is for coriolis2 experimentation