WjnddlZddlZddlZddlmZmZmZddlmZm Z dededdfdZ ede dS) N)_register_handler_Request _Response)_ExperimentalConfigprofilereqrespreturnctd}t|d}td|5}t j|dddn #1swxYwYt jdd5}||j | tj |j  d|d ddddS#1swxYwYdS) NT)profile_all_threadsduration) record_shapesexperimental_config torch_debugz.json)prefixsuffixzapplication/json)rfloat get_paramrtimesleeptempfileNamedTemporaryFileexport_chrome_tracename set_contentpathlibPath read_bytes set_status)rr rr proffs f/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/torch/distributed/debug/_handlers.py_torch_profiler$ s|- S]]:..//H t9L M M MQU 8  $M' J J Ja   ((( af--88::r*s MMMMMMMMMM77777777   d    />22222r)