From d666e38d118602f1c0f8c3e752cbaeeb200be73e Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 26 Sep 2025 18:08:24 -0700 Subject: [PATCH 1/2] Enable all-major CUDA architectures for non x86_64 builds. --- tools/gen_ort_dockerfile.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index f535229..405e9d7 100755 --- a/tools/gen_ort_dockerfile.py +++ b/tools/gen_ort_dockerfile.py @@ -323,7 +323,9 @@ def dockerfile_for_linux(output_file): else: cuda_archs = "87" else: - if os.getenv("CUDA_ARCH_LIST") is not None: + if os.uname().machine != "x86_64": + cuda_archs = "all-major" + elif os.getenv("CUDA_ARCH_LIST") is not None: print(f"[INFO] Defined CUDA_ARCH_LIST: {os.getenv('CUDA_ARCH_LIST')}") cuda_archs = ( os.getenv("CUDA_ARCH_LIST") From 94bf57ae024d8c27e3b656f85ae00da683b81973 Mon Sep 17 00:00:00 2001 From: Misha Chornyi Date: Fri, 26 Sep 2025 20:23:16 -0700 Subject: [PATCH 2/2] Hardcoding the value --- tools/gen_ort_dockerfile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tools/gen_ort_dockerfile.py b/tools/gen_ort_dockerfile.py index 405e9d7..2b56564 100755 --- a/tools/gen_ort_dockerfile.py +++ b/tools/gen_ort_dockerfile.py @@ -324,7 +324,7 @@ def dockerfile_for_linux(output_file): cuda_archs = "87" else: if os.uname().machine != "x86_64": - cuda_archs = "all-major" + cuda_archs = "80;86;90;100;110;120;121" elif os.getenv("CUDA_ARCH_LIST") is not None: print(f"[INFO] Defined CUDA_ARCH_LIST: {os.getenv('CUDA_ARCH_LIST')}") cuda_archs = (