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

View File

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

View File

@ -71,9 +71,6 @@ target_link_libraries(torch_mlir_ltc_backend
TorchMLIRAggregateCAPI
TorchMLIRJITIRImporter
${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")