init

Author: Vincent Moens

.github/scripts/linux-post-script.sh renamed to .github/scripts/linux-pre-script.sh

@@ -2,3 +2,4 @@
 
 yum update gcc
 yum update libstdc++
+export CXXFLAGS="-D_GLIBCXX_USE_CXX11_ABI=1"

.github/workflows/build-wheels-linux.yml

@@ -19,7 +19,8 @@ permissions:
 
 jobs:
   generate-matrix:
-    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@main
+    uses: pytorch/test-infra/.github/workflows/generate_binary_build_matrix.yml@release/2.6
+    test-infra-ref: release/2.6
     with:
       package-type: wheel
       os: linux
@@ -33,10 +34,11 @@ jobs:
         include:
           - repository: pytorch/tensordict
             smoke-test-script: test/smoke_test.py
-            post-script: .github/scripts/linux-post-script.sh
+            pre-script: .github/scripts/linux-pre-script.sh
             package-name: tensordict
     name: pytorch/tensordict
-    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@main
+    uses: pytorch/test-infra/.github/workflows/build_wheels_linux.yml@release/2.6
+    test-infra-ref: release/2.6
     with:
       repository: ${{ matrix.repository }}
       ref: ""

setup.py

@@ -108,6 +108,7 @@ def get_extensions():
             "-O3",
             "-std=c++17",
             "-fdiagnostics-color=always",
+            # "-D_GLIBCXX_USE_CXX11_ABI=0",
         ]
     }
     debug_mode = os.getenv("DEBUG", "0") == "1"