[MHLO] Init MHLO pooling-like op conversion (#1141)

* [MHLO] Init MHLO pooling-like op conversion and remove 'op' suffix in filenames

Co-authored-by: Bairen Yi <[email protected]>
Co-authored-by: Jiawei Wu <[email protected]>
Co-authored-by: Tianyou Guo [email protected]
Co-authored-by: Xu Yan <[email protected]>
Co-authored-by: Ziheng Jiang <[email protected]>

See RFC #999

Author: Vremold

lib/Conversion/TorchToMhlo/BasicOp.cpp renamed to lib/Conversion/TorchToMhlo/Basic.cpp

@@ -22,7 +22,6 @@
 #include "torch-mlir/Dialect/Torch/Utils/TorchUpstream.h"
 #include "torch-mlir/Dialect/Torch/Utils/Utils.h"
 #include "torch-mlir/Dialect/TorchConversion/IR/TorchConversionOps.h"
-#include "mlir-hlo/Dialect/mhlo/IR/chlo_ops.h"
 #include <iostream>
 #include <numeric>
 
@@ -618,9 +617,8 @@ LogicalResult ConvertAtenOp<AtenReluOp>::matchAndRewrite(
 namespace {
 template <>
 LogicalResult ConvertAtenOp<AtenGeluOp>::matchAndRewrite(
-    AtenGeluOp op,
-    OpAdaptor adaptor,
-    ConversionPatternRewriter& rewriter) const {
+    AtenGeluOp op, OpAdaptor adaptor,
+    ConversionPatternRewriter &rewriter) const {
   Location loc = op.getLoc();
   Value input = adaptor.self();
   auto inputTy = input.getType().template dyn_cast<RankedTensorType>();
@@ -641,7 +639,6 @@ LogicalResult ConvertAtenOp<AtenGeluOp>::matchAndRewrite(
 }
 } // namespace
 
-
 // AtenErfOp
 namespace {
 template <>

lib/Conversion/TorchToMhlo/CMakeLists.txt

@@ -1,11 +1,12 @@
 add_mlir_conversion_library(TorchMLIRTorchToMhlo
   TorchToMhlo.cpp
   MhloLegalizeUtils.cpp
-  BasicOp.cpp
-  GatherOp.cpp
+  Basic.cpp
+  Gather.cpp
   Linear.cpp
-  ViewLikeOps.cpp
-  ReductionOp.cpp
+  ViewLike.cpp
+  Reduction.cpp
+  Pooling.cpp
 
   ADDITIONAL_HEADER_DIRS
   ${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/TorchToMhlo