From 79f70f0539eccdbee4579f2b68a96097e87f45f4 Mon Sep 17 00:00:00 2001
From: zhangchenrui <zhangchenrui2@huawei.com>
Date: Tue, 4 Mar 2025 10:51:51 +0800
Subject: [PATCH] adapt cmakelists torch2.6

---
 mx_driving/csrc/CMakeLists.txt | 2 +-
 setup.py                       | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/mx_driving/csrc/CMakeLists.txt b/mx_driving/csrc/CMakeLists.txt
index 45259046..44c0bd5f 100644
--- a/mx_driving/csrc/CMakeLists.txt
+++ b/mx_driving/csrc/CMakeLists.txt
@@ -49,7 +49,7 @@ if(BUILD_STAGE EQUAL 2)
             -fstack-protector-all
             -DTORCH_API_INCLUDE_EXTENSION_H
             -DTORCH_EXTENSION_NAME=_C
-            -D_GLIBCXX_USE_CXX11_ABI=0
+            -D_GLIBCXX_USE_CXX11_ABI=${ABI}
             -D__FILENAME__=__FILE__
             ${EXT_CXX_FLAGS})
 
diff --git a/setup.py b/setup.py
index 4356937e..47b39b04 100644
--- a/setup.py
+++ b/setup.py
@@ -122,6 +122,11 @@ class ExtBuild(build_ext):
         ]
         if LooseVersion(torch.__version__) < LooseVersion("2.1.0"):
             cmake_args.append("-DCOMPILE_WITH_XLA:BOOL=ON")
+
+        if LooseVersion(torch.__version__) >= LooseVersion("2.6.0"):
+            cmake_args.append("-DABI=1")
+        else:
+            cmake_args.append("-DABI=0")
         build_args = ["--build", self.build_temp, f"-j{multiprocessing.cpu_count()}"]
 
         subprocess.check_call(
-- 
Gitee