From 6bdc6af60780f6daabe8e0345f0d59f162f943b5 Mon Sep 17 00:00:00 2001 From: Hansung Kim Date: Thu, 30 Jan 2025 01:15:57 -0800 Subject: [PATCH] Fix branch name and dims for flash script --- kernels/flash_attention/compile_flash.sh | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/kernels/flash_attention/compile_flash.sh b/kernels/flash_attention/compile_flash.sh index 95dcb233..5808508c 100755 --- a/kernels/flash_attention/compile_flash.sh +++ b/kernels/flash_attention/compile_flash.sh @@ -26,7 +26,7 @@ ln -sf input.b.rand.fp32.seqlen1024headdim64.row.bin input.b.bin ln -sf input.c.rand.fp32.seqlen1024headdim64.row.bin input.c.bin for arch in "${archs[@]}"; do - git checkout ae-$arch + git checkout ae-flash-$arch # re-compile libvortexrt.a # FIXME after restructure @@ -34,13 +34,11 @@ for arch in "${archs[@]}"; do make popd - for dim in "${dims[@]}"; do - echo "compiling flash_attn kernel for $arch with seqlen 1024, headdim 64" + echo "compiling flash_attn kernel for $arch with seqlen 1024, headdim 64" - # touch source file to force re-building, as the Makefile does not track - # binary changes - touch kernel.cpp + # touch source file to force re-building, as the Makefile does not track + # binary changes + touch kernel.cpp - make CONFIG=flash.$arch.seqlen1024.headdim64 - done + make CONFIG=flash.$arch.seqlen1024.headdim64 done