2@j 6ddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZGddZGd d Zed gd Zd ZGddZedddgZGddeZdS)N) namedtuple)Sequence)contextmanager)cached_property)configc*eZdZdZdgZdZdZdZdS)RecordLLVMPassTimingsLegacy9A helper context manager to track LLVM pass timings. _datac.tjd|S))Enables the pass timing in LLVM. T)llvmset_time_passesselfs MC:\Projects\Historical\venv\Lib\site-packages\numba/misc/llvm_pass_timings.py __enter__z%RecordLLVMPassTimingsLegacy.__enter__s T""" c^tj|_tjddS)2Reset timings and save report internally. FN)rreport_and_reset_timingsr rrexc_valexc_typeexc_tbs r__exit__z$RecordLLVMPassTimingsLegacy.__exit__s*244  U###rc*t|jSztRetrieve timing data for processing. Returns ------- timings: ProcessedPassTimings ProcessedPassTimingsr rs rgetzRecordLLVMPassTimingsLegacy.get!$DJ///rN)__name__ __module__ __qualname____doc__ __slots__rrr!rrr r sT I 00000rr c2eZdZdZddgZdZdZdZdZdS) RecordLLVMPassTimingsr r _pbc"||_d|_dSN)r+r )rpbs r__init__zRecordLLVMPassTimings.__init__1s rc8|j|S)r )r+start_pass_timingrs rrzRecordLLVMPassTimings.__enter__5s ""$$$ rcB|j|_dS)rN)r+finish_pass_timingr rs rrzRecordLLVMPassTimings.__exit__;sX0022 rc*t|jSrrrs rr!zRecordLLVMPassTimings.getAr"rN) r#r$r%r&r'r/rrr!r(rrr*r*+sf% I  00000rr*PassTimingRecord) user_time user_percent system_timesystem_percentuser_system_timeuser_system_percent wall_time wall_percent pass_name instructionc|djdksJfdfddDtd|}fd}tt||S)zAdjust timing records because of truncated information. Details: The percent information can be used to improve the timing information. Returns ------- res: List[PassTimingRecord] TotalcZ|d|dtjfd}|S)N_time_percentcD}||zdz}||<|S)z'Compute percent x total_time = adjustedg{Gz?r()dtotaladjusted percent_attr time_attr time_getter total_recs radjustz6_adjust_timings..make_adjuster..adjustns3K **Eq.5H#AiLHr)operator attrgetter)attrrNrJrKrLrMs @@@r make_adjusterz&_adjust_timings..make_adjusteris^NNN ((( ))44          rc&g|] }|Sr(r().0xrRs r z#_adjust_timings..xs0 ar)usersystem user_systemwallc*|Sr-)_asdictrUs rz!_adjust_timings..}s!))++rc<D] }||}tdi|S)Nr()r5)rGfnadj_fnss rchainedz _adjust_timings..chaineds5  B1AA$$!$$$r)r>maplist)recordsdictsrbrarRrMs @@@r_adjust_timingsrg\s I  ' ) ) ) )     "KG %%w / /E%%%%% GU## $ $$rcZeZdZdZdZdZdZdZdZdZ dd Z e d Z d Z d S)r zA class for processing raw timing report from LLVM. The processing is done lazily so we don't waste time processing unused timing information. c||_dSr- _raw_data)rraw_datas rr/zProcessedPassTimings.__init__s !rc*t|jSr-)boolrkrs r__bool__zProcessedPassTimings.__bool__sDN###rc|jS)zWReturns the raw string data. Returns ------- res: str rjrs r get_raw_dataz!ProcessedPassTimings.get_raw_datas ~rc@|djS)zhCompute the total time spend in all passes. Returns ------- res: float rA) list_recordsr<rs rget_total_timez#ProcessedPassTimings.get_total_times  ""2&00rc|jS)z{Get the processed data for the timing report. Returns ------- res: List[PassTimingRecord] ) _processedrs rrsz!ProcessedPassTimings.list_recordss rc|}tjd}tj||dd|S)aReturns the top(n) most time-consuming (by wall-time) passes. Parameters ---------- n: int This limits the maximum number of items to show. This function will show the ``n`` most time-consuming passes. Returns ------- res: List[PassTimingRecord] Returns the top(n) most time-consuming passes in descending order. r<NrA)rsrOrPheapqnlargest)rnrekeys rlist_topzProcessedPassTimings.list_topsA##%%!+..~a"s333rrc gd|zfd}|d|dd|d||D]'}|d|jdd|jd d |j(d S) aReturn a string summarizing the timing information. Parameters ---------- topn: int; optional This limits the maximum number of items to show. This function will show the ``topn`` most time-consuming passes. indent: int; optional Set the indentation level. Defaults to 0 for no indentation. Returns ------- res: str  c:|dSr-)append)argbufprefixs rapz(ProcessedPassTimings.summary..aps& JJ&'#'' ( ( ( ( (rzTotal .4fsz Top timings:z zs (5z%)  )rtr|r<r=r>join)rtopnindentrprrs @@rsummaryzProcessedPassTimings.summarysv ) ) ) ) ) )  0D'')) 0 0 0 0111 >t$$ L LA BJAKJJJJJJQ[JJ K K K Kyy~~rc*|S)ztA cached property for lazily processing the data and returning it. See ``_process()`` for details. )_processrs rrvzProcessedPassTimings._processeds }}rc`d}t||j}t|S)zParses the raw string data from LLVM timing report and attempts to improve the data by recomputing the times (See `_adjust_timings()``). c3K|}d}d|d}t|}dddddd d |D]>}tj||}|r%tjd |}fd |D}n?|d d ksJg} d} d} |dd D]J} | dkr| | z } | | d| | d| d| d| dz } Ki} t jD]} | | vr | d krd| | <| dz } |D]}tj| |}|wt| }dt| |D}| | |d }t dd |i|}|V|j dkrnd |}d|vrdS|rtd|dS)zwA generator that parses the raw_data line-by-line to extract timing information for each pass. z [a-zA-Z+ ]+z(?:\s*-+z-+)+rWrXrYrZr?r>)z User Timez System Timez User+Systemz Wall TimeInstrNamez[a-zA-Z][a-zA-Z+ ]+cDg|]}|Sr()strip)rTk header_maps rrVz@ProcessedPassTimings._process..parse..s&JJJz!''))4JJJrrAz\s*((?:[0-9]+\.)?[0-9]+)NrDrEz\s+(?:z\s*\(z%\)|-+)z\s*(.*)c<i|]\}}||t|ndS)Nr)float)rTrvs r z@ProcessedPassTimings._process..parse..s<>>> $11=uQxxxc>>>rrBrz Analysis execution timing reportz'unexpected text after parser finished: r() splitlinesiterrematchfindallrr5_fieldsrdgroupszipupdater>r ValueError)rllines colheadermulticolheaders line_iterlnm raw_headersheadersattrsrzpatrmissingdatar>rec remainingrs @rparsez,ProcessedPassTimings._process..parses''))E&I9)999OU I$',#&# J   H_b11"$*-CR"H"HKJJJJkJJJGE  2;+----E+ACSbS\ 8 8 %%1HCCLLA---LLA0007Q77Q7777CCG%- % %E>>a;&6&6!$GAJ : C  HS"%%=#AHHJJ//H>>(+E8(<(<>>>DKK((( ( I*"+/3CIII}// ),,I2Y>>  JyJJ  r)rdrkrg)rrres rrzProcessedPassTimings._processs> H H H VuuT^,,--w'''rN)r}r)r#r$r%r&r/rorqrtrsr|rrrvrr(rrr r s """$$$111444$6_R(R(R(R(R(rr NamedTimingsnametimingsceZdZdZdZedZedZdZdZ dZ e dZ dd Z d Zd Zd ZdS)PassTimingsCollectionzA collection of pass timings. This class implements the ``Sequence`` protocol for accessing the individual timing records. c"||_g|_dSr-)_name_records)rrs rr/zPassTimingsCollection.__init__Es  rc#KtjrZt5}dVdddn #1swxYwY|}|r|||dSdSdVdS)aRecord new timings and append to this collection. Note: this is mainly for internal use inside the compiler pipeline. See also ``RecordLLVMPassTimingsLegacy`` Parameters ---------- name: str Name for the records. N)rLLVM_PASS_TIMINGSr r!_append)rrrrs r record_legacyz#PassTimingsCollection.record_legacyIs  # ,.. '               ++--C ( T3''''' ( ( EEEEEs .22c#Ktjr[t|5}dVdddn #1swxYwY|}|r|||dSdSdVdS)aRecord new timings and append to this collection. Note: this is mainly for internal use inside the compiler pipeline. See also ``RecordLLVMPassTimings`` Parameters ---------- name: str Name for the records. N)rrr*r!r)rrr.rrs rrecordzPassTimingsCollection.recordbs  # &r** g               ++--C ( T3''''' ( ( EEEEEs /33cV|jt||dS)zAppend timing records Parameters ---------- name: str Name for the records. timings: ProcessedPassTimings the timing records. N)rrr)rrrs rrzPassTimingsCollection._append{s* \$8899999rcP|jrtd|jDSdS)zComputes the sum of the total time across all contained timings. Returns ------- res: float or None Returns the total number of seconds or None if no timings were recorded c3HK|]}|jVdSr-rrt)rTrs r z7PassTimingsCollection.get_total_time..s0IIaqy//11IIIIIIrN)rsumrs rrtz$PassTimingsCollection.get_total_times2 = II4=IIIII I4rc2t|jddS)zReturns the timings in descending order of total time duration. Returns ------- res: List[ProcessedPassTimings] c4|jSr-rr]s rr^z:PassTimingsCollection.list_longest_first..sAI$<$<$>$>rT)r{reverse)sortedrrs rlist_longest_firstz(PassTimingsCollection.list_longest_firsts(dm>>"$$$ $rc|j S)z rrs ris_emptyzPassTimingsCollection.is_emptys=  rr}c|jrdSg}|j}|d|j|}|d|dt |jD]o\}}|d|d|j|j|z dz}|d|d d ||j|d pd |S)aReturn a string representing the summary of the timings. Parameters ---------- topn: int; optional, default=5. This limits the maximum number of items to show. This function will show the ``topn`` most time-consuming passes. Returns ------- res: str See also ``ProcessedPassTimings.summary()`` zNo pass timings were recordedzPrinting pass timings for z Total time: rz== #rdz Percent: z.1f%)rrr) rrrrt enumeraterrrrr)rrrr overall_timeirpercents rrzPassTimingsCollection.summarys = "22CB B8DJ88 9 9 9..00L B0l000 1 1 1!$-00 ; ;1&!&&af&&''')2244|CcI.....///19$$$q$99::::99S>> !rc|j|S)zGet the i-th timing record. Returns ------- res: (name, timings) A named tuple with two fields: - name: str - timings: ProcessedPassTimings r)rrs r __getitem__z!PassTimingsCollection.__getitem__s}Qrc*t|jS)z#Length of this collection. )lenrrs r__len__zPassTimingsCollection.__len__s4=!!!rc*|Sr-)rrs r__str__zPassTimingsCollection.__str__s||~~rN)r})r#r$r%r&r/rrrrrtrpropertyrrrrrr(rrrr>s ^0^0 : : :    $ $ $!!X! """"<    """ rr)rrOrx collectionsrcollections.abcr contextlibr functoolsr numba.corerllvmlite.bindingbindingrr r*r5rgr rrr(rrrs  """"""$$$$$$%%%%%%%%%%%%00000000:00000000@:   "*%*%*%Zo(o(o(o(o(o(o(o(dz.69*=>> YYYYYHYYYYYr