test: synthetic simp_arith benchmark (#6061)

This PR adds a simp_arith benchmark.

This benchmark highlights some improvable asymptotics in `Nat.Linear`,
which
will be fixed subsequently.
This commit is contained in:
Joachim Breitner 2024-11-13 16:49:52 +01:00 committed by GitHub
parent f721f94045
commit 6b811f8c92
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 35 additions and 0 deletions

View file

@ -0,0 +1,29 @@
import Lean
open Lean Meta Elab in
elab "largeGoal%" : term =>
let n := 30 -- number of variables
let r := 3 -- number of repetitions
let mkAdd := mkApp2 (mkConst ``Nat.add)
let mkMul := mkApp2 (mkConst ``Nat.mul)
let decls := Array.ofFn fun (i : Fin n) =>
((`x).appendIndexAfter i, (fun _ => pure (mkConst ``Nat)))
withLocalDeclsD decls fun xs => do
let mut e₁ : Expr := mkNatLit 42
let mut e₂ : Expr := mkNatLit 23
for _ in [:r] do
for i in [:xs.size] do
e₁ := mkAdd (mkMul (mkNatLit i) e₁) xs[i]!
e₂ := mkAdd xs[i]! (mkMul e₂ (mkNatLit (xs.size - i)))
let goal ← mkEq e₁ e₂
let goal := mkNot goal
let goal ← mkForallFVars xs goal
return goal
set_option maxRecDepth 10000
-- manually verified: most time spent in type-checking
-- set_option trace.profiler true
example : largeGoal% := by
intros
simp_arith only

View file

@ -316,6 +316,12 @@
run_config:
<<: *time
cmd: lean reduceMatch.lean
- attributes:
description: simp_arith1
tags: [fast, suite]
run_config:
<<: *time
cmd: lean simp_arith1.lean
- attributes:
description: nat_repr
tags: [fast, suite]