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
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
(**************************************************************************)
(* Sail *)
(* *)
(* Copyright (c) 2013-2017 *)
(* Kathyrn Gray *)
(* Shaked Flur *)
(* Stephen Kell *)
(* Gabriel Kerneis *)
(* Robert Norton-Wright *)
(* Christopher Pulte *)
(* Peter Sewell *)
(* Alasdair Armstrong *)
(* Brian Campbell *)
(* Thomas Bauereiss *)
(* Anthony Fox *)
(* Jon French *)
(* Dominic Mulligan *)
(* Stephen Kell *)
(* Mark Wassell *)
(* *)
(* All rights reserved. *)
(* *)
(* This software was developed by the University of Cambridge Computer *)
(* Laboratory as part of the Rigorous Engineering of Mainstream Systems *)
(* (REMS) project, funded by EPSRC grant EP/K008528/1. *)
(* *)
(* Redistribution and use in source and binary forms, with or without *)
(* modification, are permitted provided that the following conditions *)
(* are met: *)
(* 1. Redistributions of source code must retain the above copyright *)
(* notice, this list of conditions and the following disclaimer. *)
(* 2. Redistributions in binary form must reproduce the above copyright *)
(* notice, this list of conditions and the following disclaimer in *)
(* the documentation and/or other materials provided with the *)
(* distribution. *)
(* *)
(* THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' *)
(* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED *)
(* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A *)
(* PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR *)
(* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, *)
(* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT *)
(* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF *)
(* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND *)
(* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, *)
(* OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT *)
(* OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF *)
(* SUCH DAMAGE. *)
(**************************************************************************)
open Ast
open Ast_defs
open Ast_util
open Rewriter
let rec split_at_function' id defs acc =
match defs with
| [] -> None
| ([def], env) :: defs when is_fundef id def -> Some (acc, (def, env), defs)
| (def, env) :: defs -> split_at_function' id defs ((def, env) :: acc)
let split_at_function id defs =
match split_at_function' id defs [] with
| None -> None
| Some (pre_defs, def, post_defs) ->
Some (List.rev pre_defs, def, post_defs)
let recheck (Defs defs) =
let defs = Type_check.check_with_envs Type_check.initial_env defs in
let rec find_optimizations = function
| ([DEF_pragma ("optimize", pragma, p_l)], env) :: ([DEF_spec vs as def1], _) :: defs ->
let id = id_of_val_spec vs in
let args = Str.split (Str.regexp " +") (String.trim pragma) in
begin match args with
| ["unroll"; n]->
let n = int_of_string n in
begin match split_at_function id defs with
| Some (intervening_defs, ((DEF_fundef fdef as def2, _)), defs) ->
let rw_app subst (fn, args) =
if Id.compare id fn = 0 then E_app (subst, args) else E_app (fn, args)
in
let rw_exp subst = { id_exp_alg with e_app = rw_app subst } in
let rw_defs subst = { rewriters_base with rewrite_exp = (fun _ -> fold_exp (rw_exp subst)) } in
let specs = ref [def1] in
let bodies = ref [rewrite_def (rw_defs (append_id id "_unroll_1")) def2] in
for i = 1 to n do
let current_id = append_id id ("_unroll_" ^ string_of_int i) in
let next_id = if i = n then current_id else append_id id ("_unroll_" ^ string_of_int (i + 1)) in
(* Create a valspec for the new unrolled function *)
specs := !specs @ [DEF_spec (rename_valspec current_id vs)];
(* Then duplicate it's function body and make it call the next unrolled function *)
bodies := !bodies @ [rewrite_def (rw_defs next_id) (DEF_fundef (rename_fundef current_id fdef))]
done;
!specs @ List.concat (List.map fst intervening_defs) @ !bodies @ find_optimizations defs
| _ ->
Reporting.warn "Could not find function body for unroll pragma at " p_l "";
def1 :: find_optimizations defs
end
| _ ->
Reporting.warn "Unrecognised optimize pragma at" p_l "";
def1 :: find_optimizations defs
end
| (defs, _) :: defs' ->
defs @ find_optimizations defs'
| [] -> []
in
Defs (find_optimizations defs)
|
