Disable pytorch build from source for out of tree.

breakup_python_pytorch_deps
Stella Laurenzo 2023-11-18 19:43:35 -08:00
parent a1cd142340
commit 947715b68f
3 changed files with 8 additions and 9 deletions

View File

@ -33,9 +33,9 @@ jobs:
# Exclude llvm in-tree and pytorch source
- llvm-build: in-tree
torch-binary: OFF
# Exclude llvm out-of-tree and pytorch binary
# Exclude llvm out-of-tree and pytorch source
- llvm-build: out-of-tree
torch-binary: ON
torch-binary: OFF
# Exclude macos-arm64 and llvm out-of-tree altogether
- os-arch: macos-arm64
llvm-build: out-of-tree

View File

@ -367,13 +367,11 @@ function build_out_of_tree() {
local torch_version="$3"
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
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

@ -8,6 +8,7 @@ add_library(TorchMLIRJITIRImporter STATIC
ivalue_importer.cpp
torch_to_mlir_utils.cpp
)
message(STATUS "Linking TorchMLIRJITImporter with ${TORCH_LIBRARIES}")
target_link_libraries(TorchMLIRJITIRImporter
TorchMLIRAggregateCAPI
${TORCH_LIBRARIES}