#sdy Remove unused op parameter.

Pure refactoring.

PiperOrigin-RevId: 737780450

Author: ZixuanJiang

shardy/dialect/sdy/transforms/propagation/aggressive_factor_propagation.cc

@@ -156,7 +156,7 @@ UpdateTensorShardings AggressiveFactorPropagation::propagateFactorShardings(
   bool allElementsAreEmpty = true;
   for (int64_t i = 0; i < factorSizes.size(); ++i) {
     SmallVector<AxisRefAttr>& axes = axesPerFactor.emplace_back(
-        getCompatibleMajorAxes(projection, i, directionAlongFactor(i), op));
+        getCompatibleMajorAxes(projection, i, directionAlongFactor(i)));
     if (!axes.empty()) {
       allElementsAreEmpty = false;
     }

shardy/dialect/sdy/transforms/propagation/basic_factor_propagation.cc

@@ -288,7 +288,7 @@ std::pair<SmallVector<AxisRefAttr>, bool> getCompatibleMajorAxesInternal(
 
 SmallVector<AxisRefAttr> BasicFactorPropagation::getCompatibleMajorAxes(
     const ShardingProjection& projection, int64_t factorIndex,
-    PropagationDirection direction, Operation* op) const {
+    PropagationDirection direction) const {
   if (direction == PropagationDirection::NONE) {
     return {};
   }
@@ -370,10 +370,10 @@ std::optional<AxisRefAttr> BasicFactorPropagation::compatiblePrefix(
 SmallVector<AxisRefAttr> BasicFactorPropagation::getCompatibleMajorShardingAxes(
     const ShardingProjection& projection, int64_t factorIndex,
     PropagationDirection direction, int64_t factorSize, MeshAttr mesh,
-    Operation* op, bool conservativePropagation) const {
+    bool conservativePropagation) const {
   // Finds the compatible major axes ignoring conflicts.
   SmallVector<AxisRefAttr> resultAxes =
-      getCompatibleMajorAxes(projection, factorIndex, direction, op);
+      getCompatibleMajorAxes(projection, factorIndex, direction);
 
   // Removes the major-most axis that isn't compatible w.r.t. other factors or
   // the replicated axes, and all axes that are minor to it.
@@ -391,7 +391,7 @@ SmallVector<AxisRefAttr> BasicFactorPropagation::getCompatibleMajorShardingAxes(
 UpdateTensorShardings BasicFactorPropagation::propagateFactorShardings(
     ShardingProjection& projection,
     PropagationDirectionAlongFactor directionAlongFactor,
-    ArrayRef<int64_t> factorSizes, MeshAttr mesh, Operation* op,
+    ArrayRef<int64_t> factorSizes, MeshAttr mesh, Operation*,
     bool conservativePropagation) const {
   UpdateTensorShardings result(projection.getNumOperands(),
                                projection.getNumResults());
@@ -403,7 +403,7 @@ UpdateTensorShardings BasicFactorPropagation::propagateFactorShardings(
     // tensors that aren't already sharded.
     SmallVector<AxisRefAttr> axesToPropagate = getCompatibleMajorShardingAxes(
         projection, factorIndex, directionAlongFactor(factorIndex), factorSize,
-        mesh, op, conservativePropagation);
+        mesh, conservativePropagation);
 
     // Update all shardings along this factor if possible.
     auto [updateOperandForFactor, updateResultForFactor] =

shardy/dialect/sdy/transforms/propagation/basic_factor_propagation.h

@@ -44,7 +44,7 @@ class BasicFactorPropagation : public FactorPropagation {
   UpdateTensorShardings propagateFactorShardings(
       ShardingProjection& projection,
       PropagationDirectionAlongFactor directionAlongFactor,
-      ArrayRef<int64_t> factorSizes, MeshAttr mesh, Operation* op,
+      ArrayRef<int64_t> factorSizes, MeshAttr mesh, Operation*,
       bool conservativePropagation) const override;
 
  protected:
@@ -88,7 +88,7 @@ class BasicFactorPropagation : public FactorPropagation {
   SmallVector<AxisRefAttr> getCompatibleMajorShardingAxes(
       const ShardingProjection& projection, int64_t factorIndex,
       PropagationDirection direction, int64_t factorSize, MeshAttr mesh,
-      Operation* op, bool conservativePropagation) const;
+      bool conservativePropagation) const;
 
   // Finds the longest prefix of axes that shard the given factor, such that all
   // tensors either:
@@ -99,7 +99,7 @@ class BasicFactorPropagation : public FactorPropagation {
   // This method does not resolve conflicts across factors or replicated axes.
   SmallVector<AxisRefAttr> getCompatibleMajorAxes(
       const ShardingProjection& projection, int64_t factorIndex,
-      PropagationDirection direction, Operation* op) const;
+      PropagationDirection direction) const;
 
   // Returns the largest prefix of `axisRef`, which does not overlap with
   // sharding axes and overflow axes for all other factors.