17 lines
420 B
C++
17 lines
420 B
C++
#pragma once
|
|
|
|
#include "mlir/IR/BuiltinOps.h"
|
|
#include "mlir/Pass/PassManager.h"
|
|
#include "mlir/Pass/PassRegistry.h"
|
|
|
|
#include "onnx-mlir/Compiler/OMCompilerTypes.h"
|
|
|
|
namespace onnx_mlir {
|
|
|
|
void addPassesPim(mlir::OwningOpRef<mlir::ModuleOp>& module,
|
|
mlir::PassManager& pm,
|
|
EmissionTargetType& emissionTarget,
|
|
std::string outputNameNoExt);
|
|
|
|
} // namespace onnx_mlir
|