SamplingResults.get_warmup_kernel_states()#

SamplingResults.get_warmup_kernel_states(process_state=<function asdict>)[source]#

If available, returns a dictionary of kernel states recorded during warmup, organized by kernel.

The argument process_state is a callable that is used to process the kernel states. The default kernel states in Liesel are dataclasses, which is why the default here is dataclasses.asdict.