Merge branch 'main' into xgniu/constant_weights_folding

include/gc/Analysis/DataFlow/ConstantSubgraphAnalyser.h

@@ -93,7 +93,7 @@ class ConstantSubgraphAnalyser
 public:
   using SparseForwardDataFlowAnalysis::SparseForwardDataFlowAnalysis;
 
-  void visitOperation(Operation *op,
+  LogicalResult visitOperation(Operation *op,
                       ArrayRef<const Lattice<IsConstantTensor> *> operands,
                       ArrayRef<Lattice<IsConstantTensor> *> results) override;
 

include/gc/Transforms/Passes.td

@@ -145,4 +145,29 @@ def ConstantTensorFolding : Pass<"constant-tensor-folding"> {
       "LLVM::LLVMDialect"];
 }
 
+def FoldTensorOperation : Pass<"fold-tensor-operation"> {
+  let summary = "Fold some tensor operation";
+  let description = [{
+    Remove some useless tensor operations.
+  }];
+  let dependentDialects = [
+    "::mlir::tensor::TensorDialect"
+  ];
+}
+
+def LowerToTileVector : Pass<"lower-to-tile-vector", "func::FuncOp"> {
+  let summary = "Lower tensor to tile (virtual) vector";
+  let description = [{
+    Lower operation operate on tensor to vector operation.
+  }];
+  let dependentDialects = [
+    "::mlir::func::FuncDialect",
+    "::mlir::math::MathDialect",
+    "::mlir::arith::ArithDialect",
+    "::mlir::tensor::TensorDialect",
+    "::mlir::linalg::LinalgDialect",
+    "::mlir::vector::VectorDialect",
+  ];
+}
+
 #endif // GC_DIALECT_GC_PASSES

lib/gc/Analysis/DataFlow/ConstantSubgraphAnalyser.cpp

@@ -50,7 +50,7 @@ void IsConstantTensor::print(raw_ostream &os) const {
 // ConstantSubgraphAnalyser
 //===----------------------------------------------------------------------===//
 
-void ConstantSubgraphAnalyser::visitOperation(
+LogicalResult ConstantSubgraphAnalyser::visitOperation(
     Operation *op, ArrayRef<const Lattice<IsConstantTensor> *> operands,
     ArrayRef<Lattice<IsConstantTensor> *> results) {
   LLVM_DEBUG(llvm::dbgs() << "ConstantSubgraphAnalyser: Visiting operation:\n"
@@ -89,6 +89,7 @@ void ConstantSubgraphAnalyser::visitOperation(
       propagateIfChanged(lattice, lattice->join(IsConstantTensor(true, true)));
     }
   }
+  return LogicalResult::success();
 }
 
 void ConstantSubgraphAnalyser::setToEntryState(

lib/gc/Transforms/CMakeLists.txt

@@ -18,7 +18,8 @@ gc_add_mlir_library(GcPasses
   VerifyTargetDescription.cpp
   ConstantSubgraphAnalysis.cpp
   ConstantTensorFolding.cpp
-  DeepTileContractionNamedOp.cpp
+  DecomposeAggregatedOps.cpp
+  DeepTileContractionOp.cpp
   TilingUtil.cpp
   SinkOpIntoInnerLoop.cpp
   MergeNestedForall.cpp