Hansung Kim fcd8b0b892 flash: Disable rescale flag check ...

GEMM-II finishes much earlier than softmax for this to be a problem.

2024-11-09 20:37:58 -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: Rename nowarpspec to default

2024-11-09 19:58:45 -08:00

half.hpp

Add flash attention kernel skeleton

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

kernel.cpp

flash: Update tcore kernel to use new CISC

2024-11-09 19:49:20 -08:00

kernel.gemmini.cpp

flash: Disable rescale flag check

2024-11-09 20:37:58 -08:00

kernel.gemmini.warpspec.cpp

flash: Rename nowarpspec to default

2024-11-09 19:58:45 -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: Rename nowarpspec to default

2024-11-09 19:58:45 -08:00