st/pass_strategy_synthesis
1
0
Fork 0
mirror of https://github.com/ProgramSnail/pass_strategy_synthesis.git synced 2026-03-11 18:47:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

test extension, types moved to ocanren_inject

Browse source
This commit is contained in:
ProgramSnail 2026-01-29 15:31:54 +03:00
parent e2dcf6c2a1
commit bea18df70e
2 changed files with 133 additions and 98 deletions
Download patch file Download diff file Expand all files Collapse all files

216
lib/relational_semantic_interpreter_oc.ml
View file

@ -14,84 +14,64 @@ struct
module Tag = struct module Tag = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"] [@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib [%%ocanren_inject
type nonrec t = Ref | Value type nonrec t = Ref | Value
[@@deriving gt ~options:{ show; gmap }] [@@deriving gt ~options:{ show; gmap }]
type nonrec ground = t type nonrec ground = t
] ]
let ref = inj Ref module Test = struct
let value = inj Value @type answer = logic GT.list with show
let _ =
Printf.printf "%s" @@ show(answer) (Stream.take (run q (fun q -> ocanren {q === Ref})
(fun q -> q#reify reify)))
end
end end
module Stmt = struct module Stmt = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"] [@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib [%%ocanren_inject
type nonrec ('d, 'dl) t = Call of 'd * 'dl | Read of 'd | Write of 'd type nonrec ('d, 'dl) t = Call of 'd * 'dl | Read of 'd | Write of 'd
[@@deriving gt ~options:{ show; gmap }] [@@deriving gt ~options:{ show; gmap }]
type nonrec ground = (Nat.ground, Nat.ground List.ground) t type nonrec ground = (Nat.ground, Nat.ground List.ground) t
] ]
let call f args = inj (Call (f, args))
let read id = inj (Read id)
let write id = inj (Write id)
end
module Body = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib
type nonrec ('stmt, 'l) t = T of ('stmt, 'l) List.t
[@@deriving gt ~options:{ show; gmap }]
type nonrec ground = (Stmt.ground, Stmt.ground List.ground) t
]
let make stmts = inj (T stmts)
end end
module FunDecl = struct module FunDecl = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"] [@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib [%%ocanren_inject
type nonrec ('l, 'b) t = T of 'l * 'b type nonrec ('l, 'b) t = FunDecl of 'l * 'b
[@@deriving gt ~options:{ show; gmap }] [@@deriving gt ~options:{ show; gmap }]
type nonrec ground = (Tag.ground List.ground, Body.ground) t type nonrec ground = (Tag.ground List.ground, Stmt.ground List.ground) t
] ]
let make args body = inj (T (args, body))
end end
module Prog = struct module Prog = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"] [@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib [%%ocanren_inject
type nonrec ('l, 'f) t = T of 'l * 'f type nonrec ('l, 'f) t = Prog of 'l * 'f
[@@deriving gt ~options:{ show; gmap }] [@@deriving gt ~options:{ show; gmap }]
type nonrec ground = (FunDecl.ground List.ground, FunDecl.ground) t type nonrec ground = (FunDecl.ground List.ground, FunDecl.ground) t
] ]
let make decls main_decl = inj (T (decls, main_decl))
end end
module Arg = struct module Arg = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"] [@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib [%%ocanren_inject
type nonrec 'd t = RValue | LValue of 'd type nonrec 'd t = RValue | LValue of 'd
[@@deriving gt ~options:{ show; gmap }] [@@deriving gt ~options:{ show; gmap }]
type nonrec ground = Nat.ground t type nonrec ground = Nat.ground t
] ]
let rvalue = inj RValue
let lvalue x = inj (LValue x)
end end
module Value = struct module Value = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"] [@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib [%%ocanren_inject
type nonrec t = Unit | Bot type nonrec t = Unit | Bot
[@@deriving gt ~options:{ show; gmap }] [@@deriving gt ~options:{ show; gmap }]
type nonrec ground = t type nonrec ground = t
] ]
let unit = inj Unit
let bot = inj Bot
end end
(* module Envr = struct *) (* module Envr = struct *)
@ -105,16 +85,14 @@ struct
(* let make elems = inj (T elems) *) (* let make elems = inj (T elems) *)
(* end *) (* end *)
module State = struct module St = struct
[@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"] [@@@warning "-26-27-32-33-34-35-36-37-38-39-60-66-67"]
[%%distrib [%%ocanren_inject
type nonrec ('env, 'mem, 'last_mem, 'assignments) t = T of 'env * 'mem * 'last_mem * 'assignments type nonrec ('env, 'mem, 'last_mem, 'assignments) t = St of 'env * 'mem * 'last_mem * 'assignments
[@@deriving gt ~options:{ show; gmap }] [@@deriving gt ~options:{ show; gmap }]
type nonrec ground = (((Nat.ground, Nat.ground) Pair.ground) List.ground, type nonrec ground = (((Nat.ground, Nat.ground) Pair.ground) List.ground,
Value.ground List.ground, Nat.ground, Nat.ground List.ground) t Value.ground List.ground, Nat.ground, Nat.ground List.ground) t
] ]
let make env mem last_mem assignments = inj (T (env, mem, last_mem, assignments))
end end
let rec list_replaceo xs id value ys = ocanren { let rec list_replaceo xs id value ys = ocanren {
@ -130,7 +108,8 @@ struct
list_replaceo xs' id' value ys' } list_replaceo xs' id' value ys' }
} }
let rec list_assoco a xs v' = ocanren { let rec list_assoco a xs v' =
ocanren {
fresh a', b', xs' in fresh a', b', xs' in
(Std.pair a' b') :: xs' == xs & (Std.pair a' b') :: xs' == xs &
{ a =/= a' & list_assoco a xs' v' | { a =/= a' & list_assoco a xs' v' |
@ -138,16 +117,20 @@ struct
} }
(* TODO: difference from List.assoco ?? *) (* TODO: difference from List.assoco ?? *)
let env_geto state id mem_id' = ocanren { let env_geto state id mem_id' =
let open St in
ocanren {
fresh env, mem, mem_len, assignments in fresh env, mem, mem_len, assignments in
state == State.make env mem mem_len assignments & state == St (env, mem, mem_len, assignments) &
list_assoco id env mem_id' list_assoco id env mem_id'
} }
let env_addo state id mem_id state' = ocanren { let env_addo state id mem_id state' =
let open St in
ocanren {
fresh env, env', mem, mem_len, assignments in fresh env, env', mem, mem_len, assignments in
state == State.make env mem mem_len assignments & state == St (env, mem, mem_len, assignments) &
state' == State.make env' mem mem_len assignments & state' == St (env', mem, mem_len, assignments) &
(Std.pair id mem_id) :: env == env' (Std.pair id mem_id) :: env == env'
} }
@ -172,35 +155,43 @@ struct
} }
(* TODO: use real holes *) (* TODO: use real holes *)
let mem_geto state id value' = ocanren { let mem_geto state id value' =
let open St in
ocanren {
fresh mem, mem_len, mem_id, mem_id_inv, _env, _assignments in fresh mem, mem_len, mem_id, mem_id_inv, _env, _assignments in
state == State.make _env mem mem_len _assignments & state == St (_env, mem, mem_len, _assignments) &
env_geto state id mem_id & env_geto state id mem_id &
inv_ido mem_len mem_id mem_id_inv & inv_ido mem_len mem_id mem_id_inv &
list_ntho mem mem_id_inv value' list_ntho mem mem_id_inv value'
} }
let mem_seto state id value state'= ocanren { let mem_seto state id value state'=
let open St in
ocanren {
fresh env, mem, mem_len, assignments, mem_id, inv_mem_id, mem', assignments' in fresh env, mem, mem_len, assignments, mem_id, inv_mem_id, mem', assignments' in
state == State.make env mem mem_len assignments & state == St (env, mem, mem_len, assignments) &
env_geto state id mem_id & env_geto state id mem_id &
inv_ido mem_len mem_id inv_mem_id & inv_ido mem_len mem_id inv_mem_id &
list_replaceo mem mem_id value mem' & list_replaceo mem mem_id value mem' &
assignments' == id :: assignments & assignments' == id :: assignments &
state' == State.make env mem' mem_len assignments' state' == St (env, mem', mem_len, assignments')
} }
let mem_addo state value state' = ocanren { let mem_addo state value state' =
let open St in
ocanren {
fresh env, mem, mem_len, mem_len', assignments, mem' in fresh env, mem, mem_len, mem_len', assignments, mem' in
state == State.make env mem mem_len assignments & state == St (env, mem, mem_len, assignments) &
mem' == value :: mem & mem' == value :: mem &
mem_len' == Nat.s mem_len & mem_len' == Nat.s mem_len &
state' == State.make env mem mem_len' assignments state' == St (env, mem, mem_len', assignments)
} }
let mem_checko state id state' = ocanren { let mem_checko state id state' =
mem_geto state id Value.bot & state' == state | let open Value in
mem_geto state id Value.unit & state' == state ocanren {
mem_geto state id Bot & state' == state |
mem_geto state id Unit & state' == state
} }
(* --- *) (* --- *)
@ -232,56 +223,71 @@ struct
f acc_upd x' y' acc' } f acc_upd x' y' acc' }
} }
let arg_to_valueo state arg value' = ocanren { let arg_to_valueo state arg value' =
arg == Arg.rvalue & value' == Value.unit | let open Arg in
let open Value in
ocanren {
arg == RValue & value' == Unit |
{ fresh id in { fresh id in
arg == Arg.lvalue id & arg == LValue id &
mem_geto state id value' } mem_geto state id value' }
} }
let arg_to_rvalueo _arg value' = ocanren { value' == Arg.rvalue } let arg_to_rvalueo _arg value' =
let open Arg in
ocanren { value' == RValue }
let st_mem_leno state mem_len' = ocanren { let st_mem_leno state mem_len' =
let open St in
ocanren {
fresh _env, _mem, _assignments in (* TODO: replace with real placeholder ? *) fresh _env, _mem, _assignments in (* TODO: replace with real placeholder ? *)
state == State.make _env _mem mem_len' _assignments state == St (_env, _mem, mem_len', _assignments)
} }
let st_add_argo state state_before id arg_tag arg state'' = ocanren { let st_add_argo state state_before id arg_tag arg state'' =
(* let open Nat in *)
let open Arg in
let open Tag in
ocanren {
(* arg_tag == Tag.ref & arg == Arg.rvalue & state'' == state | *) (* arg_tag == Tag.ref & arg == Arg.rvalue & state'' == state | *)
(* TODO: error, TODO: allow later ?? *) (* TODO: error, TODO: allow later ?? *)
{ fresh arg', value' in { fresh arg', value' in
arg_tag == Tag.ref & arg_tag == Ref &
arg == Arg.lvalue arg' & arg == LValue arg' &
env_geto state_before arg' value' & env_geto state_before arg' value' &
env_addo state id value' state'' } | env_addo state id value' state'' } |
{ fresh value', state', mem_len_dec' in { fresh value', state', mem_len_dec' in
arg_tag == Tag.value & arg_tag == Value &
arg_to_valueo state_before arg value' & arg_to_valueo state_before arg value' &
mem_addo state value' state' & mem_addo state value' state' &
st_mem_leno state (Nat.s mem_len_dec') & st_mem_leno state (Nat.s mem_len_dec') &
env_addo state' id mem_len_dec' state'' } env_addo state' id mem_len_dec' state'' }
} }
let st_spoil_foldero mem_len state mem id mem' = ocanren { let st_spoil_foldero mem_len state mem id mem' =
let open Value in
ocanren {
fresh mem_id', mem_id_inv' in fresh mem_id', mem_id_inv' in
env_geto state id mem_id' & env_geto state id mem_id' &
inv_ido mem_len mem_id' mem_id_inv' & inv_ido mem_len mem_id' mem_id_inv' &
list_replaceo mem mem_id_inv' Value.bot mem' list_replaceo mem mem_id_inv' Bot mem'
} }
let st_spoil_assignmentso state state' = ocanren { let st_spoil_assignmentso state state' =
let open St in
ocanren {
fresh env, mem, mem', mem_len, assignments, nil' in fresh env, mem, mem', mem_len, assignments, nil' in
state == State.make env mem mem_len assignments & state == St (env, mem, mem_len, assignments) &
list_foldlo (st_spoil_foldero mem_len state) mem assignments mem' & list_foldlo (st_spoil_foldero mem_len state) mem assignments mem' &
nil' == [] & nil' == [] &
state' == State.make env mem' mem_len nil' state' == St (env, mem', mem_len, nil')
} }
(* --- *) (* --- *)
let arg_to_lvalueo arg arg' = ocanren { let arg_to_lvalueo arg arg' =
arg' == Arg.lvalue arg let open Arg in
} ocanren { arg' == LValue arg }
let rec list_dropo n xs xs' = ocanren { let rec list_dropo n xs xs' = ocanren {
xs == [] & xs' == [] | xs == [] & xs' == [] |
@ -292,14 +298,17 @@ struct
list_dropo n' ys xs' } list_dropo n' ys xs' }
} }
let rec eval_stmto state prog stmt state' = ocanren { let rec eval_stmto state prog stmt state' =
let open Stmt in
let open Value in
ocanren {
{ fresh f_id, args, f, args' in { fresh f_id, args, f, args' in
stmt == Stmt.call f_id args & stmt == Call (f_id, args) &
list_ntho prog f_id f & list_ntho prog f_id f &
List.mapo arg_to_lvalueo args args' & List.mapo arg_to_lvalueo args args' &
eval_funo state prog f args' state' } | eval_funo state prog f args' state' } |
{ fresh id in stmt == Stmt.read id & mem_checko state id state' } | { fresh id in stmt == Read id & mem_checko state id state' } |
{ fresh id in stmt === Stmt.write id & mem_seto state id Value.unit state' } { fresh id in stmt === Write id & mem_seto state id Unit state' }
} }
and eval_body_foldero prog state stmt state' = and eval_body_foldero prog state stmt state' =
@ -310,15 +319,19 @@ struct
(* (List.fold_left (fun state stmt -> eval_stmt state prog stmt) state body) *) (* (List.fold_left (fun state stmt -> eval_stmt state prog stmt) state body) *)
(* TODO: other types on translation to ocanren ? *) (* TODO: other types on translation to ocanren ? *)
and add_arg_foldero state_before state_c arg_tag arg state_c' = ocanren { and add_arg_foldero state_before state_c arg_tag arg state_c' =
ocanren {
fresh state, id, state', id' in fresh state, id, state', id' in
state_c == (state, id) & state_c == Std.pair state id &
st_add_argo state state_before id arg_tag arg state' & st_add_argo state state_before id arg_tag arg state' &
id' == Nat.s id & id' == Nat.s id &
state_c' == (state', id') state_c' == Std.pair state' id'
} }
and eval_funo state prog decl args state' = ocanren { and eval_funo state prog decl args state' =
let open FunDecl in
let open St in
ocanren {
fresh arg_tags, body, fresh arg_tags, body,
env_before, mem_before, len_before, assignments_before, env_before, mem_before, len_before, assignments_before,
state_clean, state_clean,
@ -330,38 +343,44 @@ struct
nil_env, nil_assignments in nil_env, nil_assignments in
nil_env == [] & nil_env == [] &
nil_assignments == [] & nil_assignments == [] &
decl == FunDecl.make arg_tags body & decl == FunDecl (arg_tags, body) &
state == State.make env_before mem_before len_before assignments_before & state == St (env_before, mem_before, len_before, assignments_before) &
state_clean == State.make nil_env mem_before len_before nil_assignments & state_clean == St (nil_env, mem_before, len_before, nil_assignments) &
list_foldl2o (add_arg_foldero state) (state, Nat.o) arg_tags args (state_with_vars, _counter) & (* TODO: replace with real placeholder *) list_foldl2o (add_arg_foldero state) (Std.pair state Nat.o) arg_tags args (Std.pair state_with_vars _counter) & (* TODO: replace with real placeholder *)
eval_bodyo state_with_vars prog body state_evaled & eval_bodyo state_with_vars prog body state_evaled &
st_spoil_assignmentso state_evaled state_spoiled & st_spoil_assignmentso state_evaled state_spoiled &
state_spoiled == State.make _env mem_spoiled len _assignments & state_spoiled == St (_env, mem_spoiled, len, _assignments) &
Nat.addo len_to_drop len_before len & Nat.addo len_to_drop len_before len &
list_dropo len_to_drop mem_spoiled mem_updated & list_dropo len_to_drop mem_spoiled mem_updated &
state' == State.make env_before mem_updated len_before assignments_before state' == St (env_before, mem_updated, len_before, assignments_before)
} }
and eval_fun_empty_argso state prog decl state' = ocanren { and eval_fun_empty_argso state prog decl state' =
fresh arg_tags, args, _hole in (* TODO: replace with real placeholder *) let open FunDecl in
decl == FunDecl.make arg_tags _hole & ocanren {
fresh arg_tags, args, _body in (* TODO: replace with real placeholder *)
decl == FunDecl (arg_tags, _body) &
List.mapo arg_to_rvalueo arg_tags args & List.mapo arg_to_rvalueo arg_tags args &
eval_funo state prog decl args state' eval_funo state prog decl args state'
} }
(* --- *) (* --- *)
let empty_stateo state = ocanren { let empty_stateo state =
let open St in
ocanren {
fresh nil_env, nil_mem, nil_assignments in fresh nil_env, nil_mem, nil_assignments in
nil_env == [] & nil_env == [] &
nil_assignments == [] & nil_assignments == [] &
nil_mem == [] & nil_mem == [] &
state == State.make nil_env nil_mem Nat.o nil_assignments state == St (nil_env, nil_mem, Nat.o, nil_assignments)
} }
let eval_progo all_prog state' = ocanren { let eval_progo all_prog state' =
let open Prog in
ocanren {
fresh prog, main_decl, state in fresh prog, main_decl, state in
all_prog == Prog.make prog main_decl & all_prog == Prog (prog, main_decl) &
empty_stateo state & empty_stateo state &
eval_fun_empty_argso state prog main_decl state' eval_fun_empty_argso state prog main_decl state'
} }
@ -376,7 +395,10 @@ struct
(* let empty_prog = Prog.T (List.Nil, FunDecl.T (List.Nil, Body.T List.Nil)) *) (* let empty_prog = Prog.T (List.Nil, FunDecl.T (List.Nil, Body.T List.Nil)) *)
(* let empty_prog'' = ocanren { T ([], T ([], T [])) } *) (* let empty_prog'' = ocanren { T ([], T ([], T [])) } *)
(* let empty_prog = ocanren { Prog.make [] (FunDecl.make [] (Body.make [])) } *) (* let empty_prog = *)
(* let open FunDecl in *)
(* let open Prog in *)
(* ocanren { Prog ([], FunDecl ([], [])) } *)
(* let empty_prog = Prog.make (Std.nil ()) (FunDecl.make (Std.nil ()) (Body.make (Std.nil ()))) *) (* let empty_prog = Prog.make (Std.nil ()) (FunDecl.make (Std.nil ()) (Body.make (Std.nil ()))) *)
let eval_test = let eval_test =

15
lib/test.ml
View file

@ -6,7 +6,6 @@ module Tag = struct
ocanren type tag = Ref | Value ocanren type tag = Ref | Value
module Test = struct module Test = struct
@type answer = logic GT.list with show @type answer = logic GT.list with show
let _ = let _ =
@ -14,3 +13,17 @@ module Tag = struct
(fun q -> q#reify reify))) (fun q -> q#reify reify)))
end end
end end
module Stmt = struct
ocanren type ('d, 'dl) stmt = Call of 'd * 'dl | Read of 'd | Write of 'd
module Test = struct
@type answer1 = Nat.ground List.ground GT.list with show
@type answer = (Nat.ground, Nat.ground List.ground) ground GT.list with show
let _ = Printf.printf "%s\n" @@ show(answer1) (Stream.take (run q (fun q -> ocanren {Call (1, [2]) === Call (1, q)})
(fun q -> q#reify (List.prj_exn Nat.prj_exn))))
let _ = Printf.printf "%s\n" @@ show(answer) (Stream.take (run q (fun q -> ocanren {Call (1, [2]) === q})
(fun q -> q#reify (prj_exn Nat.prj_exn (List.prj_exn Nat.prj_exn)))))
end
end