MCPcopy Create free account
hub / github.com/zai-org/CodeGeeX / __init__

Method __init__

codegeex/megatron/mpu/random.py:148–152  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

146 """
147
148 def __init__(self):
149 # Map from a string name to the cuda rng state.
150 self.states_ = {}
151 # Seeds are just for book keeping and ensure no seed is set twice.
152 self.seeds_ = set()
153
154 def reset(self):
155 """Set to the initial state (no tracker)."""

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected