add test for back with distance zero

risc0 · Jan 13, 2025 · 33dc293 · 33dc293
1 parent deedcc1
commit 33dc293
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/zirgen/compiler/picus/picus.cpp b/zirgen/compiler/picus/picus.cpp
@@ -296,7 +296,6 @@ class PicusPrinter {
   }
 
   void visitOp(zirgen::Zhlt::BackOp back) {
-    auto callee = back.getCallee();
     size_t distance = back.getDistance().getZExtValue();
     AnySignal signal = signalize(freshName(), back.getType());
     if (distance > 0) {

diff --git a/zirgen/compiler/picus/test/back-zero.zir b/zirgen/compiler/picus/test/back-zero.zir
@@ -0,0 +1,19 @@
+// RUN: zirgen %s --emit=picus | FileCheck %s
+
+// CHECK: (prime-number 2013265921)
+// CHECK: (begin-module Count)
+// CHECK: (input x0)
+// CHECK-NEXT: (output layout_a__super__super)
+// CHECK-NEXT: (output result_a__super__super)
+// CHECK-NEXT: (output result_a_reg__super)
+// CHECK-NEXT: (call [{{.*}}] Reg [{{.*}}])
+// CHECK-NEXT: (assert (= result_a__super__super x2__super__super))
+// CHECK-NEXT: (assert (= result_a_reg__super x2_reg__super))
+// CHECK-NEXT: (end-module)
+
+#[picus]
+component Count(first: Val) {
+  public a : Reg;
+  a := Reg(a@0);
+}
+