Transpose and Refactor of Patterns

2026-05-29 13:23:31 +02:00
parent 82b44a6387
commit 832bd7f1f7
37 changed files with 285 additions and 153 deletions
@@ -3,15 +3,17 @@ mlir_tablegen(SpatialToPim.hpp.inc -gen-rewriters "-I${ONNX_MLIR_SRC_ROOT}")
 add_public_tablegen_target(SpatialToPimIncGen)

 add_pim_library(OMSpatialToPim
+  Patterns.cpp
  SpatialToPimPass.cpp
  BatchCoreLoweringPatterns.cpp
-  ChannelLoweringPatterns.cpp
  Common.cpp
  ComputeLikeRegionUtils.cpp
  CoreLoweringPatterns.cpp
-  GlobalTensorMaterialization.cpp
  ReturnPathNormalization.cpp
-  TensorPackingPatterns.cpp
+  Patterns/ChannelLowering.cpp
+  Patterns/GlobalTensorMaterialization.cpp
+  Patterns/TensorPacking.cpp
+  Patterns/Transpose.cpp

  EXCLUDE_FROM_OM_LIBS

@@ -19,6 +21,7 @@ add_pim_library(OMSpatialToPim
  SpatialToPimIncGen

  LINK_LIBS PUBLIC
+  MLIRLinalgDialect
  MLIRSCFDialect
  MLIRSCFUtils
  MLIRTransformUtils