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][chore] Adding install_tensorrt.sh script to pip wheel (#8116)

Browse Source 
Signed-off-by: Patrice Castonguay <55748270+pcastonguay@users.noreply.github.com>
This commit is contained in:
Patrice Castonguay 2025-10-02 15:47:12 -04:00 committed by GitHub
parent ab433b7228
commit 08a47918cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
scripts/build_wheel.py
View File

@ -836,6 +836,15 @@ def main(*,
install_file(build_dir / "tensorrt_llm/executor_worker/executorWorker",
bin_dir / "executorWorker")
scripts_dir = pkg_dir / "scripts"
if scripts_dir.exists():
clear_folder(scripts_dir)
scripts_dir.mkdir(parents=True, exist_ok=True)
if not on_windows:
install_file(project_dir / "docker/common/install_tensorrt.sh",
scripts_dir / "install_tensorrt.sh")
if not cpp_only:
def get_binding_lib(subdirectory, name):

3
setup.py
View File

@ -108,7 +108,8 @@ else:
'libs/nvshmem/License.txt', 'libs/nvshmem/nvshmem_bootstrap_uid.so.3',
'libs/nvshmem/nvshmem_transport_ibgda.so.103', 'bindings.*.so',
'deep_ep/LICENSE', 'deep_ep_cpp_tllm.*.so', "include/**/*",
'deep_gemm/LICENSE', 'deep_gemm/include/**/*', 'deep_gemm_cpp_tllm.*.so'
'deep_gemm/LICENSE', 'deep_gemm/include/**/*',
'deep_gemm_cpp_tllm.*.so', 'scripts/install_tensorrt.sh'
]
package_data += [