Add fence before rescale

2025-01-28 23:48:02 -08:00
parent 8c45b8b4b7
commit c26558bc93
1 changed files with 1 additions and 0 deletions
--- a/tests/regression/flash_attention/kernel.gemmini.cpp
+++ b/tests/regression/flash_attention/kernel.gemmini.cpp
@@ -615,6 +615,7 @@ void kernel_body(int task_id, kernel_arg_t *__UNIFORM__ arg) {
        //   vx_fence();
        // }
        // asm volatile("rescale_fence_read_end_%=:" ::);
+        virgo_fence(2);
 #endif

 #if 0