mirror of https://github.com/hpcaitech/ColossalAI
04aca9e55b
* Add get_cos_and_sin kernel * fix code comments * fix code typos * merge common codes of get_cos_and_sin kernel. * Fixed a typo * Changed 'asset allclose' to 'assert equal'. |
||
---|---|---|
.. | ||
__init__.py | ||
inference_ops_cuda.py |