main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\offload\base_offload_module.py
25 5 -
A
extensions\csrc\common\target.h
24 1 -
A
colossalai\legacy\inference\dynamic_batching\get_tokenizer.py
5 1 -
A
extensions\csrc\funcs\binary_functor.h
0 1 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\__init__.py
 1 -
A
extensions\csrc\funcs\reduce_function.h
4 1 -
A
colossalai\legacy\inference\dynamic_batching\io_struct.py
41 2 -
A
extensions\csrc\funcs\unary_functor.h
6 1 -
A
colossalai\auto_parallel\offload\region.py
17 3 -
A
extensions\pybind\inference\__init__.py
 1 -
A
extensions\pybind\flash_attention\flash_attention_sdpa_cuda.py
7 1 -
A
extensions\csrc\kernel\cuda\utils\nvgpu_dev_info.h
7 1 -
A
extensions\csrc\kernel\cuda\utils\gpu_launch_config.h
8 1 -
A
extensions\csrc\kernel\cuda\multi_tensor_sgd_kernel.cu
 1 -
A
extensions\csrc\kernel\cuda\decode_kv_cache_memcpy_kernel.cu
 1 1
A
extensions\csrc\kernel\cuda\activation_kernel.cu
 1 -
A
applications\ColossalChat\coati\ray\detached_trainer_base.py
44 1 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -