minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[CI] Pin the Rust TreeCore build to the resolved libtorch instead of interpreter discovery (#37696)

Browse Source 
Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
This commit is contained in:
Alex Nails
2026-09-05 15:30:32 -07:00
committed by GitHub
co-authored by Claude Opus 5
parent 77aee20259
commit 6a0c55fd6c
4 changed files with 11 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/rust_extensions/torch_build.py
+5 -1
View File
@@ -74,7 +74,11 @@ def torch_build_configuration(
cxx11_abi = bool(torch_module._C._GLIBCXX_USE_CXX11_ABI)
environment = dict(os.environ if base_environment is None else base_environment)
environment["LIBTORCH_USE_PYTORCH"] = "1"
environment.pop("LIBTORCH_USE_PYTORCH", None)
environment["LIBTORCH"] = os.fspath(torch_root)
environment["LIBTORCH_INCLUDE"] = os.fspath(torch_root)
environment["LIBTORCH_LIB"] = os.fspath(torch_root)
environment["LIBTORCH_CXX11_ABI"] = "1" if cxx11_abi else "0"
# tch 0.24 targets Torch 2.11. The compatibility header below covers the
# API removals in the supported 2.12/2.13 builds, after this explicit gate.
environment["LIBTORCH_BYPASS_VERSION_CHECK"] = "1"