torch-mlir/lib/Python/MlirInit.cpp

60 lines
1.9 KiB
C++
Raw Normal View History

//===- MlirInit.cpp -------------------------------------------------------===//
2020-04-27 06:50:23 +08:00
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "npcomp/Python/MlirInit.h"
#include "mlir/IR/Dialect.h"
2020-04-27 06:50:23 +08:00
#include "mlir/InitAllDialects.h"
2020-06-03 16:29:59 +08:00
#include "mlir/InitAllPasses.h"
#include "mlir/Pass/PassManager.h"
#include "mlir/Pass/PassRegistry.h"
#include "npcomp/Dialect/Basicpy/IR/BasicpyDialect.h"
#include "npcomp/Dialect/Numpy/IR/NumpyDialect.h"
#include "npcomp/InitAll.h"
#include "llvm/Support/CommandLine.h"
#include "llvm/Support/PrettyStackTrace.h"
#include "llvm/Support/Signals.h"
2020-04-27 06:50:23 +08:00
namespace mlir {
2020-04-27 06:50:23 +08:00
namespace npcomp {
namespace python {
bool npcompMlirInitialize() {
// Enable LLVM's signal handler to get nice stack traces.
llvm::sys::SetOneShotPipeSignalFunction(
llvm::sys::DefaultOneShotPipeSignalHandler);
llvm::sys::PrintStackTraceOnErrorSignal("npcomp");
// Register any pass manager command line options.
mlir::registerPassManagerCLOptions();
mlir::registerMLIRContextCLOptions();
std::string program_name = "npcomp";
std::vector<const char *> default_options = {program_name.c_str(), nullptr};
llvm::cl::ParseCommandLineOptions(1, default_options.data());
// Global registration.
2020-04-27 06:50:23 +08:00
::mlir::registerAllDialects();
2020-06-03 16:29:59 +08:00
::mlir::registerAllPasses();
// Local registration.
::mlir::NPCOMP::registerAllDialects();
::mlir::NPCOMP::registerAllPasses();
return true;
2020-04-27 06:50:23 +08:00
}
LogicalResult parsePassPipeline(StringRef pipeline, OpPassManager &pm,
raw_ostream &errorStream = llvm::errs()) {
return ::mlir::parsePassPipeline(pipeline, pm, errorStream);
}
2020-04-27 06:50:23 +08:00
} // namespace python
} // namespace npcomp
} // namespace mlir