guard with macro

Author: TanyoKwok

include/torch-mlir/Dialect/TorchConversion/Transforms/Passes.td

@@ -42,8 +42,10 @@ def VerifyTosaBackendContract : Pass<"torch-verify-tosa-backend-contract", "Modu
   let constructor = "mlir::torch::TorchConversion::createVerifyTosaBackendContractPass()";
 }
 
+#ifdef TORCH_MLIR_ENABLE_MHLO
 def VerifyMhloBackendContract : Pass<"torch-verify-mhlo-backend-contract", "ModuleOp"> {
   let summary = "Verifies conformity to the mhlo backend contract";
   let constructor = "mlir::torch::TorchConversion::createVerifyMhloBackendContractPass()";
 }
+#endif // TORCH_MLIR_ENABLE_MHLO
 #endif // TORCHMLIR_TORCHCONVERSION_PASSES

lib/Dialect/TorchConversion/Transforms/CMakeLists.txt

@@ -12,7 +12,6 @@ set(LinkedLibs  MLIRIR
 
 if(TORCH_MLIR_ENABLE_MHLO)
   list(APPEND LinkedLibs ChloPasses)
-  list(APPEND MhloContract VerifyMhloBackendContract.cpp)
 endif()
 
 add_mlir_library(TorchMLIRTorchConversionPasses
@@ -21,8 +20,7 @@ add_mlir_library(TorchMLIRTorchConversionPasses
   Passes.cpp
   VerifyLinalgOnTensorsBackendContract.cpp
   VerifyTosaBackendContract.cpp
-  ${MhloContract}
-
+  VerifyMhloBackendContract.cpp
 
   ADDITIONAL_HEADER_DIRS
   ${PROJECT_SOURCE_DIR}/include/torch-mlir/Dialect/TorchConversion/Transforms

lib/Dialect/TorchConversion/Transforms/VerifyMhloBackendContract.cpp

@@ -6,7 +6,7 @@
 // Also available under a BSD-style license. See LICENSE.
 //
 //===----------------------------------------------------------------------===//
-
+#ifdef TORCH_MLIR_ENABLE_MHLO
 #include "PassDetail.h"
 
 #include "mlir-hlo/Dialect/mhlo/IR/hlo_ops.h"
@@ -70,3 +70,4 @@ std::unique_ptr<OperationPass<ModuleOp>>
 mlir::torch::TorchConversion::createVerifyMhloBackendContractPass() {
   return std::make_unique<VerifyMhloBackendContractPass>();
 }
+#endif // TORCH_MLIR_ENABLE_MHLO

utils/bazel/torch-mlir-overlay/BUILD.bazel

@@ -314,7 +314,10 @@ gentbl_cc_library(
     strip_include_prefix = "include",
     tbl_outs = [
         (
-            ["-gen-pass-decls"],
+            [
+                "-gen-pass-decls",
+                "-DTORCH_MLIR_ENABLE_MHLO",
+            ],
             "include/torch-mlir/Dialect/TorchConversion/Transforms/Passes.h.inc",
         ),
     ],
@@ -502,6 +505,7 @@ cc_library(
         "lib/Dialect/TorchConversion/Transforms/Passes.cpp",
         "lib/Dialect/TorchConversion/Transforms/VerifyLinalgOnTensorsBackendContract.cpp",
         "lib/Dialect/TorchConversion/Transforms/VerifyTosaBackendContract.cpp",
+        "lib/Dialect/TorchConversion/Transforms/VerifyMhloBackendContract.cpp",
     ],
     hdrs = [
         "include/torch-mlir/Dialect/TorchConversion/Transforms/Passes.h",