[TRTLLMINF-89][feat] Make L0 retries timeout-budget aware (#14323)

dpitman-nvda · web-flow · commit 24b68fc1d00f · 2026-05-21T13:24:34.000-04:00
Signed-off-by: Derek Pitman &lt;dpitman@nvidia.com&gt;
diff --git a/jenkins/Build.groovy b/jenkins/Build.groovy
@@ -504,6 +504,7 @@ def launchStages(pipeline, cpu_arch, enableFailFast, globalVars)
 
         echo "env.globalVars is: ${env.globalVars}"
         globalVars = trtllm_utils.updateMapWithJson(pipeline, globalVars, env.globalVars, "globalVars")
+        globalVars = trtllm_utils.initializeCiBudget(pipeline, globalVars, 24, 'HOURS', "Build-${cpu_arch}")
         globalVars[ACTION_INFO] = trtllm_utils.setupPipelineDescription(pipeline, globalVars[ACTION_INFO])
     }
 
diff --git a/jenkins/L0_MergeRequest.groovy b/jenkins/L0_MergeRequest.groovy
@@ -1692,6 +1692,7 @@ pipeline {
             steps
             {
                 script {
+                    globalVars = trtllm_utils.initializeCiBudget(this, globalVars, 24, 'HOURS', 'L0_MergeRequest')
                     preparation(this, testFilter, globalVars)
                     println globalVars
                     globalVars[ACTION_INFO] = trtllm_utils.setupPipelineDescription(this, globalVars[ACTION_INFO])
diff --git a/jenkins/L0_Test.groovy b/jenkins/L0_Test.groovy

Original file line number	Diff line number	Diff line change
`@@ -504,6 +504,7 @@ def launchStages(pipeline, cpu_arch, enableFailFast, globalVars)`
`504`	`504`
`505`	`505`	`echo "env.globalVars is: ${env.globalVars}"`
`506`	`506`	`globalVars = trtllm_utils.updateMapWithJson(pipeline, globalVars, env.globalVars, "globalVars")`
	`507`	`+ globalVars = trtllm_utils.initializeCiBudget(pipeline, globalVars, 24, 'HOURS', "Build-${cpu_arch}")`
`507`	`508`	`globalVars[ACTION_INFO] = trtllm_utils.setupPipelineDescription(pipeline, globalVars[ACTION_INFO])`
`508`	`509`	`}`
`509`	`510`
Original file line number	Diff line number	Diff line change
`@@ -1692,6 +1692,7 @@ pipeline {`
`1692`	`1692`	`steps`
`1693`	`1693`	`{`
`1694`	`1694`	`script {`
	`1695`	`+ globalVars = trtllm_utils.initializeCiBudget(this, globalVars, 24, 'HOURS', 'L0_MergeRequest')`
`1695`	`1696`	`preparation(this, testFilter, globalVars)`
`1696`	`1697`	`println globalVars`
`1697`	`1698`	`globalVars[ACTION_INFO] = trtllm_utils.setupPipelineDescription(this, globalVars[ACTION_INFO])`