From 527e22a4e534465b19190b914e46e27e20e45268 Mon Sep 17 00:00:00 2001 From: Brad Larson Date: Sun, 1 Mar 2026 13:05:01 -0600 Subject: [PATCH] Add aarch64 as a viable Linux system for detection of NVIDIA or AMD GPUs. --- mojo/mojo_host_platform.bzl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mojo/mojo_host_platform.bzl b/mojo/mojo_host_platform.bzl index 8ecda5b..781004c 100644 --- a/mojo/mojo_host_platform.bzl +++ b/mojo/mojo_host_platform.bzl @@ -124,7 +124,7 @@ def _get_apple_constraint(rctx, gpu_mapping): def _impl(rctx): constraints = [] - if rctx.os.name == "linux" and rctx.os.arch == "amd64": + if rctx.os.name == "linux" and (rctx.os.arch == "amd64" or rctx.os.arch == "aarch64"): # A system may have both rocm-smi and nvidia-smi installed, check both. nvidia_smi = rctx.which("nvidia-smi")