diff --git a/kernels/flash_attention/compile_flash.sh b/kernels/flash_attention/compile_flash.sh index a2209047..42a53e07 100755 --- a/kernels/flash_attention/compile_flash.sh +++ b/kernels/flash_attention/compile_flash.sh @@ -30,7 +30,6 @@ for arch in "${archs[@]}"; do git pull # re-compile libvortexrt.a - # FIXME after restructure pushd ../../lib make popd @@ -40,6 +39,7 @@ for arch in "${archs[@]}"; do # touch source file to force re-building, as the Makefile does not track # binary changes touch kernel.cpp + touch kernel.gemmini.cpp make CONFIG=flash.$arch.seqlen1024.headdim64 done diff --git a/kernels/sgemm_gemmini_dma/compile_virgo.sh b/kernels/sgemm_gemmini_dma/compile_virgo.sh index de9b4b38..b38c13f2 100755 --- a/kernels/sgemm_gemmini_dma/compile_virgo.sh +++ b/kernels/sgemm_gemmini_dma/compile_virgo.sh @@ -4,6 +4,11 @@ git checkout ae-hopper git pull +# re-compile libvortexrt.a +pushd ../../lib +make +popd + if [ ! -f input.a.rand01.fp16.m256n256k256.row.bin ]; then echo "input binaries not found, generating operands" python3 generate_operands.py diff --git a/kernels/sgemm_tcore/compile_tcore.sh b/kernels/sgemm_tcore/compile_tcore.sh index d338521b..492ac53b 100755 --- a/kernels/sgemm_tcore/compile_tcore.sh +++ b/kernels/sgemm_tcore/compile_tcore.sh @@ -56,7 +56,6 @@ for arch in "${archs[@]}"; do git pull # re-compile libvortexrt.a - # FIXME after restructure pushd ../../lib make popd