squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Full version matrix

Browse Source
This commit is contained in:
Pierce Freeman 2023-06-02 22:47:29 -07:00
parent a372e2be1b
commit ac543b0e8d
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
.github/workflows/publish.yml vendored
View File

@ -37,18 +37,20 @@ jobs:
fail-fast: false
matrix:
os: [ubuntu-20.04, ubuntu-22.04]
#python-version: ['3.7', '3.8', '3.9', '3.10']
python-version: ['3.10']
torch-version: [1.11.0]
cuda-version: ['113', '117']
#torch-version: [1.11.0, 1.12.0, 1.12.1]
#cuda-version: ['113', '116']
python-version: ['3.7', '3.8', '3.9', '3.10']
torch-version: ['1.11.0', '1.12.0', '1.13.0']
cuda-version: ['113', '116', '117']
exclude:
# Nvidia only supports 11.7+ for ubuntu-22.04
- os: ubuntu-22.04
cuda-version: '116'
- os: ubuntu-22.04
cuda-version: '113'
# Torch only builds cuda 117 for 1.13.0+
- cuda-version: '117'
torch-version: '1.11.0'
- cuda-version: '117'
torch-version: '1.12.0'
steps:
- name: Checkout