[FRONTEND] Expose end-to-end compile to python frontend (#58)

2022-08-18 01:42:48 +08:00
parent 95bbac41e7
commit b1673caaf6
15 changed files with 228 additions and 165 deletions
--- a/bin/triton-translate.cpp
+++ b/bin/triton-translate.cpp
@@ -100,7 +100,6 @@ LogicalResult tritonTranslateMain(int argc, char **argv,
  llvm::InitLLVM y(argc, argv);

  registerAsmPrinterCLOptions();
-
  registerMLIRContextCLOptions();
  llvm::cl::ParseCommandLineOptions(argc, argv, toolName);

@@ -118,7 +117,7 @@ LogicalResult tritonTranslateMain(int argc, char **argv,
  }

  llvm::LLVMContext llvmContext;
-  auto llvmir = TranslateLLVMToLLVMIR(&llvmContext, *module);
+  auto llvmir = translateTritonGPUToLLVMIR(&llvmContext, *module);
  if (!llvmir) {
    llvm::errs() << "Translate to LLVM IR failed";
  }