diff --git a/sims/vcs/scripts/compile_designs.sh b/sims/vcs/scripts/compile_designs.sh index 5ddcb3c1..75dccbba 100755 --- a/sims/vcs/scripts/compile_designs.sh +++ b/sims/vcs/scripts/compile_designs.sh @@ -16,4 +16,7 @@ make CONFIG=VirgoFP16Config debug echo -e "\nCompiling hopper & virgo designs" make CONFIG=VirgoHopperConfig make CONFIG=VirgoHopperConfig debug +echo -e "\nCompiling flash designs" +make CONFIG=VirgoFlashConfig +make CONFIG=VirgoFlashConfig debug echo -e "\nCompilation completed" diff --git a/sims/vcs/scripts/run_sims.sh b/sims/vcs/scripts/run_sims.sh index 440f7fd6..c032bfe6 100755 --- a/sims/vcs/scripts/run_sims.sh +++ b/sims/vcs/scripts/run_sims.sh @@ -38,6 +38,8 @@ check_exists "simv-chipyard.harness-VirgoHopperConfig-debug" check_exists "$KERNELS_PATH/sgemm_tcore/kernel.radiance.gemm.tcore.volta.dim256.elf" check_exists "$KERNELS_PATH/sgemm_tcore/kernel.radiance.gemm.tcore.hopper.dim512.elf" check_exists "$KERNELS_PATH/sgemm_gemmini_dma/kernel.radiance.gemm.virgo.hopper.dim1024.elf" +check_exists "$KERNELS_PATH/flash_attention/kernel.radiance.flash.ampere.seqlen1024.headdim64.elf" +check_exists "$KERNELS_PATH/flash_attention/kernel.radiance.flash.virgo.seqlen1024.headdim64.elf" echo "Simulations will be started in parallel in 5 seconds. Please do not Ctrl+C as it kills all subprocesses." @@ -54,6 +56,9 @@ for dim in "${dims[@]}"; do suffix="" done +start_run VirgoFlashConfig flash_attention/kernel.radiance.flash.ampere.seqlen1024.headdim64.elf "ampere" "${suffix}" +start_run VirgoFlashConfig flash_attention/kernel.radiance.flash.virgo.seqlen1024.headdim64.elf "virgo" "${suffix}" + wait echo "All simulations have finished!"