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

SCM Repository

[diderot] Diff of /branches/vis15/src/compiler/high-to-mid/probe-ein.sml
ViewVC logotype

Diff of /branches/vis15/src/compiler/high-to-mid/probe-ein.sml

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

revision 3657, Thu Feb 4 19:57:40 2016 UTC revision 3658, Thu Feb 4 19:59:56 2016 UTC
# Line 73  Line 73 
73          *P is the mid-il var for the (transformation matrix)transpose          *P is the mid-il var for the (transformation matrix)transpose
74      *)      *)
75      fun handleArgs (Vid, hid, tid, args) = let      fun handleArgs (Vid, hid, tid, args) = let
     val _ =print (String.concat["\nVid:",Int.toString(Vid),"\nhid:",Int.toString(hid),  
                 "\nTid:",Int.toString(tid),  
                 "\n",String.concatWith","(List.map IR.Var.toString args),"\n"])  
76            val imgArg = List.nth (args, Vid)            val imgArg = List.nth (args, Vid)
77            val info = getImageDst imgArg            val info = getImageDst imgArg
78            val s = getKernelDst (List.nth(args, hid))            val s = getKernelDst (List.nth(args, hid))
# Line 158  Line 155 
155      * replace probe with expanded version      * replace probe with expanded version
156      *)      *)
157       fun replaceProbe ((y, IR.EINAPP(Ein.EIN{params, index, body}, args)), probe, sx) = let       fun replaceProbe ((y, IR.EINAPP(Ein.EIN{params, index, body}, args)), probe, sx) = let
      val _ =print(String.concat["\n replacing probe:",EinPP.expToString(probe)])  
158            val fid = length params            val fid = length params
159            val nid = fid+1            val nid = fid+1
160            val Pid = nid+1            val Pid = nid+1
# Line 199  Line 195 
195    
196    (* floats the reconstructed field term *)    (* floats the reconstructed field term *)
197      fun liftProbe ((y, IR.EINAPP(Ein.EIN{params, index, body}, args)), probe, sx) = let      fun liftProbe ((y, IR.EINAPP(Ein.EIN{params, index, body}, args)), probe, sx) = let
         val _ =print"\n lifting probe"  
198            val fid = length(params)            val fid = length(params)
199            val nid = fid+1            val nid = fid+1
200            val E.Probe(E.Conv(Vid, alpha, hid, dx), E.Tensor(tid, _)) = probe            val E.Probe(E.Conv(Vid, alpha, hid, dx), E.Tensor(tid, _)) = probe

Legend:
Removed from v.3657  
changed lines
  Added in v.3658

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