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

SCM Repository

[diderot] Diff of /branches/charisee_dev/src/compiler/high-to-mid/handleEin.sml
ViewVC logotype

Diff of /branches/charisee_dev/src/compiler/high-to-mid/handleEin.sml

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

branches/charisee/src/compiler/high-to-mid/handleEin.sml revision 3138, Thu Mar 26 16:27:35 2015 UTC branches/charisee_dev/src/compiler/high-to-mid/handleEin.sml revision 3503, Thu Dec 17 23:13:57 2015 UTC
# Line 1  Line 1 
1  (* Expands probe ein  (* Expands probe ein
2   *   *
3   * COPYRIGHT (c) 2012 The Diderot Project (http://diderot-language.cs.uchicago.edu)   * This code is part of the Diderot Project (http://diderot-language.cs.uchicago.edu)
4     *
5     * COPYRIGHT (c) 2015 The University of Chicago
6   * All rights reserved.   * All rights reserved.
7   *)   *)
8    
# Line 10  Line 12 
12    
13      structure E = Ein      structure E = Ein
14      structure DstIL = MidIL      structure DstIL = MidIL
     structure DstOp = MidOps  
15      structure P=Printer      structure P=Printer
     structure T=TransformEin  
16      structure MidToS=MidToString      structure MidToS=MidToString
17      structure DstV = DstIL.Var      structure L=LiftEin
18      in      in
19    
20        val liftflag=true
21        val splitflag=true
22        val splitlimit=NONE
23        val testinitial=0
24      val testing=0      val testing=0
     fun setEin(params,index,body)=Ein.EIN{params=params, index=index, body=body}  
     fun assignEinApp(y,params,index,body,args)= (y,DstIL.EINAPP(setEin(params,index,body),args))  
     fun assignEinApp(y,params,index,body,args)= (y,DstIL.EINAPP(setEin(params,index,body),args))  
       fun iTos i =Int.toString i  
25      fun testp n=(case testing      fun testp n=(case testing
26          of 0=> 1          of 0=> 1
27          | _ =>(print(String.concat n);1)          | _ =>(print(String.concat n);1)
28          (*end case*))          (*end case*))
29      fun test0 n=(*print(String.concat n)*) 1      fun test0 n=(case testinitial
30      val einappzero=DstIL.EINAPP(setEin([],[],E.Const 0),[])          of 0=>testp n
31      fun setEinZero y=  (y,einappzero)          | _ => (print(String.concat n);1))
32      fun filterSca e=Filter.filterSca e      fun toStringBind e= MidToS.toStringBind e
33      fun printEINAPP e=MidToString.printEINAPP e      fun useCount (HighIL.V{useCnt, ...}) = !useCnt
34        fun expandEinOp (srcy,bind00)=let
35            (*val _= raise Fail"development branch- do not use"*)
36    
37            val star="************"
38            val _= ("\n Start"^star)
39            val _ =test0[star,"\n Original EinApp",star,"\n\n","start get test",toStringBind bind00]
40            val (y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=bind00
41    
42    
43      fun z e=String.concat["\n\n Found: ",P.printbody e,"=> 0\n"]          (* ************* Sweep for 0's'*********** *)
44      fun sweep e= (z e;case e          val _ =testp["Pre Sweep\n",P.printbody body,"\n"]
45          of E.Tensor _           => e          val bodysweep=sweepEin.sweep body
46          | E.Add es             => E.Add(List.map sweep es)          val ein1=Ein.EIN{params=params, index=index, body=bodysweep}
47          | E.Sub(e1,e2)         => E.Sub(sweep e1,sweep e2)          val _ =testp["Post Sweep\n",P.printbody bodysweep,"\n"]
         | E.Div(e1,e2)         => E.Div(sweep e1,sweep e2)  
         | E.Sum(c ,e)          => E.Sum(c, sweep e)  
         | E.Prod es             => E.Prod(List.map sweep es)  
         | E.Neg e              => E.Neg(sweep e)  
         | E.Probe(E.Conv _,_)  => e  
         | E.Sqrt e             => E.Sqrt(sweep e)  
         | E.Cosine e           => E.Cosine(sweep e)  
         | E.ArcCosine e        => E.ArcCosine(sweep e)  
         | E.Sine e             => E.Sine(sweep e)  
         | E.ArcSine e        => E.ArcSine(sweep e)  
         | E.Const _            => e  
         | E.ConstR _           => e  
         | E.Delta _               => e  
         | E.Epsilon _             => e  
         | E.Eps2 _                => e  
         | E.Field _            => (z e;E.Const 0)  
         | E.Partial _          => (z e;E.Const 0)  
         | E.Apply _            => (z e;E.Const 0)  
         | E.Lift _             => (z e;E.Const 0)  
         | E.Conv _             => (z e;E.Const 0)  
         | E.PowInt(e ,n)       => E.PowInt(sweep e ,n)  
         | E.PowReal(e,n)       => E.PowReal(sweep e ,n)  
         | E.Probe _            =>  
         raise Fail (String.concat["\n Incorrect probe, substition was not made. Is the Field in an if statement? :",P.printbody e])  
     (*end case*))  
48    
49    
50      (*Distribute summation if needed*)          (* ************** distribute and push Summation*********** *)
51      fun distributeSummation(y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=let          val bind3 = (y,DstIL.EINAPP(SummationEin.main ein1,args))
52          val changed = ref false          val _ =testp["\n\n summation \n\n =>", toStringBind bind3]
53          fun rewrite b=(case b  
54              of E.Sum(sx,E.Tensor(id,[]))    => (changed:=true;E.Tensor(id,[]))          (* **************** lift - split ************* *)
55              | E.Sum(sx,E.Const c)           => (changed:=true;E.Const c)          val newbies5=  (case (splitflag,liftflag,splitlimit)
56              | E.Sum(sx,E.ConstR r)          => (changed:=true;E.ConstR r)              of (true,false,NONE) => Split.splitEinApp bind3
57              | E.Sum(sx,E.Neg n)             => (changed:=true;(E.Neg(E.Sum(sx,n))))              (* **************** split phase ************* *)
58              | E.Sum(sx,E.Add a)             =>              | (true,false,SOME limit) =>let
59              (changed:=true;(E.Add(List.map (fn e=> E.Sum(sx,e)) a)))                  val _ = (String.concat["\nSPLit with limit",Int.toString(limit)])
60              | E.Sum(sx,E.Sub (e1,e2))       =>                  in Split.limitSplit(bind3,limit) end
61                  (changed:=true;(E.Sub(E.Sum(sx,e1),E.Sum(sx,e2))))              | (true,true,SOME limit) =>  let
62              | E.Sum(sx,E.Div(E.Const c,e2)) =>                  (* **************** lift and split Limit ************* *)
63                  (changed:=true;(E.Div(E.Const c, E.Sum(sx,e2))))                  val (einapp1,e2a,fieldset)=L.testLift  bind3
64              | E.Sum(sx,E.Div(e1,e2))        => (changed:=true;                  val _ = (String.concat["\nSPLit with limit",Int.toString(limit),"\n **** lift Length:",Int.toString(length(e2a))])
65                  (E.Sum(sx,E.Prod[e1,E.Div(E.Const 1,rewrite e2)])))  
66              | E.Sum(sx,E.Lift e )           => (changed:=true;(E.Lift(E.Sum(sx,e))))                 (* val t2=List.map (fn t=>Split.splitEinApp t) e2*)
67              | E.Sum(sx,E.PowReal(e,n1))     => (changed:=true;(E.PowReal(E.Sum(sx,e),n1)))                  val t2=List.map (fn t =>Split.limitSplit2(t,limit,fieldset)) e2a
68              | E.Sum(sx,E.Sqrt e)            => (changed:=true;(E.Sqrt(E.Sum(sx,e))))                  val e2= List.foldr op@ [] t2
69              | E.Sum(sx,E.Sum (c2,e))        => (changed:=true; (E.Sum (sx@c2,e)))                  val newbies4=Split.limitSplit2(einapp1,limit,fieldset)
70              | E.Sum(sx,E.Prod p)            => let                  val _ = print(String.concat["\n split made pieces(",Int.toString(length(e2)-length(e2a)),") orig (", Int.toString(length(newbies4)),")"])
71                  val p'=List.map (fn e=> rewrite e) p                  in (e2@newbies4) end
72                  val (c,e)=filterSca(sx,p')              | (true,true, NONE) => let
73                  in (case c of 1=> (changed:=true; e) | _=> e ) end                  (* **************** lift and split All ************* *)
74              | E.Div(e1,e2)                  => E.Div(rewrite e1, rewrite e2)                  val (einapp1,e2a,_)=L.testLift  bind3
75              | E.Sub(e1,E.Const 0)           => (changed:=true; rewrite e1)                  val _ =  print(String.concat["\n **** lift Length"^(Int.toString(length(e2a)))])
76              | E.Sub(e1,e2)                  => E.Sub(rewrite e1, rewrite e2)                  val e2= List.foldr op@ [] (List.map (fn t=>Split.splitEinApp t) e2a)
77              | E.Add es                      => E.Add(List.map rewrite es)                  val newbies4=Split.splitEinApp einapp1
78              | E.Prod es                     => E.Prod(List.map rewrite es)                  val _ = print(String.concat["\n split made pieces(",Int.toString(length(e2)-length(e2a)),") orig (", Int.toString(length(newbies4)),")"])
79              | E.Neg e                       => E.Neg(rewrite e)                  in (e2@newbies4) end
80              | E.Sqrt e                      => E.Sqrt(rewrite e)  
81              | E.Cosine e                    => E.Cosine(rewrite e)  (*
82              | E.ArcCosine e                 => E.ArcCosine(rewrite e)              | (false,true,_) => let
83              | E.Sine e                      => E.Sine(rewrite e)                  val (einapp1,e2)=L.testLift  bind3
84              | E.ArcSine e                 => E.ArcSine(rewrite e)                  val _ = ("\n **** lift Length"^(Int.toString(length(e2))))
85              | E.Probe(e1,e2)                => E.Probe(rewrite e1, rewrite e2)                  in (e2@[einapp1]) end
86              | _                             => b              |  (false,false,_)=>Split.splitEinApp bind3
87    *)
88              (*end case*))              (*end case*))
89          fun loop body  = let          val _ = "************** TASH"
90              val body' = rewrite body          (**************** printing result ***************)
91              in          val _ =testp["\n\n Returning \n\n =>",
92                  if !changed then  (changed := false ;loop body') else  body'          String.concatWith",\n\t"(List.map toStringBind newbies5)]
93              end          val n =length(newbies5)
94            val _ = if (n > 10) then print(String.concat["\n **** SPLIT Length",Int.toString(n)]) else print ""
95          val  b = loop body           (* ************** ProbeEIN *********** *)
96          val _ =testp["\nAfter distributeSummation \n",P.printbody b]          (*gives probeEIN the set. Could split into a sepearate step*)
97          val ein=SummationEin.cleanSummation(Ein.EIN{params=params, index=index, body=b})          val fieldset= einVarSet.EinVarSet.empty
98          val b = loop(Ein.body ein)          val varset= einVarSet.VarSet.empty
99          val ein=Ein.EIN{params=Ein.params ein, index=Ein.index ein, body=b}          val isField = ref 0
100          val einapp2= (y,DstIL.EINAPP(ein,args))          val isFieldR = ref 0
101          in          val isReplaced = ref 0
102              einapp2          fun iter([],_,_)=[]
103             | iter(e1::es,fieldset,varset)= let
104                val (e2,fieldset,varset,isFieldt,isReplacedt) = ProbeEin.expandEinOp(e1,fieldset,varset)
105                val rtn=[e2]@(iter(es,fieldset,varset))
106                val isFieldRt=if ((isFieldt+isReplacedt)=2) then 1 else 0
107                in (isField:=(!isField)+isFieldt;isReplaced:=(!isReplaced )+isReplacedt;
108                isFieldR:=(!isFieldR)+isFieldRt;rtn)
109          end          end
110      |distributeSummation(y,app) =(y,app)          val code=iter(newbies5, fieldset,varset)
111            val flatcode= List.foldr op@ [] code
112    
113      fun expandEinOp einapp00=let          val n= List.length(code)
114          val star="************"          val m= List.length(flatcode)
115          val _ =test0[star,"\n Original EinApp",star,"\n\n","start get test",printEINAPP einapp00]          val _ =testp[HighIL.Var.name srcy ," Pieces",Int.toString (n),"\n"]
116          val (y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=einapp00  
117              val _ = if (n>5)
118              then  print(String.concat["\n",star,":Name:",HighIL.Var.name srcy ,
119                ":VarCount(",Int.toString(useCount srcy),")","\n\tPieces now: ",Int.toString (n),"flatcode",Int.toString (m),
120                " Fields:",Int.toString(!isField)," isReplaced:",Int.toString(!isReplaced),
121                " Replaced_Fields:",Int.toString(!isFieldR)])
122                else print ""
123    
124          (* ************* Sweep for 0's'*********** *)              val flatcode= List.foldr op@ [] code
         val bodysweep=sweep body  
         val ein1=Ein.EIN{params=params, index=index, body=bodysweep}  
         val _=testp["\nPresweep\n",P.printbody body,"\n\n Sweep\n",P.printbody bodysweep,"\n"]  
125    
         (* **************Clean Summation*********** *)  
         val ein2=SummationEin.cleanSummation(ein1)  
         val einapp2=(y,DstIL.EINAPP(ein2, args))  
         val _ =testp["\n\n******* after clean summation**",Int.toString (0)," ***** \n \t==>\n",printEINAPP(einapp2)]  
   
         (* ************** distribute Summation*********** *)  
         val einapp3 = distributeSummation einapp2  
   
         (* ************** split initial *********** *)  
         val (einapp4,newbies4)=Split.split einapp3  
         val _ =testp["\n\t===>\n",printEINAPP(einapp4),"\nand\n",(String.concatWith",\n\t"(List.map printEINAPP newbies4))]  
   
         (* ************** split Multiple *********** *)  
         val (einapp5,newbies5)= Split.iterMultiple(einapp4,newbies4)  
         val _ =testp["\n\n Returning \n\n =>",printEINAPP einapp5,  
         " newbies\n\t",String.concatWith",\n\t"(List.map printEINAPP newbies5), "\n",star]  
126    
          (* ************** ProbeEIN *********** *)  
         val code=List.map (fn e=>ProbeEin.expandEinOp e) (newbies5@[einapp5])  
         val flatcode= List.foldr op@ [] code  
127          (*val _=List.map (fn(_,DstIL.EINAPP(e,_))=>checkEin.checkEIN e | _=> 1) flatcode*)          (*val _=List.map (fn(_,DstIL.EINAPP(e,_))=>checkEin.checkEIN e | _=> 1) flatcode*)
128            val _=(star^"end")
129          in          in
130              List.map (fn (y,rator)=> DstIL.ASSGN(y,rator)) flatcode              List.map (fn (y,rator)=> DstIL.ASSGN(y,rator)) flatcode
131          end          end
132    
   
   
133    end; (* local *)    end; (* local *)
134    
135  end (* local *)  end (* local *)

Legend:
Removed from v.3138  
changed lines
  Added in v.3503

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