@fraczak/k 7.0.1 → 7.1.0

package/scripts/perf-multiplication.mjs

@@ -7,9 +7,9 @@
  * across available execution backends:
  *   - LLVM (native binary via Clang/LLVM)
  *   - WebAssembly (persistent runner via Wasm)
+ *   - PKIR (persistent runner via Wasm/PKIR)
  *   - kVM Interpreter (envelope-free bytecode engine)
  *   - Native JS (envelope-free convergence engine)
- *   - ARM64 (native binary, on supported Linux aarch64 hosts)
  *
  * Modes:
  *   --mode squaring : Successive squarings starting from --input
@@ -67,7 +67,7 @@ function printHelp() {
 
 Multiplication Algorithm & Multi-Backend Performance Harness.
 Compares multiplication algorithms ('times' vs 'karatsuba') across
-available execution backends (LLVM, Wasm, kVM, JS, ARM64).
+available execution backends (LLVM, Wasm, PKIR, kVM, JS).
 
 Options:
   --mode <squaring|sizes>
@@ -76,7 +76,8 @@ Options:
       (default: squaring)
   --backends <csv>
       Comma-separated list of execution backends to test:
-      llvm, wasm, kvm, js, arm64. (default: llvm,wasm,kvm,js)
+      llvm, llvm-pkir, wasm, pkir, kvm, js.
+      (default: llvm,llvm-pkir,wasm,pkir,kvm,js)
   --algos <csv>
       Comma-separated list of multiplication algorithms to test:
       times, karatsuba. (default: times,karatsuba)
@@ -106,11 +107,7 @@ Environment Variables:
   LLVM_PIPELINE
       Set to '1' to use persistent pipelined LLVM runner mode.
   LLVM_SPAWN_PER_CALL
-      Set to '1' to spawn a new LLVM process on each invocation.
-  ARM64_PIPELINE
-      Set to '1' to use persistent pipelined ARM64 runner mode.
-  ARM64_SPAWN_PER_CALL
-      Set to '1' to spawn a new ARM64 process on each invocation.`);
+      Set to '1' to spawn a new LLVM process on each invocation.`);
 }
 
 if (hasFlag("-h") || hasFlag("--help")) {
@@ -119,7 +116,11 @@ if (hasFlag("-h") || hasFlag("--help")) {
 }
 
 const mode = getArg("--mode", "squaring"); // 'squaring' or 'sizes'
-const backendsArg = getArg("--backends", "llvm,wasm,kvm,js");
+const defaultBackends = "llvm,llvm-pkir,wasm,pkir,kvm,js";
+const backendsArg = getArg(
+  "--backends",
+  process.env.BACKENDS || defaultBackends
+);
 const selectedBackends = backendsArg
   .split(",").map(b => b.trim().toLowerCase()).filter(Boolean);
 const algosArg = getArg("--algos", "times,karatsuba");
@@ -244,7 +245,7 @@ const cacheDir = makeCacheDir("k-mult-perf-");
 // 2. Prepare Squaring Relations & Binaries
 // For squaring: relation name `s_${algo} = {()x, ()y} ${algo}; s_${algo}`
 const algoRelations = {};
-const algoServers = { llvm: {}, wasm: {} };
+const algoServers = { llvm: {}, "llvm-pkir": {}, wasm: {}, pkir: {} };
 
 let commonState = null;
 
@@ -277,7 +278,10 @@ for (const algo of selectedAlgos) {
     const exePath = path.join(cacheDir, `perf_mult_llvm_${algo}`);
     try {
       compileObjectToExecutable(
-        algoRelations[algo].object, exePath, { clangOpt: "-O2" }
+        algoRelations[algo].object, exePath, {
+          clangOpt: "-O2",
+          inputPattern: parseIntValue("0").pattern
+        }
       );
       algoServers.llvm[algo] = new PersistentExecutable(exePath);
     } catch (e) {
@@ -287,6 +291,26 @@ for (const algo of selectedAlgos) {
     }
   }
 
+  // LLVM (via PKIR)
+  if (selectedBackends.includes("llvm-pkir")) {
+    const exePath = path.join(cacheDir, `perf_mult_llvm_pkir_${algo}`);
+    try {
+      compileObjectToExecutable(
+        algoRelations[algo].object, exePath, {
+          clangOpt: "-O2",
+          inputPattern: parseIntValue("0").pattern,
+          usePKIR: true
+        }
+      );
+      algoServers["llvm-pkir"][algo] =
+        new PersistentExecutable(exePath);
+    } catch (e) {
+      console.warn(
+        `[WARN] Failed to compile LLVM-PKIR for ${algo}: ${e.message}`
+      );
+    }
+  }
+
   // Wasm
   if (selectedBackends.includes("wasm")) {
     const wasmPath = path.join(cacheDir, `perf_mult_wasm_${algo}.wasm`);
@@ -311,15 +335,36 @@ for (const algo of selectedAlgos) {
       );
     }
   }
+
+  // PKIR
+  if (selectedBackends.includes("pkir")) {
+    const pkirPath = path.join(cacheDir, `perf_mult_pkir_${algo}.wasm`);
+    const exePath = path.join(cacheDir, `perf_mult_pkir_${algo}.exe`);
+    try {
+      const wasmBuf = await compileWasmArtifactFromObject(
+        algoRelations[algo].object, { entry: "__main__", usePKIR: true }
+      );
+      fs.writeFileSync(pkirPath, wasmBuf);
+      const runBin = fileURLToPath(
+        import.meta.resolve("../backends/wasm/bin/k-wasm-run.mjs")
+      );
+      fs.writeFileSync(
+        exePath,
+        `#!/bin/sh\nexec node "${runBin}" "${pkirPath}" "$@"\n`,
+        { mode: 0o755 }
+      );
+      algoServers.pkir[algo] = new PersistentExecutable(exePath);
+    } catch (e) {
+      console.warn(
+        `[WARN] Failed to compile PKIR for ${algo}: ${e.message}`
+      );
+    }
+  }
 }
 
 // 3. Squaring Mode Benchmark
 if (mode === "squaring") {
-  const sInitVal = valueForCode(
-    parseIntValue(squaringInputText),
-    commonState.typeAliases.int,
-    codes.find
-  );
+  const sInitVal = parseIntValue(squaringInputText);
   const { inputWire: sInitWire } = wireInput(sInitVal);
 
   const results = {};
@@ -333,8 +378,8 @@ if (mode === "squaring") {
   const stepBits = [];
 
   try {
-    // A. Run persistent backends (LLVM, Wasm)
-    const activeBackends = ["llvm", "wasm"]
+    // A. Run persistent backends (LLVM, LLVM-PKIR, Wasm, PKIR)
+    const activeBackends = ["llvm", "llvm-pkir", "wasm", "pkir"]
       .filter(b => selectedBackends.includes(b));
     for (const backend of activeBackends) {
       for (const algo of selectedAlgos) {
@@ -547,7 +592,7 @@ if (mode === "squaring") {
 
   } finally {
     // Teardown persistent servers
-    for (const backend of ["llvm", "wasm"]) {
+    for (const backend of ["llvm", "llvm-pkir", "wasm", "pkir"]) {
       for (const algo of selectedAlgos) {
         algoServers[backend]?.[algo]?.child?.kill();
       }
@@ -584,12 +629,8 @@ if (mode === "sizes") {
     const strA = randomBigIntBits(bits);
     const strB = randomBigIntBits(bits);
 
-    const valA = valueForCode(
-      parseIntValue(strA), commonState.typeAliases.int, codes.find
-    );
-    const valB = valueForCode(
-      parseIntValue(strB), commonState.typeAliases.int, codes.find
-    );
+    const valA = parseIntValue(strA);
+    const valB = parseIntValue(strB);
     const pairInput = { tag: "product", value: { x: valA, y: valB } };
 
     for (const backend of selectedBackends) {