llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Disable LTC for out-of-tree.

breakup_python_pytorch_deps
Stella Laurenzo 2023-11-18 19:12:14 -08:00
parent 9ed175e352
commit a1cd142340
2 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
build_tools/python_deploy/build_linux_packages.sh
View File

@ -364,14 +364,16 @@ function setup_venv() {
function build_out_of_tree() { function build_out_of_tree() {
local torch_from_bin="$1" local torch_from_bin="$1"
local python_version="$2" local python_version="$2"
echo ":::: Build out-of-tree Torch from binary: $torch_from_bin with Python: $python_version"
local torch_version="$3" local torch_version="$3"
local enable_ltc="ON" echo ":::: Build out-of-tree Torch from binary: $torch_from_bin with Python: $python_version ($torch_version)"
if [[ "${torch_version}" == "stable" ]]
then # Disabled due to link errors on undefined c10 and other core torch symbols.
enable_ltc="OFF" enable_ltc="OFF"
fi # local enable_ltc="ON"
# if [[ "${torch_version}" == "stable" ]]
# then
# enable_ltc="OFF"
# fi
if [ ! -d "/main_checkout/torch-mlir/llvm-build/lib/cmake/mlir/" ] if [ ! -d "/main_checkout/torch-mlir/llvm-build/lib/cmake/mlir/" ]
then then

3
projects/ltc/csrc/base_lazy_backend/CMakeLists.txt
View File

@ -71,9 +71,6 @@ target_link_libraries(torch_mlir_ltc_backend
TorchMLIRAggregateCAPI TorchMLIRAggregateCAPI
TorchMLIRJITIRImporter TorchMLIRJITIRImporter
${TORCH_LIBRARIES} ${TORCH_LIBRARIES}
# In some configurations, symbols are not found for inlines, so over-link.
torch_cpu
c10
) )
message(STATUS "TORCH_CXXFLAGS=${TORCH_CXXFLAGS} -Wno-pedantic") message(STATUS "TORCH_CXXFLAGS=${TORCH_CXXFLAGS} -Wno-pedantic")