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.