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

revision 2870, Wed Feb 25 21:47:43 2015 UTC revision 3193, Mon Apr 27 21:19:16 2015 UTC
# Line 17  Line 17 
17      structure DstV = DstIL.Var      structure DstV = DstIL.Var
18      in      in
19    
20        val testinitial=0
21      val testing=0      val testing=0
22      fun setEin(params,index,body)=Ein.EIN{params=params, index=index, body=body}      fun setEin(params,index,body)=Ein.EIN{params=params, index=index, body=body}
23      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))
# Line 26  Line 27 
27          of 0=> 1          of 0=> 1
28          | _ =>(print(String.concat n);1)          | _ =>(print(String.concat n);1)
29          (*end case*))          (*end case*))
30        fun test0 n=(case testinitial
31            of 0=>1
32            | _ => (print(String.concat n);1))
33      val einappzero=DstIL.EINAPP(setEin([],[],E.Const 0),[])      val einappzero=DstIL.EINAPP(setEin([],[],E.Const 0),[])
34      fun setEinZero y=  (y,einappzero)      fun setEinZero y=  (y,einappzero)
35        fun filterSca e=Filter.filterSca e
36        fun printEINAPP e=MidToString.printEINAPP e
37    
38        fun z e=String.concat["\n\n Found: ",P.printbody e,"=> 0\n"]
39    
40      (* isZero:ein_exp->int      (*Turn High-IL ein expressions to 0
41      *  Checks to see if the entire body is 0. If so returns 1.      * Also, sweep for 0's
42      *)      *)
43      fun isZero e=let  
44          val _=testp["\n Started Zero",P.printbody e]      fun sortadd []=[]
45          fun iterList []= 1        | sortadd(E.Const 0::es)= sortadd es
46              | iterList(e1::list)=(case (isZero e1)        | sortadd(e1::es) =[e1]@(sortadd es)
47              of 0=>0  
48              | _ => iterList list      fun sortprod( [],rest)=E.Prod rest
49              (*end case*))       | sortprod(E.Const 0::_,_)= E.Const 0
50          fun iterProdList []= 0       | sortprod(e1::es,rest) =sortprod(es,rest@[e1])
         | iterProdList [e1]=isZero e1  
         | iterProdList(e1::list)=(case (isZero e1)  
             of 1=>1  
             | _ => iterList list  
             (*end case*))  
         val z=(case e  
             of E.Tensor _         => 0  
             | E.Add e             => iterList e  
             | E.Sub(e1,e2)         => iterList [e1,e2]  
             | E.Div(e1,e2)         => iterProdList [e1,e2]  
             | E.Sum(_ ,e)          => isZero e  
             | E.Prod e             => iterProdList e  
             | E.Neg e              => isZero e  
             | E.Probe(E.Conv _,_)  => 0  
             | E.Sqrt e             => isZero e  
             | E.Const 0            => 1  
             | E.Field _            => 1  
             | E.Partial _          => 1  
             | E.Apply _            => 1  
             | E.Lift _             => 1  
             | E.Conv _             => 1  
             | E.PowInt(e ,_)            => isZero e  
             | E.PowReal(e,_)             => isZero e  
             | _                    => 0  
             (*end case*))  
         val _=testp["\n Is it Zero",P.printbody e,"--", iTos z]  
         in  
             z  
         end  
51    
52    
53      fun sweep e= (case e      fun sweep e= (case e
54          of E.Tensor _           => e          of E.Tensor _           => e
55          | E.Add es             => E.Add(List.map sweep es)          | E.Add es             => (case sortadd(List.map sweep es)
56                of []  => E.Const 0
57                | es   => E.Add es)
58          | E.Sub(e1,e2)         => E.Sub(sweep e1,sweep e2)          | E.Sub(e1,e2)         => E.Sub(sweep e1,sweep e2)
59          | E.Div(e1,e2)         => E.Div(sweep e1,sweep e2)          | E.Div(e1,e2)         => (case (sweep e1,sweep e2)
60                of (E.Const 0,e2)=> E.Const 0
61                | (e1,e2)=> E.Div(e1,e2))
62          | E.Sum(c ,e)          => E.Sum(c, sweep e)          | E.Sum(c ,e)          => E.Sum(c, sweep e)
63          | E.Prod es             => E.Prod(List.map sweep es)          | E.Prod es            => sortprod(List.map sweep es,[])
64          | E.Neg e              => E.Neg(sweep e)          | E.Neg e              => E.Neg(sweep e)
65          | E.Probe(E.Conv _,_)  => e          | E.Probe(E.Conv _,_)  => e
66          | E.Sqrt e             => E.Sqrt(sweep e)          | E.Sqrt e             => E.Sqrt(sweep e)
67            | E.Cosine e           => E.Cosine(sweep e)
68            | E.ArcCosine e        => E.ArcCosine(sweep e)
69            | E.Sine e             => E.Sine(sweep e)
70            | E.ArcSine e          => E.ArcSine(sweep e)
71          | E.Const _            => e          | E.Const _            => e
72          | E.ConstR _           => e          | E.ConstR _           => e
73          | E.Delta _               => e          | E.Delta _               => e
74          | E.Epsilon _             => e          | E.Epsilon _             => e
75          | E.Eps2 _                => e          | E.Eps2 _                => e
76          | E.Field _            => E.Const 0          | E.Field _            => (z e;E.Const 0)
77          | E.Partial _          => E.Const 0          | E.Partial _          => (z e;E.Const 0)
78          | E.Apply _            => E.Const 0          | E.Apply _            => (z e;E.Const 0)
79          | E.Lift _             => E.Const 0          | E.Lift _             => (z e;E.Const 0)
80          | E.Conv _             => E.Const 0          | E.Conv _             => (z e;E.Const 0)
81          | E.PowInt(e ,n)       => E.PowInt(sweep e ,n)          | E.PowInt(e ,n)       => E.PowInt(sweep e ,n)
82          | E.PowReal(e,n)       => E.PowReal(sweep e ,n)          | E.PowReal(e,n)       => E.PowReal(sweep e ,n)
83          | E.Probe _            =>          | E.Probe _            =>
84  raise Fail (String.concat["\n Incorrect probe, substition was not made. Is the Field in an if statement? :",P.printbody e])  raise Fail (String.concat["\n Incorrect probe, substition was not made. Is the Field in an if statement? :",P.printbody e])
85      (*end case*))      (*end case*))
86    
87    
88        (*Distribute summation if needed*)
89        fun distributeSummation(y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=let
90            val changed = ref false
91            fun rewrite b=(case b
92                of E.Sum(sx,E.Tensor(id,[]))    => (changed:=true;E.Tensor(id,[]))
93                | E.Sum(sx,E.Const c)           => (changed:=true;E.Const c)
94                | E.Sum(sx,E.ConstR r)          => (changed:=true;E.ConstR r)
95                | E.Sum(sx,E.Neg n)             => (changed:=true;(E.Neg(E.Sum(sx,n))))
96                | E.Sum(sx,E.Add a)             =>
97                (changed:=true;(E.Add(List.map (fn e=> E.Sum(sx,e)) a)))
98                | E.Sum(sx,E.Sub (e1,e2))       =>
99                    (changed:=true;(E.Sub(E.Sum(sx,e1),E.Sum(sx,e2))))
100    
101                | E.Sum(sx,E.Div(E.Const c,e2)) =>
102                    (changed:=true;(E.Div(E.Const c, E.Sum(sx,e2))))
103                | E.Sum(sx,E.Div(e1,e2))        => (changed:=true;
104                    (E.Sum(sx,E.Prod[e1,E.Div(E.Const 1,rewrite e2)])))
105                | E.Sum(sx,E.Lift e )           => (changed:=true;(E.Lift(E.Sum(sx,e))))
106                | E.Sum(sx,E.PowReal(e,n1))     => (changed:=true;(E.PowReal(E.Sum(sx,e),n1)))
107                | E.Sum(sx,E.Sqrt e)            => (changed:=true;(E.Sqrt(E.Sum(sx,e))))
108                | E.Sum(sx,E.Sum (c2,e))        => (changed:=true; (E.Sum (sx@c2,e)))
109                | E.Sum(sx,E.Prod p)            => let
110                    val p'=List.map (fn e=> rewrite e) p
111                    val (c,e)=filterSca(sx,p')
112                    in (case c of 1=> (changed:=true; e) | _=> e ) end
113                | E.Div(e1,e2)                  => E.Div(rewrite e1, rewrite e2)
114                | E.Sub(e1,E.Const 0)           => (changed:=true; rewrite e1)
115                | E.Sub(e1,e2)                  => E.Sub(rewrite e1, rewrite e2)
116                | E.Add es                      => E.Add(List.map rewrite es)
117                | E.Prod es                     => E.Prod(List.map rewrite es)
118                | E.Neg e                       => E.Neg(rewrite e)
119                | E.Sqrt e                      => E.Sqrt(rewrite e)
120                | E.Cosine e                    => E.Cosine(rewrite e)
121                | E.ArcCosine e                 => E.ArcCosine(rewrite e)
122                | E.Sine e                      => E.Sine(rewrite e)
123                | E.ArcSine e                 => E.ArcSine(rewrite e)
124                | E.Probe(e1,e2)                => E.Probe(rewrite e1, rewrite e2)
125                | _                             => b
126                (*end case*))
127            fun loop body  = let
128                val body' = rewrite body
129                in
130                    if !changed then  (changed := false ;loop body') else  body'
131                end
132    
133            val  b = loop body
134            val _ =testp["\nAfter distributeSummation \n",P.printbody b]
135            val ein=SummationEin.cleanSummation(Ein.EIN{params=params, index=index, body=b})
136            val b = loop(Ein.body ein)
137            val ein=Ein.EIN{params=Ein.params ein, index=Ein.index ein, body=b}
138            val einapp2= (y,DstIL.EINAPP(ein,args))
139            in
140                einapp2
141            end
142        |distributeSummation(y,app) =(y,app)
143    
144    
145    
146        fun expandEinOp einapp00=let
147    
148    
149            val star="************"
150            val _ =test0[star,"\n Original EinApp",star,"\n\n","start get test",printEINAPP einapp00]
151            val (y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=einapp00
152    
153    
154            (* ************* Sweep for 0's'*********** *)
155            val bodysweep=sweep body
156            val ein1=Ein.EIN{params=params, index=index, body=bodysweep}
157            val _=testp["\nPresweep\n",P.printbody body,"\n\n Sweep\n",P.printbody bodysweep,"\n"]
158    
159            (* **************Clean Summation*********** *)
160            val ein2=SummationEin.cleanSummation(ein1)
161            val einapp2=(y,DstIL.EINAPP(ein2, args))
162            val _ =testp["\n\n******* after clean summation**",Int.toString (0)," ***** \n \t==>\n",printEINAPP(einapp2)]
163    
164            (* ************** distribute Summation*********** *)
165            val einapp3 = distributeSummation einapp2
166    
167            (* **************** split phase ************* *)
168            val (newbies5)= Split.splitEinApp einapp3
169              val _ =testp["\n\n Returning \n\n =>",
170                String.concatWith",\n\t"(List.map printEINAPP newbies5)]
171               (* val _ =print(String.concat[
172                "\n",star,"Number of pieces: ",Int.toString (List.length(newbies5))])
173    *)
174             (* ************** ProbeEIN *********** *)
175    
176              (*val code=List.map (fn e=>ProbeEin.expandEinOp e) (newbies5)*)
177    
178            (*gives probeEIN the set. Could split into a sepearate step*)
179            val fieldset= einSet.EinSet.empty
180            fun iter([],_)=[]
181             | iter(e1::es,fieldset)= let
182                val (e2,fieldset) = ProbeEin.expandEinOp(e1,fieldset)
183                in [e2]@(iter(es,fieldset))
184                end
185            val code=iter(newbies5, fieldset)
186            (*val _ =print(String.concat[
187                "\n",star,"Number of pieces now: ",Int.toString (List.length(code))])
188            *)
189            val flatcode= List.foldr op@ [] code
190            (*val _=List.map (fn(_,DstIL.EINAPP(e,_))=>checkEin.checkEIN e | _=> 1) flatcode*)
191    
192            in
193                List.map (fn (y,rator)=> DstIL.ASSGN(y,rator)) flatcode
194            end
195    
196    
197    end; (* local *)    end; (* local *)
198    
199  end (* local *)  end (* local *)

Legend:
Removed from v.2870  
changed lines
  Added in v.3193

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