o Û·i£ã@s8ddlmZmZddlmZeeƒZGdd„deƒZdS)é)ÚABCÚabstractmethod)Ú init_loggerc@sleZdZdZededefdd„ƒZededBfdd„ƒZed d „ƒZede fd d „ƒZ e de fd d„ƒZ dS)Ú ProfilerBasez‚ Abstract base class for all diffusion profilers. Defines the common interface used by GPUWorker and DiffusionEngine. Útrace_path_templateÚreturncCódS)a Start profiling. Args: trace_path_template: Base path (without rank or extension). e.g. "/tmp/profiles/sdxl_run" Returns: Full path of the trace file this rank will write. N©)Úselfrr r úW/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm_omni/diffusion/profiler/base.pyÚstarts zProfilerBase.startNcCr)z’ Stop profiling and finalize/output the trace. Returns: Path to the saved trace file, or None if not active. Nr ©r r r r ÚstopszProfilerBase.stopcCr)z’ Returns a context manager that advances one profiling step. Should be a no-op (nullcontext) when profiler is not active. Nr r r r r Úget_step_context)szProfilerBase.get_step_contextcCr)z.Return True if profiling is currently running.Nr r r r r Ú is_active1szProfilerBase.is_activecCsddl}t| dd¡ƒS)NrÚRANKÚ0)ÚosÚintÚgetenv)Úclsrr r r Ú _get_rank6szProfilerBase._get_rank)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚstrr rrÚboolrÚ classmethodrrr r r r r s  rN)ÚabcrrÚ vllm.loggerrrÚloggerrr r r r Ús