o 9wi@sdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlZddlmZddlmZeeZe eeeeefZe eZe eeefZe eZGdd d eZdS) z`z6SimpleProfiler._make_report_extended..Tkeyreversecss|]}|dVqdS)Nr).0r(rrr az7SimpleProfiler._make_report_extended..) rrrritemstorchtensorlensumitemr!sort) rtotal_durationreportadd_tensorlen_dsum_d percentage_d total_callsrrr_make_report_extendedTs   z$SimpleProfiler._make_report_extendedcCs^g}|jD]\}}t|}t|}|||t||fq|jdddd|S)NcSr%)Nrr'rrrr)lr*z-SimpleProfiler._make_report..Tr+) rr2r3r4r6r7r!r5r8)rr:actionr<r=r?rrr _make_reportds zSimpleProfiler._make_reportc stjd}|jdur||jd7}|d7}|jrt|jdkrtdd|jDdtdtd td td td tf fd d }|ddddd}t| }d|}||||7}| \}}}||dd||dd7}||7}|D]\} } } }} ||| | d| |d| d7}q|||7}nRtdd|jDdtdtd td tffdd } | ddd}t| }d|}||||7}| }|D]\} } }|| | | d|d7}q||7}|7}|S)N zProfiler Reportrcs|]}t|VqdSNr5r/krrrr0yr1z)SimpleProfiler.summary..rDmean num_callstotalperrcsHd|ddd|dd}|d|dd|dd|dd7}|S)N| <s | <15 |z r)rDrMrNrOrProwmax_keyseprrlog_row_extended{s""z0SimpleProfiler.summary..log_row_extendedActionzMean duration (s)z Num callszTotal time (s)z Percentage %-Totalz.5z100 %csrHrIrJrKrrrr0r1cs*d|ddd|dd|ddS)NrQrRrSrTrUrVr)rDrMrOrXrrlog_rows*z'SimpleProfiler.summary..log_row) oslinesep_stageupperrr5rmaxstr expandtabsrBrE)r output_stringr[ header_stringoutput_string_len sep_linesreport_extendedrAr9rD mean_durationrN duration_perr_r:rrXrsummaryosP (     zSimpleProfiler.summary)NNT)__name__ __module__ __qualname____doc__rrrerboolrr rr$r_TABLE_DATA_EXTENDEDfloatrB _TABLE_DATArErn __classcell__rrrrr $s, r )rrloggingr`r collectionsrpathlibrtypingrrrrrr3typing_extensionsr $lightning.pytorch.profilers.profilerr getLoggerrologreruint_TABLE_ROW_EXTENDEDrt _TABLE_ROWrvr rrrrs