# SCM Repository

# View of /trunk/src/compiler/simplify/lift.sml

Parent Directory | Revision Log

Revision

File size: 3759 byte(s)

**227**- (**download**) (**annotate**)*Wed Aug 4 05:27:13 2010 UTC*(8 years, 9 months ago) by*jhr*File size: 3759 byte(s)

Working on lifting

(* lift.sml * * COPYRIGHT (c) 2010 The Diderot Project (http://diderot.cs.uchicago.edu) * All rights reserved. * * Lift field operations to global scope. *) structure Lift : sig val transform : Simple.program -> Simple.program end = struct structure BV = BasisVars structure S = Simple structure VSet = Var.Set structure VMap = Var.Map (* BV.op_at BV.op_D BV.fn_convolve BV.fn_load *) (* the kinds of things a variable in Simple AST can be bound to *) datatype var_binding = RHS of S.exp | Param (* identify the image load operations and lift them and their antecedents; in terms of BTA, * this phase is essentially determining what must be static in order to get the image * info needed for the rest of the compile. *) fun liftLoads block = let (* analysis to compute the set of static variables *) fun mkStatic (env, statics, x) = if VSet.member(statics, x) then statics else let val statics = VSet.add(statics, x) in case VMap.find(env, x) of SOME(S.E_Var y) => mkStatic (env, statics, y) | SOME(S.E_Tuple ys) => mkStatics (env, statics, ys) | SOME(S.E_Apply(_, _, ys, _)) => mkStatics (env, statics, ys) | SOME(S.E_Cons ys) => mkStatics (env, statics, ys) | SOME(S.E_Input(_, SOME y)) => mkStatic (env, statics, y) | SOME _ => statics | NONE => raise Fail(concat["variable ", Var.uniqueNameOf x, " has no binding"]) (* end case *) end and mkStatics (env, statics, xs) = List.foldl (fn (x, statics) => mkStatic(env, statics, x)) statics xs fun doBlock (env, statics, S.Block stms) = let fun doStmts (env, statics, []) = statics | doStmts (env, statics, stm::stms) = let val (env, statics) = doStmt (env, statics, stm) in doStmts (env, statics, stms) end in doStmts (env, statics, stms) end and doStmt (env, statics, stm) = (case stm of S.S_Assign(x, e) => let val env = VMap.insert(env, x, e) in case e of S.E_Apply(f, _, xs, _) => if Var.same(f, BV.fn_load) then (env, mkStatic(env, statics, x)) else (env, statics) | _ => (env, statics) (* end case *) end | S.S_IfThenElse(x, b1, b2) => let val statics1 = doBlock (env, statics, b1) val statics2 = doBlock (env, statics, b2) val n = VSet.numItems statics in if ((n <> VSet.numItems statics1) orelse (n <> VSet.numItems statics2)) then (env, mkStatic(env, statics, x)) else (env, statics) end | _ => (env, statics) (* end case *)) val statics = doBlock (VMap.empty, VSet.empty, block) (* lift out the static code *) fun doBlock (S.Block stms) = let fun doStmts ([], staticStms) = S.Block(List.rev staticStms) | doStmts (stm::stms, staticStms) = (case doStmt stm of SOME stm => doStmts (stms, stm::staticStms) | NONE => doStmts (stms, staticStms) (* end case *)) in doStmts (stms, []) end and doStmt stm = (case stm of S.S_Assign(x, e) => if VSet.member(statics, x) then SOME stm else NONE | S.S_IfThenElse(x, b1, b2) => if VSet.member(statics, x) then SOME(S.S_IfThenElse(x, doBlock b1, doBlock b2)) else NONE | _ => NONE (* end case *)) val staticBlock = doBlock block in print "**** static variables: "; VSet.app (fn x => print(" "^Var.uniqueNameOf x)) statics; print "\n"; staticBlock end fun transform (prog as S.Program{globals, staticInit, globalInit, actors}) = let val staticInit = liftLoads globalInit in S.Program{ globals = globals, staticInit = staticInit, globalInit = globalInit, actors = actors } end end

root@smlnj-gforge.cs.uchicago.edu | ViewVC Help |

Powered by ViewVC 1.0.0 |