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

Signed-off-by: Yanchao Lu <[email protected]>

Author: chzblych

jenkins/Build.groovy

@@ -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.13.3-1
+                    image: ${jnlpImage}
                     args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
                     resources:
                       requests:

jenkins/BuildDockerImage.groovy

@@ -79,6 +79,9 @@ def createKubernetesPodConfig(type, arch = "amd64", build_wheel = false)
         """
     }
 
+    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":
@@ -152,7 +155,7 @@ def createKubernetesPodConfig(type, arch = "amd64", build_wheel = false)
                 containers:
                   ${containerConfig}
                   - name: jnlp
-                    image: urm.nvidia.com/docker/jenkins/inbound-agent:4.13.3-1
+                    image: ${jnlpImage}
                     args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
                     resources:
                       requests:

jenkins/L0_MergeRequest.groovy

@@ -175,7 +175,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 = """
@@ -186,6 +186,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":
@@ -277,7 +280,7 @@ def createKubernetesPodConfig(image, type)
                         fieldRef:
                           fieldPath: spec.nodeName
                   - name: jnlp
-                    image: urm.nvidia.com/docker/jenkins/inbound-agent:4.13.3-1
+                    image: ${jnlpImage}
                     args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
                     resources:
                       requests:

jenkins/L0_Test.groovy

@@ -364,6 +364,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":
@@ -546,7 +549,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.13.3-1
+                    image: ${jnlpImage}
                     args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
                     resources:
                       requests:

jenkins/controlCCache.groovy

@@ -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-202506021004-9420"
 
-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.13.3-1
+                    image: ${jnlpImage}
                     args: ['\$(JENKINS_SECRET)', '\$(JENKINS_NAME)']
                     resources:
                       requests:

tests/integration/test_lists/test-db/l0_gh200.yml

@@ -33,6 +33,7 @@ l0_gh200:
       gpu:
       - '*h200*'
       linux_distribution_name: ubuntu*
+      cpu: aarch64
     terms:
       stage: post_merge
       backend: tensorrt