ColossalAI/tests/test_infer/test_kernels
Steve Luo 7806842f2d
add paged-attetionv2: support seq length split across thread block (#5707)
2024-05-14 12:46:54 +08:00
..
cuda add paged-attetionv2: support seq length split across thread block (#5707) 2024-05-14 12:46:54 +08:00
triton [Fix] Fix Inference Example, Tests, and Requirements (#5688) 2024-05-08 11:30:15 +08:00
__init__.py [Fix] Fix & Update Inference Tests (compatibility w/ main) 2024-05-05 16:28:56 +00:00