You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_infer/test_kernels/cuda
flybird11111 2ddf624a86
[shardformer] upgrade transformers to 4.39.3 (#5815)
6 months ago
..
__init__.py
test_convert_fp8.py
test_flash_decoding_attention.py Fix test import 6 months ago
test_get_cos_and_sin.py
test_kv_cache_memcpy.py [Inference] Fix flash-attn import and add model test (#5794) 6 months ago
test_rms_layernorm.py
test_rotary_embdding_unpad.py [shardformer] upgrade transformers to 4.39.3 (#5815) 6 months ago
test_silu_and_mul.py