cerebras.modelzoo.common.half_dtype

Module which provides utilities for selecting 16-bit floating point representation.

Functions

maybe_to_half_dtype	Return tensor cast to half dtype if on CSX or autocast CPU/GPU ctx.
set_half_dtype_from_params	Sets the half dtype in cstorch from the given model params.

previous

cerebras.modelzoo.common.dump_context.DumpContext

next

cerebras.modelzoo.common.half_dtype.maybe_to_half_dtype