main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalChat\tests\test_lora.py
7 3 -
A
extensions\csrc\kernel\cuda\context_kv_cache_memcpy_kernel.cu
 1 -
A
colossalai\legacy\inference\pipeline\benchmark\benchmark.py
12 2 -
A
extensions\csrc\kernel\cuda\decode_kv_cache_memcpy_kernel.cu
 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\getattr_handler.py
2 2 -
A
extensions\csrc\kernel\cuda\fused_rotary_emb_and_cache_kernel.cu
 1 -
A
colossalai\legacy\inference\pipeline\microbatch_manager.py
37 2 -
A
extensions\csrc\kernel\cuda\layer_norm_kernel.cu
 1 -
A
applications\ColossalChat\coati\experience_buffer\utils.py
15 1 -
A
extensions\pybind\flash_attention\flash_attention_sdpa_cuda.py
7 1 -
A
extensions\pybind\flash_attention\flash_attention_npu.py
7 1 -
A
extensions\csrc\kernel\cuda\utils\micros.h
0 1 -
A
extensions\csrc\kernel\cuda\scaled_upper_triang_masked_softmax_kernel.cu
 1 -
A
.github\workflows\doc_test_on_pr.yml
 8 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
applications\Colossal-LLaMA\train_sft.example.sh
 1 -
A
applications\ColossalChat\coati\utils\__init__.py
 1 -
A
applications\ColossalQA\tests\test_text_splitter.py
 1 -
A
colossalai\fx\tracer\meta_patch\patched_function\activation_function.py
1 3 -
A
colossalai\utils\multi_tensor_apply\__init__.py
 1 -