Skip to content

[pull] main from NVIDIA:main #1489

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
pull merged 5 commits into from
Jul 16, 2026
Merged

[pull] main from NVIDIA:main #1489

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
107e972
[None][infra] Handle SBSA image tags for x86 CI agents (#16503)
chzblych Jul 16, 2026
d0483a3
[https://nvbugs/6403920][fix] Move allreduce strategy test to smoke, …
MrGeva Jul 16, 2026
a273f0f
[None][infra] Check in most recent lock file from nightly pipeline
tensorrt-cicd Jul 16, 2026
a2595c0
[None][infra] Check in most recent lock file from nightly pipeline
tensorrt-cicd Jul 16, 2026
a0545b7
[None][fix] Fix Qwen image CUDA graph with CFG (#16391)
yibinl-nvidia Jul 16, 2026
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
2 changes: 1 addition & 1 deletion jenkins/Build.groovy
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ AARCH64_TRIPLE = "aarch64-linux-gnu"
LLM_DOCKER_IMAGE = env.dockerImage

// Always use x86_64 image for agent
AGENT_IMAGE = env.dockerImage.replace("aarch64", "x86_64")
AGENT_IMAGE = env.dockerImage.replace("aarch64", "x86_64").replace("sbsa", "x86_64")

POD_TIMEOUT_SECONDS_BUILD = env.podTimeoutSeconds ? env.podTimeoutSeconds : "43200"

Expand Down
7 changes: 4 additions & 3 deletions jenkins/L0_Test.groovy
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -59,6 +59,7 @@ linuxPkgName = ( env.targetArch == AARCH64_TRIPLE ? "tensorrt-llm-sbsa-release-s
// available tags can be found in: https://urm.nvidia.com/artifactory/sw-tensorrt-docker/tensorrt-llm/
// [base_image_name]-[arch]-[os](-[python_version])-[trt_version]-[torch_install_type]-[stage]-[date]-[mr_id]
LLM_DOCKER_IMAGE = env.dockerImage
X86_64_DOCKER_IMAGE = LLM_DOCKER_IMAGE.replace("aarch64", "x86_64").replace("sbsa", "x86_64")
LLM_ROCKYLINUX8_PY310_DOCKER_IMAGE = env.wheelDockerImagePy310
LLM_ROCKYLINUX8_PY312_DOCKER_IMAGE = env.wheelDockerImagePy312
LLM_WHEEL_DOCKER_IMAGE = env.wheelDockerImage
Expand Down Expand Up @@ -4895,7 +4896,7 @@ def launchTestJobs(pipeline, testFilter)
x86SlurmTestConfigs = cbtsResizeSplits(x86SlurmTestConfigs)
fullSet += x86SlurmTestConfigs.keySet()

parallelSlurmJobs = x86SlurmTestConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(LLM_DOCKER_IMAGE.replace("aarch64", "x86_64"), "slurm", "amd64"), { attemptTag, isFinalAttempt, retryContext = null ->
parallelSlurmJobs = x86SlurmTestConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(X86_64_DOCKER_IMAGE, "slurm", "amd64"), { attemptTag, isFinalAttempt, retryContext = null ->
// attemptTag comes from runKubernetesPodWithInfraRetry for the outer
// dispatcher pod (when retry is enabled — see opts below) and is
// threaded into runLLMTestlistOnSlurm so a future re-enable of outer
Expand Down Expand Up @@ -5168,7 +5169,7 @@ def launchTestJobs(pipeline, testFilter)
// singleAttempt:true disables the outer K8s pod retry; see the x86
// SLURM closure above for the full rationale (cap nested retry budget
// so consistently-timing-out tests don't burn ~36h on retry cascades).
parallelSlurmJobs = SBSASlurmTestConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(LLM_DOCKER_IMAGE.replace("aarch64", "x86_64"), "slurm", "amd64"), { attemptTag, isFinalAttempt, retryContext = null ->
parallelSlurmJobs = SBSASlurmTestConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(X86_64_DOCKER_IMAGE, "slurm", "amd64"), { attemptTag, isFinalAttempt, retryContext = null ->
// attemptTag is threaded into runLLMTestlistOnSlurm as the outer
// dispatcher pod's tag so the inner SLURM retry's postTag can't
// collide with a previous dispatcher pod's upload. See the x86
Expand All @@ -5186,7 +5187,7 @@ def launchTestJobs(pipeline, testFilter)

// Add SBSA multi node Slurm jobs
// singleAttempt:true disables the outer K8s pod retry; see above.
parallelMultiNodesSBSAJobs = multiNodesSBSAConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(LLM_DOCKER_IMAGE.replace("aarch64", "x86_64"), "slurm", "amd64"), { attemptTag, isFinalAttempt, retryContext = null ->
parallelMultiNodesSBSAJobs = multiNodesSBSAConfigs.collectEntries{key, values -> [key, [createKubernetesPodConfig(X86_64_DOCKER_IMAGE, "slurm", "amd64"), { attemptTag, isFinalAttempt, retryContext = null ->
def config = LINUX_AARCH64_CONFIG
if (key.contains("single-device")) {
config = SINGLE_DEVICE_CONFIG
Expand Down
56 changes: 24 additions & 32 deletions security_scanning/examples/auto_deploy/poetry.lock
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading
Loading