wu-arch/kernels
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

flash: Change kernel arg to contain qkv; strip stimulus gen from host code

Browse Source 
test data is now generated by the python script instead of the host
binary.
This commit is contained in:
Hansung Kim
2024-08-15 21:03:02 -07:00
parent a1858e0c80
commit e0daf226ef
4 changed files with 55 additions and 189 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/regression/flash_attention/common.h
View File

@@ -7,12 +7,12 @@
#define DEV_SMEM_START_ADDR 0xff000000
typedef struct {
uint32_t dim_m;
uint32_t dim_n;
uint32_t dim_k;
uint64_t addr_a;
uint64_t addr_b;
uint64_t addr_c;
uint32_t dim_seqlen;
uint32_t dim_headdim;
uint64_t addr_q;
uint64_t addr_k;
uint64_t addr_v;
uint64_t addr_o;
} kernel_arg_t;
#endif