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

[None][infra] Disable CU12 build to save build time (cost > 5 hours on SBSA) (#7633)

Browse Source 
Signed-off-by: ZhanruiSunCh <184402041+ZhanruiSunCh@users.noreply.github.com>
This commit is contained in:
Zhanrui Sun 2025-09-09 11:38:34 +08:00 committed by GitHub
parent 4cf9fed1e7
commit b573e07f3e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jenkins/Build.groovy
View File

@ -572,8 +572,9 @@ def launchStages(pipeline, cpu_arch, enableFailFast, globalVars)
buildConfigs = [
"Build TRT-LLM": [LLM_DOCKER_IMAGE] + prepareLLMBuild(
pipeline, cpu_arch == AARCH64_TRIPLE ? CONFIG_LINUX_AARCH64 : CONFIG_LINUX_X86_64_VANILLA),
"Build TRT-LLM CUDA12": [LLM_DOCKER_IMAGE_CU12] + prepareLLMBuild(
pipeline, cpu_arch == AARCH64_TRIPLE ? CONFIG_LINUX_AARCH64_CU12 : CONFIG_LINUX_X86_64_VANILLA_CU12),
// Disable CUDA12 build for too slow to build (cost > 5 hours on SBSA)
// "Build TRT-LLM CUDA12": [LLM_DOCKER_IMAGE_CU12] + prepareLLMBuild(
// pipeline, cpu_arch == AARCH64_TRIPLE ? CONFIG_LINUX_AARCH64_CU12 : CONFIG_LINUX_X86_64_VANILLA_CU12),
"Build TRT-LLM LLVM": [LLM_DOCKER_IMAGE] + prepareLLMBuild(
pipeline, cpu_arch == AARCH64_TRIPLE ? CONFIG_LINUX_AARCH64_LLVM : CONFIG_LINUX_X86_64_LLVM),
"Build TRT-LLM Pybind": [LLM_DOCKER_IMAGE] + prepareLLMBuild(

8
jenkins/L0_Test.groovy
View File

@ -950,10 +950,10 @@ def createKubernetesPodConfig(image, type, arch = "amd64", gpuCount = 1, perfMod
if (type == "gh200") {
hostnameMatch = """
- key: "kubernetes.io/hostname"
operator: In
operator: NotIn
values:
- "lego-cg1-qct-066.ipp3a2.colossus"
- "lego-cg1-qct-069.ipp3a2.colossus\""""
- "lego-cg1-qct-070.ipp3a2.colossus"
- "lego-cg1-qct-079.ipp3a2.colossus\""""
}
def podConfig = [
@ -2015,7 +2015,7 @@ def launchTestJobs(pipeline, testFilter)
"A100X-PyTorch-1": ["a100x", "l0_a100", 1, 1],
"L40S-PyTorch-1": ["l40s", "l0_l40s", 1, 2],
"L40S-PyTorch-2": ["l40s", "l0_l40s", 2, 2],
"H100_PCIe-PyTorch-CU12-1": ["h100-cr", "l0_h100", 1, 3],
"H100_PCIe-PyTorch-1": ["h100-cr", "l0_h100", 1, 3],
"H100_PCIe-PyTorch-2": ["h100-cr", "l0_h100", 2, 3],
"H100_PCIe-PyTorch-3": ["h100-cr", "l0_h100", 3, 3],
"H100_PCIe-CPP-1": ["h100-cr", "l0_h100", 1, 2],