From 746fe65ae1e0e810d655521db144b461e297dcc0 Mon Sep 17 00:00:00 2001
From: Vremold <xremold@gmail.com>
Date: Thu, 4 Aug 2022 10:14:26 +0800
Subject: [PATCH] fix bazel build and delete 'Op' suffix in filename

---
 lib/Conversion/TorchToMhlo/CMakeLists.txt                 | 2 +-
 lib/Conversion/TorchToMhlo/{PoolingOp.cpp => Pooling.cpp} | 0
 utils/bazel/torch-mlir-overlay/BUILD.bazel                | 1 +
 3 files changed, 2 insertions(+), 1 deletion(-)
 rename lib/Conversion/TorchToMhlo/{PoolingOp.cpp => Pooling.cpp} (100%)

diff --git a/lib/Conversion/TorchToMhlo/CMakeLists.txt b/lib/Conversion/TorchToMhlo/CMakeLists.txt
index c1fdef423ba7..33c36e84b9b3 100644
--- a/lib/Conversion/TorchToMhlo/CMakeLists.txt
+++ b/lib/Conversion/TorchToMhlo/CMakeLists.txt
@@ -5,7 +5,7 @@ add_mlir_conversion_library(TorchMLIRTorchToMhlo
   GatherOp.cpp
   ViewLikeOps.cpp
   ReductionOp.cpp
-  PoolingOp.cpp
+  Pooling.cpp
 
   ADDITIONAL_HEADER_DIRS
   ${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/TorchToMhlo
diff --git a/lib/Conversion/TorchToMhlo/PoolingOp.cpp b/lib/Conversion/TorchToMhlo/Pooling.cpp
similarity index 100%
rename from lib/Conversion/TorchToMhlo/PoolingOp.cpp
rename to lib/Conversion/TorchToMhlo/Pooling.cpp
diff --git a/utils/bazel/torch-mlir-overlay/BUILD.bazel b/utils/bazel/torch-mlir-overlay/BUILD.bazel
index 2ce5e9b7d9f6..47cac7fa5d9e 100644
--- a/utils/bazel/torch-mlir-overlay/BUILD.bazel
+++ b/utils/bazel/torch-mlir-overlay/BUILD.bazel
@@ -452,6 +452,7 @@ cc_library(
         "lib/Conversion/TorchToMhlo/GatherOp.cpp",
         "lib/Conversion/TorchToMhlo/ViewLikeOps.cpp",
         "lib/Conversion/TorchToMhlo/ReductionOp.cpp",
+        "lib/Conversion/TorchToMhlo/Pooling.cpp",
         "lib/Conversion/TorchToMhlo/MhloLegalizeUtils.h",
         "lib/Conversion/TorchToMhlo/PopulatePatterns.h",
         "lib/Conversion/PassDetail.h",