export USE_CXX11_ABI=1 for cuda12.6

Author: zewenli98

packaging/env_vars.txt

@@ -1,3 +1,3 @@
 export CI_BUILD="1"
 export RELEASE="1"
-export CXX11_ABI="1"
+export USE_CXX11_ABI="1"

setup.py

@@ -141,7 +141,9 @@ def load_dep_info():
     CXX11_ABI = True
 
 if (cxx11_abi_env_var := os.environ.get("USE_CXX11_ABI")) is not None:
-    if cxx11_abi_env_var == "1":
+    if (
+        cxx11_abi_env_var == "1" and __cuda_version__ == "12.6"
+    ):  # Only use CXX11_ABI for CUDA 12.6
         CXX11_ABI = True
 
 if platform.uname().processor == "aarch64":