Skip to content

flash-attn v2.8.0.post2 #36

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
regro-cf-autotick-bot wants to merge 2 commits into from
Closed

flash-attn v2.8.0.post2 #36

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8ac4233
updated v2.8.0.post2
regro-cf-autotick-bot Jun 15, 2025
c7bcc3f
MNT: Re-rendered with conda-build 25.5.0, conda-smithy 3.50.1, and co…
regro-cf-autotick-bot Jun 15, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 0 additions & 2 deletions ...iler_version13python3.10.____cpython.yaml → ...er_version12.6python3.10.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...iler_version13python3.11.____cpython.yaml → ...er_version12.6python3.11.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...iler_version13python3.12.____cpython.yaml → ...er_version12.6python3.12.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...mpiler_version13python3.13.____cp313.yaml → ...iler_version12.6python3.13.____cp313.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...piler_version13python3.9.____cpython.yaml → ...ler_version12.6python3.9.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...iler_version13python3.10.____cpython.yaml → ...er_version12.6python3.10.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...iler_version13python3.11.____cpython.yaml → ...er_version12.6python3.11.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...iler_version13python3.12.____cpython.yaml → ...er_version12.6python3.12.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...mpiler_version13python3.13.____cp313.yaml → ...iler_version12.6python3.13.____cp313.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
2 changes: 0 additions & 2 deletions ...piler_version13python3.9.____cpython.yaml → ...ler_version12.6python3.9.____cpython.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -39,6 +39,4 @@ target_platform:
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- cuda_compiler
- cuda_compiler_version
- docker_image
40 changes: 20 additions & 20 deletions .github/workflows/conda-build.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,55 +21,55 @@ jobs:
fail-fast: false
matrix:
include:
- CONFIG: linux_64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.10.____cpython
- CONFIG: linux_64_cuda_compiler_version12.6python3.10.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version13cuda_c_ha604d791d6', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_cuda_compiler_version12.6python_hcc941c96', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.11.____cpython
- CONFIG: linux_64_cuda_compiler_version12.6python3.11.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version13cuda_c_h4845835b7d', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_cuda_compiler_version12.6python_hdc5b00bd', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.12.____cpython
- CONFIG: linux_64_cuda_compiler_version12.6python3.12.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version13cuda_c_h2546c6389c', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_cuda_compiler_version12.6python_h82ccd7e7', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313
- CONFIG: linux_64_cuda_compiler_version12.6python3.13.____cp313
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version13cuda_c_hd698397623', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_cuda_compiler_version12.6python_h4fb0f834', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.9.____cpython
- CONFIG: linux_64_cuda_compiler_version12.6python3.9.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version13cuda_c_hd64767d203', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_cuda_compiler_version12.6python_h0449a975', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_aarch64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.10.____cpython
- CONFIG: linux_aarch64_cuda_compiler_version12.6python3.10.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_hcfd5a11850', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_cuda_compiler_version12.6p_he67a0e1c', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_aarch64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.11.____cpython
- CONFIG: linux_aarch64_cuda_compiler_version12.6python3.11.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_h9f94290832', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_cuda_compiler_version12.6p_hcb93a8ff', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_aarch64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.12.____cpython
- CONFIG: linux_aarch64_cuda_compiler_version12.6python3.12.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_h74435ed27e', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_cuda_compiler_version12.6p_h9a59e2d8', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_aarch64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.13.____cp313
- CONFIG: linux_aarch64_cuda_compiler_version12.6python3.13.____cp313
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_h137071728e', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_cuda_compiler_version12.6p_h063d0bee', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
- CONFIG: linux_aarch64_c_compiler_version13cuda_compilercuda-nvcccuda_compiler_version12.6cxx_compiler_version13python3.9.____cpython
- CONFIG: linux_aarch64_cuda_compiler_version12.6python3.9.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_c_compiler_version13c_hc1e7e6d7a9', 'linux', 'x64', 'self-hosted']
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_aarch64_cuda_compiler_version12.6p_h93b0bad1', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-x86_64:alma9
steps:

Expand Down
8 changes: 2 additions & 6 deletions build-locally.py
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

6 changes: 3 additions & 3 deletions recipe/meta.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
{% set version = "2.7.4.post1" %}
{% set version = "2.8.0.post2" %}
# TODO: Add archs 10.0 and 12.0 with CTK 12.8 release
Copy link
Member

@weiji14 weiji14 Jun 16, 2025 •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I would say let's wait the CUDA 12.9 migrator (conda-forge/conda-forge-pinning-feedstock#7476) and rebuild for Pytorch at conda-forge/pytorch-cpu-feedstock#393, and then we can try compiling with TORCH_CUDA_ARCH_LIST=...;10.0;12.0+PTX?


package:
Expand All @@ -10,13 +10,13 @@ package:
source:
# Use PYPI sdist instead of GitHub because they include a specific revended CUTLASS
- url: https://pypi.org/packages/source/f/flash-attn/flash_attn-{{ version }}.tar.gz
sha256: f03485c9a49a4d68d0733acdcad80ab0e72afa025a777fdc2966ceccf9d51765
sha256: b51d7015eb78f7ab2c332ec7c36681d7e97834f010b52eb4db1f118351982f58
# Overwrite with a simpler build script that doesn't try to revend pre-compiled binaries
- path: pyproject.toml
- path: setup.py

build:
number: 1
number: 0
script: {{ PYTHON }} -m pip install . -vvv --no-deps --no-build-isolation
script_env:
# Limit MAX_JOBS in order to prevent runners from crashing
Expand Down
Loading