conv now lowers correctly down to bufferized pim

2026-03-20 12:55:09 +01:00
parent 6e1de865bb
commit db3f52a647
4 changed files with 114 additions and 100 deletions
--- a/src/PIM/Pass/PimHostVerificationPass.cpp
+++ b/src/PIM/Pass/PimHostVerificationPass.cpp
@@ -1,3 +1,4 @@
+#include "mlir/Dialect/Arith/IR/Arith.h"
 #include "mlir/Dialect/Func/IR/FuncOps.h"
 #include "mlir/Dialect/MemRef/IR/MemRef.h"
 #include "mlir/Pass/Pass.h"
@@ -15,7 +16,8 @@ namespace onnx_mlir {
 namespace {

 static bool isAddressOnlyHostOp(Operation* op) {
-  return isa<memref::AllocOp,
+  return isa<arith::ConstantOp,
+             memref::AllocOp,
             memref::GetGlobalOp,
             memref::SubViewOp,
             memref::CastOp,