Closed yzh119 closed 4 months ago
@comaniac
Looks super clean lol It'd be better to have a test to verify the correctness
I feel we can directly use fp16 for query (but use fp8 for kv) to avoid possible accuracy loss, but let's merge this first.
@comaniac