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

SCM Repository

[diderot] Annotation of /branches/pure-cfg/src/compiler/cl-target/tree-to-cl.sml
ViewVC logotype

Annotation of /branches/pure-cfg/src/compiler/cl-target/tree-to-cl.sml

Parent Directory Parent Directory | Revision Log Revision Log


Revision 3349 - (view) (download)

1 : jhr 1632 (* tree-to-cl.sml
2 : jhr 918 *
3 : jhr 3349 * This code is part of the Diderot Project (http://diderot-language.cs.uchicago.edu)
4 :     *
5 :     * COPYRIGHT (c) 2015 The University of Chicago
6 : jhr 918 * All rights reserved.
7 :     *
8 : jhr 1632 * Translate TreeIL to the OpenCL version of CLang.
9 : jhr 918 *)
10 :    
11 :     structure TreeToCL : sig
12 :    
13 :     datatype var = V of (CLang.ty * CLang.var)
14 :    
15 :     type env = var TreeIL.Var.Map.map
16 :    
17 :     val trType : TreeIL.Ty.ty -> CLang.ty
18 :    
19 : jhr 1505 val trBlock : env * TreeIL.block -> CLang.stm
20 : jhr 918
21 : lamonts 1244 val trFragment : env * TreeIL.block -> env * CLang.stm list
22 :    
23 : jhr 918 val trAssign : env * TreeIL.var * TreeIL.exp -> CLang.stm list
24 :    
25 :     val trExp : env * TreeIL.exp -> CLang.exp
26 :    
27 : jhr 1273 (* vector indexing support. Arguments are: vector, index *)
28 :     val vecIndex : CLang.exp * int -> CLang.exp
29 : jhr 918
30 :     end = struct
31 :    
32 :     structure CL = CLang
33 :     structure RN = RuntimeNames
34 :     structure IL = TreeIL
35 :     structure Op = IL.Op
36 :     structure Ty = IL.Ty
37 :     structure V = IL.Var
38 :    
39 : jhr 1285 datatype var = datatype CLang.typed_var
40 : jhr 918
41 :     type env = var TreeIL.Var.Map.map
42 :    
43 :     fun lookup (env, x) = (case V.Map.find (env, x)
44 : lamonts 1399 of SOME(V(_, x')) => x'
45 :     | NONE => raise Fail(concat["lookup(_, ", V.name x, ")"])
46 :     (* end case *))
47 : jhr 918
48 :     (* integer literal expression *)
49 : jhr 1314 fun intExp (i : int) = CL.mkInt(IntInf.fromInt i)
50 : jhr 918
51 : jhr 1401 (* the type of an image-data pointer. *)
52 :     fun imageDataPtrTy rTy = CL.T_Qual("__global", CL.T_Ptr(CL.T_Num rTy))
53 :    
54 : jhr 918 (* translate TreeIL types to CLang types *)
55 :     fun trType ty = (case ty
56 : lamonts 1617 of Ty.BoolTy => CLang.T_Named "uint"
57 : jhr 1401 | Ty.StringTy => CL.charPtr
58 : jhr 1522 | Ty.IntTy => !RN.gIntTy
59 : jhr 1401 | Ty.TensorTy[] => !RN.gRealTy
60 :     | Ty.TensorTy[n] => CL.T_Named(RN.vecTy n)
61 :     | Ty.TensorTy[n, m] => CL.T_Named(RN.matTy(n,m))
62 : jhr 1522 | Ty.SeqTy(Ty.IntTy, n) => CL.T_Named(RN.ivecTy n)
63 :     | Ty.SeqTy(Ty.TensorTy[] , n) => CL.T_Named(RN.vecTy n)
64 : lamonts 1664 | Ty.SeqTy(ty, n) => CL.T_Array(trType ty, SOME n)
65 : jhr 1401 | Ty.AddrTy(ImageInfo.ImgInfo{ty=(_, rTy), ...}) => imageDataPtrTy rTy
66 :     | Ty.ImageTy(ImageInfo.ImgInfo{dim, ...}) => CL.T_Named(RN.imageTy dim)
67 :     | _ => raise Fail(concat["TreeToC.trType(", Ty.toString ty, ")"])
68 :     (* end case *))
69 : jhr 918
70 :     (* generate new variables *)
71 :     local
72 :     val count = ref 0
73 :     fun freshName prefix = let
74 : jhr 1401 val n = !count
75 :     in
76 :     count := n+1;
77 :     concat[prefix, "_", Int.toString n]
78 :     end
79 : jhr 918 in
80 :     fun tmpVar ty = freshName "tmp"
81 :     fun freshVar prefix = freshName prefix
82 :     end (* local *)
83 :    
84 :     (* translate IL basis functions *)
85 : jhr 1422 fun trApply (f, args) = CL.mkApply(ILBasis.toString f, args)
86 : jhr 918
87 : jhr 1273 (* vector indexing support. Arguments are: vector, index *)
88 :     local
89 :     val fields = Vector.fromList [
90 :     "s0", "s1", "s2", "s3",
91 :     "s4", "s5", "s6", "s7",
92 :     "s8", "s9", "sa", "sb",
93 :     "sc", "sd", "se", "sf"
94 :     ]
95 :     in
96 :     fun vecIndex (v, ix) = CL.mkSelect(v, Vector.sub(fields, ix))
97 :     end
98 : jhr 918
99 :     (* translate a variable use *)
100 :     fun trVar (env, x) = (case V.kind x
101 : jhr 1401 of IL.VK_Global => CL.mkIndirect(CL.E_Var RN.globalsVarName, lookup(env, x))
102 :     | IL.VK_Local => CL.mkVar(lookup(env, x))
103 :     (* end case *))
104 : jhr 918
105 : lamonts 1662 (* matrix indexing *)
106 : jhr 1669 fun matIndex (m, ix, jx) = CL.mkSelect(CL.mkSubscript(m, ix), concat["s",jx])
107 : lamonts 1662
108 : jhr 1504 (* translate a state-variable use *)
109 :     fun trStateVar (IL.SV{name, ...}) = CL.mkIndirect(CL.mkVar "selfIn", name)
110 :    
111 : jhr 1419 fun castArgs ty = List.map (fn e => CL.mkCast(ty, e))
112 :    
113 : jhr 918 (* Translate a TreeIL operator application to a CLang expression *)
114 :     fun trOp (rator, args) = (case (rator, args)
115 : jhr 1401 of (Op.Add ty, [a, b]) => CL.mkBinOp(a, CL.#+, b)
116 :     | (Op.Sub ty, [a, b]) => CL.mkBinOp(a, CL.#-, b)
117 :     | (Op.Mul ty, [a, b]) => CL.mkBinOp(a, CL.#*, b)
118 :     | (Op.Div ty, [a, b]) => CL.mkBinOp(a, CL.#/, b)
119 :     | (Op.Neg ty, [a]) => CL.mkUnOp(CL.%-, a)
120 : jhr 1522 | (Op.Abs(Ty.IntTy), args) => CL.mkApply("abs", args)
121 : jhr 1401 | (Op.Abs(Ty.TensorTy[]), args) => CL.mkApply(RN.fabs, args)
122 :     | (Op.Abs(Ty.TensorTy[_]), args) => CL.mkApply(RN.fabs, args)
123 :     | (Op.Abs ty, [a]) => raise Fail(concat["Abs<", Ty.toString ty, ">"])
124 :     | (Op.LT ty, [a, b]) => CL.mkBinOp(a, CL.#<, b)
125 :     | (Op.LTE ty, [a, b]) => CL.mkBinOp(a, CL.#<=, b)
126 :     | (Op.EQ ty, [a, b]) => CL.mkBinOp(a, CL.#==, b)
127 :     | (Op.NEQ ty, [a, b]) => CL.mkBinOp(a, CL.#!=, b)
128 :     | (Op.GTE ty, [a, b]) => CL.mkBinOp(a, CL.#>=, b)
129 :     | (Op.GT ty, [a, b]) => CL.mkBinOp(a, CL.#>, b)
130 :     | (Op.Not, [a]) => CL.mkUnOp(CL.%!, a)
131 : jhr 1419 | (Op.Max, args) => CL.mkApply(RN.max, castArgs (!RN.gRealTy) args)
132 :     | (Op.Min, args) => CL.mkApply(RN.min, castArgs (!RN.gRealTy) args)
133 : jhr 1401 | (Op.Clamp ty, [lo, hi, x]) => CL.mkApply(RN.clamp, [x, lo, hi])
134 :     | (Op.Lerp ty, args) => (case ty
135 : lamonts 1423 of Ty.TensorTy[] => CL.mkApply(RN.lerp, castArgs (!RN.gRealTy) args)
136 :     | Ty.TensorTy[n] => CL.mkApply(RN.lerp, castArgs (CL.T_Named(RN.vecTy n)) args)
137 : jhr 1401 | _ => raise Fail(concat[
138 :     "lerp<", Ty.toString ty, "> not supported"
139 :     ])
140 :     (* end case *))
141 :     | (Op.Dot d, args) => CL.E_Apply(RN.dot, args)
142 :     | (Op.MulVecMat(m, n), args) =>
143 :     if (1 < m) andalso (m < 4) andalso (m = n)
144 :     then CL.E_Apply(RN.mulVecMat(m,n), args)
145 :     else raise Fail "unsupported vector-matrix multiply"
146 :     | (Op.MulMatVec(m, n), args) =>
147 :     if (1 < m) andalso (m < 4) andalso (m = n)
148 :     then CL.E_Apply(RN.mulMatVec(m,n), args)
149 :     else raise Fail "unsupported matrix-vector multiply"
150 :     | (Op.MulMatMat(m, n, p), args) =>
151 :     if (1 < m) andalso (m < 4) andalso (m = n) andalso (n = p)
152 :     then CL.E_Apply(RN.mulMatMat(m,n,p), args)
153 :     else raise Fail "unsupported matrix-matrix multiply"
154 :     | (Op.Cross, args) => CL.E_Apply(RN.cross, args)
155 :     | (Op.Norm(Ty.TensorTy[n]), args) => CL.E_Apply(RN.length, args)
156 :     | (Op.Norm(Ty.TensorTy[m,n]), args) => CL.E_Apply(RN.norm(m,n), args)
157 :     | (Op.Normalize d, args) => CL.E_Apply(RN.normalize, args)
158 :     | (Op.Scale(Ty.TensorTy[n]), [s, v]) => CL.mkBinOp(s, CL.#*, v)
159 :     | (Op.PrincipleEvec ty, _) => raise Fail "PrincipleEvec unimplemented"
160 : jhr 1525 | (Op.Select(Ty.TupleTy tys, i), [a]) => raise Fail "Select unimplemented"
161 :     | (Op.Index(Ty.SeqTy(Ty.IntTy, n), i), [a]) => vecIndex (a, i)
162 :     | (Op.Index(Ty.TensorTy[n], i), [a]) => vecIndex (a, i)
163 : jhr 1522 | (Op.Subscript(Ty.SeqTy(Ty.IntTy, n)), [v, CL.E_Int(ix, _)]) => vecIndex (v, Int.fromLarge ix)
164 :     | (Op.Subscript(Ty.SeqTy(Ty.IntTy, n)), [v, ix]) => let
165 : jhr 1401 val unionTy = CL.T_Named(concat["union", Int.toString n, !RN.gIntSuffix, "_t"])
166 :     val vecExp = CL.mkSelect(CL.mkCast(unionTy, v), "i")
167 :     in
168 :     CL.mkSubscript(vecExp, ix)
169 :     end
170 : jhr 1669 | (Op.Subscript(Ty.SeqTy(ty, n)), [v, ix]) => CL.mkSubscript(v, ix)
171 : jhr 1401 | (Op.Subscript(Ty.TensorTy[n]), [v, CL.E_Int(ix, _)]) => vecIndex (v, Int.fromLarge ix)
172 :     | (Op.Subscript(Ty.TensorTy[n]), [v, ix]) => let
173 :     val unionTy = CL.T_Named(concat["union", Int.toString n, !RN.gRealSuffix, "_t"])
174 :     val vecExp = CL.mkSelect(CL.mkCast(unionTy, v), "r")
175 :     in
176 :     CL.mkSubscript(vecExp, ix)
177 :     end
178 :     | (Op.Subscript(Ty.TensorTy[_,_]), [m, ix, CL.E_Int(jx, _)]) =>
179 :     vecIndex(CL.mkSubscript(m, ix), Int.fromLarge jx)
180 :     | (Op.Subscript(Ty.TensorTy[_,n]), [m, ix, jx]) => let
181 :     val unionTy = CL.T_Named(concat["union", Int.toString n, !RN.gRealSuffix, "_t"])
182 :     val vecExp = CL.mkSelect(CL.mkCast(unionTy, CL.mkSubscript(m, ix)), "r")
183 :     in
184 :     CL.mkSubscript(vecExp, jx)
185 :     end
186 :     | (Op.Subscript ty, t::(ixs as _::_)) =>
187 :     raise Fail(concat["Subscript<", Ty.toString ty, "> unsupported"])
188 :     | (Op.Ceiling d, args) => CL.mkApply("ceil", args)
189 :     | (Op.Floor d, args) => CL.mkApply("floor", args)
190 :     | (Op.Round d, args) => CL.mkApply("round", args)
191 :     | (Op.Trunc d, args) => CL.mkApply("trunc", args)
192 :     | (Op.IntToReal, [a]) => CL.mkCast(!RN.gRealTy, a)
193 :     | (Op.RealToInt 1, [a]) => CL.mkCast(!RN.gIntTy, a)
194 : jhr 1504 | (Op.RealToInt d, args) => CL.mkApply(RN.vecftoi d, args)
195 : jhr 918 (* FIXME: need type info *)
196 : jhr 1401 | (Op.ImageAddress(ImageInfo.ImgInfo{ty=(_,rTy), ...}), [a as CL.E_Indirect(_,field)]) => let
197 :     val cTy = imageDataPtrTy rTy
198 :     in
199 :     CL.mkCast(cTy,
200 :     CL.mkSelect(CL.mkVar RN.globalImageDataName, RN.imageDataName field))
201 :     end
202 :     | (Op.LoadVoxels(info, 1), [a]) => let
203 :     val realTy as CL.T_Num rTy = !RN.gRealTy
204 :     val a = CL.E_UnOp(CL.%*, a)
205 :     in
206 :     if (rTy = ImageInfo.sampleTy info)
207 :     then a
208 :     else CL.E_Cast(realTy, a)
209 :     end
210 :     | (Op.LoadVoxels _, [a]) =>
211 :     raise Fail("impossible " ^ Op.toString rator)
212 :     | (Op.PosToImgSpace(ImageInfo.ImgInfo{dim, ...}), [img, pos]) =>
213 :     CL.mkApply(RN.toImageSpace dim, [CL.mkUnOp(CL.%&,img), pos])
214 :     | (Op.TensorToWorldSpace(info, ty), [v, x]) =>
215 : lamonts 1423 CL.mkApply(RN.toWorldSpace ty, [CL.mkUnOp(CL.%&,v), x])
216 : jhr 1401 | (Op.LoadImage info, [a]) =>
217 :     raise Fail("impossible " ^ Op.toString rator)
218 :     | (Op.Inside(ImageInfo.ImgInfo{dim, ...}, s), [pos, img]) =>
219 :     CL.mkApply(RN.inside dim, [pos, CL.mkUnOp(CL.%&,img), intExp s])
220 :     | (Op.Input(ty, name, desc), []) =>
221 :     raise Fail("impossible " ^ Op.toString rator)
222 :     | (Op.InputWithDefault(ty, name, desc), [a]) =>
223 :     raise Fail("impossible " ^ Op.toString rator)
224 :     | _ => raise Fail(concat[
225 :     "unknown or incorrect operator ", Op.toString rator
226 :     ])
227 :     (* end case *))
228 : jhr 918
229 :     fun trExp (env, e) = (case e
230 : jhr 1504 of IL.E_State x => trStateVar x
231 :     | IL.E_Var x => trVar (env, x)
232 : jhr 1401 | IL.E_Lit(Literal.Int n) => CL.mkIntTy(n, !RN.gIntTy)
233 :     | IL.E_Lit(Literal.Bool b) => CL.mkBool b
234 :     | IL.E_Lit(Literal.Float f) => CL.mkFlt(f, !RN.gRealTy)
235 :     | IL.E_Lit(Literal.String s) => CL.mkStr s
236 :     | IL.E_Op(rator, args) => trOp (rator, trExps(env, args))
237 :     | IL.E_Apply(f, args) => trApply(f, trExps(env, args))
238 :     | IL.E_Cons(Ty.TensorTy[n], args) => CL.mkApply(RN.mkVec n, trExps(env, args))
239 :     | IL.E_Cons(ty, _) => raise Fail(concat["E_Cons(", Ty.toString ty, ", _) in expression"])
240 :     (* end case *))
241 : jhr 918
242 :     and trExps (env, exps) = List.map (fn exp => trExp(env, exp)) exps
243 : lamonts 1462
244 : jhr 1669 (* translate an expression to a variable form; return the variable and the
245 : lamonts 1662 * (optional) declaration.
246 :     *)
247 :     fun expToVar (env, ty, name, exp) = (case trExp(env, exp)
248 :     of x as CL.E_Var _ => (x, [])
249 :     | exp => let
250 :     val x = freshVar name
251 :     in
252 :     (CL.mkVar x, [CL.mkDecl(ty, x, SOME(CL.I_Exp exp))])
253 :     end
254 :     (* end case *))
255 :    
256 : jhr 1504 fun trLHSVar (env, lhs) = (case V.kind lhs
257 :     of IL.VK_Global => CL.mkIndirect(CL.mkVar RN.globalsVarName, lookup(env, lhs))
258 :     | IL.VK_Local => CL.mkVar(lookup(env, lhs))
259 :     (* end case *))
260 : jhr 918
261 : jhr 1504 fun trLHSStateVar (IL.SV{name, ...}) = CL.mkIndirect(CL.mkVar "selfOut", name)
262 :    
263 :     fun trSet (env, lhs, rhs) = (
264 :     (* certain rhs forms, such as those that return a matrix,
265 :     * require a function call instead of an assignment
266 :     *)
267 :     case rhs
268 :     of IL.E_Op(Op.Add(Ty.TensorTy[m,n]), args) =>
269 :     [CL.mkCall(RN.addMat(m,n), lhs :: trExps(env, args))]
270 :     | IL.E_Op(Op.Sub(Ty.TensorTy[m,n]), args) =>
271 :     [CL.mkCall(RN.subMat(m,n), lhs :: trExps(env, args))]
272 :     | IL.E_Op(Op.Neg(Ty.TensorTy[m,n]), args) =>
273 :     [CL.mkCall(RN.scaleMat(m,n), lhs :: intExp ~1 :: trExps(env, args))]
274 :     | IL.E_Op(Op.Scale(Ty.TensorTy[m,n]), args) =>
275 :     [CL.mkCall(RN.scaleMat(m,n), lhs :: trExps(env, args))]
276 :     | IL.E_Op(Op.MulMatMat(m,n,p), args) =>
277 :     [CL.mkCall(RN.mulMatMat(m,n,p), lhs :: trExps(env, args))]
278 :     | IL.E_Op(Op.Identity n, args) =>
279 :     [CL.mkCall(RN.identityMat n, [lhs])]
280 :     | IL.E_Op(Op.Zero(Ty.TensorTy[m,n]), args) =>
281 :     [CL.mkCall(RN.zeroMat(m,n), [lhs])]
282 :     | IL.E_Op(Op.TensorToWorldSpace(info, ty as Ty.TensorTy[_,_]), [img,src]) =>
283 :     [CL.mkCall(RN.toWorldSpace ty, lhs :: [CL.mkUnOp(CL.%&,trExp(env, img)),trExp(env, src)] )]
284 :     | IL.E_Op(Op.LoadVoxels(info, n), [a]) =>
285 :     if (n > 1)
286 :     then let
287 :     val stride = ImageInfo.stride info
288 :     val rTy = ImageInfo.sampleTy info
289 :     val vp = freshVar "vp"
290 :     val needsCast = (CL.T_Num rTy <> !RN.gRealTy)
291 :     fun mkLoad i = let
292 :     val e = CL.mkSubscript(CL.mkVar vp, intExp(i*stride))
293 :     in
294 :     if needsCast then CL.mkCast(!RN.gRealTy, e) else e
295 :     end
296 :     in [
297 :     CL.mkDecl(imageDataPtrTy rTy, vp, SOME(CL.I_Exp(trExp(env, a)))),
298 :     CL.mkAssign(lhs,
299 :     CL.mkApply(RN.mkVec n, List.tabulate (n, mkLoad)))
300 :     ] end
301 :     else [CL.mkAssign(lhs, trExp(env, rhs))]
302 : lamonts 1662 | IL.E_Op(Op.EigenVals2x2, [m]) => let
303 :     val (m, stms) = expToVar (env, CL.T_Named(RN.matTy(2,2)), "m", m)
304 :     in
305 :     stms @ [CL.mkCall(RN.evals2x2, [
306 : lamonts 1664 CL.mkUnOp(CL.%&,lhs),
307 :     matIndex (m, CL.mkInt 0, "0"),
308 :     matIndex (m, CL.mkInt 0, "1"),
309 :     matIndex (m, CL.mkInt 1, "1")
310 : lamonts 1662 ])]
311 :     end
312 :     | IL.E_Op(Op.EigenVals3x3, [m]) => let
313 :     val (m, stms) = expToVar (env, CL.T_Named(RN.matTy(3,3)), "m", m)
314 :     in
315 :     stms @ [CL.mkCall(RN.evals3x3, [
316 : lamonts 1664 CL.mkUnOp(CL.%&,lhs),
317 :     matIndex (m, CL.mkInt 0, "0"),
318 :     matIndex (m, CL.mkInt 0, "1"),
319 :     matIndex (m, CL.mkInt 0, "2"),
320 :     matIndex (m, CL.mkInt 1, "1"),
321 :     matIndex (m, CL.mkInt 1, "2"),
322 :     matIndex (m, CL.mkInt 2, "2")
323 : lamonts 1662 ])]
324 :     end
325 :    
326 : jhr 1504 | IL.E_Cons(Ty.TensorTy[n,m], args) => let
327 :     (* matrices are represented as arrays of union<d><ty>_t vectors *)
328 :     fun doRows (_, []) = []
329 :     | doRows (i, e::es) =
330 :     CL.mkAssign(CL.mkSubscript(lhs, intExp i), e)
331 :     :: doRows (i+1, es)
332 :     in
333 :     doRows (0, trExps(env, args))
334 :     end
335 :     | IL.E_Var x => (case IL.Var.ty x
336 :     of Ty.TensorTy[n,m] => [CL.mkCall(RN.copyMat(n,m), [lhs, trVar(env, x)])]
337 :     | _ => [CL.mkAssign(lhs, trVar(env, x))]
338 :     (* end case *))
339 :     | _ => [CL.mkAssign(lhs, trExp(env, rhs))]
340 :     (* end case *))
341 :    
342 :     fun trAssign (env, lhs, rhs) = trSet (env, trLHSVar (env, lhs), rhs)
343 :    
344 : lamonts 1664 fun trMultiAssign (env, lhs, IL.E_Op(rator, args)) = (case (lhs, rator, args)
345 :     of ([vals, vecs], Op.EigenVecs2x2, [m]) => let
346 :     val (m, stms) = expToVar (env, CL.T_Named(RN.matTy(2,2)), "m", m)
347 :     in
348 :     stms @ [CL.mkCall(RN.evecs2x2, [
349 :     CL.mkUnOp(CL.%&,vals), vecs,
350 :     matIndex (m, CL.mkInt 0, "0"),
351 :     matIndex (m, CL.mkInt 0, "1"),
352 :     matIndex (m, CL.mkInt 1, "1")
353 :     ])]
354 :     end
355 :     | ([vals, vecs], Op.EigenVecs3x3, [m]) => let
356 :     val (m, stms) = expToVar (env, CL.T_Named(RN.matTy(3,3)), "m", m)
357 :     in
358 :     stms @ [CL.mkCall(RN.evecs3x3, [
359 :     CL.mkUnOp(CL.%&,vals), vecs,
360 :     matIndex (m, CL.mkInt 0, "0"),
361 :     matIndex (m, CL.mkInt 0, "1"),
362 :     matIndex (m, CL.mkInt 0, "2"),
363 :     matIndex (m, CL.mkInt 1, "1"),
364 :     matIndex (m, CL.mkInt 1, "2"),
365 :     matIndex (m, CL.mkInt 2, "2")
366 :     ])]
367 :     end
368 :     | _ => raise Fail "bogus multi-assignment"
369 :     (* end case *))
370 :     | trMultiAssign (env, lhs, rhs) = raise Fail "bogus multi-assignment"
371 :    
372 : lamonts 1244 fun trLocals (env : env, locals) =
373 : jhr 1400 List.foldl
374 :     (fn (x, env) => V.Map.insert(env, x, V(trType(V.ty x), V.name x)))
375 :     env locals
376 : lamonts 1244
377 :     (* generate code to check the status of runtime-system calls *)
378 :     fun checkSts mkDecl = let
379 : jhr 1400 val sts = freshVar "sts"
380 :     in
381 :     mkDecl sts @
382 :     [CL.mkIfThen(
383 :     CL.mkBinOp(CL.mkVar "DIDEROT_OK", CL.#!=, CL.mkVar sts),
384 :     CL.mkCall("exit", [intExp 1]))]
385 :     end
386 : lamonts 1244
387 : jhr 1505 fun trStms (env, stms) = let
388 : jhr 1400 fun trStmt (env, stm) = (case stm
389 :     of IL.S_Comment text => [CL.mkComment text]
390 : jhr 1553 | IL.S_Assign([x], exp) => trAssign (env, x, exp)
391 : lamonts 1664 | IL.S_Assign(xs, exp) =>
392 :     trMultiAssign (env, List.map (fn x => trVar (env, x)) xs, exp)
393 : jhr 1400 | IL.S_IfThen(cond, thenBlk) =>
394 : jhr 1505 [CL.mkIfThen(trExp(env, cond), trBlk(env, thenBlk))]
395 : jhr 1400 | IL.S_IfThenElse(cond, thenBlk, elseBlk) =>
396 :     [CL.mkIfThenElse(trExp(env, cond),
397 : jhr 1505 trBlk(env, thenBlk),
398 :     trBlk(env, elseBlk))]
399 : jhr 1400 | IL.S_New _ => raise Fail "new not supported yet" (* FIXME *)
400 : jhr 1553 | IL.S_Save([x], exp) => trSet (env, trLHSStateVar x, exp)
401 : jhr 1250 (* FIXME: I think that S_LoadImage should never happen in OpenCL code [jhr] *)
402 : jhr 1400 | IL.S_LoadImage(lhs, dim, name) => checkSts (fn sts => let
403 :     val lhs = lookup(env, lhs)
404 :     val name = trExp(env, name)
405 :     val imgTy = CL.T_Named(RN.imageTy dim)
406 :     val loadFn = RN.loadImage dim
407 :     in [
408 :     CL.mkDecl(
409 :     CL.T_Named RN.statusTy, sts,
410 :     SOME(CL.I_Exp(CL.E_Apply(loadFn, [name, CL.mkUnOp(CL.%&, CL.E_Var lhs)]))))
411 :     ] end)
412 : jhr 1250 (* FIXME: I think that S_Input should never happen in OpenCL code [jhr] *)
413 : lamonts 1662 | IL.S_Input(lhs, name, desc, optDflt) => checkSts (fn sts => let
414 : jhr 1400 val inputFn = RN.input(V.ty lhs)
415 :     val lhs = lookup(env, lhs)
416 :     val (initCode, hasDflt) = (case optDflt
417 : lamonts 1488 of SOME e => ([CL.mkAssign(CL.E_Var lhs, trExp(env, e))], true)
418 : jhr 1400 | NONE => ([], false)
419 :     (* end case *))
420 :     val code = [
421 :     CL.mkDecl(
422 :     CL.T_Named RN.statusTy, sts,
423 :     SOME(CL.I_Exp(CL.E_Apply(inputFn, [
424 : jhr 1669 CL.mkStr name,
425 :     CL.mkUnOp(CL.%&, CL.mkIndirect(CL.mkVar RN.globalsVarName, lhs)),
426 :     CL.mkBool hasDflt
427 :     ]))))
428 : jhr 1400 ]
429 :     in
430 :     initCode @ code
431 :     end)
432 : jhr 1505 | IL.S_Exit args => [CL.mkReturn NONE]
433 : jhr 1504 | IL.S_Active => [CL.mkReturn(SOME(CL.mkVar RN.kActive))]
434 :     | IL.S_Stabilize => [CL.mkReturn(SOME(CL.mkVar RN.kStabilize))]
435 : jhr 1400 | IL.S_Die => [CL.mkReturn(SOME(CL.mkVar RN.kDie))]
436 :     (* end case *))
437 :     in
438 :     List.foldr (fn (stm, stms) => trStmt(env, stm)@stms) [] stms
439 :     end
440 : jhr 918
441 : jhr 1505 and trBlk (env, IL.Block{locals, body}) = let
442 : jhr 1400 val env = trLocals (env, locals)
443 : jhr 1505 val stms = trStms (env, body)
444 : jhr 1400 fun mkDecl (x, stms) = (case V.Map.find (env, x)
445 :     of SOME(V(ty, x')) => CL.mkDecl(ty, x', NONE) :: stms
446 :     | NONE => raise Fail(concat["mkDecl(", V.name x, ", _)"])
447 :     (* end case *))
448 :     val stms = List.foldr mkDecl stms locals
449 :     in
450 :     CL.mkBlock stms
451 :     end
452 : lamonts 1244
453 :     fun trFragment (env, IL.Block{locals, body}) = let
454 : jhr 1400 val env = trLocals (env, locals)
455 : jhr 1505 val stms = trStms (env, body)
456 : jhr 1400 fun mkDecl (x, stms) = (case V.Map.find (env, x)
457 :     of SOME(V(ty, x')) => CL.mkDecl(ty, x', NONE) :: stms
458 :     | NONE => raise Fail(concat["mkDecl(", V.name x, ", _)"])
459 :     (* end case *))
460 :     val stms = List.foldr mkDecl stms locals
461 :     in
462 :     (env, stms)
463 :     end
464 : lamonts 1244
465 :     val trBlock = trBlk
466 :    
467 : jhr 918 end

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