1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
(**************************************************************************)
(* *)
(* OCaml *)
(* *)
(* Pierre Chambart, OCamlPro *)
(* Mark Shinwell and Leo White, Jane Street Europe *)
(* *)
(* Copyright 2013--2016 OCamlPro SAS *)
(* Copyright 2014--2016 Jane Street Group LLC *)
(* *)
(* All rights reserved. This file is distributed under the terms of *)
(* the GNU Lesser General Public License version 2.1, with the *)
(* special exception on linking described in the file LICENSE. *)
(* *)
(**************************************************************************)
[@@@ocaml.warning "+a-4-9-30-40-41-42-66"]
open! Int_replace_polymorphic_compare
let constant_field (expr:Flambda.t)
: Flambda.constant_defining_value_block_field option =
match expr with
| Let { var; defining_expr = Const c; body = Var var' ; _ } ->
assert(Variable.equal var var');
(* This must be true since var is the only variable in scope *)
Some (Flambda.Const c)
| Let { var; defining_expr = Symbol s; body = Var var' ; _ } ->
assert(Variable.equal var var');
Some (Flambda.Symbol s)
| _ ->
None
let rec loop (program : Flambda.program_body) : Flambda.program_body =
match program with
| Initialize_symbol (symbol, tag, fields, program) ->
let constant_fields = List.map constant_field fields in
begin
match Misc.Stdlib.List.some_if_all_elements_are_some constant_fields
with
| None ->
Initialize_symbol (symbol, tag, fields, loop program)
| Some fields ->
Let_symbol (symbol, Block (tag, fields), loop program)
end
| Let_symbol (symbol, const, program) ->
Let_symbol (symbol, const, loop program)
| Let_rec_symbol (defs, program) ->
Let_rec_symbol (defs, loop program)
| Effect (expr, program) ->
Effect (expr, loop program)
| End symbol ->
End symbol
let run (program : Flambda.program) =
{ program with
program_body = loop program.program_body;
}
|
