flash: Fix load addr for V tile; test with seqlen=128

2024-08-20 14:34:09 -07:00
parent df3c41aa0d
commit d8d5df64e6
2 changed files with 6 additions and 3 deletions
--- a/tests/regression/flash_attention/main.cpp
+++ b/tests/regression/flash_attention/main.cpp
@@ -99,7 +99,7 @@ int main(int argc, char *argv[]) {
  std::cout << "open device connection" << std::endl;
  RT_CHECK(vx_dev_open(&device));

-  uint32_t dim_seqlen = 64;
+  uint32_t dim_seqlen = 128;
  uint32_t dim_headdim = 64;

  using float_type = half;