MCPcopy
hub / github.com/hpcaitech/ColossalAI / FusedOptimizerLoader

Class FusedOptimizerLoader

colossalai/kernel/kernel_loader.py:98–99  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

96
97
98class FusedOptimizerLoader(KernelLoader):
99 REGISTRY = [FusedOptimizerCudaExtension]
100
101
102class InferenceOpsLoader(KernelLoader):

Callers 7

_calc_l2_normFunction · 0.90
load_fused_optimFunction · 0.90
__init__Method · 0.90
__init__Method · 0.90
__init__Method · 0.90
__init__Method · 0.90
__init__Method · 0.90

Calls

no outgoing calls

Tested by 1

__init__Method · 0.72

Used in the wild real call sites across dependent graphs

searching dependent graphs…