NiccoloN
31 lines
788 B
C++
31 lines
788 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> createSpatialLayoutPlanningPass();
|
|
std::unique_ptr<mlir::Pass> createLowerSpatialPlansPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createSpatialToPimPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createPimBufferizationPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createPimMemoryCoalescingPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createMergeComputeNodesPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createPimHostConstantFoldingPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createPimVerificationPass();
|
|
|
|
std::unique_ptr<mlir::Pass> createEmitPimCodePass();
|
|
|
|
std::unique_ptr<mlir::Pass> createMessagePass(std::string message);
|
|
|
|
} // namespace onnx_mlir
|