Dynamic gemm/conv

2026-05-28 18:00:14 +02:00
parent cbf7b235f1
commit 1ab489fe0a
17 changed files with 704 additions and 69 deletions
@@ -1,5 +1,6 @@
 #pragma once

+#include "mlir/Dialect/Affine/IR/AffineOps.h"
 #include "mlir/Dialect/Arith/IR/Arith.h"
 #include "mlir/IR/PatternMatch.h"
 #include "mlir/IR/Value.h"
@@ -29,4 +30,10 @@ mlir::Value getOrCreateHostI32Constant(mlir::Operation* anchorOp, int32_t value,

 mlir::Value getOrCreateHostI64Constant(mlir::Operation* anchorOp, int64_t value, mlir::OperationFolder& folder);

+mlir::Value createAffineApplyOrFoldedConstant(mlir::RewriterBase& rewriter,
+                                              mlir::Location loc,
+                                              mlir::AffineMap map,
+                                              mlir::ValueRange operands,
+                                              mlir::Operation* anchorOp);
+
 } // namespace onnx_mlir