From c26558bc933f6f407bb54313a96c90742c30a110 Mon Sep 17 00:00:00 2001
From: Hansung Kim <hansung_kim@berkeley.edu>
Date: Tue, 28 Jan 2025 23:48:02 -0800
Subject: [PATCH] Add fence before rescale

---
 tests/regression/flash_attention/kernel.gemmini.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/regression/flash_attention/kernel.gemmini.cpp b/tests/regression/flash_attention/kernel.gemmini.cpp
index 9208e863..b97e22cf 100644
--- 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