diff --git a/lib/Conversion/TorchToMhlo/CMakeLists.txt b/lib/Conversion/TorchToMhlo/CMakeLists.txt index f4e88ea2898..6350631520c 100644 --- a/lib/Conversion/TorchToMhlo/CMakeLists.txt +++ b/lib/Conversion/TorchToMhlo/CMakeLists.txt @@ -6,7 +6,7 @@ add_mlir_conversion_library(TorchMLIRTorchToMhlo Linear.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 d28545e951c..65639628d0f 100644 --- a/utils/bazel/torch-mlir-overlay/BUILD.bazel +++ b/utils/bazel/torch-mlir-overlay/BUILD.bazel @@ -453,6 +453,7 @@ cc_library( "lib/Conversion/TorchToMhlo/Linear.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",