Skip to content

Commit

Permalink
[Split Build] limit split build matrix generation to cuda nightly jobs (
Browse files Browse the repository at this point in the history
#5370)

Co-authored-by: PaliC <>
  • Loading branch information
PaliC authored Jun 26, 2024
1 parent a48462e commit 747cc10
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tools/scripts/generate_binary_build_matrix.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
".", "_"
Expand Down

0 comments on commit 747cc10

Please sign in to comment.