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 2870, Wed Feb 25 21:47:43 2015 UTC branches/charisee_dev/src/compiler/high-to-mid/handleEin.sml revision 3344, Fri Oct 23 04:38:49 2015 UTC
# Line 10  Line 10 
10    
11      structure E = Ein      structure E = Ein
12      structure DstIL = MidIL      structure DstIL = MidIL
     structure DstOp = MidOps  
13      structure P=Printer      structure P=Printer
     structure T=TransformEin  
14      structure MidToS=MidToString      structure MidToS=MidToString
15      structure DstV = DstIL.Var      structure L=LiftEin
16      in      in
17    
18        val testinitial=0
19      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  
20      fun testp n=(case testing      fun testp n=(case testing
21          of 0=> 1          of 0=> 1
22          | _ =>(print(String.concat n);1)          | _ =>(print(String.concat n);1)
23          (*end case*))          (*end case*))
24      val einappzero=DstIL.EINAPP(setEin([],[],E.Const 0),[])      fun test0 n=(case testinitial
25      fun setEinZero y=  (y,einappzero)          of 0=>testp n
26            | _ => (print(String.concat n);1))
27        fun toStringBind e= MidToS.toStringBind e
28        fun useCount (HighIL.V{useCnt, ...}) = !useCnt
29        fun expandEinOp (srcy,bind00)=let
30            val _=raise Fail"development branch- do not use"
31            val star="************"
32            val _=("\n Start"^star)
33            val _ =test0[star,"\n Original EinApp",star,"\n\n","start get test",toStringBind bind00]
34            val (y,einapp as DstIL.EINAPP(Ein.EIN{params, index, body},args))=bind00
35    
36    
37            (* ************* Sweep for 0's'*********** *)
38            val bodysweep=sweepEin.sweep body
39            val ein1=Ein.EIN{params=params, index=index, body=bodysweep}
40            val _=testp["Post Sweep\n",P.printbody bodysweep,"\n"]
41    
42    
43            (* ************** distribute and push Summation*********** *)
44            val bind3 = (y,DstIL.EINAPP(SummationEin.main ein1,args))
45            val _ =testp["\n\n summation \n\n =>", toStringBind bind3]
46    
47    (*
48                val (einapp1,e2)=L.testLift  bind3
49                val newbies5=Split.iterAll(einapp1,e2)
50    
     (* isZero:ein_exp->int  
     *  Checks to see if the entire body is 0. If so returns 1.  
51      *)      *)
52      fun isZero e=let  
53          val _=testp["\n Started Zero",P.printbody e]           val (einapp1,e2)=L.testLift  bind3
54          fun iterList []= 1          val newbies5=  Split.iterAll2(einapp1,e2)
55              | iterList(e1::list)=(case (isZero e1)          val n =length(newbies5)
56              of 0=>0          val _ = if (n > 10) then print("\n **** SPLIT Length"^(Int.toString(n))) else print ""
57              | _ => iterList list  
58              (*end case*))  (*
59          fun iterProdList []= 0           (* **************** lift field phase ************* *)
60          | iterProdList [e1]=isZero e1          (*val newbies5 = L.testLift  bind3*)
61          | iterProdList(e1::list)=(case (isZero e1)          (*List.rev(  [einapp]@code'*)
62              of 1=>1          val (einapp1,e2)=L.testLift  bind3
63              | _ => iterList list  
64              (*end case*))          val newbies5piece= Split.splitEinApp(einapp1,5)
65          val z=(case e          val newbies5= e2 @ newbies5piece
66              of E.Tensor _         => 0          val n =length(newbies5)
67              | E.Add e             => iterList e          val _ = if (n > 10) then print("\n **** SPLIT Length"^(Int.toString(n))) else print ""
68              | E.Sub(e1,e2)         => iterList [e1,e2]          (* **************** split phase ************* *)
69              | E.Div(e1,e2)         => iterProdList [e1,e2]          (*
70              | E.Sum(_ ,e)          => isZero e          val (newbies5)= Split.splitEinApp(bind3,1)
71              | E.Prod e             => iterProdList e          val _ =test0["\n\n Returning \n\n =>",
72              | E.Neg e              => isZero e          String.concatWith",\n\t"(List.map toStringBind newbies5)]
73              | E.Probe(E.Conv _,_)  => 0          *)
74              | E.Sqrt e             => isZero e  *)
75              | E.Const 0            => 1           (* ************** ProbeEIN *********** *)
76              | E.Field _            => 1          (*gives probeEIN the set. Could split into a sepearate step*)
77              | E.Partial _          => 1          val fieldset= einSet.EinSet.empty
78              | E.Apply _            => 1          val isField = ref 0
79              | E.Lift _             => 1          val isFieldR = ref 0
80              | E.Conv _             => 1          val isReplaced = ref 0
81              | E.PowInt(e ,_)            => isZero e          fun iter([],_)=[]
82              | E.PowReal(e,_)             => isZero e           | iter(e1::es,fieldset)= let
83              | _                    => 0              val (e2,fieldset,isFieldt,isReplacedt) = ProbeEin.expandEinOp(e1,fieldset)
84              (*end case*))              val rtn=[e2]@(iter(es,fieldset))
85          val _=testp["\n Is it Zero",P.printbody e,"--", iTos z]              val isFieldRt=if ((isFieldt+isReplacedt)=2) then 1 else 0
86          in              in (isField:=(!isField)+isFieldt;isReplaced:=(!isReplaced )+isReplacedt;
87              z              isFieldR:=(!isFieldR)+isFieldRt;rtn)
88          end          end
89            val code=iter(newbies5, fieldset)
90    
91            val n= List.length(code)
92    
93              val _ = if (n>5)
94             then  (String.concat["\n\n",star,":Name:",HighIL.Var.name srcy ,
95                ":VarCount(",Int.toString(useCount srcy),")",
96                    "\n\t",star,"Pieces now: ",Int.toString (n),
97                " Fields:",Int.toString(!isField)," isReplaced:",Int.toString(!isReplaced),
98                " Replaced_Fields:",Int.toString(!isFieldR)
99                   ,toStringBind bind3])
100                else  ""
101    
102            val flatcode= List.foldr op@ [] code
103            (*val _=List.map (fn(_,DstIL.EINAPP(e,_))=>checkEin.checkEIN e | _=> 1) flatcode*)
104            val _=(star^"end")
105            in
106                List.map (fn (y,rator)=> DstIL.ASSGN(y,rator)) flatcode
107            end
108    
     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*))  
109    end; (* local *)    end; (* local *)
110    
111  end (* local *)  end (* local *)

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

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