o I&i@sUddlmZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z ddlmZddlmZe rCddlmZmZmZmZdadadagad ed <gZd ed <dadadad dZ iZ!eGdddZ"e"#dgde"#dgdddZ$edddZ%ddZ&ddZ'dS) ) 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_runtimescCs,dadadattdadadadS)Nr) generated_kernel_countgenerated_cpp_vec_kernel_countnum_bytes_accessedrclearrir_nodes_pre_fusioncpp_to_dtype_countdisable_cpp_wrapperrrBC:\wamp64\www\opt\env\Lib\site-packages\torch/_inductor/metrics.pyreset-src@sVeZdZUded<ded<dZded<dd Zd d Zd d ZddZe ddZ dS) MetricTablestr table_namez List[str] column_namesrintnum_rows_addedcs|jtvrdS|t|jtks"Jt|jdtt|jtks=Jt|jdttg}|fdd|jD7}||dS)Nz v.s. csg|]}|qSrr).0Z column_nameZrow_dictrr Xsz'MetricTable.add_row..)renabled_metric_tableslenrsetkeysr _write_row)selfZrow_fnrowrr#radd_rowIs"   zMetricTable.add_rowcCsd|jdS)NZ metric_table_z.csv)r)r*rrroutput_filename[szMetricTable.output_filenamecCsX|}t|d}tj|dd}|dg|jWddS1s%wYdS)Nw lineterminatorZ model_name)r-opencsvwriterwriterowr)r*filenamefdr4rrr write_header^s  "zMetricTable.write_headercCs|}|jdkrtj|s||jd7_t|D]\}}t|tr,|d}n |dur3d}n|}|||<qt |d}t j |dd}| |WddS1sWwYdS)Nrz.6far/r0) r-r!ospathexistsr8 enumerate isinstancefloatr2r3r4r5)r*r+r6idxZorig_valnew_valr7r4rrrr)ds     "zMetricTable._write_rowcCst||}|t|<dSN)rREGISTERED_METRIC_TABLES)namertablerrrregister_tablexs  zMetricTable.register_tableN) __name__ __module__ __qualname____annotations__r!r,r-r8r) staticmethodrHrrrrrBs  rZ slow_fusion)Z kernel1_pathZkernel1_latencyZ kernel2_pathZkernel2_latencyZfused_kernel_pathZfused_kernel_latencyZslow_down_ratioZ graph_stats)Zgraph_idZnum_nodes_before_fusionZnum_nodes_after_fusioncCsFtD]\}}|tvr |}tj|rt||qdS)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) rEitemsr%r-r<r=r>unlinkr8)rFrGr6rrrpurge_old_log_filess   rPreturnSet[str]cCsPtj}t}|dD]}|}|sq |tvs Jd|d||q |S)N,zMetric table name z is not registered)r r%r'splitstriprEadd)Z config_strenabledrFrrrr%s   r%cCs |tvSrD)r%rFrrris_metric_table_enableds rYcCs |tvs Jd|dt|S)Nz Metric table z is not defined)rErXrrrget_metric_tablesrZ)rQrR)( __future__rr3r< dataclassesr functoolsrtypingrrrrr Ztorch._inductorr Ztorch._inductor.utilsr Ztorch._inductor.schedulerr r rrrrrrrLrrrrrrErrHrPr%rYrZrrrrsH     ;