nnicolosi/Raptor
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix output paths

Browse Source 
add gemm test
This commit is contained in:
NiccoloN
2026-02-25 17:24:31 +01:00
parent d036c02160
commit ae6e815c7b
14 changed files with 86 additions and 106 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
src/PIM/Pass/MessagePass.cpp
View File

@@ -1,4 +1,5 @@
#include "mlir/Pass/Pass.h"
#include "src/Compiler/CompilerUtils.hpp"
using namespace mlir;
@@ -7,21 +8,15 @@ namespace onnx_mlir {
namespace {
struct MessagePass : public PassWrapper<MessagePass, OperationPass<ModuleOp>> {
struct MessagePass : PassWrapper<MessagePass, OperationPass<ModuleOp>> {
MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(MessagePass)
StringRef getArgument() const override { return "message-pass"; }
StringRef getDescription() const override { return "Print compilation status"; }
StringRef getDescription() const override {
return "Lower ONNX ops to Spatial ops.";
}
MessagePass(std::string message)
: message(message) {}
MessagePass(const MessagePass& pass) {}
// Make sure that we have a valid default constructor and copy
// constructor to make sure that the options are initialized properly.
MessagePass(std::string message) : message(message) {}
MessagePass(const MessagePass &pass)
: PassWrapper<MessagePass, OperationPass<ModuleOp>>() {}
void runOnOperation() final { showCompilePhase(message); }
private:
@@ -30,8 +25,6 @@ private:
} // namespace
std::unique_ptr<Pass> createMessagePass(std::string message) {
return std::make_unique<MessagePass>(message);
}
std::unique_ptr<Pass> createMessagePass(std::string message) { return std::make_unique<MessagePass>(message); }
} // namespace onnx_mlir
} // namespace onnx_mlir