Skip to content

Commit 731b2eb

Browse files
authored
[TRTLLM-5312][infra] Add triton trigger rules (#6440)
Signed-off-by: Yiqing Yan <[email protected]>
1 parent cee7071 commit 731b2eb

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

jenkins/L0_MergeRequest.groovy

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -640,6 +640,8 @@ def getAutoTriggerTagList(pipeline, testFilter, globalVars) {
640640
}
641641
def specialFileToTagMap = [
642642
"tensorrt_llm/_torch/models/modeling_deepseekv3.py": ["-DeepSeek-"],
643+
"tests/integration/defs/triton_server/": ["-Triton-"],
644+
"triton_backend/": ["-Triton-"],
643645
"cpp/kernels/fmha_v2/": ["-FMHA-"],
644646
"tensorrt_llm/_torch/models/modeling_gpt_oss.py": ["-GptOss-"],
645647
]

jenkins/L0_Test.groovy

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3266,6 +3266,7 @@ def launchTestJobs(pipeline, testFilter)
32663266
"pytorch": "-PyTorch-",
32673267
"tensorrt": "-TensorRT-",
32683268
"cpp": "-CPP-",
3269+
"triton": "-Triton-",
32693270
"fmha": "-FMHA-",
32703271
]
32713272
def backendModeList = backendMode.collect { changeMap.get(it) }.flatten()
@@ -3290,7 +3291,7 @@ def launchTestJobs(pipeline, testFilter)
32903291
} else {
32913292
echo "ONLY_ONE_GROUP_CHANGED mode is true. The group is: ${testFilter[(ONLY_ONE_GROUP_CHANGED)]}."
32923293
def excludedBackends = new HashMap()
3293-
excludedBackends["PyTorch"] = ["-CPP-", "-TensorRT-", "-Triton-", "-FMHA-"]
3294+
excludedBackends["PyTorch"] = ["-CPP-", "-TensorRT-", "-FMHA-"] // Only pytorch file change also need to run triton tests
32943295
excludedBackends["Triton"] = ["-PyTorch-", "-CPP-", "-TensorRT-", "-FMHA-"]
32953296
excludedBackends["FMHA"] = ["-PyTorch-", "-CPP-", "-TensorRT-", "-Triton-"]
32963297
def group = testFilter[(ONLY_ONE_GROUP_CHANGED)]

0 commit comments

Comments
 (0)