faster pim host constant folding

2026-04-14 19:58:26 +02:00
parent 95ae93e07d
commit ae93d1c563
4 changed files with 94 additions and 41 deletions
--- a/src/PIM/Pass/PimCodegen/HostConstantFolding/Common.hpp
+++ b/src/PIM/Pass/PimCodegen/HostConstantFolding/Common.hpp
@@ -30,6 +30,10 @@ mlir::memref::GlobalOp createFoldedGlobal(mlir::ModuleOp moduleOp,
                                          llvm::StringRef nameStem,
                                          mlir::IntegerAttr alignment = {});

+llvm::FailureOr<mlir::DenseElementsAttr> foldDenseSubview(mlir::DenseElementsAttr denseAttr,
+                                                          llvm::ArrayRef<int64_t> staticOffsets,
+                                                          llvm::ArrayRef<int64_t> resultShape);
+
 llvm::FailureOr<mlir::DenseElementsAttr> getDenseGlobalValue(mlir::ModuleOp moduleOp, mlir::Value value);

 llvm::FailureOr<StaticSubviewInfo> getStaticSubviewInfo(mlir::Value value);