Skip to main content

Module prof

Module prof 

Source
Expand description

Wall-clock phase counters for the bake profiler (CMF_BAKE_PROF=1 prints them). Written from worker threads, so atomics; nanoseconds.

Statics§

ATTN_FWD
BWD
FFN_FWD
GEMM
GEMM_CALLS
SHAPES
(n, k, m) → (calls, ns). A handful of shapes dominate a training step; naming them decides which ones are worth a GPU submit and which are cheaper computed in place.

Functions§

add
gemm_shape
shape_report
take
Read and reset, in seconds.