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

SCM Repository

[diderot] Diff of /branches/vis15/src/compiler/simplify/inliner.sml
ViewVC logotype

Diff of /branches/vis15/src/compiler/simplify/inliner.sml

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

revision 3451, Sat Nov 21 21:11:21 2015 UTC revision 3485, Sun Dec 6 13:48:48 2015 UTC
# Line 39  Line 39 
39                    S.Block(List.rev stms)                    S.Block(List.rev stms)
40                  end                  end
41            and doStmt (env, isTop, stm) = (case stm            and doStmt (env, isTop, stm) = (case stm
42                   of S.S_Var x => let                   of S.S_Var(x, optE) => let
43                        val x' = V.copy x                        val x' = V.copy(x, V.kindOf x)
44                          val optE' = Option.map (doExp env) optE
45                        in                        in
46                          (V.Map.insert(env, x, x'), S.S_Var x')                          (V.Map.insert(env, x, x'), S.S_Var(x', optE'))
47                        end                        end
48                    | S.S_Assign(x, e) => (case V.Map.find(env, x)                    | S.S_Assign(x, e) => (env, S.S_Assign(rename env x, doExp env e))
                        of SOME x' => (env, S.S_Assign(x', doExp env e))  
                         | NONE => let  
                             val x' = V.copy x  
                             in  
                               (V.Map.insert(env, x, x'), S.S_Assign(x', doExp env e))  
                             end  
                       (* end case *))  
49                    | S.S_IfThenElse(x, b1, b2) =>                    | S.S_IfThenElse(x, b1, b2) =>
50                        (env, S.S_IfThenElse(rename env x, doBlock(env, false, b1), doBlock(env, false, b2)))                        (env, S.S_IfThenElse(rename env x, doBlock(env, false, b1), doBlock(env, false, b2)))
51                      | S.S_Foreach(x, xs, blk) =>
52                          (env, S.S_Foreach(rename env x, rename env xs, doBlock(env, false, blk)))
53                    | S.S_New(strnd, xs) => (env, S.S_New(strnd, List.map (rename env) xs))                    | S.S_New(strnd, xs) => (env, S.S_New(strnd, List.map (rename env) xs))
54                      | S.S_Continue => (env, stm)
55                    | S.S_Die => (env, stm)                    | S.S_Die => (env, stm)
56                    | S.S_Stabilize => (env, stm)                    | S.S_Stabilize => (env, stm)
57                    | S.S_Return x => (                    | S.S_Return x => (
58                        if not isTop then needsLHSPreDecl := true else ();                        if not isTop then needsLHSPreDecl := true else ();
59                        (env, S.S_Assign(lhs, S.E_Var(rename env x))))                        (env, S.S_Assign(lhs, S.E_Var(rename env x))))
60                    | S.S_Print xs => (env, S.S_Print(List.map (rename env) xs))                    | S.S_Print xs => (env, S.S_Print(List.map (rename env) xs))
61                      | S.S_MapReduce _ => raise Fail "unexpected MapReduce in function"
62                  (* end case *))                  (* end case *))
63            and doExp env exp = (case exp            and doExp env exp = (case exp
64                   of S.E_Var x => S.E_Var(rename env x)                   of S.E_Var x => S.E_Var(rename env x)
65                      | S.E_Select(x, fld) => S.E_Select(rename env x, fld)
66                    | S.E_Lit _ => exp                    | S.E_Lit _ => exp
67                    | S.E_Apply(f, xs, ty) => S.E_Apply(f, List.map (rename env) xs, ty)                    | S.E_Apply(f, xs, ty) => S.E_Apply(f, List.map (rename env) xs, ty)
68                    | S.E_Prim(f, tys, xs, ty) =>                    | S.E_Prim(f, tys, xs, ty) =>
69                        S.E_Prim(f, tys, List.map (rename env) xs, ty)                        S.E_Prim(f, tys, List.map (rename env) xs, ty)
70                    | S.E_Cons(xs, ty) => S.E_Cons(List.map (rename env) xs, ty)                    | S.E_Tensor(xs, ty) => S.E_Tensor(List.map (rename env) xs, ty)
71                    | S.E_Seq(xs, ty) => S.E_Seq(List.map (rename env) xs, ty)                    | S.E_Seq(xs, ty) => S.E_Seq(List.map (rename env) xs, ty)
72                    | S.E_Slice(x, xs, ty) =>                    | S.E_Slice(x, xs, ty) =>
73                        S.E_Slice(rename env x, List.map (Option.map (rename env)) xs, ty)                        S.E_Slice(rename env x, List.map (Option.map (rename env)) xs, ty)
# Line 84  Line 83 
83            val blk as S.Block stms = doBlock (env, true, body)            val blk as S.Block stms = doBlock (env, true, body)
84            in            in
85              if !needsLHSPreDecl              if !needsLHSPreDecl
86                then S.Block(S.S_Var lhs :: stms)                then S.Block(S.S_Var(lhs, NONE) :: stms)
87                else blk                else blk
88            end            end
89    
# Line 135  Line 134 
134    
135      fun transform (prog as S.Program{funcs=[], ...}) = prog      fun transform (prog as S.Program{funcs=[], ...}) = prog
136        | transform prog = let        | transform prog = let
137            val S.Program{props, inputs, globals, funcs, init, strand, create, update} = prog            val S.Program{props, consts, inputs, constInit, globals, funcs, init, strand, create, update} = prog
138          (* a table that maps function names to their definitions *)          (* a table that maps function names to their definitions *)
139            val funcTbl = V.Tbl.mkTable (List.length funcs, Fail "funcTbl")            val funcTbl = V.Tbl.mkTable (List.length funcs, Fail "funcTbl")
140          (* first we inline expand the function bodies in definition order *)          (* first we inline expand the function bodies in definition order *)
# Line 144  Line 143 
143            in            in
144              S.Program{              S.Program{
145                  props = props,                  props = props,
146                    consts = consts,
147                  inputs = inputs,                  inputs = inputs,
148                    constInit = constInit,
149                  globals = globals,                  globals = globals,
150                  init = expandBlock init,                  init = expandBlock init,
151                  funcs = [],                  funcs = [],
152                  strand = expandStrand funcTbl strand,                  strand = expandStrand funcTbl strand,
153                  create = expandBlock create,                  create = (case create
154                       of S.Create{dim, code} => S.Create{dim = dim, code = expandBlock code}
155                      (* end case *)),
156                  update = Option.map expandBlock update                  update = Option.map expandBlock update
157                }                }
158            end            end

Legend:
Removed from v.3451  
changed lines
  Added in v.3485

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