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

Class CPUAdamLoader

colossalai/kernel/kernel_loader.py:86–87  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

84
85
86class CPUAdamLoader(KernelLoader):
87 REGISTRY = [CpuAdamX86Extension, CpuAdamArmExtension]
88
89
90class LayerNormLoader(KernelLoader):

Callers 2

__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…