init

Author: vmoens

.github/scripts/linux-post-script.sh .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,12 +19,12 @@ 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
     with:
       package-type: wheel
       os: linux
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
+      test-infra-ref: release/2.6
   build:
     needs: generate-matrix
     strategy:
@@ -33,15 +33,15 @@ 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
     with:
       repository: ${{ matrix.repository }}
       ref: ""
       test-infra-repository: pytorch/test-infra
-      test-infra-ref: main
+      test-infra-ref: release/2.6
       build-matrix: ${{ needs.generate-matrix.outputs.matrix }}
       package-name: ${{ matrix.package-name }}
       smoke-test-script: ${{ matrix.smoke-test-script }}

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"