fix bazel build and delete 'Op' suffix in filename

llvm · Aug 4, 2022 · 746fe65 · 746fe65
1 parent 144c4e9
commit 746fe65
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 1 deletion.
diff --git 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 → lib/Conversion/TorchToMhlo/Pooling.cpp b/lib/Conversion/TorchToMhlo/PoolingOp.cpp → lib/Conversion/TorchToMhlo/Pooling.cpp
diff --git 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",