mirror of https://github.com/llvm/torch-mlir
This reverts commit f85ae9c685
.
pull/1187/head
parent
351f15424e
commit
3e97a33c80
|
@ -87,6 +87,7 @@ jobs:
|
|||
externals/llvm-project/llvm
|
||||
cmake --build llvm-build
|
||||
|
||||
# TODO: Reenable LTC once OOT build is successful (https://github.com/llvm/torch-mlir/issues/1154)
|
||||
cmake -GNinja -Bbuild \
|
||||
-DCMAKE_LINKER=lld \
|
||||
-DCMAKE_C_COMPILER_LAUNCHER=ccache -DCMAKE_CXX_COMPILER_LAUNCHER=ccache \
|
||||
|
@ -96,6 +97,7 @@ jobs:
|
|||
-DMLIR_ENABLE_BINDINGS_PYTHON=OFF \
|
||||
-DTORCH_MLIR_ENABLE_MHLO=ON \
|
||||
-DTORCH_MLIR_USE_INSTALLED_PYTORCH=${{ matrix.torch-binary }} \
|
||||
-DTORCH_MLIR_ENABLE_LTC=OFF \
|
||||
-DPython3_EXECUTABLE=$(which python) \
|
||||
.
|
||||
|
||||
|
|
|
@ -10,7 +10,15 @@ include_directories(BEFORE
|
|||
)
|
||||
link_directories("${TORCH_INSTALL_PREFIX}/lib")
|
||||
|
||||
add_library(TorchMLIRJITIRImporter SHARED
|
||||
# TODO: Currently, out-of-tree build fails when LIBRARY_TYPE is set to SHARED, so we have this toggle.
|
||||
# See https://github.com/llvm/torch-mlir/issues/1154 for more details.
|
||||
if(TORCH_MLIR_ENABLE_LTC)
|
||||
set(LIBRARY_TYPE "SHARED")
|
||||
else()
|
||||
set(LIBRARY_TYPE "MODULE")
|
||||
endif()
|
||||
|
||||
add_library(TorchMLIRJITIRImporter ${LIBRARY_TYPE}
|
||||
class_annotator.cpp
|
||||
class_annotator_pybind.cpp
|
||||
get_registered_ops.cpp
|
||||
|
|
2
setup.py
2
setup.py
|
@ -76,6 +76,8 @@ class CMakeBuild(build_py):
|
|||
f"-DMLIR_ENABLE_BINDINGS_PYTHON=ON",
|
||||
f"-DLLVM_ENABLE_PROJECTS=mlir",
|
||||
f"-DTORCH_MLIR_ENABLE_MHLO=ON",
|
||||
# TODO: Reenable LTC once JIT importer linkage issue is fixed (https://github.com/llvm/torch-mlir/issues/1154)
|
||||
f"-DTORCH_MLIR_ENABLE_LTC=OFF",
|
||||
f"-DLLVM_EXTERNAL_PROJECTS=torch-mlir;torch-mlir-dialects",
|
||||
f"-DLLVM_EXTERNAL_TORCH_MLIR_SOURCE_DIR={src_dir}",
|
||||
f"-DLLVM_EXTERNAL_TORCH_MLIR_DIALECTS_SOURCE_DIR={src_dir}/externals/llvm-external-projects/torch-mlir-dialects",
|
||||
|
|
Loading…
Reference in New Issue