[TRTINFRA-7379][infra] Change SLURM config access to use resolvePlatform (#11006)

mlefeb01 · web-flow · commit c26a8f764cbf · 2026-01-27T12:33:16.000-08:00
Signed-off-by: Matt Lefebvre &lt;mlefebvre@nvidia.com&gt;
diff --git a/jenkins/L0_Test.groovy b/jenkins/L0_Test.groovy
@@ -548,7 +548,7 @@ def cleanUpNodeResources(def pipeline, SlurmCluster cluster, String nodeName, St
 
 def runLLMTestlistWithAgent(pipeline, platform, testList, config=VANILLA_CONFIG, perfMode=false, stageName="Undefined", splitId=1, splits=1, gpuCount=1, skipInstallWheel=false, cpver="cp312")
 {
-    SlurmPartition partition = SlurmConfig.partitionConfig[platform] as SlurmPartition
+    SlurmPartition partition = SlurmConfig.resolvePlatform(platform)
     SlurmCluster cluster = SlurmConfig.clusterConfig[partition.clusterName]
 
     def entrypoint = SlurmConfig.containerRuntimeToEntrypoint[cluster.containerRuntime]
@@ -912,7 +912,7 @@ def getMountListForSlurmTest(SlurmCluster cluster, boolean useSbatch = false)
 
 def runLLMTestlistWithSbatch(pipeline, platform, testList, config=VANILLA_CONFIG, perfMode=false, stageName="Undefined", splitId=1, splits=1, gpuCount=1, nodeCount=1, skipInstallWheel=false, cpver="cp312")
 {
-    SlurmPartition partition = SlurmConfig.partitionConfig[platform] as SlurmPartition
+    SlurmPartition partition = SlurmConfig.resolvePlatform(platform)
     SlurmCluster cluster = SlurmConfig.clusterConfig[partition.clusterName]
 
     // Create a unique suffix for the job name

Original file line number	Diff line number	Diff line change
`@@ -548,7 +548,7 @@ def cleanUpNodeResources(def pipeline, SlurmCluster cluster, String nodeName, St`
`548`	`548`
`549`	`549`	`def runLLMTestlistWithAgent(pipeline, platform, testList, config=VANILLA_CONFIG, perfMode=false, stageName="Undefined", splitId=1, splits=1, gpuCount=1, skipInstallWheel=false, cpver="cp312")`
`550`	`550`	`{`
`551`		`- SlurmPartition partition = SlurmConfig.partitionConfig[platform] as SlurmPartition`
	`551`	`+ SlurmPartition partition = SlurmConfig.resolvePlatform(platform)`
`552`	`552`	`SlurmCluster cluster = SlurmConfig.clusterConfig[partition.clusterName]`
`553`	`553`
`554`	`554`	`def entrypoint = SlurmConfig.containerRuntimeToEntrypoint[cluster.containerRuntime]`
`@@ -912,7 +912,7 @@ def getMountListForSlurmTest(SlurmCluster cluster, boolean useSbatch = false)`
`912`	`912`
`913`	`913`	`def runLLMTestlistWithSbatch(pipeline, platform, testList, config=VANILLA_CONFIG, perfMode=false, stageName="Undefined", splitId=1, splits=1, gpuCount=1, nodeCount=1, skipInstallWheel=false, cpver="cp312")`
`914`	`914`	`{`
`915`		`- SlurmPartition partition = SlurmConfig.partitionConfig[platform] as SlurmPartition`
	`915`	`+ SlurmPartition partition = SlurmConfig.resolvePlatform(platform)`
`916`	`916`	`SlurmCluster cluster = SlurmConfig.clusterConfig[partition.clusterName]`
`917`	`917`
`918`	`918`	`// Create a unique suffix for the job name`