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 3189, Thu Apr 2 18:49:21 2015 UTC branches/charisee_dev/src/compiler/high-to-mid/handleEin.sml revision 3304, Fri Oct 16 18:51:09 2015 UTC
# Line 19  Line 19 
19    
20      val testinitial=0      val testinitial=0
21      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  
22      fun testp n=(case testing      fun testp n=(case testing
23          of 0=> 1          of 0=> 1
24          | _ =>(print(String.concat n);1)          | _ =>(print(String.concat n);1)
25          (*end case*))          (*end case*))
26      fun test0 n=(case testinitial      fun test0 n=(case testinitial
27          of 0=>1          of 0=>testp n
28          | _ => (print(String.concat n);1))          | _ => (print(String.concat n);1))
29      val einappzero=DstIL.EINAPP(setEin([],[],E.Const 0),[])      fun toStringBind e= MidToString.toStringBind e
30      fun setEinZero y=  (y,einappzero)      fun useCount (HighIL.V{useCnt, ...}) = !useCnt
31      fun filterSca e=Filter.filterSca e      fun expandEinOp (srcy,bind00)=let
     fun printEINAPP e=MidToString.printEINAPP e  
   
     fun z e=String.concat["\n\n Found: ",P.printbody e,"=> 0\n"]  
     fun sweep e= (case e  
         of E.Tensor _          => e  
         | E.Add es             => E.Add(List.map sweep es)  
         | E.Sub(e1,e2)         => E.Sub(sweep e1,sweep e2)  
         | 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*))  
   
   
     (*Distribute summation if needed*)  
     fun distributeSummation(y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=let  
         val changed = ref false  
         fun rewrite b=(case b  
             of E.Sum(sx,E.Tensor(id,[]))    => (changed:=true;E.Tensor(id,[]))  
             | E.Sum(sx,E.Const c)           => (changed:=true;E.Const c)  
             | E.Sum(sx,E.ConstR r)          => (changed:=true;E.ConstR r)  
             | E.Sum(sx,E.Neg n)             => (changed:=true;(E.Neg(E.Sum(sx,n))))  
             | E.Sum(sx,E.Add a)             =>  
             (changed:=true;(E.Add(List.map (fn e=> E.Sum(sx,e)) a)))  
             | E.Sum(sx,E.Sub (e1,e2))       =>  
                 (changed:=true;(E.Sub(E.Sum(sx,e1),E.Sum(sx,e2))))  
             | E.Sum(sx,E.Div(E.Const c,e2)) =>  
                 (changed:=true;(E.Div(E.Const c, E.Sum(sx,e2))))  
             | E.Sum(sx,E.Div(e1,e2))        => (changed:=true;  
                 (E.Sum(sx,E.Prod[e1,E.Div(E.Const 1,rewrite e2)])))  
             | E.Sum(sx,E.Lift e )           => (changed:=true;(E.Lift(E.Sum(sx,e))))  
             | E.Sum(sx,E.PowReal(e,n1))     => (changed:=true;(E.PowReal(E.Sum(sx,e),n1)))  
             | E.Sum(sx,E.Sqrt e)            => (changed:=true;(E.Sqrt(E.Sum(sx,e))))  
             | E.Sum(sx,E.Sum (c2,e))        => (changed:=true; (E.Sum (sx@c2,e)))  
             | E.Sum(sx,E.Prod p)            => let  
                 val p'=List.map (fn e=> rewrite e) p  
                 val (c,e)=filterSca(sx,p')  
                 in (case c of 1=> (changed:=true; e) | _=> e ) end  
             | E.Div(e1,e2)                  => E.Div(rewrite e1, rewrite e2)  
             | E.Sub(e1,E.Const 0)           => (changed:=true; rewrite e1)  
             | E.Sub(e1,e2)                  => E.Sub(rewrite e1, rewrite e2)  
             | E.Add es                      => E.Add(List.map rewrite es)  
             | E.Prod es                     => E.Prod(List.map rewrite es)  
             | E.Neg e                       => E.Neg(rewrite e)  
             | E.Sqrt e                      => E.Sqrt(rewrite e)  
             | E.Cosine e                    => E.Cosine(rewrite e)  
             | E.ArcCosine e                 => E.ArcCosine(rewrite e)  
             | E.Sine e                      => E.Sine(rewrite e)  
             | E.ArcSine e                 => E.ArcSine(rewrite e)  
             | E.Probe(e1,e2)                => E.Probe(rewrite e1, rewrite e2)  
             | _                             => b  
             (*end case*))  
         fun loop body  = let  
             val body' = rewrite body  
             in  
                 if !changed then  (changed := false ;loop body') else  body'  
             end  
   
         val  b = loop body  
         val _ =testp["\nAfter distributeSummation \n",P.printbody b]  
         val ein=SummationEin.cleanSummation(Ein.EIN{params=params, index=index, body=b})  
         val b = loop(Ein.body ein)  
         val ein=Ein.EIN{params=Ein.params ein, index=Ein.index ein, body=b}  
         val einapp2= (y,DstIL.EINAPP(ein,args))  
         in  
             einapp2  
         end  
     |distributeSummation(y,app) =(y,app)  
   
   
   
     fun expandEinOp einapp00=let  
   
32    
33          val star="************"          val star="************"
34          val _ =test0[star,"\n Original EinApp",star,"\n\n","start get test",printEINAPP einapp00]          val _ =test0[star,"\n Original EinApp",star,"\n\n","start get test",toStringBind bind00]
35          val (y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=einapp00          val (y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=bind00
36    
37    
38          (* ************* Sweep for 0's'*********** *)          (* ************* Sweep for 0's'*********** *)
39          val bodysweep=sweep body          val bodysweep=sweepEin.sweep body
40          val ein1=Ein.EIN{params=params, index=index, body=bodysweep}          val ein1=Ein.EIN{params=params, index=index, body=bodysweep}
41          val _=testp["\nPresweep\n",P.printbody body,"\n\n Sweep\n",P.printbody bodysweep,"\n"]          val _=testp["Post Sweep\n",P.printbody bodysweep,"\n"]
42    
         (* **************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)]  
43    
44          (* ************** distribute Summation*********** *)          (* ************** distribute and push Summation*********** *)
45          val einapp3 = distributeSummation einapp2          val bind3 = (y,DstIL.EINAPP(SummationEin.main ein1,args))
46            val _ =testp["\n\n summation \n\n =>", toStringBind bind3]
47    
48          (* **************** split phase ************* *)          (* **************** split phase ************* *)
49          val (newbies5)= Split.splitEinApp einapp3          val (newbies5)= Split.splitEinApp bind3
50            val _ =testp["\n\n Returning \n\n =>",          val _ =test0["\n\n Returning \n\n =>",
51              String.concatWith",\n\t"(List.map printEINAPP newbies5)]              String.concatWith",\n\t"(List.map toStringBind newbies5)]
52             (* val _ =print(String.concat[  
             "\n",star,"Number of pieces: ",Int.toString (List.length(newbies5))])  
 *)  
          (* ************** ProbeEIN *********** *)  
53    
54             (* ************** ProbeEIN *********** *)
55            (*val code=List.map (fn e=>ProbeEin.expandEinOp e) (newbies5)*)            (*val code=List.map (fn e=>ProbeEin.expandEinOp e) (newbies5)*)
56    
57          (*gives probeEIN the set. Could split into a sepearate step*)          (*gives probeEIN the set. Could split into a sepearate step*)
58          val fieldset= einSet.EinSet.empty          val fieldset= einSet.EinSet.empty
59            val isField = ref 0
60            val isFieldR = ref 0
61            val isReplaced = ref 0
62          fun iter([],_)=[]          fun iter([],_)=[]
63           | iter(e1::es,fieldset)= let           | iter(e1::es,fieldset)= let
64              val (e2,fieldset) = ProbeEin.expandEinOp(e1,fieldset)              val (e2,fieldset,isFieldt,isReplacedt) = ProbeEin.expandEinOp(e1,fieldset)
65              in [e2]@(iter(es,fieldset))              val rtn=[e2]@(iter(es,fieldset))
66                val isFieldRt=if ((isFieldt+isReplacedt)=2) then 1 else 0
67                in (isField:=(!isField)+isFieldt;isReplaced:=(!isReplaced )+isReplacedt;
68                isFieldR:=(!isFieldR)+isFieldRt;rtn)
69              end              end
70          val code=iter(newbies5, fieldset)          val code=iter(newbies5, fieldset)
71          (*val _ =print(String.concat[  
72              "\n",star,"Number of pieces now: ",Int.toString (List.length(code))])          val n= List.length(code)
73          *)  
74              val _ = if (n>5)
75             then  (String.concat["\n\n",star,":Name:",HighIL.Var.name srcy ,
76                ":VarCount(",Int.toString(useCount srcy),")",
77                    "\n\t",star,"Pieces now: ",Int.toString (n),
78                " Fields:",Int.toString(!isField)," isReplaced:",Int.toString(!isReplaced),
79                " Replaced_Fields:",Int.toString(!isFieldR)
80                   ,toStringBind bind3])
81                else  ""
82    
83          val flatcode= List.foldr op@ [] code          val flatcode= List.foldr op@ [] code
84          (*val _=List.map (fn(_,DstIL.EINAPP(e,_))=>checkEin.checkEIN e | _=> 1) flatcode*)          (*val _=List.map (fn(_,DstIL.EINAPP(e,_))=>checkEin.checkEIN e | _=> 1) flatcode*)
85    
# Line 174  Line 87 
87              List.map (fn (y,rator)=> DstIL.ASSGN(y,rator)) flatcode              List.map (fn (y,rator)=> DstIL.ASSGN(y,rator)) flatcode
88          end          end
89    
   
90    end; (* local *)    end; (* local *)
91    
92  end (* local *)  end (* local *)

Legend:
Removed from v.3189  
changed lines
  Added in v.3304

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