Skip to content

[Dev] Fix when build local version with isolated build #1487

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
LeiWang1999 merged 5 commits into from
Dec 25, 2025
Merged

[Dev] Fix when build local version with isolated build #1487

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
ab50f79
fix when build local version with isolated build
oraluben Dec 21, 2025
c627724
fix
oraluben Dec 21, 2025
b75cfb9
trivial update
oraluben Dec 22, 2025
bfd14a0
Merge remote-tracking branch 'upstream/main' into local-build-z3
oraluben Dec 25, 2025
e5d125d
upd
oraluben Dec 25, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 11 additions & 8 deletions CMakeLists.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -300,10 +300,15 @@ install(TARGETS tilelang_cython_wrapper
RUNTIME DESTINATION tilelang/lib
ARCHIVE DESTINATION tilelang/lib)

# add python z3 lib path to rpath for running tests and dev in current folder
# Copy libz3.so to build folder to workaround isolated build env issue
if(USE_Z3 AND USE_PYPI_Z3)
set_property(TARGET tvm APPEND PROPERTY BUILD_RPATH ${Python3_SITELIB}/z3/lib)
set_property(TARGET tvm APPEND PROPERTY BUILD_RPATH ${Python3_SITELIB}/z3/bin)
get_target_property(Z3_LIBRARY_PATH z3::libz3 IMPORTED_LOCATION)
install(FILES "${Z3_LIBRARY_PATH}" DESTINATION "${CMAKE_BINARY_DIR}/tvm")
if(APPLE)
set_target_properties(tvm PROPERTIES BUILD_RPATH "@loader_path")
else()
set_target_properties(tvm PROPERTIES BUILD_RPATH "\$ORIGIN")
endif()
endif()

if(APPLE)
Expand All @@ -320,11 +325,9 @@ elseif(UNIX)
endif()
endif()

# let libtilelang to search tvm/tvm_runtime in same dir
set_target_properties(tilelang PROPERTIES INSTALL_RPATH "${TILELANG_INSTALL_RPATH}")
set_target_properties(tilelang_module PROPERTIES INSTALL_RPATH "${TILELANG_INSTALL_RPATH}")
set_target_properties(tvm PROPERTIES INSTALL_RPATH "${TILELANG_INSTALL_RPATH}")
set_target_properties(tvm_runtime PROPERTIES INSTALL_RPATH "${TILELANG_INSTALL_RPATH}")
set_target_properties(
tilelang tilelang_module tvm tvm_runtime
PROPERTIES INSTALL_RPATH "${TILELANG_INSTALL_RPATH}")

install(
TARGETS tvm tvm_runtime tilelang_module tilelang
Expand Down
7 changes: 2 additions & 5 deletions pyproject.toml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -27,11 +27,8 @@ classifiers = [
]
dynamic = ["version"]
dependencies = [
"apache-tvm-ffi~=0.1.0",
# Extra constraint to tvm-ffi for abi issue,
# should be removed after our tvm's update.
# See discussion in tilelang#1373 and apache/tvm-ffi#307
"apache-tvm-ffi>=0.1.6",
# Memory issue: tilelang#1502
"apache-tvm-ffi~=0.1.0,>=0.1.6",
# torch-c-dlpack-ext provides prebuilt torch extensions.
# Without it, TVM FFI may require JIT compilation on first import.
"torch-c-dlpack-ext",
Expand Down
2 changes: 1 addition & 1 deletion requirements-dev.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# Requirements to run local build with `--no-build-isolation` or other developments

apache-tvm-ffi>=0.1.6
apache-tvm-ffi~=0.1.0,>=0.1.6
build
cmake>=3.26
cython>=3.0.0
Expand Down
3 changes: 2 additions & 1 deletion requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
# Runtime requirements
apache-tvm-ffi>=0.1.6

apache-tvm-ffi~=0.1.0,>=0.1.6
torch-c-dlpack-ext
cloudpickle
ml-dtypes
Expand Down
Loading