spdl.pipeline.ProfileHook

class ProfileHook

A hook object that can be used to execute custom code before and after each stage and pipeline profiling.

Methods

pipeline_profile_hook()	A context manager that is executed at the beginning and the end of profile_pipeline() function.
stage_profile_hook(stage, concurrency)	A context manager that is executed around each stage profiling.

abstract pipeline_profile_hook() → Iterator[None]

A context manager that is executed at the beginning and the end of profile_pipeline() function.

abstract stage_profile_hook(stage: str, concurrency: int) → Iterator[None]

A context manager that is executed around each stage profiling.