Don't do libtorch pre-cxx11-abi for ROCm

pytorch · Dec 20, 2024 · b4bb616 · b4bb616
1 parent c922275
commit b4bb616
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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: