Try to make JITIRImporter lib SHARED

llvm · Aug 19, 2022 · f544d57 · f544d57
1 parent 5fa7151
commit f544d57
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/CMakeLists.txt b/python/torch_mlir/dialects/torch/importer/jit_ir/csrc/CMakeLists.txt
@@ -11,7 +11,7 @@ include_directories(BEFORE
 link_directories("${TORCH_INSTALL_PREFIX}/lib")
 
 # Static library with core functionality.
-add_library(TorchMLIRJITIRImporter STATIC
+add_library(TorchMLIRJITIRImporter SHARED
   class_annotator.cpp
   function_importer.cpp
   node_importer.cpp
@@ -27,9 +27,10 @@ set_target_properties(TorchMLIRJITIRImporter PROPERTIES
   LIBRARY_OUTPUT_DIRECTORY "${TORCH_MLIR_PYTHON_PACKAGES_DIR}/torch_mlir/torch_mlir/_mlir_libs"
   OUTPUT_NAME lib_jit_ir_importer
   PREFIX ""
-  SUFFIX ".a"
+  SUFFIX ".so"
   CXX_VISIBILITY_PRESET "default"
   COMPILE_FLAGS "${TORCH_CXXFLAGS}"
+  LINK_FLAGS "-rdynamic"
   )
 
 # Separate Pybind MODULE due to issues with a SHARED library.