Revert "Implement reduction avg op end to end (#271)" (#282)

This reverts commit 1ddbd0e due to build break.

include/ttmlir/Dialect/TTIR/IR/TTIROps.td

@@ -217,13 +217,6 @@ def TTIR_SumOp : TTIR_ReductionOp<"sum"> {
     }];
 }
 
-def TTIR_AvgOp : TTIR_ReductionOp<"avg"> {
-  let summary = "Average reduction op.";
-  let description = [{
-    Average reduction op.
-  }];
-}
-
 def TTIR_SoftmaxOp : TTIR_DPSOp<"softmax"> {
     let summary = "Softmax operation.";
     let description = [{

include/ttmlir/Dialect/TTNN/IR/TTNNOps.td

@@ -124,13 +124,6 @@ def TTNN_SumOp : TTNN_ReductionOp<"sum"> {
     }];
 }
 
-def TTNN_AvgOp : TTNN_ReductionOp<"avg"> {
-  let summary = "Average reduction op.";
-  let description = [{
-    Average reduction op.
-  }];
-}
-
 def TTNN_ReluOp : TTNN_ElementwiseOp<"relu"> {
     let summary = "Eltwise ReLU.";
     let description = [{

include/ttmlir/Target/TTNN/program.fbs

@@ -40,7 +40,6 @@ table EltwiseOp {
 
 enum ReductionOpType: uint32 {
   Sum = 0,
-  Avg = 1,
 }
 
 table ReductionOp {

lib/Conversion/TTIRToTTNN/TTIRToTTNN.cpp

@@ -144,7 +144,6 @@ void populateTTIRToTTNNPatterns(MLIRContext *ctx, RewritePatternSet &patterns,
            ElementwiseBinaryOpConversionPattern<ttir::GreaterEqualOp, ttnn::GreaterEqualOp>,
            ElementwiseBinaryOpConversionPattern<ttir::ReluOp, ttnn::ReluOp>,
            ReductionOpConversionPattern<ttir::SumOp, ttnn::SumOp>,
-           ReductionOpConversionPattern<ttir::AvgOp, ttnn::AvgOp>,
            SoftmaxOpConversionPattern,
            MatmulOpConversionPattern
            >(typeConverter, ctx);

lib/Conversion/TTNNToEmitC/TTNNToEmitC.cpp

@@ -119,7 +119,6 @@ void populateTTNNToEmitCPatterns(mlir::MLIRContext *ctx,
   // Reduction ops
   //
   patterns.add<DefaultOpConversionPattern<ttnn::SumOp>>(typeConverter, ctx);
-  patterns.add<DefaultOpConversionPattern<ttnn::AvgOp>>(typeConverter, ctx);
 }
 
 } // namespace mlir::tt

lib/Dialect/TTIR/Transforms/Passes.cpp

@@ -534,9 +534,7 @@ class TTIRLayout : public impl::TTIRLayoutBase<TTIRLayout> {
           TTIRLayoutOperandsRewriter<MultiplyOp>,
           TTIRLayoutOperandsRewriter<SubtractOp>,
           TTIRLayoutOperandsRewriter<GreaterEqualOp>,
-          TTIRLayoutOperandsRewriter<ReluOp>,
-          TTIRLayoutOperandsRewriter<SumOp>,
-          TTIRLayoutOperandsRewriter<AvgOp>,
+          TTIRLayoutOperandsRewriter<ReluOp>, TTIRLayoutOperandsRewriter<SumOp>,
           TTIRLayoutOperandsRewriter<SoftmaxOp>,
           TTIRLayoutOperandsRewriter<MatmulOp>, TTIRLayoutFuncReturnRewriter>(
           &getContext());

lib/Target/TTNN/TTNNToFlatbuffer.cpp

@@ -142,8 +142,6 @@ createReductionOp(FlatbufferObjectCache &cache, ReductionOp op) {
   ::tt::target::ttnn::ReductionOpType type;
   if constexpr (std::is_same_v<ReductionOp, SumOp>) {
     type = ::tt::target::ttnn::ReductionOpType::Sum;
-  } else if constexpr (std::is_same_v<ReductionOp, AvgOp>) {
-    type = ::tt::target::ttnn::ReductionOpType::Avg;
   } else {
     llvm_unreachable("unhandled ReductionOp");
   }
@@ -211,9 +209,6 @@ emitTTNNOperation(FlatbufferObjectCache &cache, Operation *op,
   if (auto sumOp = dyn_cast<SumOp>(op); sumOp) {
     return createOperation(cache, createReductionOp(cache, sumOp), debugString);
   }
-  if (auto avgOp = dyn_cast<AvgOp>(op); avgOp) {
-    return createOperation(cache, createReductionOp(cache, avgOp), debugString);
-  }
   if (auto softmaxOp = dyn_cast<SoftmaxOp>(op); softmaxOp) {
     return createOperation(cache, createSoftmaxOp(cache, softmaxOp),
                            debugString);

runtime/lib/ttnn/program.cpp

@@ -138,20 +138,6 @@ run(::tt::target::ttnn::ReductionOp const *op, ::ttnn::Device &device,
     liveTensors.try_emplace(op->out()->global_id(), &tensorPool.back());
     break;
   }
-  case ::tt::target::ttnn::ReductionOpType::Avg: {
-    auto &in = *liveTensors.at(op->in()->global_id());
-
-    const auto *dim_arg_fb_ptr = op->dim_arg();
-    std::optional<vector<int>> dim_arg =
-        dim_arg_fb_ptr ? std::make_optional(std::vector<int>(
-                             dim_arg_fb_ptr->begin(), dim_arg_fb_ptr->end()))
-                       : std::nullopt;
-
-    tensorPool.push_back(::ttnn::avg(in, dim_arg, op->keep_dim()));
-
-    liveTensors.try_emplace(op->out()->global_id(), &tensorPool.back());
-    break;
-  }
   }
 }