cerebras.modelzoo.common.dump_context

Provides DumpContext, a debug utility for dumping activations and gradients on a CPU/GPU run, and setting up debug names for dumped WSE activations to be automatically correlated.

Classes

DumpContext

A debug utility context manager.

previous

cerebras.modelzoo.common.checkpoint_utils.CkptInfo

next

cerebras.modelzoo.common.dump_context.DumpContext