From f7da0c8d2628e1b202649bf26ca155a48b37f01c Mon Sep 17 00:00:00 2001 From: Ting Lu Date: Tue, 2 Jul 2024 10:28:01 -0700 Subject: [PATCH] add CUDA GPU Runner --- tools/scripts/generate_binary_build_matrix.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 1c3f57360f..1061a8394c 100644 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -81,6 +81,7 @@ LINUX_GPU_RUNNER = "linux.g5.4xlarge.nvidia.gpu" LINUX_CPU_RUNNER = "linux.2xlarge" LINUX_AARCH64_RUNNER = "linux.arm64.2xlarge" +LINUX_AARCH64_GPU_RUNNER = "linux.arm64.m7g.4xlarge" WIN_GPU_RUNNER = "windows.8xlarge.nvidia.gpu" WIN_CPU_RUNNER = "windows.4xlarge" MACOS_M1_RUNNER = "macos-m1-stable" @@ -117,7 +118,10 @@ def validation_runner(arch_type: str, os: str) -> str: else: return LINUX_CPU_RUNNER elif os == LINUX_AARCH64: - return LINUX_AARCH64_RUNNER + if arch_type == CUDA_AARCH64: + return LINUX_AARCH64_GPU_RUNNER + else: + return LINUX_AARCH64_RUNNER elif os == WINDOWS: if arch_type == CUDA: return WIN_GPU_RUNNER