Robust/src/Main/Main.java

   1 package Main;
   2
   3 import java.io.FileOutputStream;
   4 import java.io.PrintStream;
   5 import java.io.Reader;
   6 import java.io.BufferedReader;
   7 import java.io.FileReader;
   8 import java.io.FileInputStream;
   9 import java.util.Hashtable;
  10 import java.util.Iterator;
  11 import java.util.Set;
  12 import java.util.Vector;
  13
  14 import IR.Tree.ParseNode;
  15 import IR.Tree.BuildIR;
  16 import IR.Tree.SemanticCheck;
  17 import IR.Flat.BuildCodeMultiCore;
  18 import IR.Flat.BuildFlat;
  19 import IR.Flat.BuildCode;
  20 import IR.ClassDescriptor;
  21 import IR.State;
  22 import IR.TaskDescriptor;
  23 import IR.TypeUtil;
  24 import Analysis.Scheduling.MCImplSynthesis;
  25 import Analysis.Scheduling.Schedule;
  26 import Analysis.Scheduling.ScheduleAnalysis;
  27 import Analysis.Scheduling.ScheduleSimulator;
  28 import Analysis.TaskStateAnalysis.TaskAnalysis;
  29 import Analysis.TaskStateAnalysis.TaskTagAnalysis;
  30 import Analysis.TaskStateAnalysis.TaskGraph;
  31 import Analysis.CallGraph.CallGraph;
  32 import Analysis.TaskStateAnalysis.FEdge;
  33 import Analysis.TaskStateAnalysis.FlagState;
  34 import Analysis.TaskStateAnalysis.TagAnalysis;
  35 import Analysis.TaskStateAnalysis.GarbageAnalysis;
  36 import Analysis.TaskStateAnalysis.ExecutionGraph;
  37 import Analysis.TaskStateAnalysis.SafetyAnalysis;
  38 import Analysis.Locality.LocalityAnalysis;
  39 import Analysis.Locality.GenerateConversions;
  40 import Analysis.Prefetch.PrefetchAnalysis;
  41 import Analysis.FlatIRGraph.FlatIRGraph;
  42 import Analysis.OwnershipAnalysis.OwnershipAnalysis;
  43 import Analysis.MLP.MLPAnalysis;
  44 import Analysis.Loops.*;
  45 import IR.MethodDescriptor;
  46 import IR.Flat.FlatMethod;
  47 import Interface.*;
  48 import Util.GraphNode;
  49 import Util.GraphNode.DFS;
  50 import Util.GraphNode.SCC;
  51
  52 public class Main {
  53
  54   /** Main method for the compiler.  */
  55
  56   public static void main(String args[]) throws Exception {
  57     String ClassLibraryPrefix="./ClassLibrary/";
  58     State state=new State();
  59     Vector sourcefiles=new Vector();
  60     state.classpath.add(".");
  61
  62     String outputdir = null;
  63     boolean isDistributeInfo = false;
  64
  65     for(int i=0; i<args.length; i++) {
  66       String option=args[i];
  67       if (option.equals("-precise"))
  68         IR.Flat.BuildCode.GENERATEPRECISEGC=true;
  69       else if (option.equals("-prefetch"))
  70         state.PREFETCH=true;
  71       else if (option.equals("-dir"))
  72         IR.Flat.BuildCode.PREFIX=args[++i]+"/";
  73       else if (option.equals("-fastcheck"))
  74         state.FASTCHECK=true;
  75       else if (option.equals("-selfloop"))
  76         state.selfloops.add(args[++i]);
  77           else if (option.equals("-outputdir"))
  78         state.outputdir = args[++i];
  79       else if (option.equals("-excprefetch"))
  80         state.excprefetch.add(args[++i]);
  81       else if (option.equals("-classlibrary"))
  82         state.classpath.add(args[++i]);
  83       else if(option.equals("-numcore")) {
  84         ++i;
  85         state.CORENUM = Integer.parseInt(args[i]);
  86       } else if (option.equals("-mainclass"))
  87         state.main=args[++i];
  88       else if (option.equals("-trueprob")) {
  89         state.TRUEPROB=Double.parseDouble(args[++i]);
  90       } else if (option.equals("-printflat"))
  91         State.PRINTFLAT=true;
  92       else if (option.equals("-printscheduling"))
  93         State.PRINTSCHEDULING=true;
  94       else if (option.equals("-printschedulesim"))
  95         State.PRINTSCHEDULESIM=true;
  96       else if (option.equals("-printcriticalpath"))
  97           State.PRINTCRITICALPATH=true;
  98       else if (option.equals("-struct"))
  99         state.structfile=args[++i];
 100       else if (option.equals("-conscheck"))
 101         state.CONSCHECK=true;
 102       else if (option.equals("-task"))
 103         state.TASK=true;
 104       else if (option.equals("-abortreaders"))
 105         state.ABORTREADERS=true;
 106       else if (option.equals("-taskstate"))
 107         state.TASKSTATE=true;
 108       else if (option.equals("-tagstate"))
 109         state.TAGSTATE=true;
 110       else if (option.equals("-flatirtasks")) {
 111         state.FLATIRGRAPH=true;
 112         state.FLATIRGRAPHTASKS=true;
 113       } else if (option.equals("-flatirusermethods")) {
 114         state.FLATIRGRAPH=true;
 115         state.FLATIRGRAPHUSERMETHODS=true;
 116       } else if (option.equals("-flatirlibmethods")) {
 117         state.FLATIRGRAPH=true;
 118         state.FLATIRGRAPHLIBMETHODS=true;
 119       } else if (option.equals("-multicore"))
 120         state.MULTICORE=true;
 121       else if (option.equals("-ownership"))
 122         state.OWNERSHIP=true;
 123       else if (option.equals("-ownallocdepth")) {
 124         state.OWNERSHIPALLOCDEPTH=Integer.parseInt(args[++i]);
 125       } else if (option.equals("-ownwritedots")) {
 126         state.OWNERSHIPWRITEDOTS=true;
 127         if (args[++i].equals("all")) {
 128           state.OWNERSHIPWRITEALL=true;
 129         }
 130       } else if (option.equals("-ownaliasfile"))
 131         state.OWNERSHIPALIASFILE=args[++i];
 132       else if (option.equals("-optional"))
 133         state.OPTIONAL=true;
 134       else if (option.equals("-optimize"))
 135         state.OPTIMIZE=true;
 136       else if (option.equals("-raw"))
 137         state.RAW=true;
 138       else if (option.equals("-scheduling"))
 139         state.SCHEDULING=true;
 140       else if (option.equals("-distributioninfo"))
 141         isDistributeInfo=true;
 142       else if (option.equals("-useprofile"))
 143         state.USEPROFILE=true;
 144       else if (option.equals("-thread"))
 145         state.THREAD=true;
 146       else if (option.equals("-dsm"))
 147         state.DSM=true;
 148       else if (option.equals("-singleTM"))
 149         state.SINGLETM=true;
 150       else if (option.equals("-webinterface"))
 151         state.WEBINTERFACE=true;
 152       else if (option.equals("-instructionfailures"))
 153         state.INSTRUCTIONFAILURE=true;
 154       else if (option.equals("-abcclose"))
 155         state.ARRAYBOUNDARYCHECK=false;
 156       else if (option.equals("-mlp"))
 157         state.MLP=true;
 158       else if (option.equals("-help")) {
 159         System.out.println("-classlibrary classlibrarydirectory -- directory where classlibrary is located");
 160         System.out.println("-selfloop task -- this task doesn't self loop its parameters forever");
 161         System.out.println("-dir outputdirectory -- output code in outputdirectory");
 162         System.out.println("-struct structfile -- output structure declarations for repair tool");
 163         System.out.println("-mainclass -- main function to call");
 164         System.out.println("-dsm -- distributed shared memory support");
 165         System.out.println("-singleTM -- single machine committing transactions");
 166         System.out.println("-abortreaders -- abort readers");
 167         System.out.println("-precise -- use precise garbage collection");
 168         System.out.println("-conscheck -- turn on consistency checking");
 169         System.out.println("-task -- compiler for tasks");
 170         System.out.println("-fastcheck -- fastcheckpointing for Bristlecone");
 171         System.out.println("-thread -- threads");
 172         System.out.println("-trueprob <d> -- probability of true branch");
 173         System.out.println("-printflat -- print out flat representation");
 174         System.out.println("-instructionfailures -- insert code for instruction level failures");
 175         System.out.println("-taskstate -- do task state analysis");
 176         System.out.println("-flatirtasks -- create dot files for flat IR graphs of tasks");
 177         System.out.println("-flatirusermethods -- create dot files for flat IR graphs of user methods");
 178         System.out.println("-flatirlibmethods -- create dot files for flat IR graphs of library class methods");
 179         System.out.println("  note: -flatirusermethods or -flatirlibmethods currently generate all class method flat IR graphs");
 180         System.out.println("-ownership -- do ownership analysis");
 181         System.out.println("-ownallocdepth <d> -- set allocation depth for ownership analysis");
 182         System.out.println("-ownwritedots <all/final> -- write ownership graphs; can be all results or just final results");
 183         System.out.println("-ownaliasfile <filename> -- write a text file showing all detected aliases in program tasks");
 184         System.out.println("-optimize -- enable optimizations");
 185         System.out.println("-optional -- enable optional arguments");
 186         System.out.println("-abcclose close the array boundary check");
 187         System.out.println("-scheduling do task scheduling");
 188         System.out.println("-mlp build mlp code");
 189         System.out.println("-multicore generate multi-core version binary");
 190         System.out.println("-numcore set the number of cores (should be used together with -multicore), defaultly set as 1");
 191         System.out.println("-raw generate raw version binary (should be used together with -multicore)");
 192         System.out.println("-interrupt generate raw version binary with interruption (should be used togethere with -raw)");
 193         System.out.println("-rawconfig config raw simulator as 4xn (should be used together with -raw)");
 194         System.out.println("-rawpath print out execute path information for raw version (should be used together with -raw)");
 195         System.out.println("-useprofile use profiling data for scheduling (should be used together with -raw)");
 196         System.out.println("-threadsimulate generate multi-thread simulate version binary");
 197         System.out.println("-rawuseio use standard io to output profiling data (should be used together with -raw and -profile), it only works with single core version");
 198         System.out.println("-printscheduling -- print out scheduling graphs");
 199         System.out.println("-printschedulesim -- print out scheduling simulation result graphs");
 200         System.out.println("-webinterface -- enable web interface");
 201         System.out.println("-help -- print out help");
 202         System.exit(0);
 203       } else {
 204         sourcefiles.add(args[i]);
 205       }
 206     }
 207
 208     //add default classpath
 209     if (state.classpath.size()==1)
 210       state.classpath.add(ClassLibraryPrefix);
 211
 212     BuildIR bir=new BuildIR(state);
 213     TypeUtil tu=new TypeUtil(state, bir);
 214
 215
 216     SemanticCheck sc=new SemanticCheck(state,tu);
 217
 218     for(int i=0;i<sourcefiles.size();i++)
 219       loadClass(state, bir,(String)sourcefiles.get(i));
 220
 221     //Stuff the runtime wants to see
 222     sc.getClass("String");
 223     sc.getClass("Math");
 224     sc.getClass("File");
 225     sc.getClass("Socket");
 226     sc.getClass("ServerSocket");
 227     sc.getClass("FileInputStream");
 228     sc.getClass("FileOutputStream");
 229     if (state.TASK) {
 230       sc.getClass("TagDescriptor");
 231     }
 232     if (state.THREAD||state.DSM||state.SINGLETM) {
 233       sc.getClass("Thread");
 234     }
 235
 236     sc.semanticCheck();
 237
 238     tu.createFullTable();
 239
 240     BuildFlat bf=new BuildFlat(state,tu);
 241     bf.buildFlat();
 242     SafetyAnalysis sa=null;
 243     PrefetchAnalysis pa=null;
 244
 245     if (state.OPTIMIZE) {
 246       CallGraph callgraph=new CallGraph(state);
 247       CopyPropagation cp=new CopyPropagation();
 248       DeadCode dc=new DeadCode();
 249       GlobalFieldType gft=new GlobalFieldType(callgraph, state, tu.getMain());
 250       CSE cse=new CSE(gft, tu);
 251       localCSE lcse=new localCSE(gft, tu);
 252       LoopOptimize lo=new LoopOptimize(gft, tu);
 253       Iterator classit=state.getClassSymbolTable().getDescriptorsIterator();
 254       while(classit.hasNext()) {
 255         ClassDescriptor cn=(ClassDescriptor)classit.next();
 256         Iterator methodit=cn.getMethods();
 257         while(methodit.hasNext()) {
 258           /* Classify parameters */
 259           MethodDescriptor md=(MethodDescriptor)methodit.next();
 260           FlatMethod fm=state.getMethodFlat(md);
 261           cp.optimize(fm);
 262           dc.optimize(fm);
 263           lo.optimize(fm);
 264           lcse.doAnalysis(fm);
 265           cse.doAnalysis(fm);
 266           cp.optimize(fm);
 267           dc.optimize(fm);
 268         }
 269       }
 270     }
 271
 272     if (state.FLATIRGRAPH) {
 273       FlatIRGraph firg = new FlatIRGraph(state,
 274                                          state.FLATIRGRAPHTASKS,
 275                                          state.FLATIRGRAPHUSERMETHODS,
 276                                          state.FLATIRGRAPHLIBMETHODS);
 277     }
 278
 279     if (state.OWNERSHIP && !state.MLP) {
 280       CallGraph callGraph = new CallGraph(state);
 281       OwnershipAnalysis oa = new OwnershipAnalysis(state,
 282                                                    tu,
 283                                                    callGraph,
 284                                                    state.OWNERSHIPALLOCDEPTH,
 285                                                    state.OWNERSHIPWRITEDOTS,
 286                                                    state.OWNERSHIPWRITEALL,
 287                                                    state.OWNERSHIPALIASFILE);
 288     }
 289
 290     if (state.MLP) {
 291       // gotta run this to have mlp turned on
 292       assert state.OWNERSHIP;
 293
 294       CallGraph callGraph = new CallGraph(state);
 295       OwnershipAnalysis oa = new OwnershipAnalysis(state,
 296                                                    tu,
 297                                                    callGraph,
 298                                                    state.OWNERSHIPALLOCDEPTH,
 299                                                    state.OWNERSHIPWRITEDOTS,
 300                                                    state.OWNERSHIPWRITEALL,
 301                                                    state.OWNERSHIPALIASFILE);
 302       MLPAnalysis mlpa = new MLPAnalysis(state,
 303                                          tu,
 304                                          callGraph,
 305                                          oa);
 306     }
 307
 308     if (state.TAGSTATE) {
 309       CallGraph callgraph=new CallGraph(state);
 310       TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
 311       TaskTagAnalysis tta=new TaskTagAnalysis(state, taganalysis, tu);
 312     }
 313
 314     if (state.TASKSTATE) {
 315       CallGraph callgraph=new CallGraph(state);
 316       TagAnalysis taganalysis=new TagAnalysis(state, callgraph);
 317       TaskAnalysis ta=new TaskAnalysis(state, taganalysis, tu);
 318       ta.taskAnalysis();
 319       TaskGraph tg=new TaskGraph(state, ta);
 320       tg.createDOTfiles();
 321
 322       if (state.OPTIONAL) {
 323         ExecutionGraph et=new ExecutionGraph(state, ta);
 324         et.createExecutionGraph();
 325         sa = new SafetyAnalysis(et.getExecutionGraph(), state, ta);
 326         sa.doAnalysis();
 327         state.storeAnalysisResult(sa.getResult());
 328         state.storeOptionalTaskDescriptors(sa.getOptionalTaskDescriptors());
 329       }
 330
 331       if (state.WEBINTERFACE) {
 332         GarbageAnalysis ga=new GarbageAnalysis(state, ta);
 333         WebInterface wi=new WebInterface(state, ta, tg, ga, taganalysis);
 334         JhttpServer serve=new JhttpServer(8000,wi);
 335         serve.run();
 336       }
 337
 338       if (state.SCHEDULING) {
 339         // Use ownership analysis to get alias information
 340         CallGraph callGraph = new CallGraph(state);
 341         OwnershipAnalysis oa = null;/*new OwnershipAnalysis(state,
 342                                                      tu,
 343                                                      callGraph,
 344                                                      state.OWNERSHIPALLOCDEPTH,
 345                                                      state.OWNERSHIPWRITEDOTS,
 346                                                      state.OWNERSHIPWRITEALL,
 347                                                      state.OWNERSHIPALIASFILE);*/
 348
 349         // synthesis a layout according to target multicore processor
 350         MCImplSynthesis mcImplSynthesis = new MCImplSynthesis(state,
 351                                                               ta,
 352                                                               oa);
 353         if(isDistributeInfo) {
 354             mcImplSynthesis.distribution();
 355         } else {
 356             mcImplSynthesis.setScheduleThreshold(20);
 357             mcImplSynthesis.setProbThreshold(0);
 358             mcImplSynthesis.setGenerateThreshold(30);
 359             Vector<Schedule> scheduling = mcImplSynthesis.synthesis();
 360
 361             // generate multicore codes
 362             if(state.MULTICORE) {
 363                 BuildCodeMultiCore bcm=new BuildCodeMultiCore(state,
 364                                                               bf.getMap(),
 365                                                               tu,
 366                                                               sa,
 367                                                               scheduling,
 368                                                               mcImplSynthesis.getCoreNum(),
 369                                                               pa);
 370                 bcm.setOwnershipAnalysis(oa);
 371                 bcm.buildCode();
 372             }
 373             scheduling.clear();
 374             scheduling = null;
 375         }
 376       }
 377     }
 378     if(!state.MULTICORE) {
 379       if (state.DSM||state.SINGLETM) {
 380         CallGraph callgraph=new CallGraph(state);
 381         if (state.PREFETCH) {
 382           //speed up prefetch generation using locality analysis results
 383           LocalityAnalysis la=new LocalityAnalysis(state, callgraph, tu);
 384           pa=new PrefetchAnalysis(state, callgraph, tu, la);
 385         }
 386         LocalityAnalysis la=new LocalityAnalysis(state, callgraph, tu);
 387         GenerateConversions gc=new GenerateConversions(la, state);
 388         BuildCode bc=new BuildCode(state, bf.getMap(), tu, la, pa);
 389         bc.buildCode();
 390       } else {
 391         BuildCode bc=new BuildCode(state, bf.getMap(), tu, sa, pa);
 392         bc.buildCode();
 393       }
 394     }
 395
 396     System.out.println("Lines="+state.lines);
 397     System.exit(0);
 398   }
 399
 400   public static void loadClass(State state, BuildIR bir, String sourcefile) {
 401     ParseNode pn=readSourceFile(state, sourcefile);
 402     bir.buildtree(pn, null);
 403   }
 404
 405   /** Reads in a source file and adds the parse tree to the state object. */
 406
 407   public static ParseNode readSourceFile(State state, String sourcefile) {
 408     try {
 409       Reader fr= new BufferedReader(new FileReader(sourcefile));
 410       Lex.Lexer l = new Lex.Lexer(fr);
 411       java_cup.runtime.lr_parser g;
 412       g = new Parse.Parser(l);
 413       ParseNode p=null;
 414       try {
 415         p=(ParseNode) g./*debug_*/parse().value;
 416       } catch (Exception e) {
 417         System.err.println("Error parsing file:"+sourcefile);
 418         e.printStackTrace();
 419         System.exit(-1);
 420       }
 421       state.addParseNode(p);
 422       if (l.numErrors()!=0) {
 423         System.out.println("Error parsing "+sourcefile);
 424         System.exit(l.numErrors());
 425       }
 426       state.lines+=l.line_num;
 427       return p;
 428
 429     } catch (Exception e) {
 430       throw new Error(e);
 431     }
 432   }
 433 }