MCPcopy Create free account
hub / github.com/XueFuzhao/OpenMoE / Capsule

Class Capsule

analysis/colossalai_replace/layer.py:272–274  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

270
271 @dataclasses.dataclass
272 class Capsule:
273 data: torch.Tensor
274 handle: Any = None
275
276 NUM_CHUNK = 4
277 NUM_STAGES = 4

Callers 2

_ep_processMethod · 0.85
_tp_processMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected