Fixes for LP384 in some make_xx.py, etc

1 files changed, 9 insertions, 4 deletions

diff --git a/icefuzz/make_mesh.py b/icefuzz/make_mesh.py
index 96e6044..1a69ca4 100644
--- a/icefuzz/make_mesh.py
+++ b/icefuzz/make_mesh.py
@@ -9,15 +9,20 @@ os.mkdir("work_mesh")
 
 for idx in range(num):
     with open("work_mesh/mesh_%02d.v" % idx, "w") as f:
-        print("module top(input [39:0] a, output [39:0] y);", file=f)
+        if os.getenv('ICE384PINS'):
+            print("module top(input [9:0] a, output [9:0] y);", file=f)
+        else:
+            print("module top(input [39:0] a, output [39:0] y);", file=f)
         print("  assign y = a;", file=f)
         print("endmodule", file=f)
     with open("work_mesh/mesh_%02d.pcf" % idx, "w") as f:
         p = np.random.permutation(pins)
-        for i in range(40):
+        if os.getenv('ICE384PINS'): r = 10
+        else: r = 40
+        for i in range(r):
             print("set_io a[%d] %s" % (i, p[i]), file=f)
-        for i in range(40):
-            print("set_io y[%d] %s" % (i, p[40+i]), file=f)
+        for i in range(r):
+            print("set_io y[%d] %s" % (i, p[r+i]), file=f)
 
 with open("work_mesh/Makefile", "w") as f:
     print("all: %s" % " ".join(["mesh_%02d.bin" % i for i in range(num)]), file=f)