File tree Expand file tree Collapse file tree 1 file changed +7
-6
lines changed
.github/workflows/mlir-tensorrt Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change 1616 },
1717 ],
1818 "test" : [
19- {
20- "cuda" : "12.9" ,
21- "trt" : "10.12" ,
22- },
2319 {
2420 "cuda" : "13.0" ,
2521 "trt" : "10.13" ,
3733 ],
3834}
3935
36+ ARCH_LIST_DICT = {
37+ "test" : ["x86_64" ],
38+ "release" : ["x86_64" , "aarch64" ],
39+ "nightly" : ["x86_64" , "aarch64" ],
40+ }
4041
4142GH_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" ]
You can’t perform that action at this time.
0 commit comments