Minor fix.

llvm · Jul 20, 2022 · 0f7e490 · 0f7e490
1 parent 3f7dade
commit 0f7e490
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/include/torch-mlir/Conversion/TorchToMhlo/TorchToMhlo.h b/include/torch-mlir/Conversion/TorchToMhlo/TorchToMhlo.h
@@ -10,13 +10,11 @@
 #ifndef TORCHMLIR_CONVERSION_TORCHTOMHLO_TORCHTOMHLO_H
 #define TORCHMLIR_CONVERSION_TORCHTOMHLO_TORCHTOMHLO_H
 
+#include "mlir/Dialect/Func/IR/FuncOps.h"
 #include "mlir/Pass/Pass.h"
 #include <memory>
 
 namespace mlir {
-namespace func {
-class FuncOp;
-} // namespace func
 namespace torch {
 std::unique_ptr<OperationPass<func::FuncOp>> createConvertTorchToMhloPass();
 } // namespace torch

diff --git a/lib/Dialect/TorchConversion/Transforms/Passes.cpp b/lib/Dialect/TorchConversion/Transforms/Passes.cpp
@@ -20,7 +20,9 @@
 #include "torch-mlir/Conversion/TorchToStd/TorchToStd.h"
 #include "torch-mlir/Conversion/TorchToTMTensor/TorchToTMTensor.h"
 #include "torch-mlir/Conversion/TorchToTosa/TorchToTosa.h"
+#ifdef TORCH_MLIR_ENABLE_MHLO
 #include "torch-mlir/Conversion/TorchToMhlo/TorchToMhlo.h"
+#endif
 #include "torch-mlir/Dialect/Torch/Transforms/Passes.h"
 
 using namespace mlir;