ColossalAI/tests/test_infer_ops
Cuiqing Li 64110b12c0
combine codes (#4509)
2023-08-24 16:43:39 +08:00
..
cuda [Kernels] add necessary kernels (llama & bloom) for attention forward and kv-cache manager (#4485) 2023-08-24 16:30:02 +08:00
triton combine codes (#4509) 2023-08-24 16:43:39 +08:00