kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

[None][feat] precompiled installation from local src dir with fnmatch only (#10430)

Browse Source 
Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>
This commit is contained in:
Lucas Liebenwein 2026-01-06 15:31:59 -05:00 committed by GitHub
parent 77be1b7572
commit 00355b24b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
setup.py
View File

@ -173,7 +173,6 @@ def extract_from_precompiled(precompiled_location: str, package_data: List[str],
- Remote URL: Downloads and extracts from URL (wheel or tar.gz)
"""
import fnmatch
import glob
import shutil
import tarfile
import zipfile
@ -183,6 +182,7 @@ def extract_from_precompiled(precompiled_location: str, package_data: List[str],
# Handle local directory (assuming repo structure)
if os.path.isdir(precompiled_location):
precompiled_location = os.path.abspath(precompiled_location)
print(
f"Using local directory as precompiled source: {precompiled_location}"
)
@ -192,14 +192,13 @@ def extract_from_precompiled(precompiled_location: str, package_data: List[str],
f"Directory {precompiled_location} does not contain a tensorrt_llm folder."
)
for pattern in package_data:
full_pattern = os.path.join(source_tensorrt_llm, pattern)
for src_file in glob.glob(full_pattern, recursive=True):
if not os.path.isfile(src_file):
continue
# Walk through all files and match using fnmatch (consistent with wheel extraction)
for root, dirs, files in os.walk(source_tensorrt_llm):
for filename in files:
src_file = os.path.join(root, filename)
# Get path relative to precompiled_location (e.g., "tensorrt_llm/libs/libtensorrt_llm.so")
rel_path = os.path.relpath(src_file, precompiled_location)
dst_file = rel_path # e.g., "tensorrt_llm/libs/libtensorrt_llm.so"
dst_file = rel_path
# Skip yaml files
if dst_file.endswith(".yaml"):
@ -214,6 +213,14 @@ def extract_from_precompiled(precompiled_location: str, package_data: List[str],
if not any(dst_file.startswith(d) for d in allowed_dirs):
continue
# Check if file matches any pattern using fnmatch (same as wheel extraction)
for filename_pattern in package_data:
if fnmatch.fnmatchcase(rel_path,
f"tensorrt_llm/{filename_pattern}"):
break
else:
continue
dst_dir = os.path.dirname(dst_file)
if dst_dir:
os.makedirs(dst_dir, exist_ok=True)