pytorch
diff --git a/‎.github/scripts/generate_binary_build_matrix.py‎
Lines changed: 8 additions & 2 deletions b/‎.github/scripts/generate_binary_build_matrix.py‎
Lines changed: 8 additions & 2 deletions
diff --git a/‎.github/workflows/generated-linux-binary-manywheel-nightly.yml‎
Lines changed: 285 additions & 0 deletions b/‎.github/workflows/generated-linux-binary-manywheel-nightly.yml‎
Lines changed: 285 additions & 0 deletions
@@ -19,11 +19,15 @@
 ROCM_ARCHES = ["5.1.1", "5.2"]
 
 
+CPU_CXX11_ABI_ARCH = ['cpu-cxx11-abi']
+
 def arch_type(arch_version: str) -> str:
     if arch_version in CUDA_ARCHES:
         return "cuda"
     elif arch_version in ROCM_ARCHES:
         return "rocm"
+    elif arch_version in CPU_CXX11_ABI_ARCH:
+        return "cpu-cxx11-abi"
     else:  # arch_version should always be "cpu" in this case
         return "cpu"
 
@@ -38,6 +42,7 @@ def arch_type(arch_version: str) -> str:
         for gpu_arch in ROCM_ARCHES
     },
     "cpu": "pytorch/manylinux-builder:cpu",
+    "cpu-cxx11-abi": "pytorch/manylinuxcxx11-abi-builder:cpu-cxx11-abi",
 }
 
 CONDA_CONTAINER_IMAGES = {
@@ -77,6 +82,7 @@ def arch_type(arch_version: str) -> str:
 def translate_desired_cuda(gpu_arch_type: str, gpu_arch_version: str) -> str:
     return {
         "cpu": "cpu",
+        "cpu-cxx11-abi": "cpu-cxx11-abi",
         "cuda": f"cu{gpu_arch_version.replace('.', '')}",
         "rocm": f"rocm{gpu_arch_version}",
     }.get(gpu_arch_type, gpu_arch_version)
@@ -192,15 +198,15 @@ def generate_wheels_matrix(os: str,
         # Define default compute archivectures
         arches = ["cpu"]
         if os == "linux":
-            arches += CUDA_ARCHES + ROCM_ARCHES
+            arches += CPU_CXX11_ABI_ARCH + CUDA_ARCHES + ROCM_ARCHES
         elif os == "windows":
             arches += CUDA_ARCHES
 
     ret: List[Dict[str, str]] = []
     for python_version in python_versions:
         for arch_version in arches:
             gpu_arch_type = arch_type(arch_version)
-            gpu_arch_version = "" if arch_version == "cpu" else arch_version
+            gpu_arch_version = "" if arch_version == "cpu" or arch_version == "cpu-cxx11-abi" else arch_version
             # Skip rocm 3.11 binaries for now as the docker image are not correct
             if python_version == "3.11" and gpu_arch_type == "rocm":
                 continue