fix: export USE_CXX11_ABI=1 for cuda12.6 (#3319)

pytorch · Dec 12, 2024 · a945aeb · a945aeb
1 parent e3ec185
commit a945aeb
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/packaging/env_vars.txt b/packaging/env_vars.txt
@@ -1,2 +1,7 @@
 export CI_BUILD="1"
-export RELEASE="1"
+export RELEASE="1"
+if [ $CU_VERSION == "cu126" ]; then
+    export USE_CXX11_ABI="1"
+else
+    export USE_CXX11_ABI="0"
+fi