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
58
59
60
61
62
63
64
65
|
(**************************************************************************)
(* *)
(* 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. *)
(* *)
(**************************************************************************)
(** Helper module for adding specialised arguments to sets of closures. *)
module Definition : sig
type t =
| Existing_inner_free_var of Variable.t
| Projection_from_existing_specialised_arg of Projection.t
end
module What_to_specialise : sig
type t
val create
: set_of_closures:Flambda.set_of_closures
-> t
val new_specialised_arg
: t
-> fun_var:Variable.t
-> group:Variable.t
-> definition:Definition.t (* [projecting_from] "existing inner vars" *)
-> t
val make_direct_call_surrogate_for : t -> fun_var:Variable.t -> t
end
module type S = sig
val pass_name : string
val what_to_specialise
: env:Inline_and_simplify_aux.Env.t
-> set_of_closures:Flambda.set_of_closures
-> What_to_specialise.t
end
module Make (_ : S) : sig
(** [duplicate_function] should be
[Inline_and_simplify.duplicate_function]. *)
val rewrite_set_of_closures
: env:Inline_and_simplify_aux.Env.t
-> duplicate_function:(
env:Inline_and_simplify_aux.Env.t
-> set_of_closures:Flambda.set_of_closures
-> fun_var:Variable.t
-> new_fun_var:Variable.t
-> Flambda.function_declaration
* Flambda.specialised_to Variable.Map.t)
-> set_of_closures:Flambda.set_of_closures
-> (Flambda.expr * Inlining_cost.Benefit.t) option
end
|
