From b4bb6168a782dbbb3be30f416e04dfe63fc9bd89 Mon Sep 17 00:00:00 2001 From: Sergii Dymchenko Date: Fri, 20 Dec 2024 12:50:21 -0800 Subject: [PATCH] Don't do libtorch pre-cxx11-abi for ROCm --- tools/scripts/generate_binary_build_matrix.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py index 25f15ab414..757d06b469 100755 --- a/tools/scripts/generate_binary_build_matrix.py +++ b/tools/scripts/generate_binary_build_matrix.py @@ -353,7 +353,10 @@ def generate_libtorch_matrix( if os == WINDOWS: abi_versions = [RELEASE, DEBUG] elif os == LINUX: - abi_versions = [PRE_CXX11_ABI, CXX11_ABI] + if with_rocm == ENABLE: + abi_versions = [CXX11_ABI] + else: + abi_versions = [PRE_CXX11_ABI, CXX11_ABI] elif os in [MACOS_ARM64]: abi_versions = [CXX11_ABI] else: