3e2f1faebf
This PR introduces support for user-defined fallback code in the `grind` tactic. The fallback code can be utilized to inspect the state of failing `grind` subgoals and/or invoke user-defined automation. Users can now write `grind on_failure <code>`, where `<code>` should have the type `GoalM Unit`. See the modified tests in this PR for examples.
46 lines
1.5 KiB
Text
46 lines
1.5 KiB
Text
import Lean.Meta.Tactic.Grind
|
||
|
||
def f (α : Type) [Add α] (a : α) := a + a + a
|
||
|
||
open Lean Meta Grind in
|
||
def fallback : Fallback := do
|
||
let nodes ← filterENodes fun e => return e.self.isAppOf ``Lean.Grind.nestedProof
|
||
logInfo (nodes.toList.map (·.self))
|
||
let nodes ← filterENodes fun e => return e.self.isAppOf ``GetElem.getElem
|
||
let [_, n, _] := nodes.toList | unreachable!
|
||
logInfo (← getEqc n.self)
|
||
(← get).mvarId.admit
|
||
|
||
set_option grind.debug true
|
||
set_option grind.debug.proofs true
|
||
|
||
/-
|
||
Recall that array access terms, such as `a[i]`, have nested proofs.
|
||
The following test relies on `grind` `nestedProof` wrapper to
|
||
detect equalities between array access terms.
|
||
-/
|
||
|
||
/-
|
||
info: [Lean.Grind.nestedProof (i < a.toList.length) (_example.proof_1 i j a b h1 h2),
|
||
Lean.Grind.nestedProof (j < a.toList.length) h1,
|
||
Lean.Grind.nestedProof (j < b.toList.length) h]
|
||
---
|
||
info: [a[i], b[j], a[j]]
|
||
---
|
||
warning: declaration uses 'sorry'
|
||
-/
|
||
-- #guard_msgs in
|
||
|
||
example (i j : Nat) (a b : Array Nat) (h1 : j < a.size) (h : j < b.size) (h2 : i ≤ j) : a[i] < a[j] + b[j] → i = j → a = b → False := by
|
||
grind on_failure fallback
|
||
|
||
/--
|
||
info: [Lean.Grind.nestedProof (i < a.toList.length) (_example.proof_1 i j a b h1 h2),
|
||
Lean.Grind.nestedProof (j < a.toList.length) h1,
|
||
Lean.Grind.nestedProof (j < b.toList.length) h]
|
||
---
|
||
info: [a[i], a[j]]
|
||
-/
|
||
#guard_msgs (info) in
|
||
example (i j : Nat) (a b : Array Nat) (h1 : j < a.size) (h : j < b.size) (h2 : i ≤ j) : a[i] < a[j] + b[j] → i = j → False := by
|
||
grind on_failure fallback
|