çg%ddlZddlmZddlZddlZddlZddlm Z m Z dZ dZ dZ dZedd d gZed d d dddZedd dddZedd dddZidejDdejDZdZdZdZdZedkr edSdS)N) namedtuple)COMPUTE_METADATA_SCOPE_NAME TritonHookc"g}|r~|D]z}|}|D]a}|dd}|||fvr||dznb{n |dgdz}|S)N(r (inc))lowersplitstripappend)metrics raw_metricsretmetric raw_metricraw_metric_no_units R/var/www/html/ai-engine/env/lib/python3.11/site-packages/triton/profiler/viewer.pymatch_available_metricsr s C *  F\\^^F)   %/%5%5c%:%:1%=%C%C%E%E%K%K%M%M"j*<===JJzH4555E>  1~) Jctj|}|d}tj|}|||fS)N)jsonloadpopht GraphFrame from_literalshow_metric_columns)filedatabase device_infogfs rget_raw_metricsr#sLyH,,q//K  # #H - -B r%%'' 44rc tjd|jdg}|D]2}||D]%}|||d}|||d}|||d}tjD]}|d|k} || } d|| jvr#d } |d kr<|d kr d |d z z } ni|dkr d|d z z } nZ|dkr|dz |zdz dzdz|d z z } n<|dkr$|dkr d|d z z } n'|dks|dkrd|d z z } nt d||j| dfxx| d|d | z z cc<ڐ'4|S)Nmin_timeindexcolumnsarchnum_sms clock_rateDeviceIdflopsrCUDA80g2C89gbB90rgx:AimBHIPgfx90ag//bBgfx941gfx942g?y"CzUnsupported device type: ) pd DataFramer(r flops_widthr) ValueErrorlocfillna) dfr!min_time_flops device_type device_indexr*r+r,widthidx device_frames max_flopss rget_min_time_flopsrH s\#RX |LLLN"ll ' 4 l lL{+L9&AD!+.|) r@r!min_time_bytesrBrCrErFrJrKpeak_bandwidths rget_min_time_bytesrPAs\#RX |LLLN"[[ ' 4 [ [LZ.L0CsGM +K 8 FGZ [ #K0>{KI]->>DqHN  sJ / / /=3IN3Z Z / / / /  [ r FactorDictnamefactortimergMbP?gư>g& .>)time/sztime/msztime/usztime/nsr.geAr5)zflop/szgflop/sztflop/srM)zbyte/szgbyte/sztbyte/sci|] }|t S)flops_factor_dict.0keys r r\T+333  333rci|] }|t SrW)bytes_factor_dictrYs rr\r\Vr]rcg}g}ttjg|d}tjdz|ddddz}|D]}|dkrt |j|} t |j|} |j|tj|tjdz z} | d| dt| z |jd <| d |tvrt|} | j} | j}t| g|d}|j||j|tj|zz ||z |j|d <| |d E|tjvrtjdz|ddz}|j|tj|tj|z z|j|d <| |d | ||rt||}||zS) Nr/rr)utilrUr&z util (inc)r) rtime_factor_dictrRr rP dataframerHrScombinemaxr derivable_metrics)r"r rr!derived_metricsoriginal_metricstime_metric_name time_unitrrNrAtime_secderiveable_metric metric_namemetric_factor_dictmatched_metric_namemetric_time_units rderive_metricsrs[sO.0@0E/F TTUVW!&,/?/E/Ec/J/J1/M/S/STW/X/XYZ/[[I,, V  / kJJN/ kJJN|$459I9PQZ9[9I9PQY9Z:[\H)7 )C)K)KN[eLfhk)l)low)wBL &  " "< 0 0 0 0 ( ( ( 1& 9 +0K!2!9 "9;-"U"UVW"X /1|= F) metric_column expand_namedepth render_header) openr#lenupdate_inclusive_columnsrsfilterrprinttree) r filenameincludeexclude thresholdrzfr"rr! name_filterquerys rparser~s h  b'6q'9'9$K;!###%I### ##%%% Wk;GG  /g /18E-----gK6;/0E5..BA!<AAABC YYuTY * *  /71:'8Y'8'89:E5..B bggGUZ_g``aaa#bbbbbbbbbbbbbbbbbbsC5DDDcNt|d5}t|\}}}td|rV|D]S}|dd}td|T ddddS#1swxYwYdS)NruzAvailable metrics:rrz- )r|r#rr r r ) file_namer_rrrs r show_metricsrs i  +A..; "###  1) 1 1 %/%5%5c%:%:1%=%C%C%E%E%K%K%M%M"/-//0000sA) / 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_argsr}r r rrrrzr=listrr) argparserargs target_argsrr rrrrzs rmainrs)'B 5I              N    N   |    / "2244D+ {  q "? AI)-?dl  %%%4GlGlGI JED7DBCCC yFY F gy'7IuEEEEEFFr__main__)r collectionsrrpandasr:hatchetrtriton.profiler.hookrrrr#rHrPrQrdrXr_rSkeysrhrsrrr__name__rWrrrs"""""" HHHHHHHH   555B   Z vx&8 9 9 :ftPTae&f&fggJw1QU(V(VWWJw1QU(V(VWW33#*//1133333#*//11333 . . .Fbbb*JFJFJFZ zDFFFFFr