sg.4ddlZddlZddlZddlZddlZddlmZddlZddlm Z ddl m Z dZ dZ djZdZdd Zd jZdd Zd Zd jZddZdjZdZdZdZedk(reyy)N)Dict)profiler) get_env_infoc0|ddtjddyN)sysargv)new_argvs R/var/www/html/venv/lib/python3.12/site-packages/torch/utils/bottleneck/__main__.py redirect_argvr s1+CHHQKc:|jrd|jSy)Nzcompiled w/ CUDA znot compiled w/ CUDA)cuda_compiled_version)sysinfos r compiled_with_cudars"$$"7#@#@"ABB !r a\ -------------------------------------------------------------------------------- Environment Summary -------------------------------------------------------------------------------- PyTorch {pytorch_version}{debug_str} {cuda_compiled} Running with Python {py_version} and {cuda_runtime} `{pip_version} list` truncated output: {pip_list_output} ctdt}i}d}|jrd}d}|jr|j}|d|z}nd}|j }|j }|d}||jt|tjddtjd |||d }tjd i|S) NzRunning environment analysis...z DEBUGzCUDA zCUDA unavailablezUnable to fetchr.) debug_strpytorch_version cuda_compiled py_version cuda_runtime pip_versionpip_list_output) printris_debug_buildis_cuda_availablecuda_runtime_versionr pip_packages torch_versionrr version_info env_summaryformat)inforesultr cuda_availcudarrs r run_env_analysisr+$s +, >DFI  J ((   4J!""K''O+--+D1))!,-Qs/?/?/B.CD""*F    ' ''r ctdtj}|jt ||d|j |S)Nz!Running your script with cProfile)rcProfileProfileenableexecdisable)codeglobslaunch_blockingprofs r run_cprofiler6Hs< -.    DKKMudLLN Kr z -------------------------------------------------------------------------------- cProfile output -------------------------------------------------------------------------------- ctttj|j |}|j |yr)r cprof_summarypstatsStats sort_stats print_stats)r5sortbytopkcprofile_statss r print_cprofile_summaryr@Xs1 -\\$'226:Nt$r cdfd }td|dg}tjjr|j |d|S|j d|S)NFcztj|5}tdddd|S#1swYSxYw)Nuse_cuda)rprofiler0)rDr5r2r3s r run_profz#run_autograd_prof..run_prof_s9   x 0 $D ud # $  $ s0:z1Running your script with the autograd profiler...rCTF)rtorchr* is_availableappend)r2r3rFr(s`` r run_autograd_profrK^s[  =>& 'F zz  h-. M  d Mr z -------------------------------------------------------------------------------- autograd profiler output ({mode} mode) -------------------------------------------------------------------------------- {description} {cuda_warning} {output} cJgd}|vrd}t|jd|dk(rd}nd}t|jfdd }|d|}|d |d tj j j||d } ttjd i| y)N)cpu_time cuda_timecpu_time_totalcuda_time_totalcountzWARNING: invalid sorting option for autograd profiler results: {} Expected `cpu_time`, `cpu_time_total`, or `count`. Defaulting to `cpu_time`.rMCUDAz Because the autograd profiler uses the CUDA event API, the CUDA time column reports approximately max(cuda_time, cpu_time). Please ignore this output if your code does not use CUDA. rct|Sr)getattr)xr=s r z-print_autograd_prof_summary..sF);r T)keyreverseztop z events sorted by )mode descriptionoutput cuda_warningr) rr&sortedfunction_eventsrHautograd profiler_util _build_tableautograd_prof_summary) r5rYr=r> valid_sortbywarnr\ sorted_events topk_eventsr(s ` r print_autograd_prof_summaryrgxsZL \!, dkk&!" v~X  4//;TKM&KdV#5fX>....;;KH$ F  & & 0 01r a `bottleneck` is a tool that can be used as an initial step for debugging bottlenecks in your program. It summarizes runs of your script with the Python profiler and PyTorch's autograd profiler. Because your script will be profiled, please ensure that it exits in a finite amount of time. For more complicated uses of the profilers, please see https://docs.python.org/3/library/profile.html and https://pytorch.org/docs/main/autograd.html#profiler for more information. ctjt}|jdtd|jdttj d|j S)N)rZ scriptfilezGPath to the script to be run. Usually run with `python path/to/script`.)typehelpargsz2Command-line arguments to be passed to the script.)rjnargsrk)argparseArgumentParserdescript add_argumentstr REMAINDER parse_args)parsers r rtrtsd  $ $ :F  3DE S0B0BQS    r c:td|jDS)Nc34K|]}|jywr)rO).0events r z!cpu_time_total..sOu##Os)sumr^) autograd_profs r rOrOs O1N1NO OOr ct}|j}|jgn |j}|jd|d}d}d}d}t |t j jdtj j|t|d5}t|j|d}ddd|dddd} ttt} tj j#rtj j%t'| } t)|| \} } t| t+| ||tj j#st-| d ||yt/| }t1| j2dkDr/t/| }||z |z }t5|d kDrt-| d ||t-| d ||y#1swY'xYw) NrtottimerOrbr0__main__)__file____name__ __package__ __cached__CPUg?rR)rtrirlinsertr rpathosdirnameopencompilereadrrpr+rHr*rIinitr6rKr@rgrOlenr^abs)rlri scriptargscprofile_sortby cprofile_topkautograd_prof_sortbyautograd_prof_topkstreamr2r3r% cprofile_profautograd_prof_cpuautograd_prof_cudacuda_prof_exec_timecpu_prof_exec_timepct_diffs r mainrs RTfg));<  , ,-1+,=>'*<<@SS x=4  '(95BVXj k 2Frs  #0"  EG !(H EG %   EG2<  EG P1ff zFr