o 8wi@sdZddlZddlZddlZddlmZddlmZddlm Z m Z ddl Z ddl m Z ddlmZeeZeeeeeefZeeZeeeefZeeZGdd d eZdS) z`z6SimpleProfiler._make_report_extended..Tkeyreversecss|]}|dVqdS)Nr).0r%rrr az7SimpleProfiler._make_report_extended..) rrrritemstorchtensorlensumitemrsort) 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() rr/r0r1r3r4rr2r5)rr7actionr9r: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|VqdSNr2r,krrrr-yr.z)SimpleProfiler.summary..rAmean num_callstotalperr csHd|ddd|dd}|d|dd|dd|dd7}|S)N| <s | <15 |z r)rArJrKrLrMrowmax_keyseprrlog_row_extended{s""z0SimpleProfiler.summary..log_row_extendedActionzMean duration (s)z Num callszTotal time (s)z Percentage %-Totalz.5z100 %csrErFrGrHrrrr-r.cs*d|ddd|dd|ddS)NrNrOrPrQrRrSr)rArJrLrUrrlog_rows*z'SimpleProfiler.summary..log_row) oslinesep_stageupperr r2rmaxstr expandtabsr?rB)r output_stringrX header_stringoutput_string_len sep_linesreport_extendedr>r6rA mean_durationrK duration_perr\r7rrUrsummaryosP (     zSimpleProfiler.summary)NNT)__name__ __module__ __qualname____doc__rrrbrboolrrrr!tuple_TABLE_DATA_EXTENDEDfloatr? _TABLE_DATArBrk __classcell__rrrrr$s, r)rologgingr]r collectionsrpathlibrtypingrrr0typing_extensionsr$pytorch_lightning.profilers.profilerr getLoggerrllogrqrbrsint_TABLE_ROW_EXTENDEDrrr _TABLE_ROWrtrrrrrs