Build for PyTorch 2.7.1

ghstack-source-id: 687672fe86b9b7db141537adc27658fa9bb245e7
Pull-Request: fairinternal/xformers#1399

__original_commit__ = fairinternal/xformers@96f264c

Author: danthe3rd

.github/workflows/rocm_build.yml

@@ -22,7 +22,7 @@ jobs:
       matrix:
         os: ['ubuntu-alola']
         python: ['3.11']
-        torch_version: ['2.7.0']
+        torch_version: ['2.7.1']
         toolkit_type: ['rocm']
         toolkit_short_version: ['6.2.4', '6.3']
 

.github/workflows/wheels.yml

@@ -38,7 +38,7 @@ jobs:
 
         include = []
         for os in ['8-core-ubuntu', 'windows-8-core']:
-          for torch_version in ['2.7.0']:
+          for torch_version in ['2.7.1']:
             # CUDA builds
             for cuda_short_version in CU_VERSIONS:
               if cuda_short_version < "124" and "windows" in os:
@@ -88,7 +88,7 @@ jobs:
     uses: ./.github/workflows/wheels_upload_pip.yml
     with:
       twine_username: __token__
-      filter: "*torch2.7.0+cu126*"
+      filter: "*torch2.7.1+cu126*"
       execute: ${{ github.repository == 'facebookresearch/xformers' && github.event_name != 'pull_request' }}
     secrets:
       twine_password: ${{ secrets.PYPI_TOKEN }}
@@ -109,5 +109,5 @@ jobs:
       aws_role: "arn:aws:iam::749337293305:role/pytorch_bot_uploader_role"
       s3_path: s3://pytorch/whl/${{ matrix.suffix }}/
       aws_s3_cp_extra_args: --acl public-read
-      filter: "*torch2.7.0+${{ matrix.suffix }}*"
+      filter: "*torch2.7.1+${{ matrix.suffix }}*"
       execute: ${{ github.repository == 'facebookresearch/xformers' && github.ref_type == 'tag' }}

.github/workflows/win-build.yml

@@ -73,7 +73,7 @@ jobs:
 
       - name: Install build dependencies
         run: |
-          $PY -m pip install wheel setuptools ninja torch==2.7.0 -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126
+          $PY -m pip install wheel setuptools ninja torch==2.7.1 -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126
           git config --global --add safe.directory "*"
           $PY -c "import torch; print('torch', torch.__version__)"
           $PY -c "import torch; print('torch.cuda', torch.version.cuda)"

CHANGELOG.md

@@ -4,10 +4,17 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [0.0.31] - 2025-??-??
+## [0.0.32] - 2025-??-??
+
+## [0.0.31] - 2025-06-25
+Pre-built binary wheels are available for PyTorch 2.7.1.
+### Added
+- xFormers wheels are now python-version agnostic: this means that the same wheel can be used for python 3.9, 3.10, ... 3.13
+- Added support for Flash-Attention 3 on Ampere GPUs
 ### Removed
 - We will no longer support V100 or older GPUs, following PyTorch (pytorch/pytorch#147607)
-- Removed support for building Flash-Attention 2 as part of xFormers. For Ampere GPUs, we now use Flash-Attention 3. On Linux, Flash-Attention 2 can still be used through PyTorch.
+- Deprecated support for building Flash-Attention 2 as part of xFormers. For Ampere GPUs, we now use Flash-Attention 3 on windows, and Flash-Attention 2 can still be used through PyTorch on linux.
+
 ## [0.0.30] - 2025-04-28
 Pre-built binary wheels are available for PyTorch 2.7.0. Following PyTorch, we build wheels for CUDA 11.8, 12.6, and 12.8 only (we no longer build for CUDA 12.4).
 xFormers now requires PyTorch >= 2.7

version.txt

@@ -1 +1 @@
-0.0.31
+0.0.32