|
| 1 | +// Deopt-aware benchmark suite (mitata) for the MeTTa TS hot paths. Pure TypeScript, no native. |
| 2 | +// Run after building core: `node packages/node/bench/suite.mjs` |
| 3 | +import { run, bench, group, summary } from "mitata"; |
| 4 | +import { |
| 5 | + runProgram, |
| 6 | + matchAtoms, |
| 7 | + buildEnv, |
| 8 | + evalAtom, |
| 9 | + stdTable, |
| 10 | + sym, |
| 11 | + variable, |
| 12 | + expr, |
| 13 | + gint, |
| 14 | + parseAll, |
| 15 | + standardTokenizer, |
| 16 | + preludeAtoms, |
| 17 | +} from "../../core/dist/index.js"; |
| 18 | +import { readFileSync, readdirSync } from "node:fs"; |
| 19 | +import { resolve } from "node:path"; |
| 20 | + |
| 21 | +const tk = standardTokenizer(); |
| 22 | +const CORPUS = resolve(process.cwd(), "corpus"); |
| 23 | +const corpus = readdirSync(CORPUS) |
| 24 | + .filter((f) => f.endsWith(".metta") && f !== "c2_spaces_kb.metta") |
| 25 | + .map((f) => readFileSync(resolve(CORPUS, f), "utf8")); |
| 26 | + |
| 27 | +// Hot-path microbenchmarks |
| 28 | +const deep = expr([variable("x"), expr([variable("y"), sym("a")]), gint(3)]); |
| 29 | +const deepT = expr([sym("p"), expr([sym("q"), sym("a")]), gint(3)]); |
| 30 | + |
| 31 | +group("matcher", () => { |
| 32 | + summary(() => { |
| 33 | + bench("matchAtoms (nested, binds 2 vars)", () => matchAtoms(deep, deepT)); |
| 34 | + bench("matchAtoms (symbol mismatch)", () => matchAtoms(sym("a"), sym("b"))); |
| 35 | + }); |
| 36 | +}); |
| 37 | + |
| 38 | +// A space with 1000 facts, query one |
| 39 | +const facts = Array.from({ length: 1000 }, (_, i) => expr([sym("Edge"), gint(i), gint(i + 1)])); |
| 40 | +const big = buildEnv([...preludeAtoms(), ...facts], stdTable()); |
| 41 | +const query = expr([sym("match"), sym("&self"), expr([sym("Edge"), gint(500), variable("y")]), variable("y")]); |
| 42 | +group("space", () => { |
| 43 | + bench("match over 1000-atom space", () => evalAtom(big, query)); |
| 44 | +}); |
| 45 | + |
| 46 | +// Evaluation |
| 47 | +const fibEnv = buildEnv( |
| 48 | + [ |
| 49 | + ...preludeAtoms(), |
| 50 | + ...parseAll( |
| 51 | + "(= (fib $n) (unify $n 0 0 (unify $n 1 1 (+ (fib (- $n 1)) (fib (- $n 2))))))", |
| 52 | + tk, |
| 53 | + ).map((t) => t.atom), |
| 54 | + ], |
| 55 | + stdTable(), |
| 56 | +); |
| 57 | +const fib15 = parseAll("(fib 15)", tk)[0].atom; |
| 58 | +group("eval", () => { |
| 59 | + bench("fib(15) (~1.2k recursive calls)", () => evalAtom(fibEnv, fib15)); |
| 60 | + bench("stdlib load + (+ 1 2)", () => runProgram("!(+ 1 2)")); |
| 61 | +}); |
| 62 | + |
| 63 | +group("oracle", () => { |
| 64 | + bench("full 270-assertion corpus", () => { |
| 65 | + for (const src of corpus) runProgram(src); |
| 66 | + }); |
| 67 | +}); |
| 68 | + |
| 69 | +await run(); |
0 commit comments