Roy b6dcb4d442

[Misc] Fix flash attention backend log (#4368)

2024-04-25 12:43:32 -07:00

..

backends

[Mypy] Part 3 fix typing for nested directories for most of directory (#4161)

2024-04-22 21:32:44 -07:00

ops

[Bugfix][Kernel] allow non-power-of-two head sizes in prefix prefill (#4128)

2024-04-18 00:51:28 -07:00

__init__.py

[Core][5/N] Fully working chunked prefill e2e (#3884)

2024-04-10 17:56:48 -07:00

layer.py

[Core][5/N] Fully working chunked prefill e2e (#3884)

2024-04-10 17:56:48 -07:00

selector.py

[Misc] Fix flash attention backend log (#4368)

2024-04-25 12:43:32 -07:00