Revert "[MLIR][Mesh] Add sharding propagation pass (#69665)"

This reverts commit 9d9400d.
This reverts commit bda763a.

The buildbot is broken and tests are failing.

Author: joker-eph

mlir/include/mlir/Dialect/Mesh/CMakeLists.txt

@@ -1,3 +1 @@
-add_subdirectory(Interfaces)
 add_subdirectory(IR)
-add_subdirectory(Transforms)

mlir/include/mlir/Dialect/Mesh/IR/MeshBase.td

@@ -49,22 +49,6 @@ def Mesh_Partial : I32EnumAttr<"Partial", "partial type of a distributed tensor"
   let cppNamespace = "::mlir::mesh";
 }
 
-// Mesh_IteratorType and Mesh_Partial are used to annotate different aspects of
-// distributed tensors. Mesh_IteratorType annotates loops in an operation, while
-// Mesh_Partial indicates whether a tensor is sharded on a specific dimension or
-// is partial.
-def Mesh_IteratorType : I32EnumAttr<"IteratorType", "Iterator type", [
-  I32EnumAttrCase<"Parallel", 1, "parallel">,
-  I32EnumAttrCase<"ReductionSum", 2, "reduction_sum">,
-  I32EnumAttrCase<"ReductionMax", 3, "reduction_max">,
-  I32EnumAttrCase<"ReductionMin", 4, "reduction_min">,
-  I32EnumAttrCase<"ReductionGeneric", 5, "reduction_generic">,
-  I32EnumAttrCase<"Invalid", 100, "invalid">
-]> {
-    let genSpecializedAttr = 0;
-    let cppNamespace = "::mlir::mesh";
-}
-
 //===----------------------------------------------------------------------===//
 // Mesh Attribute
 //===----------------------------------------------------------------------===//
@@ -138,24 +122,6 @@ def MeshSharding : AttrDef<Mesh_Dialect, "MeshSharding"> {
        $partial_axes^ `]`)? `>`
   }];
 
-  let builders = [
-    AttrBuilder<(ins "SymbolRefAttr":$cluster, 
-                     "ArrayRef<SmallVector<int32_t>>":$split_axes,
-                     "ArrayRef<int32_t>": $partial_axes,
-                     "mesh::Partial": $partial_type), [{
-      SmallVector<DenseI32ArrayAttr> splitAxesAttr = llvm::map_to_vector(
-                  split_axes, [&](ArrayRef<int32_t> array) {
-          return DenseI32ArrayAttr::get($_ctxt, array);
-      });
-      return $_get($_ctxt, cluster, splitAxesAttr, partial_axes,
-                   partial_type);
-    }]>,
-    AttrBuilder<(ins "SymbolRefAttr":$cluster, 
-                     "ArrayRef<SmallVector<int32_t>>":$split_axes), [{
-      return MeshShardingAttr::get($_ctxt, cluster, split_axes, {}, Partial::Sum);
-    }]>
-  ];
-
   let genVerifyDecl = 1;
 }
 

mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h

@@ -24,22 +24,4 @@
 #define GET_OP_CLASSES
 #include "mlir/Dialect/Mesh/IR/MeshOps.h.inc"
 
-namespace mlir {
-namespace mesh {
-
-bool isReductionLoop(IteratorType iType);
-
-bool areReductionAndPartialMatch(IteratorType iType, Partial partial);
-
-template <typename T>
-void removeTrailingEmptySubArray(SmallVector<SmallVector<T>> &array) {
-  while (!array.empty() && array.back().empty())
-    array.pop_back();
-}
-
-Partial getPartialTypeFromReduction(IteratorType iType);
-
-} // namespace mesh
-} // namespace mlir
-
 #endif // MLIR_DIALECT_MESH_IR_MESHOPS_H

mlir/include/mlir/Dialect/Mesh/Interfaces/CMakeLists.txt

@@ -104,18 +104,6 @@ class AffineMap {
   static AffineMap getPermutationMap(ArrayRef<unsigned> permutation,
                                      MLIRContext *context);
 
-  /// Returns an affine map with `numDims` input dimensions and results
-  /// specified by `targets`.
-  ///
-  /// Examples:
-  /// * getMultiDimMapWithTargets(3, [0, 2, 1])
-  ///       -> affine_map<(d0, d1, d2) -> (d0, d2, d1)>
-  /// * getMultiDimMapWithTargets(3, [2, 1])
-  ///       -> affine_map<(d0, d1, d2) -> (d2, d1)>
-  static AffineMap getMultiDimMapWithTargets(unsigned numDims,
-                                             ArrayRef<unsigned> targets,
-                                             MLIRContext *context);
-
   /// Returns a vector of AffineMaps; each with as many results as
   /// `exprs.size()`, as many dims as the largest dim in `exprs` and as many
   /// symbols as the largest symbol in `exprs`.

mlir/include/mlir/Dialect/Mesh/Interfaces/ShardingInterface.h

@@ -79,7 +79,6 @@
 #include "mlir/Dialect/Tensor/TransformOps/TensorTransformOps.h"
 #include "mlir/Dialect/Tensor/Transforms/BufferizableOpInterfaceImpl.h"
 #include "mlir/Dialect/Tensor/Transforms/SubsetInsertionOpInterfaceImpl.h"
-#include "mlir/Dialect/Tosa/IR/ShardingInterfaceImpl.h"
 #include "mlir/Dialect/Tosa/IR/TosaOps.h"
 #include "mlir/Dialect/Transform/IR/TransformDialect.h"
 #include "mlir/Dialect/Transform/PDLExtension/PDLExtension.h"
@@ -172,7 +171,6 @@ inline void registerAllDialects(DialectRegistry &registry) {
   tensor::registerSubsetOpInterfaceExternalModels(registry);
   tensor::registerTilingInterfaceExternalModels(registry);
   tensor::registerValueBoundsOpInterfaceExternalModels(registry);
-  tosa::registerShardingInterfaceExternalModels(registry);
   vector::registerBufferizableOpInterfaceExternalModels(registry);
   vector::registerSubsetOpInterfaceExternalModels(registry);
   NVVM::registerNVVMTargetInterfaceExternalModels(registry);