Skip to content

Commit 70d8b87

Browse files
authored
Don't do libtorch pre-cxx11-abi for ROCm (#6099)
1 parent 393f0be commit 70d8b87

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tools/scripts/generate_binary_build_matrix.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -353,7 +353,10 @@ def generate_libtorch_matrix(
353353
if os == WINDOWS:
354354
abi_versions = [RELEASE, DEBUG]
355355
elif os == LINUX:
356-
abi_versions = [PRE_CXX11_ABI, CXX11_ABI]
356+
if with_rocm == ENABLE:
357+
abi_versions = [CXX11_ABI]
358+
else:
359+
abi_versions = [PRE_CXX11_ABI, CXX11_ABI]
357360
elif os in [MACOS_ARM64]:
358361
abi_versions = [CXX11_ABI]
359362
else:

0 commit comments

Comments
 (0)