From 747cc1086b493d5f500446f0feb47d4540d9ca1c Mon Sep 17 00:00:00 2001
From: Sahan Paliskara <sahancpal@gmail.com>
Date: Wed, 26 Jun 2024 09:48:05 -0700
Subject: [PATCH] [Split Build] limit split build matrix generation to cuda
 nightly jobs (#5370)

Co-authored-by: PaliC <>
---
 tools/scripts/generate_binary_build_matrix.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
index 70465f3152..960544accd 100644
--- a/tools/scripts/generate_binary_build_matrix.py
+++ b/tools/scripts/generate_binary_build_matrix.py
@@ -545,7 +545,7 @@ def generate_wheels_matrix(
                     "use_split_build": False,
                 }
             ret.append(entry)
-            if use_split_build:
+            if use_split_build and (gpu_arch_version in CUDA_ARCHES) and (os == LINUX) and (channel == NIGHTLY):
                 entry = entry.copy()
                 entry["build_name"] = f"{package_type}-py{python_version}-{gpu_arch_type}{gpu_arch_version}-split".replace(
                     ".", "_"