Skip to content
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.

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

Rebuild for pytorch 2.5, add python 3.13, update to flash-attn 2.7.0.post2 #20

Merged
Merged
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.10.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.11.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.12.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
c_compiler:
- gcc
c_compiler_version:
- '11'
c_stdlib:
- sysroot
c_stdlib_version:
- '2.17'
cdt_name:
- conda
channel_sources:
- conda-forge
channel_targets:
- conda-forge main
cuda_compiler:
- nvcc
cuda_compiler_version:
- '11.8'
cxx_compiler:
- gxx
cxx_compiler_version:
- '11'
docker_image:
- quay.io/condaforge/linux-anvil-cuda:11.8
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.13.* *_cp313
pytorch:
- '2.5'
target_platform:
- linux-64
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cuda_compiler
- cuda_compiler_version
- docker_image
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.9.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.10.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.11.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.12.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
c_compiler:
- gcc
c_compiler_version:
- '12'
c_stdlib:
- sysroot
c_stdlib_version:
- '2.17'
cdt_name:
- conda
channel_sources:
- conda-forge
channel_targets:
- conda-forge main
cuda_compiler:
- cuda-nvcc
cuda_compiler_version:
- '12.0'
cxx_compiler:
- gxx
cxx_compiler_version:
- '12'
docker_image:
- quay.io/condaforge/linux-anvil-cos7-x86_64
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.13.* *_cp313
pytorch:
- '2.5'
target_platform:
- linux-64
zip_keys:
- - c_compiler_version
- cxx_compiler_version
- c_stdlib_version
- cuda_compiler
- cuda_compiler_version
- docker_image
Original file line number Diff line number Diff line change
Expand Up @@ -25,15 +25,15 @@ docker_image:
github_actions_labels:
- cirun-openstack-cpu-xlarge
libtorch:
- '2.4'
- '2.5'
pin_run_as_build:
python:
min_pin: x.x
max_pin: x.x
python:
- 3.9.* *_cpython
pytorch:
- '2.4'
- '2.5'
target_platform:
- linux-64
zip_keys:
Expand Down
10 changes: 10 additions & 0 deletions .ci_support/migrations/pytorch25.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
__migrator:
build_number: 1
commit_message: Rebuild for pytorch 2.5
kind: version
migration_number: 1
libtorch:
- '2.5'
migrator_ts: 1730666768.682698
pytorch:
- '2.5'
10 changes: 10 additions & 0 deletions .github/workflows/conda-build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,11 @@ jobs:
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version11cuda_c_h58c8c5aa26', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-cuda:11.8
- CONFIG: linux_64_c_compiler_version11cuda_compilernvcccuda_compiler_version11.8cxx_compiler_version11python3.13.____cp313
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version11cuda_c_hba688eb4eb', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-cuda:11.8
- CONFIG: linux_64_c_compiler_version11cuda_compilernvcccuda_compiler_version11.8cxx_compiler_version11python3.9.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
Expand All @@ -56,6 +61,11 @@ jobs:
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version12cuda_c_h11ed5249c8', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-cos7-x86_64
- CONFIG: linux_64_c_compiler_version12cuda_compilercuda-nvcccuda_compiler_version12.0cxx_compiler_version12python3.13.____cp313
UPLOAD_PACKAGES: True
os: ubuntu
runs_on: ['cirun-openstack-cpu-xlarge--${{ github.run_id }}-linux_64_c_compiler_version12cuda_c_hee2a70c1f0', 'linux', 'x64', 'self-hosted']
DOCKER_IMAGE: quay.io/condaforge/linux-anvil-cos7-x86_64
- CONFIG: linux_64_c_compiler_version12cuda_compilercuda-nvcccuda_compiler_version12.0cxx_compiler_version12python3.9.____cpython
UPLOAD_PACKAGES: True
os: ubuntu
Expand Down
9 changes: 4 additions & 5 deletions recipe/meta.yaml
Original file line number Diff line number Diff line change
@@ -1,28 +1,27 @@
{% set version = "2.6.3" %}
{% set version = "2.7.0.post2" %}

package:
name: flash-attn-split
version: {{ version }}

source:
- url: https://pypi.org/packages/source/f/flash-attn/flash_attn-{{ version }}.tar.gz
sha256: 5bfae9500ad8e7d2937ebccb4906f3bc464d1bf66eedd0e4adabd520811c7b52
sha256: 62244de20dabad9a42d0636f6b2ea4c051d33df179a0b3bea4352502ebca1466
# Overwrite with a simpler build script that doesn't try to revend pre-compiled binaries
- path: pyproject.toml
- path: setup.py

build:
number: 2
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
- MAX_JOBS=4
- TORCH_CUDA_ARCH_LIST=8.0;8.6;8.9;9.0+PTX
skip: true # [cuda_compiler_version in (undefined, "None")]
skip: true # [not linux]
skip: true # [py==313] # Skip until pytorch dependency on setuptools is fixed
# debugging skips below
# skip: true # [py!=312]
# skip: true # [py!=313]
# skip: true # [cuda_compiler_version != "11.8"]

requirements:
Expand Down
Loading