diff --git a/lib/Conversion/Utils/CMakeLists.txt b/lib/Conversion/Utils/CMakeLists.txt index f4baf8634..6b352bdc5 100644 --- a/lib/Conversion/Utils/CMakeLists.txt +++ b/lib/Conversion/Utils/CMakeLists.txt @@ -3,4 +3,9 @@ add_mlir_conversion_library(TorchMLIRConversionUtils ADDITIONAL_HEADER_DIRS ${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/Utils + + LINK_LIBS PUBLIC + MLIRArithmeticDialect + MLIRLinalgDialect + TorchMLIRTorchDialect )