finally fast googlenet with correct latency artifacts for fair comparison

2026-07-29 18:20:44 +02:00
parent 060a21172e
commit 1b4f070bef
74 changed files with 2773 additions and 1311 deletions
@@ -2,10 +2,14 @@

 #include "mlir/Pass/Pass.h"

+#include <cstddef>
 #include <memory>
 #include <string>

 namespace onnx_mlir {
+namespace spatial {
+struct SchedulingTarget;
+}

 std::unique_ptr<mlir::Pass> createONNXToSpatialPass();
 std::unique_ptr<mlir::Pass> createSpatialLayoutPlanningPass();
@@ -16,8 +20,11 @@ std::unique_ptr<mlir::Pass> createSpatialToPimPass();
 std::unique_ptr<mlir::Pass> createPimBufferizationPass();

 std::unique_ptr<mlir::Pass> createMergeComputeNodesPass();
+std::unique_ptr<mlir::Pass> createMergeComputeNodesPass(const spatial::SchedulingTarget& target);

 std::unique_ptr<mlir::Pass> createTrivialGraphComputeMergePass();
+std::unique_ptr<mlir::Pass> createTrivialGraphComputeMergePass(
+  size_t residentWeightCapacity);

 std::unique_ptr<mlir::Pass> createPimHostConstantFoldingPass();