Hansung Kim 4e087a8aab flash: Fix loop iteration for gemmini ...

Kernel is software-pipelined around 2 GEMMs and softmax; it requires two
iterations to fully complete a tile.

2024-11-08 16:43:08 -08:00

..

.gitignore

flash: gitignore

2024-08-15 21:04:59 -07:00

common.h

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

2024-08-15 21:03:02 -07:00

flash_impl.hpp

flash: Optimize smem alloc for tcore for 8banks

2024-09-19 21:31:39 -07:00

half.hpp

Add flash attention kernel skeleton

2024-08-14 20:46:09 -07:00

kernel.cpp

flash: Fix tcore kernel for CISC arg field changes

2024-11-08 16:40:16 -08:00

kernel.gemmini.cpp

flash: Fix loop iteration for gemmini

2024-11-08 16:43:08 -08:00

main.cpp

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

2024-08-20 14:34:09 -07:00

Makefile

flash: Fix online softmax for DMA layout

2024-09-07 23:21:28 -07:00