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'. |
||
---|---|---|
.. | ||
inference.cpp | ||
layer_norm.cpp | ||
moe.cpp | ||
optimizer.cpp | ||
scaled_masked_softmax.cpp | ||
scaled_upper_triang_masked_softmax.cpp |