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 3586, Thu Jan 14 14:49:31 2016 UTC revision 3657, Thu Feb 4 19:57:40 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
76        val _ =print (String.concat["\nVid:",Int.toString(Vid),"\nhid:",Int.toString(hid),
77                    "\nTid:",Int.toString(tid),
78                    "\n",String.concatWith","(List.map IR.Var.toString args),"\n"])
79            val imgArg = List.nth (args, Vid)            val imgArg = List.nth (args, Vid)
80            val info = getImageDst imgArg            val info = getImageDst imgArg
81            val s = getKernelDst (List.nth(args, hid))            val s = getKernelDst (List.nth(args, hid))
# Line 155  Line 158 
158      * replace probe with expanded version      * replace probe with expanded version
159      *)      *)
160       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
161         val _ =print(String.concat["\n replacing probe:",EinPP.expToString(probe)])
162            val fid = length params            val fid = length params
163            val nid = fid+1            val nid = fid+1
164            val Pid = nid+1            val Pid = nid+1
# Line 195  Line 199 
199    
200    (* floats the reconstructed field term *)    (* floats the reconstructed field term *)
201      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
202            val _ =print"\n lifting probe"
203            val fid = length(params)            val fid = length(params)
204            val nid = fid+1            val nid = fid+1
205            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.3586  
changed lines
  Added in v.3657

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