Code
Hub
Workspaces
Connect
Indexed graphs
Engine
MCP
copy
hub
/
github.com/hpcaitech/ColossalAI
/ load
Method
load
extensions/base_extension.py:81–82 ·
view source on GitHub ↗
(self)
Source
from the content-addressed store, hash-verified
79
80
@abstractmethod
81
def
load(self) -> Callable:
82
pass
Callers
14
__init__
Method · 0.45
__init__
Method · 0.45
test_smoothquant_linear.py
File · 0.45
test_llama_mlp.py
File · 0.45
test_flash_attn_func
Function · 0.45
test_rotary_embdding_unpad.py
File · 0.45
test_flash_decoding_attention.py
File · 0.45
test_get_cos_and_sin.py
File · 0.45
test_silu_and_mul.py
File · 0.45
test_convert_fp8.py
File · 0.45
test_rms_layernorm.py
File · 0.45
test_kv_cache_memcpy.py
File · 0.45
Calls
no outgoing calls
Tested by
5
__init__
Method · 0.36
__init__
Method · 0.36
test_flash_attn_func
Function · 0.36
exam_lazy_from_pretrained
Function · 0.36
run_moe_dispatch_combine_fwd_bwd
Function · 0.36