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

SCM Repository

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

Diff of /branches/vis15/src/compiler/low-to-tree/low-to-tree.sml

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

revision 3843, Tue May 10 15:10:58 2016 UTC revision 3855, Fri May 13 20:47:16 2016 UTC
# Line 8  Line 8 
8    
9  structure LowToTree : sig  structure LowToTree : sig
10    
11      val translate : LowIR.program * (int -> TreeTypes.vec_layout) -> TreeIR.program      val translate : LowIR.program * Env.target_info -> TreeIR.program
12    
13    end = struct    end = struct
14    
# Line 64  Line 64 
64      val {getFn = getStateVar, ...} = SV.newProp mkStateVar      val {getFn = getStateVar, ...} = SV.newProp mkStateVar
65      end      end
66    
67      fun mkBlock stms = T.Block{locals=[], body=stms}    (* for variables that are in an equivalence class (see UnifyVars), we use a single
68       * TreeIR variable (or vector of variables) to represent them.
69       *)
70        datatype eq_var_rep = NOEQ | VAR of TV.t | VEC of TV.t list
71        local
72          val {peekFn : IR.var -> eq_var_rep option, setFn, ...} =
73                V.newProp (fn x => raise Fail(V.toString x))
74        in
75        fun eqClassRepOf (env, x) = (case UnifyVars.eqClassOf x
76               of SOME x' => (case peekFn x'
77                     of SOME b => b
78                      | NONE => let
79                          val rep = (case V.ty x'
80                                 of Ty.TensorTy[d] => VEC(Util.newVectorVars(Env.layoutVec env d))
81                                  | ty => VAR(Util.newLocalVar x)
82                                (* end case *))
83                          in
84                            setFn (x, rep);
85                            rep
86                          end
87                    (* end case *))
88                | NONE => NOEQ
89              (* end case *))
90        end
91    
92        fun mkBlock stms = T.Block{locals = ref [], body = stms}
93      fun mkIf (x, stms, []) = T.S_IfThen(x, mkBlock stms)      fun mkIf (x, stms, []) = T.S_IfThen(x, mkBlock stms)
94        | mkIf (x, stms1, stms2) = T.S_IfThenElse(x, mkBlock stms1, mkBlock stms2)        | mkIf (x, stms1, stms2) = T.S_IfThenElse(x, mkBlock stms1, mkBlock stms2)
95    
# Line 73  Line 98 
98        | cvtScalarTy (Ty.TensorTy[]) = TTy.realTy        | cvtScalarTy (Ty.TensorTy[]) = TTy.realTy
99        | cvtScalarTy ty = raise Fail(concat["cvtScalarTy(", Ty.toString ty, ")"])        | cvtScalarTy ty = raise Fail(concat["cvtScalarTy(", Ty.toString ty, ")"])
100    
101      (* get a variable's binding as a single argument expression.  This means that
102       * if x is bound to a vector of expressions, then we need to pack it.
103       *)
104        fun singleArg env x = (case Env.useVar env x
105               of Env.TREE e => e
106                | Env.VEC(layout, es) => T.E_Pack(layout, es)
107              (* end case *))
108    
109        fun scalarArg env x = (case Env.useVar env x
110               of Env.TREE e => e
111                | _ => raise Fail("expected scalar binding for " ^ IR.Var.toString x)
112              (* end case *))
113    
114      fun vectorArg (env, x) = (case Env.useVar env x      fun vectorArg (env, x) = (case Env.useVar env x
115             of Env.TREE e => (case IR.Var.ty x             of Env.TREE e => (case V.ty x
116                   of Ty.TensorTy[d] => let                   of Ty.TensorTy[d] => let
117                        val layout = Env.layoutVec env d                        val layout = Env.layoutVec env d
118  (* QUESTION: can "e" be a complicated expression or are we guaranteed that it will just  (* QUESTION: can "e" be a complicated expression or are we guaranteed that it will just
# Line 86  Line 124 
124                        in                        in
125                          (layout, es)                          (layout, es)
126                        end                        end
127                    | ty => raise Fail("expected TensorTy[_], but found " ^ Ty.toString ty)                    | ty => raise Fail(concat[
128                            "expected ", V.toString x, " : TensorTy[_], but found " ^ Ty.toString ty
129                          ])
130                  (* end case *))                  (* end case *))
131              | Env.VEC(layout, es) => (layout, es)              | Env.VEC(layout, es) => (layout, es)
132            (* end case *))            (* end case *))
# Line 107  Line 147 
147              (layout, List.map List.rev argLists)              (layout, List.map List.rev argLists)
148            end            end
149    
150      fun trOp (env, lhs, srcRator, args) = let      fun trOp (env, srcRator, args) = let
151            fun bindOp rator = let  (* FIXME: if the use count of lhs is > 1, then we should bind to a local variable *)
152                  fun getBinding x = (case Env.useVar env x            fun bindOp rator = Env.TREE(T.E_Op(rator, List.map (scalarArg env) args))
                        of Env.TREE e => e  
                         | _ => raise Fail("expected single binding for " ^ IR.Var.toString x)  
                       (* end case *))  
                 in  
                   Env.TREE(T.E_Op(rator, List.map getBinding args))  
                 end  
153            fun bindVOp rator = let            fun bindVOp rator = let
154                  val (layout, argss) = vectorArgs (env, args)                  val (layout, argss) = vectorArgs (env, args)
155                  val exps = ListPair.map                  val exps = ListPair.map
# Line 152  Line 186 
186                | Op.RLerp => bindOp TOp.RLerp                | Op.RLerp => bindOp TOp.RLerp
187                | Op.VAdd _ => bindVOp TOp.VAdd                | Op.VAdd _ => bindVOp TOp.VAdd
188                | Op.VSub _ => bindVOp TOp.VSub                | Op.VSub _ => bindVOp TOp.VSub
189                | Op.VScale _ => bindVOp TOp.VScale                | Op.VScale _ => let
190                      val [s, v] = args
191                      val s = scalarArg env s
192                      val (layout, vs) = vectorArg (env, v)
193                      val exps = ListPair.map
194                          (fn (w, x) => T.E_Op(TOp.VScale w, [s, x]))
195                            (#pieces layout, vs)
196                      in
197                        Env.VEC(layout, exps)
198                      end
199                | Op.VMul _ => bindVOp TOp.VMul                | Op.VMul _ => bindVOp TOp.VMul
200                | Op.VNeg _ => bindVOp TOp.VNeg                | Op.VNeg _ => bindVOp TOp.VNeg
201                | Op.VSum _ => ??                | Op.VSum _ => raise Fail "FIXME: VSum"
202                | Op.VIndex(d, i) => ??                | Op.VIndex(_, i) => let
203                | Op.VClamp n => bindVOp TOp.VClamp  (* FIXME: more efficient to lookup the variable and avoid expanding TREE args *)
204                      val ({pieces, ...}, es) = vectorArg (env, hd args)
205                      fun select (i, w::ws, e::es) =
206                            if (i < w)
207    (* FIXME: what if lhs is used more than once? *)
208                              then Env.TREE(T.E_Op(TOp.VIndex(w, i), [e]))
209                              else select (i-w, ws, es)
210                        | select _ = raise Fail("bogus " ^ Op.toString srcRator)
211                      in
212                        select (i, pieces, es)
213                      end
214                  | Op.VClamp n => let
215                      val [v, lo, hi] = args
216                      val (layout, vs) = vectorArg (env, v)
217                      val lo = scalarArg env lo
218                      val hi = scalarArg env hi
219                      val exps = ListPair.map
220                          (fn (w, x) => T.E_Op(TOp.VClamp w, [x, lo, hi]))
221                            (#pieces layout, vs)
222                      in
223                        Env.VEC(layout, exps)
224                      end
225                | Op.VMapClamp n => bindVOp TOp.VMapClamp                | Op.VMapClamp n => bindVOp TOp.VMapClamp
226                | Op.VLerp n => bindVOp TOp.VLerp                | Op.VLerp n => bindVOp TOp.VLerp
227                | Op.TensorIndex(ty, idxs) => ??                | Op.TensorIndex(ty, idxs) => bindOp(TOp.TensorIndex(Util.trType ty, idxs))
228                | Op.ProjectLast(ty, idxs) => ??                | Op.ProjectLast(ty, idxs) => bindOp(TOp.ProjectLast(Util.trType ty, idxs))
229                | Op.EigenVecs2x2 => ??                | Op.Zero ty => bindOp (TOp.Zero(Util.trType ty))
230                | Op.EigenVecs3x3 => ??                | Op.Select(ty, i) => bindOp (TOp.Select(Util.trType ty, i))
231                | Op.EigenVals2x2 => ??                | Op.Subscript ty => bindOp (TOp.Subscript(Util.trType ty))
232                | Op.EigenVals3x3 => ??                | Op.MkDynamic(ty, n) => bindOp (TOp.MkDynamic(Util.trType ty, n))
233                | Op.Zero ty => ??                | Op.Append ty => bindOp (TOp.Append(Util.trType ty))
234                | Op.Select(Ty.TupleTy tys, i) => ??                | Op.Prepend ty => bindOp (TOp.Prepend(Util.trType ty))
235                | Op.Subscript(Ty.SeqTy(ty, NONE)) => ??                | Op.Concat ty => bindOp (TOp.Concat(Util.trType ty))
236                | Op.Subscript(Ty.SeqTy(ty, SOME _)) => ??                | Op.Range => bindOp TOp.Range
237                | Op.MkDynamic(ty, n) => ??                | Op.Length ty => bindOp (TOp.Length(Util.trType ty))
238                | Op.Append ty => ??                | Op.SphereQuery(ty1, ty2) => raise Fail "FIXME: SphereQuery"
               | Op.Prepend ty => ??  
               | Op.Concat ty => ??  
               | Op.Range => ??  
               | Op.Length ty => ??  
               | Op.SphereQuery(ty1, ty2) => ??  
239                | Op.Sqrt => bindOp TOp.Sqrt                | Op.Sqrt => bindOp TOp.Sqrt
240                | Op.Cos => bindOp TOp.Cos                | Op.Cos => bindOp TOp.Cos
241                | Op.ArcCos => bindOp TOp.ArcCos                | Op.ArcCos => bindOp TOp.ArcCos
# Line 185  Line 244 
244                | Op.Tan => bindOp TOp.Tan                | Op.Tan => bindOp TOp.Tan
245                | Op.ArcTan => bindOp TOp.ArcTan                | Op.ArcTan => bindOp TOp.ArcTan
246                | Op.Ceiling 1 => bindOp (TOp.Ceiling 1)                | Op.Ceiling 1 => bindOp (TOp.Ceiling 1)
247                | Op.Ceiling d => ??                | Op.Ceiling d => bindVOp TOp.Ceiling
248                | Op.Floor 1 => bindOp (TOp.Floor 1)                | Op.Floor 1 => bindOp (TOp.Floor 1)
249                | Op.Floor d => ??                | Op.Floor d => bindVOp TOp.Floor
250                | Op.Round 1 => bindOp (TOp.Floor 1)                | Op.Round 1 => bindOp (TOp.Round 1)
251                | Op.Round d => ??                | Op.Round d => bindVOp TOp.Round
252                | Op.Trunc 1 => bindOp (TOp.Trunc 1)                | Op.Trunc 1 => bindOp (TOp.Trunc 1)
253                | Op.Trunc d => ??                | Op.Trunc d => bindVOp TOp.Trunc
254                | Op.IntToReal => bindOp TOp.IntToReal                | Op.IntToReal => bindOp TOp.IntToReal
255                | Op.RealToInt 1 => ??                | Op.RealToInt 1 => bindOp (TOp.RealToInt 1)
256                | Op.RealToInt d => ??                | Op.RealToInt d => Env.TREE(T.E_Op(TOp.RealToInt d, List.map (singleArg env) args))
257    (* FIXME
258                | Op.R_All ty => ??                | Op.R_All ty => ??
259                | Op.R_Exists ty => ??                | Op.R_Exists ty => ??
260                | Op.R_Max ty => ??                | Op.R_Max ty => ??
# Line 203  Line 263 
263                | Op.R_Product ty => ??                | Op.R_Product ty => ??
264                | Op.R_Mean ty => ??                | Op.R_Mean ty => ??
265                | Op.R_Variance ty => ??                | Op.R_Variance ty => ??
266                | Op.Transform info => ??  *)
267                | Op.Translate info => ??                | Op.Transform info => bindOp (TOp.Transform info)
268                | Op.ControlIndex(info, ctl, d) => ??                | Op.Translate info => bindOp (TOp.Translate info)
269                | Op.LoadVoxel info => ??                | Op.ControlIndex(info, ctl, d) => bindOp (TOp.ControlIndex(info, ctl, d))
270                | Op.Inside(info, s) => ??                | Op.LoadVoxel info => bindOp (TOp.LoadVoxel info)
271                  | Op.Inside(info, s) => bindOp (TOp.Inside(info, s))
272                | Op.ImageDim(info, d) => bindOp(TOp.ImageDim(info, d))                | Op.ImageDim(info, d) => bindOp(TOp.ImageDim(info, d))
273                | Op.LoadSeq(ty, file) => ??                | Op.LoadSeq(ty, file) => bindOp(TOp.LoadSeq(Util.trType ty, file))
274                | Op.LoadImage(ty, file) => ??                | Op.LoadImage(ty, file) => bindOp(TOp.LoadImage(Util.trType ty, file))
275                | Op.MathFn f => bindOp (TOp.MathFn f)                | Op.MathFn f => bindOp (TOp.MathFn f)
276                | rator => raise Fail("bogus operator " ^ Op.toString rator)                | rator => raise Fail("bogus operator " ^ Op.toString srcRator)
277              (* end case *)              (* end case *)
278            end            end
279    
280    (* cases:
281            x in EqClass
282                    issue assignment; lhs is binding of representative (could be multiple vars)
283            useCount(x) > 1 and rhs is not simple
284            rhs is simple
285            rhs is vector
286    *)
287      fun trAssign (env, lhs, rhs) = let      fun trAssign (env, lhs, rhs) = let
288          (* simple binding for lhs variable; we check to see if it is part of an merged          (* simple binding for lhs variable; we check to see if it is part of an merged
289           * equivalence class, in which case we need to generate the assigment.           * equivalence class, in which case we need to generate the assigment.
290           *)           *)
291            fun bindSimple rhs = (case UnifyVars.eqClassOf lhs            fun bindSimple rhs = (case eqClassRepOf(env, lhs)
292                   of SOME x => ??                   of NOEQ => (Env.bindSimple (env, lhs, Env.TREE rhs); [])
293                    | NONE => (Env.bindSimple (env, lhs, rhs); (env, []))                    | VAR x' => [T.S_Assign(x', rhs)]
294                      | VEC _ => raise Fail "unexpected VEC"
295                  (* end case *))                  (* end case *))
296              fun assignOp (rator, args) = let
297    (* FIXME: what if lhs is EqClass var? *)
298                    val t = Util.newLocalVar lhs
299                    val stm = T.S_Assign(t, T.E_Op(rator, List.map (scalarArg env) args))
300                    in
301                      Env.bindSimple (env, lhs, Env.TREE(T.E_Var t));
302                      [stm]
303                    end
304            in            in
305              case rhs              case rhs
306               of IR.GLOBAL x => bindSimple (T.E_Global(getGlobalVar x))               of IR.GLOBAL x => bindSimple (T.E_Global(mkGlobalVar x))
307                | IR.STATE x => bindSimple (T.E_State(getStateVar x))                | IR.STATE(NONE, fld) =>
308                | IR.VAR x => (case Env.useVar env x                    bindSimple (T.E_State(NONE, getStateVar fld))
309                     of Env.TREE e => ??                | IR.STATE(SOME x, fld) =>
310                      | Env.VEC(layout, es) => ??                    bindSimple (T.E_State(SOME(scalarArg env x), getStateVar fld))
311                    (* end case *))                | IR.VAR x => raise Fail "FIXME: VAR"
312                | IR.LIT lit => bindSimple (env, lhs, T.E_Lit lit)                | IR.LIT lit => bindSimple (T.E_Lit lit)
313                | IR.OP(rator, args) => ??                | IR.OP(Op.EigenVecs2x2, args) => assignOp (TOp.EigenVecs2x2, args)
314                | IR.CONS(args, ty) => ??                | IR.OP(Op.EigenVecs3x3, args) => assignOp (TOp.EigenVecs3x3, args)
315                | IR.SEQ(args, ty) => ??                | IR.OP(Op.EigenVals2x2, args) => assignOp (TOp.EigenVals2x2, args)
316                | IR.EINAPP _ => raise Fail "unexpected EINAPP in LowIR code"                | IR.OP(Op.EigenVals3x3, args) => assignOp (TOp.EigenVals3x3, args)
317                  | IR.OP(rator, args) => let
318                      val rhs = trOp (env, rator, args)
319                      val needAssignment = (V.useCount lhs > 1) orelse (Env.isInlineOp env rator)
320                      in
321                        case (rhs, eqClassRepOf(env, lhs), needAssignment)
322                         of (_, NOEQ, false) => (Env.bindSimple (env, lhs, rhs); [])
323                          | (Env.TREE e, NOEQ, true) => let
324                              val t = Util.newLocalVar lhs
325                              in
326                                Env.bindSimple (env, lhs, Env.TREE(T.E_Var t));
327                                [T.S_Assign(t, e)]
328                              end
329                          | (Env.TREE e, VAR x', _) => [T.S_Assign(x', e)]
330                          | (Env.VEC(layout, es), NOEQ, true) => let
331                              val vs = Util.newVectorVars layout
332                              in
333                                Env.bindSimple (env, lhs, Env.VEC(layout, List.map T.E_Var vs));
334                                ListPair.mapEq T.S_Assign (vs, es)
335                              end
336                          | (Env.VEC(layout, es), VEC xs, _) => ListPair.mapEq T.S_Assign (xs, es)
337                          | _ => raise Fail "inconsistent"
338                        (* end case *)
339                      end
340                  | IR.CONS(args, Ty.TensorTy[d]) => let
341                      val layout = Env.layoutVec env d
342                      fun mkVecs (args, w::ws) = let
343    (* FIXME: what about padding? *)
344                            fun take (0, args, es) = T.E_Vec(List.rev es) :: mkVecs (args, ws)
345                              | take (i, [], es) = if #padded layout andalso null ws
346                                  then let
347                                    val zero = T.E_Lit(Literal.Real(RealLit.zero false))
348                                    in
349                                      [T.E_Vec(List.rev(List.tabulate(i, fn _ => zero)))]
350                                    end
351                                  else raise Fail "too few arguments for CONS"
352                              | take (i, arg::args, es) = take (i-1, args, scalarArg env arg :: es)
353                            in
354                              take (w, args, [])
355                            end
356                        | mkVecs ([], []) = []
357                        | mkVecs (_, []) = raise Fail "too many arguments for CONS"
358                      val es = mkVecs (args, #pieces layout)
359                      in
360                        case (eqClassRepOf(env, lhs), V.useCount lhs > 1)
361                         of (NOEQ, false) => (Env.bindVar(env, lhs, Env.VEC(layout, es)); [])
362                          | (NOEQ, true) => let
363                              val vs = Util.newVectorVars layout
364                              in
365                                Env.bindSimple (env, lhs, Env.VEC(layout, List.map T.E_Var vs));
366                                ListPair.mapEq T.S_Assign (vs, es)
367                              end
368                          | (VEC xs, _) => ListPair.mapEq T.S_Assign (xs, es)
369                          | _ => raise Fail "inconsistent"
370                        (* end case *)
371                      end
372                  | IR.CONS(args, Ty.TensorTy shape) => raise Fail "FIXME: CONS"
373                  | IR.SEQ(args, ty) => raise Fail "FIXME: SEQ"
374                  | rhs => raise Fail(concat["unexpected ", IR.RHS.toString rhs, " in LowIR code"])
375              (* end case *)              (* end case *)
376            end            end
377    
# Line 255  Line 390 
390       *)       *)
391        | ELSE_BR of T.stm list * T.exp * T.stm list * IR.node_kind        | ELSE_BR of T.stm list * T.exp * T.stm list * IR.node_kind
392    
393      fun trCFGWithEnv (env, prefix, cfg) = let      fun trCFGWithEnv (env, cfg) = let
394            fun useScalar x = (case Env.useVar env x            fun useScalar x = (case Env.useVar env x
395                   of Env.TREE e => e                   of Env.TREE e => e
396                    | _ => raise Fail("expected scalar binding for " ^ V.toString x)                    | _ => raise Fail("expected scalar binding for " ^ V.toString x)
397                  (* end case *))                  (* end case *))
398            val _ = UnifyVars.analyze cfg            val _ = UnifyVars.analyze cfg
399          (* join (env, stk, stms, k): handle a control-flow join, where env is the          (* join (stk, stms, k): handle a control-flow join, where env is the
400           * current environment, stk is the stack of open ifs (the top of stk specifies           * current environment, stk is the stack of open ifs (the top of stk specifies
401           * which branch we are in), stms are the TreeIL statements preceding the join           * which branch we are in), stms are the TreeIL statements preceding the join
402           * on the current path, and k is the kind of the join node (either JOIN or EXIT).           * on the current path, and k is the kind of the join node (either JOIN or EXIT).
403           *)           *)
404            fun join (env, [], _, IR.JOIN _) = raise Fail "JOIN with no open if"            fun join ([], _, IR.JOIN _) = raise Fail "JOIN with no open if"
405              | join (env, [], stms, _) = Env.endScope (env, prefix @ List.rev stms)              | join ([], stms, _) = mkBlock (List.rev stms)
406              | join (env, THEN_BR(stms1, cond, elseBr)::stk, thenBlk, k) = let              | join (THEN_BR(stms1, cond, elseBr)::stk, thenBlk, k) = let
407                  val (env, thenBlk) = Env.flushPending (env, thenBlk)                  val thenBlk = Env.flushPending (env, thenBlk)
408                  in                  in
409                    doNode (env, elseBr, ELSE_BR(stms1, cond, thenBlk, k)::stk, [])                    doNode (elseBr, ELSE_BR(stms1, cond, thenBlk, k)::stk, [])
410                  end                  end
411              | join (env, ELSE_BR(stms, cond, thenBlk, k1)::stk, elseBlk, k2) = let              | join (ELSE_BR(stms, cond, thenBlk, k1)::stk, elseBlk, k2) = let
412                  val (env, elseBlk) = Env.flushPending (env, elseBlk)                  val elseBlk = Env.flushPending (env, elseBlk)
413                  in                  in
414                    case (k1, k2)                    case (k1, k2)
415                     of (IR.JOIN{phis, succ, ...}, IR.JOIN _) => let                     of (IR.JOIN{succ, ...}, IR.JOIN _) => let
                         val (env, [thenBlk, elseBlk]) =  
                               List.foldl doPhi (env, [thenBlk, elseBlk]) (!phis)  
416                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)
417                          in                          in
418                            doNode (env, !succ, stk, stm::stms)                            doNode (!succ, stk, stm::stms)
419                          end                          end
420                      | (IR.JOIN{phis, succ, ...}, _) => let                      | (IR.JOIN{succ, ...}, _) => let
                         val (env, [thenBlk]) = List.foldl doPhi (env, [thenBlk]) (!phis)  
421                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)
422                          in                          in
423                            doNode (env, !succ, stk, stm::stms)                            doNode (!succ, stk, stm::stms)
424                          end                          end
425                      | (_, IR.JOIN{phis, succ, ...}) => let                      | (_, IR.JOIN{succ, ...}) => let
                         val (env, [elseBlk]) = List.foldl doPhi (env, [elseBlk]) (!phis)  
426                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)
427                          in                          in
428                            doNode (env, !succ, stk, stm::stms)                            doNode (!succ, stk, stm::stms)
429                          end                          end
430                      | (_, _) => let                      | (_, _) => let
431                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)                          val stm = mkIf(cond, List.rev thenBlk, List.rev elseBlk)
432                          in                          in
433                            Env.endScope (env, prefix @ List.rev(stm::stms))                            mkBlock (List.rev(stm::stms))
434                          end                          end
435                    (* end case *)                    (* end case *)
436                  end                  end
437            and doNode (env, nd : IR.node, ifStk : open_if list, stms) = (case IR.Node.kind nd            and doNode (nd : IR.node, ifStk : open_if list, stms) = (case IR.Node.kind nd
438                   of IR.NULL => raise Fail "unexpected NULL"                   of IR.NULL => raise Fail "unexpected NULL"
439                    | IR.ENTRY{succ} => doNode (env, !succ, ifStk, stms)                    | IR.ENTRY{succ} => doNode (!succ, ifStk, stms)
440                    | k as IR.JOIN _ => join (env, ifStk, stms, k)                    | k as IR.JOIN _ => join (ifStk, stms, k)
441                    | IR.COND{cond, trueBranch, falseBranch, ...} => let                    | IR.COND{cond, trueBranch, falseBranch, ...} => let
442                        val cond = useScalar (!cond)                        val cond = useScalar (!cond)
443                        val (env, stms) = Env.flushPending (env, stms)                        val stms = Env.flushPending (env, stms)
444                        in                        in
445                          doNode (env, !trueBranch, THEN_BR(stms, cond, !falseBranch)::ifStk, [])                          doNode (!trueBranch, THEN_BR(stms, cond, !falseBranch)::ifStk, [])
446                        end                        end
447                    | IR.FOREACH{var, src, bodyEntry, succ, ...} => let                    | IR.FOREACH{var, src, bodyEntry, succ, ...} => let
448                        val src = useScalar (!src)                        val src = useScalar (!src)
449                        val var = ??                        val var = Util.newIterVar var
450                        val (env, stms) = Env.flushPending (env, stms)                        val stms = Env.flushPending (env, stms)
451                        val body = doNode (env, !bodyEntry, [], [])                        val body = doNode (!bodyEntry, [], [])
452                        val stm = T.S_Foreach(var, src, body)                        val stm = T.S_Foreach(var, src, body)
453                        in                        in
454                          doNode (env, !succ, ifStk, stm::stms)                          doNode (!succ, ifStk, stm::stms)
455                        end                        end
456                    | IR.NEXT _ => Env.endScope (env, List.rev stms)                    | IR.NEXT _ => mkBlock (List.rev stms)
457                    | IR.COM {text, succ, ...} =>                    | IR.COM {text, succ, ...} =>
458                        doNode (env, !succ, ifStk, T.S_Comment text :: stms)                        doNode (!succ, ifStk, T.S_Comment text :: stms)
459                    | IR.ASSIGN{stm=(lhs, rhs), succ, ...} => let                    | IR.ASSIGN{stm=(lhs, rhs), succ, ...} => let
460                        val (env, stms') = doAssign (env, lhs, rhs)                        val stms' = trAssign (env, lhs, rhs)
461                        in                        in
462                          doNode (env, !succ, ifStk, stms' @ stms)                          doNode (!succ, ifStk, stms' @ stms)
463                        end                        end
464                    | IR.MASSIGN{stm=(ys, rator, xs), succ, ...} => let                    | IR.MASSIGN{stm=(ys, rator, xs), succ, ...} => let
465                        fun doit () = let                        fun doLHSVar (y, ys) = let
466                              fun doLHSVar (y, (env, ys)) = let                              val t = Util.newLocalVar y
                                   val t = newLocal y  
467                                    in                                    in
468                                      (rename (addLocal(env, t), y, t), t::ys)                                Env.bindSimple (env, y, Env.TREE(T.E_Var t));
469                                  t::ys
470                                    end                                    end
471                              val (env, ys) = List.foldr doLHSVar (env, []) ys                        val ys = List.foldr doLHSVar [] ys
472  (* FIXME: need to translate the operator *)                        val rator = (case rator
473                              val exp = T.E_Op(rator, List.map (useVar env) xs)                               of Op.Print tys => TOp.Print(List.map Util.trType tys)
474                              val stm = T.S_Assign(ys, exp)                                | _ => raise Fail(concat[
475                                        "unexepected operator ", Op.toString rator, " for MASSIGN"
476                                      ])
477                                (* end case *))
478                          val stm = T.S_MAssign(ys, T.E_Op(rator, List.map (singleArg env) xs))
479                              in                              in
480                                doNode (env, !succ, ifStk, stm :: stms)                          doNode (!succ, ifStk, stm :: stms)
481                              end                              end
482                      | IR.GASSIGN{lhs, rhs, succ, ...} => let
483                          val stm = T.S_GAssign(mkGlobalVar lhs, singleArg env rhs)
484                        in                        in
485                          case rator                          doNode (!succ, ifStk, stm::stms)
                          of Op.Print _ => if Target.supportsPrinting()  
                               then doit ()  
                               else doNode (env, !succ, ifStk, stms)  
                           | _ => doit()  
                         (* end case *)  
486                        end                        end
487                    | IR.GASSIGN{lhs, rhs, succ, ...} => let                    | IR.NEW{strand, args, succ, ...} => let
488                        val stm = (case useAsInput(env, rhs)                        val stm = T.S_New(strand, List.map (singleArg env) args)
                              of NONE => T.S_GAssign(getGlobalVar lhs, useVar env rhs)  
                               | SOME mkStm => mkStm(getGlobalVar lhs)  
                             (* end case *))  
489                        in                        in
490                          doNode (env, !succ, ifStk, stm::stms)                          doNode (!succ, ifStk, stm::stms)
491                        end                        end
                   | IR.NEW{strand, args, succ, ...} => raise Fail "NEW unimplemented"  
492                    | IR.SAVE{lhs, rhs, succ, ...} => let                    | IR.SAVE{lhs, rhs, succ, ...} => let
493                        val stm = T.S_Save([getStateVar lhs], useVar env rhs)                        val stm = T.S_Save(getStateVar lhs, singleArg env rhs)
494                        in                        in
495                          doNode (env, !succ, ifStk, stm::stms)                          doNode (!succ, ifStk, stm::stms)
496                        end                        end
497                    | k as IR.EXIT{kind, succ, ...} => (case (!succ, kind)                    | k as IR.EXIT{kind, succ, ...} => (case (!succ, kind)
498                         of (NONE, ExitKind.RETURN) => let                         of (NONE, ExitKind.RETURN) => mkBlock (List.rev(T.S_Exit :: stms))
499                              val suffix = [T.S_Exit]                          | (NONE, ExitKind.ACTIVE) => mkBlock (List.rev(T.S_Active :: stms))
                             in  
                               Env.endScope (env, prefix @ List.revAppend(stms, suffix))  
                             end  
                         | (NONE, ExitKind.ACTIVE) => let  
                             val suffix = [T.S_Active]  
                             in  
                               Env.endScope  (env, prefix @ List.revAppend(stms, suffix))  
                             end  
500                          | (NONE, ExitKind.STABILIZE) => let                          | (NONE, ExitKind.STABILIZE) => let
501                              val stms = T.S_Stabilize :: stms                              val stms = T.S_Stabilize :: stms
502                              in                              in
503                                join (env, ifStk, stms, k)                                join (ifStk, stms, k)
504                              end                              end
505                          | (NONE, ExitKind.DIE) => join (env, ifStk, T.S_Die :: stms, k)                          | (NONE, ExitKind.DIE) => join (ifStk, T.S_Die :: stms, k)
506                          | (NONE, ExitKind.UNREACHABLE) => join (env, ifStk, stms, k)                          | (NONE, ExitKind.UNREACHABLE) => join (ifStk, stms, k)
507                          | (SOME nd, ExitKind.ACTIVE) => doNode (env, nd, ifStk, T.S_Active :: stms)                          | (SOME nd, ExitKind.ACTIVE) => doNode (nd, ifStk, T.S_Active :: stms)
508                          | (SOME nd, ExitKind.STABILIZE) => doNode (env, nd, ifStk, T.S_Stabilize :: stms)                          | (SOME nd, ExitKind.STABILIZE) => doNode (nd, ifStk, T.S_Stabilize :: stms)
509                          | (SOME nd, ExitKind.DIE) => doNode (env, nd, ifStk, T.S_Die :: stms)                          | (SOME nd, ExitKind.DIE) => doNode (nd, ifStk, T.S_Die :: stms)
510                          | (SOME nd, ExitKind.UNREACHABLE) => doNode (env, nd, ifStk, stms)                          | (SOME nd, ExitKind.UNREACHABLE) => doNode (nd, ifStk, stms)
511                          | _ => raise Fail("unexpected continuation edge from "^IR.Node.toString nd)                          | _ => raise Fail("unexpected continuation edge from "^IR.Node.toString nd)
512                        (* end case *))                        (* end case *))
513                  (* end case *))                  (* end case *))
514            in            in
515              doNode (env, CFG.entry cfg, [], [])              ScopeVars.assignScopes (doNode (IR.CFG.entry cfg, [], []))
516            end            end
517    
518      fun trCFG vecLayout (prefix, cfg) = trCFGWithEnv (Env.new vecLayout, prefix, cfg)      fun trCFG info cfg = trCFGWithEnv (Env.new info, cfg)
519    
520      fun trStrand strand = let      fun trStrand info strand = let
521              val trCFG = trCFG info
522            val IR.Strand{name, params, state, stateInit, initM, updateM, stabilizeM} = strand            val IR.Strand{name, params, state, stateInit, initM, updateM, stabilizeM} = strand
523            val params' = List.map newParam params            val params' = List.map Util.newParamVar params
524            val env = ListPair.foldlEq            val env = Env.new info
525                  (fn (x, x', env) => rename(env, x, x')) ?? (params, params')            val () = ListPair.appEq
526                    (fn (x, x') => Env.bindSimple (env, x, Env.TREE(T.E_Var x'))) (params, params')
527            val state' = List.map getStateVar state            val state' = List.map getStateVar state
528            in            in
529              T.Strand{              T.Strand{
# Line 409  Line 531 
531                  params = params',                  params = params',
532                  state = state',                  state = state',
533                  stateInit = trCFGWithEnv (env, stateInit),                  stateInit = trCFGWithEnv (env, stateInit),
534                  initM = Option.map (fn cfg => trCFG ([], cfg)) initM,                  initM = Option.map trCFG initM,
535                  updateM = trCFG ([], updateM),                  updateM = trCFG updateM,
536                  stabilizeM = Option.map (fn cfg => trCFG ([], cfg)) stabilizeM                  stabilizeM = Option.map trCFG stabilizeM
537                }                }
538            end            end
539    
540      fun translate (prog, vecLayout) = let      fun translate (prog, info) = let
541            val LowIR.Program{            val LowIR.Program{
542                    props, consts, inputs, constInit, globals, globalInit, strand, create, update                    props, consts, inputs, constInit, globals, globalInit, strand, create, update
543                  } = prog                  } = prog
544              val trCFG = trCFG info
545            in            in
546              TreeIR.Program{              TreeIR.Program{
547                  props = props,                  props = props,
548                  consts = ??,                  consts = List.map mkGlobalVar consts,
549                  inputs = ??,                  inputs = List.map (Inputs.map mkGlobalVar) inputs,
550                  constInit = ??,                  constInit = trCFG constInit,
551                  globals = ??,                  globals = List.map mkGlobalVar globals,
552                  globalInit = ??,                  globalInit = trCFG globalInit,
553                  strand = ??,                  strand = trStrand info strand,
554                  create = ??,                  create = let
555                  update = ??                    val IR.Create{dim, code} = create
556                      in
557                        T.Create{dim = dim, code = trCFG code}
558                      end,
559                    update = Option.map trCFG update
560                }                }
561            end            end
562    

Legend:
Removed from v.3843  
changed lines
  Added in v.3855

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