mirror of https://github.com/hpcaitech/ColossalAI
![]() * fused the gate and up proj in mlp * fix code styles * opt auto_grad * rollback test_inference_engine.py * modifications based on the review feedback. * fix bugs in flash attn * Change reshape to view * fix test_rmsnorm_triton.py |
||
---|---|---|
.. | ||
__init__.py | ||
attention.py |