Skip to content

Commit

Permalink
Translate non recursive let to nontac expr when possible
Browse files Browse the repository at this point in the history
Fix #16 (by being compatible with Tac2intern.is_value)

I'd like to have a more direct implementation than using `is_nontac`
but it's not coming to me.
  • Loading branch information
SkySkimmer committed Dec 4, 2023
1 parent 4e1a3f7 commit 8514cd6
Show file tree
Hide file tree
Showing 3 changed files with 70 additions and 10 deletions.
72 changes: 62 additions & 10 deletions src/tac2compile.ml
Original file line number Diff line number Diff line change
Expand Up @@ -258,6 +258,7 @@ type nontac_expr =
| Opn of SpilledKn.t * nontac_expr list
| PrjV of nontac_expr * int (* non-mutable projection *)
| Prim of ml_tactic_name
| ValLetNoRec of (Name.t * nontac_expr) list * nontac_expr

(** evaluates to a valexpr tactic *)
and tac_expr_head =
Expand Down Expand Up @@ -439,6 +440,25 @@ let precompiled_prim ml =
if ml.mltac_plugin <> "coq-core.plugins.ltac2" then None
else CString.Map.find_opt ml.mltac_tactic Tac2compiledPrim.registered

let rec is_nontac = function
| GTacAtm _ | GTacVar _ | GTacRef _ | GTacFun _
| GTacPrm _
-> true
| GTacCst (kn, _, l) ->
is_pure_ctor kn && List.for_all is_nontac l
| GTacOpn (_, l) -> List.for_all is_nontac l
| GTacPrj (typ, sube, i) ->
not (is_mutable_proj typ i) && is_nontac sube

| GTacLet (false, bnd, e) ->
List.for_all (fun (_, e) -> is_nontac e) bnd
&& is_nontac e

| GTacApp _ | GTacLet (true, _, _) | GTacCse _
| GTacSet _ | GTacWth _ | GTacFullMatch _
| GTacExt _
-> false

let rec nontac_expr env ((cnt, nonvals) as acc) e = match e with
| GTacAtm a -> acc, Atm a
| GTacVar x ->
Expand Down Expand Up @@ -475,6 +495,19 @@ let rec nontac_expr env ((cnt, nonvals) as acc) e = match e with
acc, Ref (LocalKn ("Tac2compiledPrim."^ml.mltac_tactic, info))
end

| GTacLet (false, bnd, sube) when is_nontac e ->
let envbnd, nas' = push_env push_user_names (List.map fst bnd) Valexpr env in
let bnd = List.map2 (fun (_, e) na' ->
let acc', e = nontac_expr env acc e in
assert (acc' == acc);
(na', e))
bnd
nas'
in
let acc', sube = nontac_expr envbnd acc sube in
assert (acc' == acc);
acc, ValLetNoRec (bnd, sube)

| GTacApp _ | GTacLet _ | GTacCse _
| GTacPrj _ | GTacSet _ | GTacWth _ | GTacFullMatch _
| GTacExt _ | GTacCst _ ->
Expand Down Expand Up @@ -541,16 +574,21 @@ and tac_expr env e =
acc, LetRec (lets, e)

(* XXX detect when a let can be nontac_expr *)
| GTacLet (false, bnd, e) ->
let envbnd, nas' = push_env push_user_names (List.map fst bnd) Valexpr env in
let bnd = List.map2 (fun (_, e) na' ->
let e = tac_expr env e in
(na', e))
bnd
nas'
in
let e = tac_expr envbnd e in
acc, LetNoRec (bnd, e)
| GTacLet (false, bnd, sube) ->
if is_nontac e then
let acc', e = nontac_expr env acc e in
assert (acc == acc');
acc, Return e
else
let envbnd, nas' = push_env push_user_names (List.map fst bnd) Valexpr env in
let bnd = List.map2 (fun (_, e) na' ->
let e = tac_expr env e in
(na', e))
bnd
nas'
in
let sube = tac_expr envbnd sube in
acc, LetNoRec (bnd, sube)

| GTacCse (e, _, esInt, esBlk) ->
let acc, e = nontac_expr env acc e in
Expand Down Expand Up @@ -743,6 +781,20 @@ let rec pp_nontac_expr = function
hov 2 (str "[|" ++ pp_val_list es ++ str "|]")))
| Prim ml ->
surround (str "Tac2env.interp_primitive" ++ spc() ++ pp_ml_name ml)
| ValLetNoRec (bnd, e) ->
let pr_one_let (na,e) = Name.print na ++ str " = " ++ spc() ++ pp_nontac_expr e in
let x1, rest = match bnd with
| [] -> assert false
| x :: rest -> x, rest
in
hv 2
(str "(" ++
hov 2 (str "let " ++ pr_one_let x1) ++ spc() ++
prlist (fun x -> hov 2 (str "and " ++ pr_one_let x) ++ spc()) rest ++
str "in" ++ spc() ++
pp_nontac_expr e ++
str ")")


(* produce a ocaml term of type valexpr tactic *)
and pp_expr e =
Expand Down
1 change: 1 addition & 0 deletions tests/_CoqProject
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ minibench.v
compiler_bug_4.v
compiler_bug_6.v
compiler_bug_14.v
compiler_bug_16.v
compiler_bug_17.v

-I src
Expand Down
7 changes: 7 additions & 0 deletions tests/compiler_bug_16.v
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
From Ltac2 Require Import Ltac2 RedFlags.
From Ltac2Compiler Require Import Ltac2Compiler.

Ltac2 beta_iota :=
let x := RedFlags.all in x.

Ltac2 Compile beta_iota.

0 comments on commit 8514cd6

Please sign in to comment.