31 lines
712 B
C++
31 lines
712 B
C++
#pragma once
|
|
|
|
#include "mlir/Pass/Pass.h"
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace onnx_mlir {
|
|
|
|
std::unique_ptr<mlir::Pass> createONNXToSpatialPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createSpatialToGraphvizPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createSpatialToPimPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createBufferizePimPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createConstantFoldingPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createMaterializeConstantsPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createVerificationPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createEmitPimJsonPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createMessagePass(std::string message);
|
|
|
|
std::unique_ptr<mlir::Pass> createCountInstructionPass();
|
|
|
|
} // namespace onnx_mlir
|