mirror of https://github.com/hpcaitech/ColossalAI
![]() * adapted to rotary_embedding * adapted to nopad rms norm * fix bugs in benchmark * fix flash_decoding.py |
||
---|---|---|
.. | ||
cuda_native | ||
jit | ||
triton | ||
__init__.py | ||
op_builder |