mirror of https://github.com/hpcaitech/ColossalAI
![]() * 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'. |
||
---|---|---|
.. | ||
arm | ||
common | ||
cuda | ||
x86 | ||
__init__.py | ||
scaled_softmax.py |