subblock_stats

Modules

modelopt.torch.puzzletron.subblock_stats.calc_subblock_stats

Calc subblock stats to compute memory and runtime statistics for subblocks.

Subblock statistics collection for Puzzletron.