squiggle/packages/squiggle-lang/scripts/bench-map-reduce.mjs

#!/usr/bin/env node
import { SqProject } from "@quri/squiggle-lang";
import { measure } from "./lib.mjs";

const maxP = 5;

for (let p = 0; p <= maxP; p++) {
  const size = Math.pow(10, p);
  const project = SqProject.create();
  project.setSource(
    "main",
    `List.upTo(1, ${size}) |> map({|x| List.upTo(1, 100) |> reduce(0, {|a,b|a+b})})`
  );
  const time = measure(() => {
    project.run("main");
  });
  console.log(`1e${p}`, "\t", time);
}