o 灛i%@sddlZddlmZddlZddlZddlZddlm Z m Z ddZ ddZ dd Z d d Zed d dgZeddddddZedddddZedddddZiddejDddejDZddZd d!Zd"d#Zd$d%Zed&kredSdS)'N) namedtuple)COMPUTE_METADATA_SCOPE_NAME TritonHookcCsng}|r.|D]%}|}|D]}|dd}|||fvr*||dnqq|S|dgd}|S)N(r (inc))lowersplitstripappend)metrics raw_metricsretmetric raw_metricraw_metric_no_unitrX/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/triton/profiler/viewer.pymatch_available_metrics s rcCs.t|}|d}tj|}|||fS)N)jsonloadpopht GraphFrame from_literalshow_metric_columns)filedatabase device_infogfrrrget_raw_metricss   r c Cshtjd|jdgd}|D]}||D]}|||d}|||d}|||d}tjD]}|d|k} || } d|| jvrDq/d } |d kru|d krUd |d } nE|dkr`d|d } n:|dkrt|d|ddd|d } n%|dkr|dkrd|d } n|dks|dkrd|d } ntd||j| df| d|d | 7<q/qq |S)Nmin_timeindexcolumnsarchnum_sms clock_rateDeviceIdflopsrCUDA80g2C89gbB90rgx:AimBHIPgfx90ag//bBgfx941gfx942g?y"CzUnsupported device type: ) pd DataFramer$r flops_widthr% ValueErrorlocfillna) dfrmin_time_flops device_type device_indexr&r'r(widthidx device_frames max_flopsrrrget_min_time_flops s>     ,rDc Cstjd|jdgd}|D]<}||D]5}|d|k}||}|||d}|||d}d||dd } |j|df|d | 7<qq |S) Nr!r"r#r)memory_clock_rate bus_widthg@@r-bytes)r6r7r$r:) r<rmin_time_bytesr>r?rArBrErFpeak_bandwidthrrrget_min_time_bytesAs   rK FactorDictnamefactortimergMbP?gư>g& .>)time/sztime/msztime/usztime/nsr*geAr1)zflop/szgflop/sztflop/srH)zbyte/szgbyte/sztbyte/scCi|]}|tqSr)flops_factor_dict.0keyrrr T rVcCrQr)bytes_factor_dictrSrrrrVVrWcCsg}g}ttjg|d}tjd|dddd}|D]}|dkrVt|j|} t|j|} |j|tj|tjd} | d| dt | |jd <| d q |t vrt |} | j} | j}t| g|d}|j||j|tj||||j|d <| |d q |tjvrtjd|dd}|j|tj|tj||j|d <| |d q | |q |rt||}||S) Nr/rr)utilrPr"z util (inc)r) rtime_factor_dictrMrrK dataframerDrNcombinemaxr derivable_metrics)rr r rderived_metricsoriginal_metricstime_metric_name time_unitrrIr=time_secderiveable_metric metric_namemetric_factor_dictmatched_metric_namemetric_time_unitrrrderive_metrics[sD"      rkc Cst|do}t|\}}} t|dksJd|t|||| }|s&|r=|r.d|dn|} dd| ig} |j| dd }ddtdig} |j| dd }|rad|dd |ig} |j| dd }t|j|d|d d WddS1swwYdS) Nrrz"No metrics found in the input filez^(?!z).**rMT)squashz>= F) metric_column expand_namedepth render_header) openr lenupdate_inclusive_columnsrkfilterrprinttree) r filenameincludeexclude thresholdrqfrr r name_filterqueryrrrparse~s  "rcCszt|d.}t|\}}}td|r*|D]}|dd}td|q WddS1s6wYdS)NrlzAvailable metrics:rrz- )rsr rwrr r) file_namer}_r rrrrr show_metricss $rc Cstjdtjd}|jddddd|jdd tdd d |jd d tddd |jddtddd |jddtddd |jddtddd |\}}t|dksSJd|d}|j r`|j dnd}|j }|j }|j }|j}|rv|rvtd|jrt|dS|rt||||||dSdS)Nz,Performance data viewer for proton profiles.) descriptionformatter_classz-lz--list store_truea`List available metrics. Metric names are case insensitive and ignore units. Derived metrics can be created when source metrics are available. - time/s, time/ms, time/us, time/ns: time - flop/s, gflop/s, tflop/s: flops / time - byte/s, gbyte/s, tbyte/s: bytes / time - util: max(sum(flops) / peak_flops_time, bytes / peak_bandwidth_time)) )actionhelpz-mz --metricszAt maximum two metrics can be specified, separated by comma. There are two modes: 1) Choose the output metric to display. It's case insensitive and ignore units. 2) Derive a new metric from existing metrics. )typedefaultrz-iz --includez?Include frames(kernels) that match the given regular expressionz-ez --excludez?Exclude frames(kernels) that match the given regular expressionz-tz --thresholdzrExclude frames(kernels) whose metrics are below the given threshold. This filter only applies on the first metric.z-dz--depthdz The depth of the tree to displayrzMust specify a file to readr,z'Cannot specify both include and exclude)argparseArgumentParserRawTextHelpFormatter add_argumentstrfloatintparse_known_argsrtr rrzr{r|rqr9listrr) argparserargs target_argsrr rzr{r|rqrrrmainsx     r__main__)r collectionsrrpandasr6hatchetrtriton.profiler.hookrrrr rDrKrLr\rRrXrNkeysr`rkrrr__name__rrrrs< ! # M