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

2024-04-22 21:32:44 -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

[Test] Add xformer and flash attn tests (#3961)

2024-04-11 03:09:50 +00:00