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 3203, Wed Jul 8 16:30:56 2015 UTC
# Line 17  Line 17 
17      structure DstV = DstIL.Var      structure DstV = DstIL.Var
18      in      in
19    
20      val testing=0      val testinitial=0
21      fun setEin(params,index,body)=Ein.EIN{params=params, index=index, body=body}      val testing=1
     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      val einappzero=DstIL.EINAPP(setEin([],[],E.Const 0),[])      fun test0 n=(case testinitial
27      fun setEinZero y=  (y,einappzero)          of 0=>testp n
28            | _ => (print(String.concat n);1))
29        fun printEINAPP e=MidToString.printEINAPP e
30    
31      (* isZero:ein_exp->int  
32      *  Checks to see if the entire body is 0. If so returns 1.  
33    
34        fun expandEinOp bind00=let
35    
36            val star="************"
37            val _ =test0[star,"\n Original EinApp",star,"\n\n","start get test",printEINAPP bind00]
38            val (y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=bind00
39    
40    
41            (* ************* Sweep for 0's'*********** *)
42            val bodysweep=sweepEin.sweep body
43            val ein1=Ein.EIN{params=params, index=index, body=bodysweep}
44            val _=testp["Post Sweep\n",P.printbody bodysweep,"\n"]
45    
46    
47            (* ************** distribute and push Summation*********** *)
48            val bind3 = (y,DstIL.EINAPP(SummationEin.main ein1,args))
49            val _ =testp["\n\n summation \n\n =>", printEINAPP bind3]
50    
51            (* **************** split phase ************* *)
52            val (newbies5)= Split.splitEinApp bind3
53            val _ =testp["\n\n Returning \n\n =>",
54                String.concatWith",\n\t"(List.map printEINAPP newbies5)]
55    
56    
57             (* ************** ProbeEIN *********** *)
58            (*val code=List.map (fn e=>ProbeEin.expandEinOp e) (newbies5)*)
59    
60            (*gives probeEIN the set. Could split into a sepearate step*)
61            val fieldset= einSet.EinSet.empty
62            fun iter([],_)=[]
63             | iter(e1::es,fieldset)= let
64                val (e2,fieldset) = ProbeEin.expandEinOp(e1,fieldset)
65                in [e2]@(iter(es,fieldset))
66                end
67            val code=iter(newbies5, fieldset)
68            (*val _ =print(String.concat[
69                "\n",star,"Number of pieces now: ",Int.toString (List.length(code))])
70      *)      *)
71      fun isZero e=let          val flatcode= List.foldr op@ [] code
72          val _=testp["\n Started Zero",P.printbody e]          (*val _=List.map (fn(_,DstIL.EINAPP(e,_))=>checkEin.checkEIN e | _=> 1) flatcode*)
73          fun iterList []= 1  
             | iterList(e1::list)=(case (isZero e1)  
             of 0=>0  
             | _ => iterList list  
             (*end case*))  
         fun iterProdList []= 0  
         | 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]  
74          in          in
75              z              List.map (fn (y,rator)=> DstIL.ASSGN(y,rator)) flatcode
76          end          end
77    
78    
     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.Const _            => e  
         | E.ConstR _           => e  
         | E.Delta _               => e  
         | E.Epsilon _             => e  
         | E.Eps2 _                => e  
         | E.Field _            => E.Const 0  
         | E.Partial _          => E.Const 0  
         | E.Apply _            => E.Const 0  
         | E.Lift _             => E.Const 0  
         | E.Conv _             => 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*))  
79    end; (* local *)    end; (* local *)
80    
81  end (* local *)  end (* local *)

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

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