Home My Page Projects Code Snippets Project Openings diderot
Summary Activity Tracker Tasks SCM

SCM Repository

[diderot] Diff of /branches/vis15/src/compiler/tree-ir/tree-ir.sml
ViewVC logotype

Diff of /branches/vis15/src/compiler/tree-ir/tree-ir.sml

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 3848, Thu May 12 14:30:19 2016 UTC revision 4380, Mon Aug 8 17:05:58 2016 UTC
# Line 15  Line 15 
15      structure Op = TreeOps      structure Op = TreeOps
16      structure Ty = TreeTypes      structure Ty = TreeTypes
17    
18        type target_info = {
19            layout : int -> TreeTypes.vec_layout,
20            isInline : LowOps.rator -> bool
21          }
22    
23      datatype program = Program of {      datatype program = Program of {
24          props : Properties.t list,          props : Properties.t list,
25            target : target_info,           (* target-specific info about representation invariants *)
26          consts : global_var list,       (* large constant variables *)          consts : global_var list,       (* large constant variables *)
27          inputs : input list,            (* global input variables *)          inputs : input list,            (* global input variables *)
28          constInit : block,              (* code that initializes constants and inputs *)          constInit : block,              (* code that initializes constants and inputs *)
29          globals : global_var list,      (* other global variables *)          globals : global_var list,      (* other global variables *)
30          globalInit : block,             (* CFG to initialize other globals (if any) *)          funcs : func_def list,          (* user-defined functions *)
31            globInit : block,               (* CFG to initialize other globals (if any) *)
32          strand : strand,                (* the strand definition *)          strand : strand,                (* the strand definition *)
33          create : create,                (* initial strand creation *)          create : create,                (* initial strand creation *)
34          update : block option           (* optional update code. *)          init : block option,            (* optional global initially code *)
35            update : block option           (* optional global update code. *)
36          }
37    
38        and func_def = Func of {
39            name : func,
40            params : var list,
41            body : block
42        }        }
43    
44      and strand = Strand of {      and strand = Strand of {
45          name : Atom.atom,          name : Atom.atom,
46          params : var list,          params : var list,
47            spatialDim : int option,
48          state : state_var list,          state : state_var list,
49          stateInit : block,          stateInit : method,
50          initM : block option,          initM : method option,
51          updateM : block,          updateM : method,
52          stabilizeM : block option          stabilizeM : method option
53        }        }
54    
55      and create = Create of {      and method = Method of {
56          dim : int option,               (* grid dimension; NONE for collections *)            needsW : bool,                (* does the method need the world (e.g., to print) *)
57          code : block                    (* the loop nest for creating the strands *)            hasG : bool,                  (* does the method contain references to globals? *)
58              body : block                  (* the method body *)
59        }        }
60    
61      and block = Block of {      and block = Block of {
# Line 49  Line 65 
65    
66      and stm      and stm
67        = S_Comment of string list        = S_Comment of string list
68        | S_Unpack of var list * exp      (* unpack vector pieces into local variables *)        | S_Assign of bool * var * exp    (* assignment; the boolean is true for var decls *)
       | S_Assign of var * exp  
69        | S_MAssign of var list * exp        | S_MAssign of var list * exp
70        | S_GAssign of global_var * exp        | S_GAssign of global_var * exp
71        | S_IfThen of exp * block        | S_IfThen of exp * block
72        | S_IfThenElse of exp * block * block        | S_IfThenElse of exp * block * block
73          | S_For of var * exp * exp * block
74        | S_Foreach of var * exp * block        | S_Foreach of var * exp * block
75      (* special Diderot forms *)      (* special Diderot forms *)
76        | S_LoadNrrd of var * string        | S_MapReduce of (var * Reductions.t * block) list * var * StrandSets.t
77          | S_LoadNrrd of var * APITypes.t * string
78        | S_Input of global_var * string * string option * exp option (* get input *)        | S_Input of global_var * string * string option * exp option (* get input *)
79        | S_InputNrrd of global_var * string * string option * string option (* get image/seq input *)        | S_InputNrrd of global_var * string * string option * string option (* get image/seq input *)
80        | S_New of Atom.atom * exp list   (* new strand creation *)        | S_New of Atom.atom * exp list   (* new strand creation *)
81        | S_Save of state_var * exp  (* save strand state *)        | S_Save of state_var * exp  (* save strand state *)
       | S_Exit  
82        | S_Print of Ty.t list * exp list        | S_Print of Ty.t list * exp list
83          | S_Return of exp
84      (* return functions for methods *)      (* return functions for methods *)
85        | S_Active        | S_Active
86        | S_Stabilize        | S_Stabilize
# Line 71  Line 88 
88    
89      and exp      and exp
90        = E_Global of global_var        = E_Global of global_var
91        | E_State of var option * state_var        | E_State of exp option * state_var
92        | E_Var of var        | E_Var of var
93        | E_Lit of Literal.t        | E_Lit of Literal.t
94        | E_Op of Op.rator * exp list        | E_Op of Op.rator * exp list
95        | E_Cons of exp list * Ty.t        | E_Apply of func * exp list              (* user-defined function application *)
96          | E_Vec of int * int * exp list           (* vector value; the ints are the width and the
97                                                     * padded width
98                                                     *)
99          | E_Cons of exp list * Ty.t               (* in-memory tensor value *)
100        | E_Seq of exp list * Ty.t        | E_Seq of exp list * Ty.t
101        | E_Pack of exp list              (* pack vector pieces into composite vector value *)        | E_Pack of Ty.vec_layout * exp list      (* pack vector pieces into composite vector value *)
102        | E_VLoad of Ty.vec_layout * exp * int (* load a piece of a composite vector value *)        | E_VLoad of Ty.vec_layout * exp * int    (* load a piece of a composite vector value from
103                                                     * a tensor.  The integer specifies the index of
104                                                     * the piece.
105                                                     *)
106    
107      and global_var = GV of {      and global_var = GV of {
108          name : string,                  (* name (should be unique) *)          name : string,                  (* name (should be unique) *)
# Line 102  Line 126 
126          ty : Ty.t                       (* type *)          ty : Ty.t                       (* type *)
127        }        }
128    
129        and func = FV of {
130            name : string,                  (* name (should be unique) *)
131            id : Stamp.stamp,               (* unique ID *)
132            ty : Ty.t,                      (* return type *)
133            paramTys : Ty.t list,           (* parameter types *)
134            needsW : bool,                  (* does the function need the world (e.g., to print) *)
135            hasG : bool                     (* does the function contain references to globals? *)
136          }
137    
138      withtype input = global_var Inputs.input      withtype input = global_var Inputs.input
139             and create = block Create.t
140    
141        fun emptyBlk (Block{body=[], ...}) = true
142          | emptyBlk _ = false
143    
144    end    end

Legend:
Removed from v.3848  
changed lines
  Added in v.4380

root@smlnj-gforge.cs.uchicago.edu
ViewVC Help
Powered by ViewVC 1.0.0