MCPcopy Create free account
hub / github.com/THUDM/GLM / cb

Function cb

mpu/random.py:96–98  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

94 if hasattr(_C, '_cuda_setRNGState') and callable(_C._cuda_setRNGState):
95 # older PyTorch
96 def cb():
97 with device_ctx_manager(device):
98 _C._cuda_setRNGState(new_state)
99 else:
100 # newer PyTorch
101 if device == -1:

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected