Skip to content

Commit 4aa0153

Browse files
committed
Renames environment variables for sparse attention build configuration
1 parent 92c0fad commit 4aa0153

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

.github/workflows/_build.yml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,12 +172,12 @@ jobs:
172172
173173
export MAX_JOBS=$([ "$MATRIX_CUDA_VERSION" == "129" ] && echo 1 || echo 2)
174174
export NVCC_THREADS=2
175-
export FLASH_DMATTN_FORCE_BUILD="TRUE"
176-
export FLASH_DMATTN_FORCE_CXX11_ABI=${{ inputs.cxx11_abi }}
175+
export FLASH_SPARSE_ATTENTION_FORCE_BUILD="TRUE"
176+
export FLASH_SPARSE_ATTENTION_FORCE_CXX11_ABI=${{ inputs.cxx11_abi }}
177177
178178
# If specified, limit to a single compute capability to speed up build
179179
if [ -n "${MATRIX_ARCH}" ]; then
180-
export FLASH_DMATTN_CUDA_ARCHS="${MATRIX_ARCH}"
180+
export FLASH_SPARSE_ATTENTION_CUDA_ARCHS="${MATRIX_ARCH}"
181181
fi
182182
183183
# GH allows max 6h

0 commit comments

Comments
 (0)