MCPcopy
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.pyFile · 0.45
test_llama_mlp.pyFile · 0.45
test_flash_attn_funcFunction · 0.45
test_rotary_embdding_unpad.pyFile · 0.45
test_flash_decoding_attention.pyFile · 0.45
test_get_cos_and_sin.pyFile · 0.45
test_silu_and_mul.pyFile · 0.45
test_convert_fp8.pyFile · 0.45
test_rms_layernorm.pyFile · 0.45
test_kv_cache_memcpy.pyFile · 0.45

Calls

no outgoing calls

Tested by 5

__init__Method · 0.36
__init__Method · 0.36
test_flash_attn_funcFunction · 0.36
exam_lazy_from_pretrainedFunction · 0.36
run_moe_dispatch_combine_fwd_bwdFunction · 0.36