ColossalAI/colossalai/inference/engine
Cuiqing Li (李崔卿) bce919708f
[Kernels]added flash-decoidng of triton (#5063)
* added flash-decoidng of triton based on lightllm kernel

* add req

* clean

* clean

* delete build.sh

---------

Co-authored-by: cuiqing.li <lixx336@gmail.com>
2023-11-20 13:58:29 +08:00
..
modeling [Kernels]added flash-decoidng of triton (#5063) 2023-11-20 13:58:29 +08:00
policies [inference] Refactor inference architecture (#5057) 2023-11-19 21:05:05 +08:00
__init__.py [inference] Refactor inference architecture (#5057) 2023-11-19 21:05:05 +08:00
engine.py [inference] Refactor inference architecture (#5057) 2023-11-19 21:05:05 +08:00
microbatch_manager.py [inference] Refactor inference architecture (#5057) 2023-11-19 21:05:05 +08:00