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