Skip to content

Commit c26cb60

Browse files
committed
[CI] Only test on x86-64 for PR
1 parent df0a3f8 commit c26cb60

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

.github/workflows/mlir-tensorrt/generate-matrix.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,6 @@
1616
},
1717
],
1818
"test": [
19-
{
20-
"cuda": "12.9",
21-
"trt": "10.12",
22-
},
2319
{
2420
"cuda": "13.0",
2521
"trt": "10.13",
@@ -37,6 +33,11 @@
3733
],
3834
}
3935

36+
ARCH_LIST_DICT = {
37+
"test": ["x86_64"],
38+
"release": ["x86_64", "aarch64"],
39+
"nightly": ["x86_64", "aarch64"],
40+
}
4041

4142
GH_RUNNER_DICT = {
4243
"x86_64": "linux-amd64-gpu-h100-latest-1",
@@ -86,9 +87,9 @@ def main(args: list[str]) -> None:
8687
cuda_trt_versions = CUDA_TRT_VERSIONS_DICT[channel]
8788
docker_images = DOCKER_IMAGE_DICT[channel]
8889
cmake_preset = CMAKE_PRESET_DICT[channel]
89-
90+
arch_list = ARCH_LIST_DICT[channel]
9091
matrix_dict = {"include": []}
91-
for arch in ("x86_64", "aarch64"):
92+
for arch in arch_list:
9293
gh_runner = GH_RUNNER_DICT[arch]
9394
for cuda_trt_version in cuda_trt_versions:
9495
cuda_version = cuda_trt_version["cuda"]

0 commit comments

Comments
 (0)