comptime, check files, blog post

Author: ngmachado

ora-example/comptime/comptime_basics.ora

@@ -0,0 +1,24 @@
+contract ComptimeBasics {
+    pub fn basic_arithmetic() -> u256 {
+        const a: u256 = 10;
+        const b: u256 = 20;
+        const c: u256 = a * b + 5;
+        return c;
+    }
+
+    pub fn refined_constants() {
+        const x: MinValue<u256, 100> = 200;
+        const y: MinValue<u256, 50> = 100;
+        const prod = x * y; // comptime: 20000
+
+        const r: InRange<u256, 20, 200> = 55;
+        const sum = r + 5; // comptime: 60
+    }
+
+    pub fn comptime_block_example() -> u256 {
+        const res: u256 = comptime {
+            return 42;
+        };
+        return res;
+    }
+}

ora-example/comptime/comptime_folds.ora

@@ -0,0 +1,82 @@
+struct Pair {
+    a: u256;
+    b: u256;
+}
+
+enum Status {
+    A,
+    B,
+    C,
+}
+
+contract ComptimeFolds {
+    pub fn compare_bool() -> bool {
+        const a: u256 = 10;
+        const b: u256 = 20;
+        const ok: bool = a < b;
+        return ok;
+    }
+
+    pub fn bitwise_and_shift() -> u256 {
+        const x: u256 = 6;  // 110
+        const y: u256 = 3;  // 011
+        const z: u256 = (x & y) | (x ^ y) << 1;
+        return z;
+    }
+
+    pub fn shift_ops() -> u256 {
+        const a: u256 = 1 << 8;
+        const b: u256 = 256 >> 3;
+        return a + b;
+    }
+
+    pub fn switch_expr() -> u256 {
+        const x: u256 = 7;
+        const y: u256 = switch (x) {
+            0...5 => 10,
+            6...10 => 20,
+            else => 30,
+        };
+        return y;
+    }
+
+    pub fn array_literal_sum() -> u256 {
+        const arr: [u256; 3] = [1, 2, 3];
+        const sum: u256 = arr[0] + arr[1] + arr[2];
+        return sum;
+    }
+
+    pub fn struct_field_access() -> u256 {
+        const v: u256 = (Pair { a: 10, b: 32 }).a + (Pair { a: 10, b: 32 }).b;
+        return v;
+    }
+
+    pub fn enum_compare() -> bool {
+        const ok: bool = Status.B == Status.B;
+        return ok;
+    }
+
+    pub fn enum_switch() -> u256 {
+        const v: u256 = switch (Status.B) {
+            Status.A => 1,
+            Status.B => 2,
+            Status.C => 3,
+        };
+        return v;
+    }
+
+    pub fn unary_ops() -> u256 {
+        const a: u256 = 1;
+        const b: u256 = ~a;
+        const c: bool = !(a == 2);
+        return b + (if (c) 1 else 0);
+    }
+
+    pub fn cast_ops() -> u256 {
+        const a: u256 = (5 as u256);
+        const b: u256 = (10 as! u256);
+        return a + b;
+    }
+
+    // builtin divs removed for now
+}

ora-example/refinements/arithmetic_test.ora

@@ -58,9 +58,9 @@ contract ArithmeticRefinements {
         let diff4 = e - f; // Type: MaxValue<u256, 500> (1000 - 500)
 
         // InRange - InRange = InRange
-        let g: InRange<u256, 10, 100> = 50;
-        let h: InRange<u256, 20, 200> = 100;
-        let diff5 = g - h; // Type: InRange<u256, 0, 80> (min: 10-200, max: 100-20)
+        let g: InRange<u256, 100, 100> = 100;
+        let h: InRange<u256, 20, 50> = 50;
+        let diff5 = g - h; // Type: InRange<u256, 50, 80> (min: 100-50, max: 100-20)
     }
 
     // ============================================================================
@@ -147,4 +147,3 @@ contract ArithmeticRefinements {
         let result2 = result1 - b; // Type: MinValue<u256, 500> (conservative: 1000 - 500)
     }
 }
-

scripts/generate-mlir-auto-checks.py

@@ -0,0 +1,105 @@
+#!/usr/bin/env python3
+import re
+import subprocess
+from pathlib import Path
+
+REPO = Path(__file__).resolve().parents[1]
+ORA_BIN = REPO / "zig-out" / "bin" / "ora"
+CHECKS_DIR = REPO / "tests" / "mlir"
+
+SKIP_MARKERS = ["/invalid/", "/negative_tests/"]
+
+
+def should_skip(path: Path) -> bool:
+    s = str(path)
+    if any(m in s for m in SKIP_MARKERS):
+        return True
+    name = path.name
+    if name.startswith("fail_") or "/fail_" in s or "fail_region_err" in s:
+        return True
+    if "/logs/log_fail_" in s:
+        return True
+    return False
+
+
+def main() -> int:
+    if not ORA_BIN.exists():
+        print(f"error: ora binary not found at {ORA_BIN}. Run 'zig build' first.")
+        return 1
+
+    CHECKS_DIR.mkdir(parents=True, exist_ok=True)
+
+    # collect already covered inputs
+    covered = set()
+    for check in CHECKS_DIR.glob("*.check"):
+        try:
+            text = check.read_text()
+        except Exception:
+            continue
+        m = re.search(r"^// INPUT:\s*(.+)$", text, re.M)
+        if m:
+            covered.add(m.group(1).strip())
+
+    ora_files = sorted((REPO / "ora-example").rglob("*.ora"))
+
+    created = 0
+    failed = []
+
+    for path in ora_files:
+        rel = path.relative_to(REPO).as_posix()
+        if should_skip(path):
+            continue
+        if rel in covered:
+            continue
+
+        try:
+            proc = subprocess.run(
+                [str(ORA_BIN), "--emit-mlir", str(path)],
+                cwd=REPO,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True,
+                timeout=30,
+            )
+        except Exception as e:
+            failed.append((rel, f"exception {e}"))
+            continue
+        if proc.returncode != 0:
+            failed.append((rel, f"exit {proc.returncode}"))
+            continue
+
+        out = proc.stdout
+        marker = "Ora MLIR (before conversion)"
+        if marker not in out:
+            failed.append((rel, "missing Ora MLIR marker"))
+            continue
+
+        ora = out.split(marker, 1)[1]
+        func_names = re.findall(r"func\.func\s+@([A-Za-z0-9_]+)", ora)
+
+        check_name = "auto__" + rel.replace("/", "__").replace(".ora", ".check")
+        check_path = CHECKS_DIR / check_name
+
+        lines = []
+        lines.append("// AUTO-GENERATED. Do not edit by hand.")
+        lines.append(f"// INPUT: {rel}")
+        lines.append("// CHECK-NOT: builtin.unrealized_conversion_cast")
+        if func_names:
+            for name in func_names:
+                lines.append(f"// CHECK-LABEL: func.func @{name}")
+        else:
+            lines.append("// CHECK: module")
+
+        check_path.write_text("\n".join(lines) + "\n")
+        created += 1
+
+    print(f"created {created} checks")
+    if failed:
+        print("failed:")
+        for rel, reason in failed:
+            print(f"  {rel}: {reason}")
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

scripts/run-mlir-checks.sh

@@ -0,0 +1,46 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+FILECHECK_BIN=${FILECHECK:-./vendor/llvm-project/build-mlir/bin/FileCheck}
+ORA_BIN=${ORA_BIN:-./zig-out/bin/ora}
+
+if ! command -v "$FILECHECK_BIN" >/dev/null 2>&1; then
+  echo "error: FileCheck not found. Set FILECHECK=/path/to/FileCheck" >&2
+  exit 1
+fi
+
+if [ ! -x "$ORA_BIN" ]; then
+  echo "error: ora binary not found at $ORA_BIN" >&2
+  exit 1
+fi
+
+fail=0
+while IFS= read -r -d '' check; do
+  input=$(rg -m1 "^// INPUT:" "$check" | sed -E 's#^// INPUT: ##')
+  if [ -z "$input" ]; then
+    echo "error: missing // INPUT: in $check" >&2
+    fail=1
+    continue
+  fi
+  if [[ "$input" == path/to/* ]]; then
+    continue
+  fi
+
+  if [ ! -f "$input" ]; then
+    echo "error: input file not found: $input (from $check)" >&2
+    fail=1
+    continue
+  fi
+
+  echo "[mlir-check] $check -> $input"
+  "$ORA_BIN" --emit-mlir "$input" 2>/dev/null | \
+    awk '
+      /Ora MLIR \(before conversion\)/ {f=1; print; next}
+      /^module / {if (!f) f=1}
+      f {print}
+    ' | \
+    "$FILECHECK_BIN" "$check" || fail=1
+
+done < <(find tests/mlir -name "*.check" -print0 | sort -z)
+
+exit $fail

src/ast/expressions.zig

@@ -431,6 +431,7 @@ pub const CastExpr = struct {
 pub const ComptimeExpr = struct {
     block: BlockNode,
     span: SourceSpan,
+    type_info: TypeInfo = TypeInfo.unknown(),
 };
 
 /// old() expressions for postconditions