Michał Moskal e8cc7967ff [Bugfix][Kernel] allow non-power-of-two head sizes in prefix prefill (#4128)		2024-04-18 00:51:28 -07:00
..
backends	[Test] Test multiple attn backend for chunked prefill. (#4023)	2024-04-12 09:56:57 -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