try to fix by changing cmake

Signed-off-by: raayandhar <[email protected]>

Author: raayandhar

build_tools/cmake/TorchMLIRPyTorch.cmake

@@ -68,16 +68,49 @@ function(TorchMLIRConfigurePyTorch)
     message(STATUS "PyTorch C++ Dual ABI setting: \"${_use_cxx11_abi}\"")
 
     # Check ABI compatibility version
+    # Note: _PYBIND11_BUILD_ABI was removed in PyTorch 2.5+
+    # First, try the new method (checking libtorch_python.so)
     execute_process(
       COMMAND ${Python3_EXECUTABLE}
-      -c "import torch; import sys; abi=torch._C._PYBIND11_BUILD_ABI; abi.startswith('_cxxabi10') or sys.exit(1); sys.stdout.write(str(abi[-2:]))"
+      -c "import torch; import sys; abi = getattr(torch._C, '_PYBIND11_BUILD_ABI', None); sys.stdout.write(str(abi) if abi else '')"
       RESULT_VARIABLE _result
       WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
-      OUTPUT_VARIABLE _cxx_abi_version)
-    if(_result)
-      message(FATAL_ERROR "Failed to determine C++ ABI version")
+      OUTPUT_VARIABLE _pybind11_abi)
+
+    if(_result EQUAL "0" AND _pybind11_abi AND _pybind11_abi MATCHES "^_cxxabi10")
+      # Old method worked - extract ABI version from the string
+      string(REGEX REPLACE "^_cxxabi10(..)$" "\\1" _cxx_abi_version "${_pybind11_abi}")
+      message(STATUS "PyTorch C++ ABI version (from _PYBIND11_BUILD_ABI): \"${_cxx_abi_version}\"")
+    else()
+      # Fallback: try to extract from libtorch_python.so
+      message(STATUS "PyTorch _PYBIND11_BUILD_ABI not available, checking libtorch_python.so...")
+      execute_process(
+        COMMAND ${Python3_EXECUTABLE}
+        -c "import torch; import os; print(os.path.join(os.path.dirname(torch.__file__), 'lib', 'libtorch_python.so'), end='')"
+        RESULT_VARIABLE _result
+        WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
+        OUTPUT_VARIABLE _libtorch_python_path)
+
+      if(_result EQUAL "0" AND EXISTS "${_libtorch_python_path}")
+        execute_process(
+          COMMAND bash "-c" "strings ${_libtorch_python_path} | grep -E '^_cxxabi[0-9]{4}' | head -1 | grep -oE '..$'"
+          RESULT_VARIABLE _result
+          OUTPUT_VARIABLE _cxx_abi_version
+          OUTPUT_STRIP_TRAILING_WHITESPACE)
+
+        if(_result EQUAL "0" AND _cxx_abi_version)
+          message(STATUS "PyTorch C++ ABI version (from libtorch_python.so): \"${_cxx_abi_version}\"")
+        else()
+          # Final fallback: use a sensible default (ABI version 11 is most common)
+          set(_cxx_abi_version "11")
+          message(WARNING "Could not determine PyTorch C++ ABI version, using default: \"${_cxx_abi_version}\"")
+        endif()
+      else()
+        # Final fallback: use a sensible default
+        set(_cxx_abi_version "11")
+        message(WARNING "Could not find libtorch_python.so, using default ABI version: \"${_cxx_abi_version}\"")
+      endif()
     endif()
-    message(STATUS "PyTorch C++ ABI version: \"${_cxx_abi_version}\"")
 
     # Specialize compile flags for compiler
     if(${CMAKE_CXX_COMPILER_ID} STREQUAL "GNU")