o i@sZddlmZddlZddlmZddlmZddlmZej dddd Z ddd dZ dS)) annotationsN)Callablewraps)Any save_file str | Nonec csddl}|}|zdVW||r#|dkr#||dS|jdddS||r;|dkr;||w|jddw)zRun a cprofile Args: save_file: path to save the profile result. "1" or None will result in printing to stdout. rN1cumtime)sort)cProfileProfileenabledisable dump_stats print_stats)rr profrJ/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm/utils/profiling.pycprofile_context s   rTenabledboolcsdfdd }|S)zDecorator to profile a Python method using cProfile. Args: save_file: Path to save the profile result. If "1", None, or "", results will be printed to stdout. enabled: Set to false to turn this into a no-op funcrcstdfdd }|S)NargsrkwargscsLs |i|St|i|WdS1swYdSN)r)rr)rrrrrwrapper-s   $z,cprofile..decorator..wrapper)rrrrr)rrrr)rr decorator,szcprofile..decoratorN)rrr)rrrrrrcprofile#s  rr)rr)NT)rrrr) __future__r contextlibcollections.abcr functoolsrtypingrcontextmanagerrrrrrrs