From 426b38ed247d70a5dd9effa09ecfd7641bdfdfa1 Mon Sep 17 00:00:00 2001
From: Vincent Moens <vmoens@meta.com>
Date: Tue, 4 Feb 2025 17:26:36 +0000
Subject: [PATCH] init

---
 .../{linux-post-script.sh => linux-pre-script.sh}      |  1 +
 .github/workflows/build-wheels-linux.yml               | 10 +++++-----
 setup.py                                               |  1 +
 3 files changed, 7 insertions(+), 5 deletions(-)
 rename .github/scripts/{linux-post-script.sh => linux-pre-script.sh} (52%)

diff --git a/.github/scripts/linux-post-script.sh b/.github/scripts/linux-pre-script.sh
similarity index 52%
rename from .github/scripts/linux-post-script.sh
rename to .github/scripts/linux-pre-script.sh
index 537989045..978deeb5d 100755
--- a/.github/scripts/linux-post-script.sh
+++ b/.github/scripts/linux-pre-script.sh
@@ -2,3 +2,4 @@
 
 yum update gcc
 yum update libstdc++
+export CXXFLAGS="-D_GLIBCXX_USE_CXX11_ABI=1"
diff --git a/.github/workflows/build-wheels-linux.yml b/.github/workflows/build-wheels-linux.yml
index 8bf29802c..9d1de68eb 100644
--- a/.github/workflows/build-wheels-linux.yml
+++ b/.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 }}
diff --git a/setup.py b/setup.py
index 986752654..cd910737a 100644
--- a/setup.py
+++ b/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"