Expect _GLIBCXX_USE_CXX11_ABI=1 for ROCm manylinux2_28 wheels (#2043)

pytorch · Nov 23, 2024 · 5354a64 · 5354a64
1 parent 37ac645
commit 5354a64
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/check_binary.sh b/check_binary.sh
@@ -80,7 +80,7 @@ fi
 echo "Checking that the gcc ABI is what we expect"
 if [[ "$(uname)" != 'Darwin' ]]; then
   function is_expected() {
-    if [[ "$DESIRED_DEVTOOLSET" == *"cxx11-abi"* ]]; then
+    if [[ "$DESIRED_DEVTOOLSET" == *"cxx11-abi"* || "$DESIRED_CUDA" == *"rocm"* ]]; then
       if [[ "$1" -gt 0 || "$1" == "ON " ]]; then
         echo 1
       fi