Michał Moskal 32881f3f31 [kernel] fix sliding window in prefix prefill Triton kernel (#4405) ... Co-authored-by: SangBin Cho <rkooo567@gmail.com>		2024-05-02 11:23:37 -07:00
..
allclose_default.py	[ROCm] Fix some kernels failed unit tests (#2498)	2024-02-05 14:25:36 -08:00
conftest.py	[CI] Try introducing isort. (#3495)	2024-03-25 07:59:47 -07:00
test_activation.py	[CI] Try introducing isort. (#3495)	2024-03-25 07:59:47 -07:00
test_attention.py	[Misc] Add indirection layer for custom ops (#3913)	2024-04-10 20:26:07 -07:00
test_cache.py	[Misc] Add indirection layer for custom ops (#3913)	2024-04-10 20:26:07 -07:00
test_layernorm.py	[Kernel] Layernorm performance optimization (#3662)	2024-03-30 14:26:38 -07:00
test_moe.py	[Core] Set linear_weights directly on the layer (#3977)	2024-04-11 16:35:51 -04:00
test_pos_encoding.py	[CI] Try introducing isort. (#3495)	2024-03-25 07:59:47 -07:00
test_prefix_prefill.py	[kernel] fix sliding window in prefix prefill Triton kernel (#4405)	2024-05-02 11:23:37 -07:00
test_rand.py	[CI] Try introducing isort. (#3495)	2024-03-25 07:59:47 -07:00
test_sampler.py	[CI] Try introducing isort. (#3495)	2024-03-25 07:59:47 -07:00