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 2922, Tue Mar 3 03:55:09 2015 UTC revision 2976, Fri Mar 6 16:37:08 2015 UTC
# Line 199  Line 199 
199                                              [assignEin(y, mk.dotimes(d1, dd@[d2]),xs)]),                                              [assignEin(y, mk.dotimes(d1, dd@[d2]),xs)]),
200                  (BV.op_Ddot,         fn (y, [_, Ty.DIM d1,  Ty.SHAPE dd, Ty.DIM d2], xs) =>                  (BV.op_Ddot,         fn (y, [_, Ty.DIM d1,  Ty.SHAPE dd, Ty.DIM d2], xs) =>
201                                          [assignEin(y, mk.divergence(d1, dd),xs)] ),                                          [assignEin(y, mk.divergence(d1, dd),xs)] ),
202                  (BV.op_norm_tt,            fn (y, [sv], xs) => let                  (BV.op_norm_t,            fn (y, [sv], xs) => let
203                                              val ty=shapeVarToTensor sv                                              val ty=shapeVarToTensor sv
204                                              in (case shapeVarToTensor sv                                              in (case shapeVarToTensor sv
205                                                  of DstTy.TensorTy[]     => assign(y, Op.Abs DstTy.realTy, xs)                                                  of DstTy.TensorTy[]     => assign(y, Op.Abs DstTy.realTy, xs)
# Line 209  Line 209 
209                                                  | _ => raise Fail"unsupported norm of TensorTy"                                                  | _ => raise Fail"unsupported norm of TensorTy"
210                                                  (* end case *))                                                  (* end case *))
211                                              end),                                              end),
212                  (BV.op_norm_ff,            fn (y, [ _,Ty.DIM d1, Ty.SHAPE dd], xs) => (case dd                  (BV.op_norm_f,            fn (y, [ _,Ty.DIM d1, Ty.SHAPE dd], xs) => (case dd
213                                              of []  =>  [assignEin(y, mk.magnitudeFF(d1, dd),xs)]                                              of []  =>  [assignEin(y, mk.magnitudeFF(d1, dd),xs)]
214                                              | [i]  =>  [assignEin(y, mk.magnitudeFF(d1, dd),xs@xs)]                                              | [_]  =>  [assignEin(y, mk.magnitudeFF(d1, dd),xs@xs)]
215                                                | [_,_]  =>  [assignEin(y, mk.magnitudeFF(d1, dd),xs@xs)]
216                                              | _ => raise Fail"unsupported norm of fieldTy"                                              | _ => raise Fail"unsupported norm of fieldTy"
217                                              (* end case *))),                                              (* end case *))),
218                  (BV.op_not,             simpleOp Op.Not),                  (BV.op_not,             simpleOp Op.Not),
# Line 230  Line 231 
231                                              [assignEin(y, (mk.innerTT(dd1,dd2)),xs)]                                              [assignEin(y, (mk.innerTT(dd1,dd2)),xs)]
232                                          end),                                          end),
233    
234                  (BV.op_inner_ff,      fn (y,  [_,_,Ty.SHAPE dd1,Ty.DIM d,Ty.SHAPE dd2,_], xs) =>                  (BV.op_inner_tf,      fn (y,  [_,Ty.DIM d,sh1,Ty.SHAPE dd2,_], xs) =>let
235                                              [assignEin(y, mk.innerFF(dd1,d,dd2),xs)]),                                              val ty1 as DstTy.TensorTy dd1 = shapeVarToTensor sh1
236                                            in
237                                                    [assignEin(y, mk.innerTF(dd1,d,dd2),xs)]
238                                            end ),
239                    (BV.op_inner_ft,      fn (y,  [_,Ty.DIM d,Ty.SHAPE dd1,sh2,_], xs) =>let
240                                                val ty1 as DstTy.TensorTy dd2 = shapeVarToTensor sh2
241                                            in
242                                                    [assignEin(y, mk.innerFT(dd1,d,dd2),xs)]
243                                            end ),
244    
245                  (BV.op_inner_ft,      fn (y,  [_,Ty.DIM d,Ty.SHAPE dd1,_,_], xs) =>                  (BV.op_inner_ff,      fn (y,  [_,_,Ty.DIM d,Ty.SHAPE dd1,Ty.SHAPE dd2,_], xs) =>
246                                                  [assignEin(y, mk.cross3FF,xs)]),                                              [assignEin(y, mk.innerFF(dd1,d,dd2),xs)]),
247    
248                  (BV.op_colon_tt,           fn (y, [sh1, sh2, _], xs) => let                  (BV.op_colon_tt,           fn (y, [sh1, sh2, _], xs) => let
249                                              val ty1 as DstTy.TensorTy dd1 = shapeVarToTensor sh1                                              val ty1 as DstTy.TensorTy dd1 = shapeVarToTensor sh1
# Line 270  Line 279 
279                  (*  modulate is vector * vector pointwise multiplication *)                  (*  modulate is vector * vector pointwise multiplication *)
280                  (BV.fn_modulate,        fn (y,[Ty.DIM dd1], xs) =>                  (BV.fn_modulate,        fn (y,[Ty.DIM dd1], xs) =>
281                                  [assignEin(y, (mk.modulate dd1),xs)]),                                  [assignEin(y, (mk.modulate dd1),xs)]),
282                  (BV.fn_normalize_tt,   vectorOp Op.Normalize),                  (BV.fn_normalize_t,   vectorOp Op.Normalize),
283                  (BV.fn_normalize_ff,             fn (y, [ _,Ty.DIM d1, Ty.SHAPE dd], xs) =>                  (BV.fn_normalize_f,             fn (y, [ _,Ty.DIM d1, Ty.SHAPE dd], xs) =>
284                      (case dd                      (case dd
285                          of [] => [assignEin(y, mk.normalizeFF(d1, []),xs)]                          of [] => [assignEin(y, mk.normalizeFF(d1, []),xs)]
286                          | [i] => [assignEin(y, mk.normalizeFF(d1, [i]),xs@xs)]                          | [i] => [assignEin(y, mk.normalizeFF(d1, [i]),xs@xs)]

Legend:
Removed from v.2922  
changed lines
  Added in v.2976

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