yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

[WIP] add aws code build

Browse Source
This commit is contained in:
Junya Morioka
2025-08-08 18:07:35 +09:00
parent ed81f89d0a
commit abdb3f4431
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-code-build.yml
+3
View File
@@ -5,6 +5,7 @@ on:
jobs: jobs:
build_wheels: build_wheels:
name: Build wheels and Upload name: Build wheels and Upload
timeout-minutes: 3600
# runs-on: windows-latest # runs-on: windows-latest
runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }} runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }}
steps: steps:
@@ -26,6 +27,7 @@ jobs:
- name: Install VisualStudio BuildTools - name: Install VisualStudio BuildTools
shell: pwsh shell: pwsh
timeout-minutes: 1200
run: | run: |
choco install -y visualstudio2022buildtools ` choco install -y visualstudio2022buildtools `
--version=117.14.1 --params "--add Microsoft.VisualStudio.Component.VC.Tools.x86.x64 --includeRecommended --includeOptional" --version=117.14.1 --params "--add Microsoft.VisualStudio.Component.VC.Tools.x86.x64 --includeRecommended --includeOptional"
@@ -46,6 +48,7 @@ jobs:
pip install -U pip setuptools==75.8.0 wheel setuptools packaging psutil ninja numpy pip install -U pip setuptools==75.8.0 wheel setuptools packaging psutil ninja numpy
- name: Build wheels - name: Build wheels
timeout-minutes: 1200
shell: pwsh shell: pwsh
env: env:
MAX_JOBS: 4 MAX_JOBS: 4