Michael Goin 978aed5300 [Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

..

backends

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

ops

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

__init__.py

[Core][2/N] Model runner refactoring part 2. Combine prepare prefill / decode to a single API (#4681)

2024-05-15 14:00:10 +09:00

layer.py

[Kernel][Attention] Separate Attention.kv_scale into k_scale and v_scale (#6081)

2024-07-16 15:31:32 -07:00

selector.py

[Misc] Remove flashinfer warning, add flashinfer tests to CI (#6351)

2024-07-12 01:32:06 +00:00