mirror of https://github.com/hpcaitech/ColossalAI
ce7ade3882
* add chatglm2 * add * gather needed kernels * fix some bugs * finish context forward * finish context stage * fix * add * pause * add * fix bugs * finish chatglm * fix bug * change some logic * fix bugs * change some logics * add * add * add * fix * fix tests * fix |
||
---|---|---|
.. | ||
cuda_native | ||
jit | ||
triton | ||
__init__.py | ||
op_builder |