kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-02-04 18:21:52 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

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

Browse Source 
Signed-off-by: Matt Lefebvre <mlefebvre@nvidia.com>
This commit is contained in:
Matt Lefebvre 2026-01-27 12:33:16 -08:00 committed by GitHub
parent 6c1862fb33
commit c26a8f764c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jenkins/L0_Test.groovy
View File

@ -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