a dh@sdZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlZddlmZeeZe eeeeefZe eZe eeefZe eZGdddeZdS) zRsz8SimpleProfiler._make_report_extended..cSs|dS)NrxrrrVz6SimpleProfiler._make_report_extended..Tkeyreversecss|]}|dVqdS)Nr)r'r/rrr Wr1z7SimpleProfiler._make_report_extended..)rrrritemssortr&)rreport total_callsrr*r_make_report_extendedPs z$SimpleProfiler._make_report_extendedcCs*dd|jD}|jdddd|S)NcSs&g|]\}}|t|t|fqSr)r#r$r&)r'actionr)rrrr,[r1z/SimpleProfiler._make_report..cSs|dS)Nrr.rrrr0\r1z-SimpleProfiler._make_report..Tr2)rr7r8)rr9rrr _make_reportZszSimpleProfiler._make_reportc stjd}|jdur(||jd7}|d7}|jr2t|jdkrtdd|jDt t t t t t dfdd }|d d d d d}t| }d|}||||7}| \}}}||dd||dd7}||7}|D]0\} } } }} ||| | d| |d| d7}q||7}ntdd|jDt t t t dfdd } | d d d }t| }d|}||||7}| }|D]$\} } }|| | | d|d7}q||7}|7}|S)N zProfiler Reportrcss|]}t|VqdSNr%r'krrrr6ir1z)SimpleProfiler.summary..)r<r$ num_callstotalperrcsHd|ddd|dd}|d|dd|dd|dd7}|S)N| <s | <15 |z r)r<r$rErFrGrowZmax_keyseprrlog_row_extendedks""z0SimpleProfiler.summary..log_row_extendedActionzMean duration (s)z Num callszTotal time (s)z Percentage %-ZTotalz.5z100 %css|]}t|VqdSrArBrCrrrr6r1)r<r$rFrcs*d|ddd|dd|ddS)NrHrIrJrKrLrMr)r<r$rFrOrrlog_rowsz'SimpleProfiler.summary..log_row) oslinesepZ_stageupperr r%rmaxkeysstr expandtabsr;r>)rZ output_stringrQZ header_stringZoutput_string_lenZ sep_linesZreport_extendedr:r+r<Z mean_durationrEZ duration_perrTr9rrOrsummary_sN      zSimpleProfiler.summary)NNT)__name__ __module__ __qualname____doc__rrrZrboolrrr"r_TABLE_DATA_EXTENDEDfloatr; _TABLE_DATAr>r\ __classcell__rrrrr "s r )r`loggingrUr collectionsrpathlibrtypingrrrrrnumpyr#Z$pytorch_lightning.profilers.profilerr getLoggerr]logrZrcintZ_TABLE_ROW_EXTENDEDrbZ _TABLE_ROWrdr rrrrs