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 3767, Mon Apr 18 18:42:10 2016 UTC revision 3995, Sat Jun 18 20:00:19 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) *)          globInit : block,               (* CFG to initialize other globals (if any) *)
31          strand : strand,                (* the strand definition *)          strand : strand,                (* the strand definition *)
32          create : create,                (* initial strand creation *)          create : create,                (* initial strand creation *)
33          update : block option           (* optional update code. *)          init : block option,            (* optional global initially code *)
34            update : block option           (* optional global update code. *)
35        }        }
36    
37      and strand = Strand of {      and strand = Strand of {
38          name : Atom.atom,          name : Atom.atom,
39          params : var list,          params : var list,
40          state : state_var list,          state : state_var list,
41          stateInit : block,          stateInit : method,
42          initM : block option,          initM : method option,
43          updateM : block,          updateM : method,
44          stabilizeM : block option          stabilizeM : method option
45          }
46    
47        and method = Method of {
48              usesGlobals : bool,           (* does the method contain references to globals? *)
49              needsWorld : bool,            (* does the method need the world (e.g., to print) *)
50              body : block                  (* the method body *)
51        }        }
52    
53      and create = Create of {      and create = Create of {
# Line 43  Line 56 
56        }        }
57    
58      and block = Block of {      and block = Block of {
59          locals : var list,          locals : var list ref,          (* the local variables that are used in this block *)
60          body : stm list          body : stm list
61        }        }
62    
63      and stm      and stm
64        = S_Comment of string list        = S_Comment of string list
65        | 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 *)
66        | S_Assign of var * exp        | S_MAssign of var list * exp
67        | S_GAssign of global_var * exp        | S_GAssign of global_var * exp
68        | S_IfThen of exp * block        | S_IfThen of exp * block
69        | S_IfThenElse of exp * block * block        | S_IfThenElse of exp * block * block
70          | S_For of var * exp * exp * block
71        | S_Foreach of var * exp * block        | S_Foreach of var * exp * block
72      (* special Diderot forms *)      (* special Diderot forms *)
73        | S_LoadNrrd of var * string        | S_LoadNrrd of var * APITypes.t * string
74        | S_Input of global_var * string * string option * exp option (* get input *)        | S_Input of global_var * string * string option * exp option (* get input *)
75        | 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 *)
76        | S_New of Atom.atom * exp list   (* new strand creation *)        | S_New of Atom.atom * exp list   (* new strand creation *)
77        | S_Save of state_var * exp  (* save strand state *)        | S_Save of state_var * exp  (* save strand state *)
78        | S_Exit of exp list  (* QUESTION: do we need S_Exit? *)
79          | S_Exit
80          | S_Print of Ty.t list * exp list
81      (* return functions for methods *)      (* return functions for methods *)
82        | S_Active        | S_Active
83        | S_Stabilize        | S_Stabilize
# Line 69  Line 85 
85    
86      and exp      and exp
87        = E_Global of global_var        = E_Global of global_var
88        | E_State of state_var        | E_State of exp option * state_var
89        | E_Var of var        | E_Var of var
90        | E_Lit of Literal.t        | E_Lit of Literal.t
91        | E_Op of Op.rator * exp list        | E_Op of Op.rator * exp list
92        | E_Cons of exp list * Ty.ty        | E_Vec of int * int * exp list           (* vector value; the ints are the width and the
93        | E_Seq of exp list * Ty.ty                                                   * padded width
94        | E_Pack of exp list              (* pack vector pieces into composite vector value *)                                                   *)
95          | E_Cons of exp list * Ty.t               (* in-memory tensor value *)
96          | E_Seq of exp list * Ty.t
97          | E_Pack of Ty.vec_layout * exp list      (* pack vector pieces into composite vector value *)
98          | E_VLoad of Ty.vec_layout * exp * int    (* load a piece of a composite vector value from
99                                                     * a tensor.  The integer specifies the index of
100                                                     * the piece.
101                                                     *)
102    
103      and global_var = GV of {      and global_var = GV of {
104          name : string,          (* name (should be unique) *)          name : string,          (* name (should be unique) *)
105          ty : Ty.ty,             (* type *)          ty : Ty.t,                      (* type *)
106          input : bool            (* is an input variable *)          xty : APITypes.t option,        (* external type (for inputs and outputs) *)
107            input : bool,                   (* is an input variable *)
108            output : bool,                  (* is the output value of the strand *)
109            varying : bool                  (* varies over the execution of the program *)
110        }        }
111    
112      and state_var = SV of {      and state_var = SV of {
113          name : string,          (* name (should be unique) *)          name : string,          (* name (should be unique) *)
114          ty : Ty.ty,             (* type *)          ty : Ty.t,                      (* type *)
115          varying : bool,         (* varies over the lifetime of the strand *)          xty : APITypes.t option,        (* external type (for outputs) *)
116          output : bool           (* is the output value of the strand *)          varying : bool                  (* varies over the lifetime of the strand *)
117        }        }
118    
119      and var = V of {      and var = V of {
120          name : string,                  (* name (should be unique) *)          name : string,                  (* name (should be unique) *)
121          id : Stamp.stamp,               (* unique ID *)          id : Stamp.stamp,               (* unique ID *)
122          ty : Ty.ty                      (* type *)          ty : Ty.t                       (* type *)
123        }        }
124    
125      withtype input = global_var Inputs.input      withtype input = global_var Inputs.input
126    
     structure GlobalVar : sig  
         val name : global_var -> string  
         val ty : global_var -> Ty.ty  
         val isInput : global_var -> bool  
         val toString : global_var -> string  
       end = struct  
         fun name (GV{name, ...}) = name  
         fun ty (GV{ty, ...}) = ty  
         fun isInput (GV{input, ...}) = input  
         fun toString (GV{name, ...}) = "globals." ^ name  
       end  
   
     structure StateVar : sig  
         val name : state_var -> string  
         val ty : state_var -> Ty.ty  
         val isOutput : state_var -> bool  
         val isVarying : state_var -> bool  
         val toString : state_var -> string  
       end = struct  
         fun name (SV{name, ...}) = name  
         fun ty (SV{ty, ...}) = ty  
         fun isOutput (SV{output, ...}) = output  
         fun isVarying (SV{varying, ...}) = varying  
         fun toString (SV{name, ...}) = "self." ^ name  
       end  
   
     structure Var : sig  
   
         val new : string * Ty.ty -> var  
         val name : var -> string  
         val toString : var -> string  
         val ty : var -> Ty.ty  
   
         structure Set : ORD_SET where type Key.ord_key = var  
         structure Map : ORD_MAP where type Key.ord_key = var  
   
       end = struct  
         fun new (name, ty) = V{  
                 name = name,  
                 id = Stamp.new(),  
                 ty = ty  
               }  
         fun name (V{name, ...}) = name  
         fun toString (V{name, id, ...}) = concat[name, "$", Stamp.toString id]  
         fun ty (V{ty, ...}) = ty  
         local  
           structure VarOrd =  
             struct  
               type ord_key = var  
               fun compare (V{id=a, ...}, V{id=b, ...}) = Stamp.compare(a, b)  
             end  
         in  
         structure Set = RedBlackSetFn (VarOrd)  
         structure Map = RedBlackMapFn (VarOrd)  
         end (* local *)  
       end  
   
127    end    end

Legend:
Removed from v.3767  
changed lines
  Added in v.3995

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