From 80ce167168b782ca52b513b3a0ee19256dfb7a98 Mon Sep 17 00:00:00 2001
From: Sunidhi-Gaonkar1 <sunidhi.gaonkar@ibm.com>
Date: Thu, 20 Nov 2025 18:30:16 +0530
Subject: [PATCH] Update build.py to skip libnvshmem3-cuda-13 for cpu only
 build.

---
 build.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/build.py b/build.py
index 950bf48968..0b4640df4f 100755
--- a/build.py
+++ b/build.py
@@ -1509,8 +1509,9 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
 """
 
     if target_platform() not in ["igpu", "windows", "rhel"]:
-        repo_arch = "sbsa" if target_machine == "aarch64" else "x86_64"
-        df += f"""
+        if FLAGS.triton_wheels_dependencies_group != "cpu" :
+           repo_arch = "sbsa" if target_machine == "aarch64" else "x86_64"
+           df += f"""
 RUN curl -o /tmp/cuda-keyring.deb \\
         https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/{repo_arch}/cuda-keyring_1.1-1_all.deb \\
       && apt install /tmp/cuda-keyring.deb \\