cilkgraph computation DAG generation tool

Author: kennyzzhang

clang/include/clang/Basic/LangOptions.h

@@ -398,7 +398,8 @@ class LangOptions : public LangOptionsBase {
     Cilktool_None = 0,
     Cilktool_Cilkscale,
     Cilktool_Cilkscale_InstructionCount,
-    Cilktool_Cilkscale_Benchmark
+    Cilktool_Cilkscale_Benchmark,
+    Cilktool_Cilkgraph
   };
 
   enum CilkVersion {

clang/lib/CodeGen/BackendUtil.cpp

@@ -300,6 +300,20 @@ static CSIOptions getCSIOptionsForCilkscaleBenchmark() {
   return Options;
 }
 
+static CSIOptions getCSIOptionsForCilkgraph() {
+  CSIOptions Options;
+  // Disable CSI hooks that Cilkgraph doesn't need.
+  Options.InstrumentLoops = false;
+//  Options.InstrumentBasicBlocks = false;
+  Options.InstrumentMemoryAccesses = false;
+  Options.InstrumentCalls = false;
+  Options.InstrumentAtomics = false;
+  Options.InstrumentMemIntrinsics = false;
+  Options.InstrumentAllocas = false;
+  Options.InstrumentAllocFns = false;
+  return Options;
+}
+
 static TargetLibraryInfoImpl *createTLII(llvm::Triple &TargetTriple,
                                          const CodeGenOptions &CodeGenOpts) {
   TargetLibraryInfoImpl *TLII = new TargetLibraryInfoImpl(TargetTriple);
@@ -1064,6 +1078,15 @@ void EmitAssemblyHelper::RunOptimizationPipeline(
               MPM.addPass(PB.buildPostCilkInstrumentationPipeline(Level));
             });
         break;
+      case LangOptions::CilktoolKind::Cilktool_Cilkgraph:
+          PB.registerTapirLoopEndEPCallback(
+            [&PB](ModulePassManager &MPM, OptimizationLevel Level) {
+              MPM.addPass(CSISetupPass(getCSIOptionsForCilkgraph()));
+              MPM.addPass(ComprehensiveStaticInstrumentationPass(
+                  getCSIOptionsForCilkgraph()));
+              MPM.addPass(PB.buildPostCilkInstrumentationPipeline(Level));
+            });
+        break;
       }
     }
     // Register the CSI pass.

clang/lib/Frontend/CompilerInvocation.cpp

@@ -1287,6 +1287,7 @@ parseCilktoolKind(StringRef FlagName, ArgList &Args, DiagnosticsEngine &Diags) {
                   LangOptions::Cilktool_Cilkscale_InstructionCount)
             .Case("cilkscale-benchmark",
                   LangOptions::Cilktool_Cilkscale_Benchmark)
+            .Case("cilkgraph", LangOptions::Cilktool_Cilkgraph)
             .Default(LangOptions::Cilktool_None);
     if (ParsedCilktool == LangOptions::Cilktool_None)
       Diags.Report(diag::err_drv_invalid_value) << FlagName << Val;
@@ -1309,6 +1310,9 @@ serializeCilktoolKind(LangOptions::CilktoolKind K) {
   case LangOptions::Cilktool_Cilkscale_Benchmark:
     CilktoolStr = "cilkscale-benchmark";
     break;
+  case LangOptions::Cilktool_Cilkgraph:
+    CilktoolStr = "cilkgraph";
+    break;
   case LangOptions::Cilktool_None:
     break;
   }

clang/lib/Frontend/InitPreprocessor.cpp

@@ -1316,6 +1316,9 @@ static void InitializePredefinedMacros(const TargetInfo &TI,
   case LangOptions::CilktoolKind::Cilktool_Cilkscale_Benchmark:
     Builder.defineMacro("__cilkscale__");
     break;
+  case LangOptions::CilktoolKind::Cilktool_Cilkgraph:
+    Builder.defineMacro("__cilkgraph__");
+    break;
   default: break;
   }