mirror of
https://github.com/gorules/zen.git
synced 2026-10-04 08:02:28 +00:00
stefan-gorules
a4cafd0346
* chore: add benchmarks for tier 1 languages * improve fixtures * fix cargo fmt
40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
import json
|
|
import os
|
|
import sys
|
|
import time
|
|
|
|
import zen
|
|
|
|
root = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
|
|
fixtures = os.path.join(root, "fixtures")
|
|
manifest = json.load(open(os.path.join(root, "manifest.json")))
|
|
iters = int(os.environ.get("BENCH_ITERS", "2000"))
|
|
out_path = sys.argv[1] if len(sys.argv) > 1 else None
|
|
|
|
|
|
def loader(key):
|
|
with open(os.path.join(fixtures, key)) as f:
|
|
return f.read()
|
|
|
|
|
|
engine = zen.ZenEngine({"loader": loader})
|
|
|
|
results = []
|
|
for e in manifest:
|
|
for trace in (False, True):
|
|
opts = {"trace": trace}
|
|
engine.evaluate(e["file"], e["input"], opts)
|
|
start = time.perf_counter_ns()
|
|
for _ in range(iters):
|
|
engine.evaluate(e["file"], e["input"], opts)
|
|
per = (time.perf_counter_ns() - start) / iters
|
|
suffix = " +trace" if trace else ""
|
|
results.append({"name": f'{e["name"]} ({e["kind"]}){suffix}', "unit": "ns/op", "value": per})
|
|
|
|
if out_path:
|
|
with open(out_path, "w") as f:
|
|
f.write(json.dumps(results, indent=2))
|
|
else:
|
|
for r in results:
|
|
print(f'{r["name"]:<44} {r["value"]:>12.0f} ns/op')
|