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