Commit Graph

3 Commits

Author SHA1 Message Date
Hansung Kim
d8d5df64e6 flash: Fix load addr for V tile; test with seqlen=128 2024-08-20 14:34:09 -07:00
Hansung Kim
e0daf226ef flash: Change kernel arg to contain qkv; strip stimulus gen from host code
test data is now generated by the python script instead of the host
binary.
2024-08-15 21:03:02 -07:00
Hansung Kim
692d028afd Add flash attention kernel skeleton 2024-08-14 20:46:09 -07:00