mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-07 02:54:10 +00:00
* fix * fix * fix rotate embedding test * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> |
||
---|---|---|
.. | ||
__init__.py | ||
test_convert_fp8.py | ||
test_flash_decoding_attention.py | ||
test_get_cos_and_sin.py | ||
test_kv_cache_memcpy.py | ||
test_rms_layernorm.py | ||
test_rotary_embdding_unpad.py | ||
test_silu_and_mul.py |