chuanqiw
2024-12-06 15:10:47 +00:00
committed by PyTorch MergeBot
parent 34033cce4d
commit b64a537993
2 changed files with 16 additions and 1 deletions

View File

@ -427,7 +427,7 @@ def generate_wheels_matrix(
"use_split_build": "True" if use_split_build else "False",
"devtoolset": (
"cxx11-abi"
if (arch_version in ["cpu-cxx11-abi", "cpu-aarch64"])
if (arch_version in ["cpu-cxx11-abi", "cpu-aarch64", "xpu"])
or gpu_arch_type == "rocm"
else ""
),