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

SCM Repository

[diderot] Diff of /branches/charisee/src/compiler/translate/translate-basis.sml
ViewVC logotype

Diff of /branches/charisee/src/compiler/translate/translate-basis.sml

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

revision 2377, Mon Jun 3 19:41:56 2013 UTC revision 2378, Thu Jun 6 21:02:40 2013 UTC
# Line 25  Line 25 
25      structure TU = TypeUtil      structure TU = TypeUtil
26      structure MV = MetaVar      structure MV = MetaVar
27      structure VTbl = Var.Tbl      structure VTbl = Var.Tbl
28        structure EinOp= Examples
29    
30      fun pruneTy tv = (case TU.prune(MV.toType tv)      fun pruneTy tv = (case TU.prune(MV.toType tv)
31             of (ty as Ty.T_Var _) => raise Fail("unresolved type " ^ TU.toString ty)             of (ty as Ty.T_Var _) => raise Fail("unresolved type " ^ TU.toString ty)
# Line 82  Line 83 
83                  end                  end
84            end            end
85    
86        fun assignEin(*hmm*)
87    
88        fun createScalarEin rator(y, [],xs)= assignEin (y,rator, xs)
89    
90        fun createEin rator(y, [sv], xs)= assignEin(y, rator, xs)
91    
92    
93    
94    
95    (* build a table that maps Basis variables to their translation functions *)    (* build a table that maps Basis variables to their translation functions *)
96      val tbl : ((IL.var * Ty.meta_var list * IL.var list) -> IL.assignment list) VTbl.hash_table = let      val tbl : ((IL.var * Ty.meta_var list * IL.var list) -> IL.assignment list) VTbl.hash_table = let
97            val tbl = VTbl.mkTable (128, Fail "Translate table")            val tbl = VTbl.mkTable (128, Fail "Translate table")
# Line 104  Line 114 
114                  (BV.neq_ii,             simpleOp(Op.NEQ DstTy.IntTy)),                  (BV.neq_ii,             simpleOp(Op.NEQ DstTy.IntTy)),
115                  (BV.neq_ss,             simpleOp(Op.NEQ DstTy.StringTy)),                  (BV.neq_ss,             simpleOp(Op.NEQ DstTy.StringTy)),
116                  (BV.neq_rr,             simpleOp(Op.NEQ(DstTy.realTy))),                  (BV.neq_rr,             simpleOp(Op.NEQ(DstTy.realTy))),
117                  (BV.add_ii,             simpleOp(Op.Add DstTy.IntTy)),  
118                  (BV.add_tt,             tensorOp Op.Add),                      (* Changed*)
119    
120                    (BV.add_ii,             createScalarEin(EinOp.addScalars)), (* DstTy.IntTy*)
121                    (BV.add_tt,             createEin(Ein.Op.addTensors)),
122    
123                         (* UnChanged*)
124                  (BV.add_ff,             fn (y, _, [f, g]) => assign(y, Op.AddField, [f, g])),                  (BV.add_ff,             fn (y, _, [f, g]) => assign(y, Op.AddField, [f, g])),
125                  (BV.add_fr,             fn (y, _, [f, s]) => assign(y, Op.OffsetField, [f, s])),                  (BV.add_fr,             fn (y, _, [f, s]) => assign(y, Op.OffsetField, [f, s])),
126                  (BV.add_rf,             fn (y, _, [s, f]) => assign(y, Op.OffsetField, [f, s])),                  (BV.add_rf,             fn (y, _, [s, f]) => assign(y, Op.OffsetField, [f, s])),
127                  (BV.sub_ii,             simpleOp(Op.Sub DstTy.IntTy)),                        (* Changed*)
128                  (BV.sub_tt,             tensorOp Op.Sub),                  (BV.sub_ii,             createScalarEin(EinOp.subScalars)), (*DstTy.IntTy*)
129                    (BV.sub_tt,             createEin(EinOp.subTensor)),
130    
131                            (* UnChanged*)
132                  (BV.sub_ff,             fn (y, _, [f, g]) => assign(y, Op.SubField, [f, g])),                  (BV.sub_ff,             fn (y, _, [f, g]) => assign(y, Op.SubField, [f, g])),
133                  (BV.sub_fr,             fn (y, _, [f, s]) => let                  (BV.sub_fr,             fn (y, _, [f, s]) => let
134                                            val s' = IL.Var.copy s                                            val s' = IL.Var.copy s
# Line 118  Line 136 
136                                              IL.ASSGN(s', IL.OP(Op.Neg DstTy.realTy, [s])),                                              IL.ASSGN(s', IL.OP(Op.Neg DstTy.realTy, [s])),
137                                              IL.ASSGN(y, IL.OP(Op.OffsetField, [f, s']))                                              IL.ASSGN(y, IL.OP(Op.OffsetField, [f, s']))
138                                            ] end),                                            ] end),
139    
140                  (BV.sub_rf,             fn (y, _, [s, f]) => let                  (BV.sub_rf,             fn (y, _, [s, f]) => let
141                                            val f' = IL.Var.copy f                                            val f' = IL.Var.copy f
142                                            in [                                            in [
143                                              IL.ASSGN(f', IL.OP(Op.NegField, [f])),                                              IL.ASSGN(f', IL.OP(Op.NegField, [f])),
144                                              IL.ASSGN(y, IL.OP(Op.OffsetField, [f', s]))                                              IL.ASSGN(y, IL.OP(Op.OffsetField, [f', s]))
145                                            ] end),                                            ] end),
146                  (BV.mul_ii,             simpleOp(Op.Mul DstTy.IntTy)),                           (* Changed*)
147                  (BV.mul_rr,             simpleOp(Op.Mul(DstTy.realTy))),                  (BV.mul_ii,             createScalarEin(EinOp.scalarxscalar)), (* DstTy.IntTy)),*)
148                  (BV.mul_rt,             tensorOp Op.Scale),                  (BV.mul_rr,             createScalarEin(EinOp.scalarxscalar)), (*(DstTy.realTy))),*)
149                  (BV.mul_tr,             fn (y, sv, [t, r]) => tensorOp Op.Scale (y, sv, [r, t])),                  (BV.mul_rt,             createEin(EinOp.scaleTensor)),    (*tensorOp Op.Scale),*)
150                    (BV.mul_tr,             fn (y, sv, [t, r]) => createEin EinOp.scaleTensor (y, sv, [r, t])),
151    
152    
153    
154    
155                             (* UnChanged*)
156                  (BV.mul_rf,             fn (y, _, [s, f]) => assign(y, Op.ScaleField, [s, f])),                  (BV.mul_rf,             fn (y, _, [s, f]) => assign(y, Op.ScaleField, [s, f])),
157                  (BV.mul_fr,             fn (y, _, [f, s]) => assign(y, Op.ScaleField, [s, f])),                  (BV.mul_fr,             fn (y, _, [f, s]) => assign(y, Op.ScaleField, [s, f])),
158    
159    
160    
161                  (BV.div_ii,             simpleOp(Op.Div DstTy.IntTy)),                  (BV.div_ii,             simpleOp(Op.Div DstTy.IntTy)),
162                  (BV.div_rr,             simpleOp(Op.Div DstTy.realTy)),                  (BV.div_rr,             simpleOp(Op.Div DstTy.realTy)),
163                  (BV.div_tr,             fn (y, [sv], [x, s]) => let                  (BV.div_tr,             fn (y, [sv], [x, s]) => let
# Line 150  Line 178 
178                                            ] end),                                            ] end),
179                  (BV.exp_ri,             simpleOp(Op.Power)),                  (BV.exp_ri,             simpleOp(Op.Power)),
180                  (BV.exp_rr,             basisFn MathFuns.pow),                  (BV.exp_rr,             basisFn MathFuns.pow),
181    
182    
183    
184                  (BV.curl2D,             fn (y, _, xs) => assign(y, Op.CurlField 2, xs)),                  (BV.curl2D,             fn (y, _, xs) => assign(y, Op.CurlField 2, xs)),
185                  (BV.curl3D,             fn (y, _, xs) => assign(y, Op.CurlField 3, xs)),                  (BV.curl3D,             fn (y, _, xs) => assign(y, Op.CurlField 3, xs)),
186    
187    
188    
189                  (BV.convolve_vk,        fn (y, [_, dv, _], xs) =>                  (BV.convolve_vk,        fn (y, [_, dv, _], xs) =>
190                                            assign(y, Op.Field(pruneDim(MV.toDim dv)), xs)),                                            assign(y, Op.Field(pruneDim(MV.toDim dv)), xs)),
191                  (BV.convolve_kv,        fn (y, [_, dv, _], [k, v]) =>                  (BV.convolve_kv,        fn (y, [_, dv, _], [k, v]) =>

Legend:
Removed from v.2377  
changed lines
  Added in v.2378

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