flash: Add begin end markers

2024-11-09 10:16:40 -08:00
parent 1e3d476e70
commit 365b1d8e67
3 changed files with 11 additions and 0 deletions
--- a/tests/regression/flash_attention/flash_impl.hpp
+++ b/tests/regression/flash_attention/flash_impl.hpp
@@ -4,6 +4,9 @@
 #include <vx_spawn.h>
 #include <float.h>

+#define MARK_BEG() asm volatile ("slti x0, x1, -1047")
+#define MARK_END() asm volatile ("slti x0, x1, -499")
+
 #define B_ROW 64
 #define B_COL 64
 #define HEADDIM 64