Skip to content

Commit 1f2761e

Browse files
authored
[None][feat] Enable gpt oss on DGX H100. (#6775)
Signed-off-by: Tracin <10434017+Tracin@users.noreply.github.com>
1 parent 9f1d9b7 commit 1f2761e

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

jenkins/L0_MergeRequest.groovy

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -630,6 +630,7 @@ def getAutoTriggerTagList(pipeline, testFilter, globalVars) {
630630
def specialFileToTagMap = [
631631
"tensorrt_llm/_torch/models/modeling_deepseekv3.py": ["-DeepSeek-"],
632632
"cpp/kernels/fmha_v2/": ["-FMHA-"],
633+
"tensorrt_llm/_torch/models/modeling_gpt_oss.py": ["-GptOss-"],
633634
]
634635
for (file in changedFileList) {
635636
for (String key : specialFileToTagMap.keySet()) {

jenkins/L0_Test.groovy

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1274,6 +1274,8 @@ def getMakoArgsFromStageName(stageName, parseSysinfo=false) {
12741274
}
12751275
if (stageName.contains("-DeepSeek-")) {
12761276
makoArgs += ["auto_trigger=deepseek"]
1277+
} else if (stageName.contains("-GptOss-")) {
1278+
makoArgs += ["auto_trigger=gpt_oss"]
12771279
} else {
12781280
makoArgs += ["auto_trigger=others"]
12791281
}
@@ -2039,6 +2041,7 @@ def launchTestJobs(pipeline, testFilter)
20392041
"DGX_H100-4_GPUs-PyTorch-DeepSeek-1": ["dgx-h100-x4", "l0_dgx_h100", 1, 2, 4],
20402042
"DGX_H100-4_GPUs-PyTorch-DeepSeek-2": ["dgx-h100-x4", "l0_dgx_h100", 2, 2, 4],
20412043
"DGX_H100-2_GPUs-PyTorch-Others-1": ["dgx-h100-x2", "l0_dgx_h100", 1, 1, 2],
2044+
"DGX_H100-4_GPUs-PyTorch-GptOss-1": ["dgx-h100-x4", "l0_dgx_h100", 1, 1, 4],
20422045
"DGX_H100-4_GPUs-PyTorch-Others-1": ["dgx-h100-x4", "l0_dgx_h100", 1, 1, 4],
20432046
"DGX_H100-4_GPUs-CPP-1": ["dgx-h100-x4", "l0_dgx_h100", 1, 1, 4],
20442047
"A10-PyTorch-1": ["a10", "l0_a10", 1, 1],

0 commit comments

Comments
 (0)