ColossalAI/tests/test_infer/test_kernels/cuda
flybird11111 04516bb756
[upgrade]Upgrade vit (#6308)
* 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>
2025-05-21 16:14:20 +08:00
..
__init__.py
test_convert_fp8.py
test_flash_decoding_attention.py Fix test import 2024-06-10 02:03:30 +00:00
test_get_cos_and_sin.py
test_kv_cache_memcpy.py [Inference] Fix flash-attn import and add model test (#5794) 2024-06-12 14:13:50 +08:00
test_rms_layernorm.py
test_rotary_embdding_unpad.py [upgrade]Upgrade vit (#6308) 2025-05-21 16:14:20 +08:00
test_silu_and_mul.py