commit ddad8f47b80304bc7bf36add65317e929ad2339f
author Kévin Petit <kpet@free.fr> Mon Sep 30 15:12:08 2019 +0100
committer alan-baker <alanbaker@google.com> Mon Sep 30 10:12:08 2019 -0400
tree fb5c61e90152e7d95be80eb8c3744a2ed76738d0
parent 214254f3d77fa1f5ff04229022ef3ab0097a6445

Add support for taking LLVM IR as input (#430)

Add a -x option to select the input type, defaulting to OpenCL C.

Signed-off-by: Kévin Petit <kpet@free.fr>

lib/Compiler.cpp

diff --git a/lib/Compiler.cpp b/lib/Compiler.cpp
index 9963427..ad28085 100644
--- a/lib/Compiler.cpp
+++ b/lib/Compiler.cpp
@@ -120,6 +120,12 @@
     InputFilename(llvm::cl::Positional, llvm::cl::desc("<input .cl file>"),
                   llvm::cl::init("-"));
 
+static llvm::cl::opt<clang::Language> InputLanguage(
+    "x", llvm::cl::desc("Select input type"),
+    llvm::cl::init(clang::Language::OpenCL),
+    llvm::cl::values(clEnumValN(clang::Language::OpenCL, "cl", "OpenCL source"),
+                     clEnumValN(clang::Language::LLVM_IR, "ir", "LLVM IR")));
+
 static llvm::cl::opt<std::string>
     OutputFilename("o", llvm::cl::desc("Override output filename"),
                    llvm::cl::value_desc("filename"));
@@ -782,12 +788,19 @@
   // If we are reading our input file from stdin.
   if ("-" == InputFilename) {
     // We need to overwrite the file name we use.
-    overiddenInputFilename = "stdin.cl";
+    switch (InputLanguage) {
+    case clang::Language::OpenCL:
+      overiddenInputFilename = "stdin.cl";
+      break;
+    case clang::Language::LLVM_IR:
+      overiddenInputFilename = "stdin.ll";
+      break;
+    }
   }
 
   clang::CompilerInstance instance;
-  clang::FrontendInputFile kernelFile(
-      overiddenInputFilename, clang::InputKind(clang::Language::OpenCL));
+  clang::FrontendInputFile kernelFile(overiddenInputFilename,
+                                      clang::InputKind(InputLanguage));
   std::string log;
   llvm::raw_string_ostream diagnosticsStream(log);
   if (auto error = SetCompilerInstanceOptions(