Skip to content

vllm_omni.diffusion.profiler.diffusion_pipeline_profiler

logger module-attribute

logger = init_logger(__name__)

DiffusionPipelineProfilerMixin

stage_durations property

stage_durations: dict[str, float]

clear_profiler_records

clear_profiler_records() -> None

setup_diffusion_pipeline_profiler

setup_diffusion_pipeline_profiler(
    profiler_targets: list[str] | None = None,
    enable_diffusion_pipeline_profiler: bool = False,
) -> None

profiler

profiler(
    name: str, func: Callable, instance: Any
) -> Callable

Timing a function execution.

wrap_methods_by_paths

wrap_methods_by_paths(
    root_obj: Any, method_paths: list[str]
) -> None

Wrap specified methods of an object with profiler.