PhZUddlmZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZe r ddlmZmZmZmZdadadagad ed <gZd ed <dadadad Z iZ!eGddZ"e"jGdgde"jGdgddZ$eddZ%dZ&dZ'y)) annotationsN) dataclass) lru_cache)ListSetTuple TYPE_CHECKINGUnion)config)get_benchmark_name)BaseSchedulerNodeExternKernelSchedulerNodeNopKernelSchedulerNode SchedulerNodezYList[Tuple[Union[NopKernelSchedulerNode, SchedulerNode, ExternKernelSchedulerNode], int]]nodes_num_elemz%List[Tuple[BaseSchedulerNode, float]] node_runtimescldadadatj t j dadaday)Nr) generated_kernel_countgenerated_cpp_vec_kernel_countnum_bytes_accessedrclearrir_nodes_pre_fusioncpp_to_dtype_countdisable_cpp_wrapperbC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_inductor/metrics.pyresetr-s@%&"rcXeZdZUded<ded<dZded<dZd Zd Zd Ze d Z y ) MetricTablestr table_namez List[str] column_namesrintnum_rows_addedc|jtvry|}t|jt|k(s(Jt|jdt|t |jt |j k(s6Jt |jdt |j t g}||jDcgc]}|| c}z }|j|ycc}w)Nz v.s. )r"enabled_metric_tableslenr#setkeysr _write_row)selfrow_fnrow_dictrow column_names radd_rowzMetricTable.add_rowIs  ??"7"9 9 84$$% *   <$##$ %VCM? ; < 4$$% MMO*   C$##$ %VC ,@+A B C    9J9JK9J+%9JKK Ls' D c"d|jdS)N metric_table_z.csv)r")r,s routput_filenamezMetricTable.output_filename[st/t44rc|j}t|d5}tj|d}|j dg|j zdddy#1swYyxYw)Nw lineterminator model_name)r4opencsvwriterwriterowr#)r,filenamefdr=s r write_headerzMetricTable.write_header^sQ'') (C BZZ48F OO\NT->->> ?! s 7AA&c|j}|jdk(r/tjj |s|j |xjdz c_t |D]&\}}t|tr|d}n|d}n|}|||<(t|d5}tj|d}|j|dddy#1swYyxYw)Nrz.6far7r8) r4r%ospathexistsrA enumerate isinstancefloatr;r<r=r>)r,r/r?idxorig_valnew_valr@r=s rr+zMetricTable._write_rowds'')   ! #BGGNN8,D     q &s^MC(E*%cN!"CH,(C BZZ48F OOC ! s $)CCc.t||}|t|<yN)r REGISTERED_METRIC_TABLES)namer#tables rregister_tablezMetricTable.register_tablexsD,/). &rN) __name__ __module__ __qualname____annotations__r%r1r4rAr+ staticmethodrTrrrr r Bs@ONC$5@ !(//rr slow_fusion) kernel1_pathkernel1_latency kernel2_pathkernel2_latencyfused_kernel_pathfused_kernel_latencyslow_down_ratio graph_stats)graph_idnum_nodes_before_fusionnum_nodes_after_fusionctjD]f\}}|tvs|j}tj j |rt j||jhy)z Purge the old log file at the beginning when the benchmark script runs. Should do it in the parent process rather than the child processes running each individual model. N) rQitemsr'r4rFrGrHunlinkrA)rRrSr?s rpurge_old_log_filesris] 0557 e (* *,,.Hww~~h' (#    8rctj}t}|jdD]9}|j }|s|t vs Jd|d|j |;|S)N,zMetric table name z is not registered)r r'r)splitstriprQadd) config_strenabledrRs rr'r'sp--JeG  %zz|  , , 9 v%7 8 9 , D& Nrc|tvSrP)r'rRs ris_metric_table_enabledrss (* **rc:|tvs Jd|dt|S)Nz Metric table z is not defined)rQrrs rget_metric_tablerus* + +R}TF/-RR + #D ))r)returnzSet[str])( __future__rr<rF dataclassesr functoolsrtypingrrrr r torch._inductorr torch._inductor.utilsr torch._inductor.schedulerr rrrrrrrrXrrrrrrQr rTrir'rsrurrrr~s" !99"4!"   8: 49$ 8/8/ 8/v    !    +*r