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

fix aws codebuild

Browse Source
This commit is contained in:
Junya Morioka
2025-08-15 02:29:27 +09:00
parent 0b49ce54e3
commit 7a8b2d4bce
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build_linux_self_host.yml
+2 -2
View File
@@ -36,7 +36,7 @@ jobs:
env: env:
DEBIAN_FRONTEND: noninteractive DEBIAN_FRONTEND: noninteractive
TERM: xterm-256color TERM: xterm-256color
timeout-minutes: 10000 timeout-minutes: 2160
steps: steps:
- name: Install tools - name: Install tools
shell: bash shell: bash
@@ -101,7 +101,7 @@ jobs:
export LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH export LD_LIBRARY_PATH=/usr/local/nvidia/lib64:/usr/local/cuda/lib64:$LD_LIBRARY_PATH
- name: Build wheels - name: Build wheels
timeout-minutes: 10000 timeout-minutes: 2160
id: build_wheels id: build_wheels
shell: bash shell: bash
env: env:
.github/workflows/build_windows_code_build.yml
+2 -2
View File
@@ -26,7 +26,7 @@ on:
jobs: jobs:
build_wheels: build_wheels:
name: Build wheels and Upload (Windows x86_64, AWS CodeBuild, ${{ inputs.flash-attn-version }}, ${{ inputs.python-version }}, ${{ inputs.torch-version }}, ${{ inputs.cuda-version }}) name: Build wheels and Upload (Windows x86_64, AWS CodeBuild, ${{ inputs.flash-attn-version }}, ${{ inputs.python-version }}, ${{ inputs.torch-version }}, ${{ inputs.cuda-version }})
timeout-minutes: 129600 timeout-minutes: 2160
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 }}
# Large Instance # Large Instance
env: env:
@@ -74,7 +74,7 @@ jobs:
- name: Build wheels - name: Build wheels
shell: pwsh shell: pwsh
timeout-minutes: 129600 timeout-minutes: 2160
run: | run: |
.\build_windows.ps1 -FlashAttnVersion "${{ inputs.flash-attn-version }}" -PythonVersion "${{ inputs.python-version }}" -TorchVersion "${{ inputs.torch-version }}" -CudaVersion "${{ inputs.cuda-version }}" .\build_windows.ps1 -FlashAttnVersion "${{ inputs.flash-attn-version }}" -PythonVersion "${{ inputs.python-version }}" -TorchVersion "${{ inputs.torch-version }}" -CudaVersion "${{ inputs.cuda-version }}"
$wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name } $wheelName = Get-ChildItem -Path "flash-attention\dist\*.whl" | Select-Object -First 1 | ForEach-Object { $_.Name }