mirror of
https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git
synced 2026-07-01 01:27:54 -04:00
fix aws codebuild
This commit is contained in:
+3
-3
@@ -65,9 +65,9 @@ Write-Host "Environment variables:"
|
||||
Write-Host " MAX_JOBS: $env:MAX_JOBS"
|
||||
Write-Host " NVCC_THREADS: $env:NVCC_THREADS"
|
||||
$env:FLASH_ATTENTION_FORCE_BUILD = "TRUE"
|
||||
$env:NVCC_FLAGS = "-w --disable-warnings -diag-suppress=all_warnings"
|
||||
$env:CXXFLAGS = "/w /wd9025"
|
||||
$env:CFLAGS = "/w /wd9025"
|
||||
$env:NVCC_FLAGS = "-w --disable-warnings"
|
||||
$env:CXXFLAGS = "/w"
|
||||
$env:CFLAGS = "/w"
|
||||
|
||||
cd flash-attention
|
||||
python setup.py bdist_wheel --dist-dir=dist
|
||||
|
||||
Reference in New Issue
Block a user