.`ilddlmZddlZddlmZddlmZddlmZej d ddZ ddd Z dS)) annotationsN)Callablewraps)Any save_file str | Nonec#Kddl}|}| dV||r|dkr||dS|ddS#||r|dkr||w|dwxYw)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)rrprofs h/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/utils/profiling.pycprofile_contextr sOOO     DKKMMM-    -c)) OOI & & & & &   )  , , , , ,   -c)) OOI & & & &   )  , , , ,s A??A C Tenabledboolcdfd }|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 funcrcDtdfd }|S)Nargsrkwargsczs|i|St5|i|cdddS#1swxYwYdSN)r)rrrrrs rwrapperz,cprofile..decorator..wrapper-s -tT,V,,,!),, - -tT,V,, - - - - - - - - - - - - - - - - - -s 044)rrrrr)rr rrs` r decoratorzcprofile..decorator,sC t - - - - - - -  -)rr)rrr!s`` rcprofiler$#s/        r"r)rr )NT)rr rr) __future__r contextlibcollections.abcr functoolsrtypingrcontextmanagerrr$r#r"rr+s#"""""$$$$$$ -----,r"