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

[Infra] - Update JNLP container config (#5009)

Browse Source 
Signed-off-by: Yanchao Lu <yanchaol@nvidia.com>
This commit is contained in:
Yanchao Lu 2025-06-08 18:09:35 +08:00 committed by GitHub
parent a6e53bf4e0
commit 9f45e806b2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jenkins/Build.groovy
View File

@ -142,6 +142,9 @@ def createKubernetesPodConfig(image, type, arch = "amd64")
def jobName = getShortenedJobName(env.JOB_NAME)
def buildID = env.BUILD_ID
def archSuffix = arch == "arm64" ? "arm" : "amd"
def jnlpImage = "urm.nvidia.com/sw-ipp-blossom-sre-docker-local/lambda/custom_jnlp_images_${archSuffix}_linux:jdk17"
switch(type)
{
case "build":
@ -229,7 +232,7 @@ def createKubernetesPodConfig(image, type, arch = "amd64")
fieldRef:
fieldPath: spec.nodeName
- name: jnlp
image: urm.nvidia.com/docker/jenkins/inbound-agent:4.11-1-jdk11
image: ${jnlpImage}
args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
resources:
requests:

5
jenkins/BuildDockerImage.groovy
View File

@ -24,6 +24,9 @@ def createKubernetesPodConfig(type, arch = "amd64")
def targetCould = "kubernetes-cpu"
def containerConfig = ""
def archSuffix = arch == "arm64" ? "arm" : "amd"
def jnlpImage = "urm.nvidia.com/sw-ipp-blossom-sre-docker-local/lambda/custom_jnlp_images_${archSuffix}_linux:jdk17"
switch(type)
{
case "agent":
@ -81,7 +84,7 @@ def createKubernetesPodConfig(type, arch = "amd64")
containers:
${containerConfig}
- name: jnlp
image: urm.nvidia.com/docker/jenkins/inbound-agent:4.11-1-jdk11
image: ${jnlpImage}
args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
resources:
requests:

7
jenkins/L0_MergeRequest.groovy
View File

@ -168,7 +168,7 @@ String getShortenedJobName(String path)
return parts.join('-').toLowerCase()
}
def createKubernetesPodConfig(image, type)
def createKubernetesPodConfig(image, type, arch = "amd64")
{
def targetCould = "kubernetes-cpu"
def selectors = """
@ -179,6 +179,9 @@ def createKubernetesPodConfig(image, type)
def jobName = getShortenedJobName(env.JOB_NAME)
def buildID = env.BUILD_ID
def archSuffix = arch == "arm64" ? "arm" : "amd"
def jnlpImage = "urm.nvidia.com/sw-ipp-blossom-sre-docker-local/lambda/custom_jnlp_images_${archSuffix}_linux:jdk17"
switch(type)
{
case "agent":
@ -270,7 +273,7 @@ def createKubernetesPodConfig(image, type)
fieldRef:
fieldPath: spec.nodeName
- name: jnlp
image: urm.nvidia.com/docker/jenkins/inbound-agent:4.11-1-jdk11
image: ${jnlpImage}
args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
resources:
requests:

5
jenkins/L0_Test.groovy
View File

@ -360,6 +360,9 @@ def createKubernetesPodConfig(image, type, arch = "amd64", gpuCount = 1, perfMod
def jobName = getShortenedJobName(env.JOB_NAME)
def buildID = env.BUILD_ID
def archSuffix = arch == "arm64" ? "arm" : "amd"
def jnlpImage = "urm.nvidia.com/sw-ipp-blossom-sre-docker-local/lambda/custom_jnlp_images_${archSuffix}_linux:jdk17"
switch(type)
{
case "agent":
@ -524,7 +527,7 @@ def createKubernetesPodConfig(image, type, arch = "amd64", gpuCount = 1, perfMod
fieldRef:
fieldPath: spec.nodeName
- name: jnlp
image: urm.nvidia.com/docker/jenkins/inbound-agent:4.11-1-jdk11
image: ${jnlpImage}
args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
resources:
requests:

7
jenkins/controlCCache.groovy
View File

@ -3,8 +3,11 @@ import java.lang.InterruptedException
DOCKER_IMAGE = "urm.nvidia.com/sw-tensorrt-docker/tensorrt-llm:pytorch-25.04-py3-x86_64-ubuntu24.04-trt10.10.0.31-skip-tritondevel-202505292346-4931"
def createKubernetesPodConfig(image)
def createKubernetesPodConfig(image, arch = "amd64")
{
def archSuffix = arch == "arm64" ? "arm" : "amd"
def jnlpImage = "urm.nvidia.com/sw-ipp-blossom-sre-docker-local/lambda/custom_jnlp_images_${archSuffix}_linux:jdk17"
def podConfig = [
cloud: "kubernetes-cpu",
namespace: "sw-tensorrt",
@ -39,7 +42,7 @@ def createKubernetesPodConfig(image)
ephemeral-storage: 25Gi
imagePullPolicy: Always
- name: jnlp
image: urm.nvidia.com/docker/jenkins/inbound-agent:4.11-1-jdk11
image: ${jnlpImage}
args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
resources:
requests: