o i24@sddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZGdddZGd d d Zed gd Zd dZGdddZedddgZGdddeZdS)N) namedtuple)Sequence)contextmanager)cached_property)configc@s.eZdZdZdgZddZddZddZd S) RecordLLVMPassTimingsLegacy9A helper context manager to track LLVM pass timings. _datacCstd|S))Enables the pass timing in LLVM. T)llvmset_time_passesselfrZ/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/numba/misc/llvm_pass_timings.py __enter__ z%RecordLLVMPassTimingsLegacy.__enter__cCst|_tddS)2Reset timings and save report internally. FN)r report_and_reset_timingsr r rexc_valexc_typeexc_tbrrr__exit__s  z$RecordLLVMPassTimingsLegacy.__exit__cC t|jSztRetrieve timing data for processing. Returns ------- timings: ProcessedPassTimings ProcessedPassTimingsr r rrrget! zRecordLLVMPassTimingsLegacy.getN)__name__ __module__ __qualname____doc__ __slots__rrrrrrrrs  rc@s8eZdZdZddgZddZddZdd Zd d Zd S) RecordLLVMPassTimingsrr _pbcCs||_d|_dSN)r&r )rpbrrr__init__1 zRecordLLVMPassTimings.__init__cCs|j|S)r )r&start_pass_timingr rrrr5rzRecordLLVMPassTimings.__enter__cCs|j|_dS)rN)r&finish_pass_timingr rrrrr;s zRecordLLVMPassTimings.__exit__cCrrrr rrrrArzRecordLLVMPassTimings.getN) r r!r"r#r$r)rrrrrrrr%+s r%PassTimingRecord) user_time user_percent system_timesystem_percentuser_system_timeuser_system_percent wall_time wall_percent pass_name instructioncs\|djdks JfddfdddDtdd |}fd d }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] Totalcs4|d|dtfdd}|S)N_time_percentcs$}||d}||<|S)z'Compute percent x total_time = adjustedg{Gz?r)dtotaladjusted) percent_attr time_attr time_getter total_recrradjustnsz6_adjust_timings..make_adjuster..adjust)operator attrgetter)attrrC)rB)r?r@rAr make_adjusteris   z&_adjust_timings..make_adjustercsg|]}|qSrr).0x)rGrr xsz#_adjust_timings..)usersystem user_systemwallcS|Sr')_asdictrIrrr}sz!_adjust_timings..cs D]}||}qtdi|S)Nr)r-)r<fn)adj_fnsrrchaineds z _adjust_timings..chained)r6maplist)recordsdictsrUr)rTrGrBr_adjust_timings\s    rZc@s^eZdZdZddZddZddZdd Zd d Zd d Z dddZ e ddZ ddZ dS)rzA class for processing raw timing report from LLVM. The processing is done lazily so we don't waste time processing unused timing information. cCs ||_dSr' _raw_data)rraw_datarrrr) zProcessedPassTimings.__init__cCrr')boolr\r rrr__bool__r^zProcessedPassTimings.__bool__cC|jS)zWReturns the raw string data. Returns ------- res: str r[r rrr get_raw_dataz!ProcessedPassTimings.get_raw_datacCs|djS)zhCompute the total time spend in all passes. Returns ------- res: float r8) list_recordsr4r rrrget_total_timesz#ProcessedPassTimings.get_total_timecCra)z{Get the processed data for the timing report. Returns ------- res: List[PassTimingRecord] ) _processedr rrrrdrcz!ProcessedPassTimings.list_recordscCs(|}td}t||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. r4Nr8)rdrDrEheapqnlargest)rnrXkeyrrrlist_tops zProcessedPassTimings.list_toprcsvgd|fdd}|d|dd|d||D]}|d|jdd |jd d |jq!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  cs|dSr')append)argbufprefixrrapsz(ProcessedPassTimings.summary..apzTotal .4fsz Top timings:z zs (5z%)  )rerkr4r5r6join)rtopnindentrsprrprsummarys& zProcessedPassTimings.summarycCrO)ztA cached property for lazily processing the data and returning it. See ``_process()`` for details. )_processr rrrrfszProcessedPassTimings._processedcCsdd}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()``). c3s|}d}d|d}t|}dddddd d |D]}t||}|r7td |}fd d |D}nq|dd ks@Jg} d} d} |ddD]&} | dkrW| | 7} qL| | d| | d| d| d| d7} qLi} tjD]} | | vr| d krd| | <qx| d7} |D]8}t| |}|durt| }ddt | |D}| | |d}tdd |i|}|V|j dkrnqd |}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-+)+rKrLrMrNr7r6)z User Timez System Timez User+Systemz Wall TimeInstrNamez[a-zA-Z][a-zA-Z+ ]+csg|]}|qSr)strip)rHk header_maprrrJsz@ProcessedPassTimings._process..parse..r8z\s*((?:[0-9]+\.)?[0-9]+)Nr:r;z\s+(?:z\s*\(z%\)|-+)z\s*(.*)cSs&i|]\}}||durt|ndqS)Nr)float)rHrvrrr sz@ProcessedPassTimings._process..parse..r9rwz Analysis execution timing reportz'unexpected text after parser finished: r) splitlinesiterrematchfindallrnr-_fieldsrWgroupszipupdater6rx ValueError)r]lines colheadermulticolheaders line_iterlnm raw_headersheadersattrsripatrmissingdatar6rec remainingrrrparsesz          z,ProcessedPassTimings._process..parse)rWr\rZ)rrrXrrrr}sKzProcessedPassTimings._processN)rlr)r r!r"r#r)r`rbrerdrkr|rrfr}rrrrrs     r NamedTimingsnametimingsc@sveZdZdZddZeddZeddZdd Zd d Z d d Z e ddZ dddZ ddZddZddZdS)PassTimingsCollectionzA collection of pass timings. This class implements the ``Sequence`` protocol for accessing the individual timing records. cCs||_g|_dSr')_name_records)rrrrrr)Er*zPassTimingsCollection.__init__ccs^tjr*t }dVWdn1swY|}|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_TIMINGSrr_append)rrrrrrr record_legacyIs  z#PassTimingsCollection.record_legacyccs`tjr+t| }dVWdn1swY|}|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%rr)rrr(rrrrrrecordbs  zPassTimingsCollection.recordcCs|jt||dS)zAppend timing records Parameters ---------- name: str Name for the records. timings: ProcessedPassTimings the timing records. N)rrnr)rrrrrrr{s zPassTimingsCollection._appendcCs|jr tdd|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 css|]}|jVqdSr'rre)rHrrrr sz7PassTimingsCollection.get_total_time..N)rsumr rrrres z$PassTimingsCollection.get_total_timecCst|jddddS)zReturns the timings in descending order of total time duration. Returns ------- res: List[ProcessedPassTimings] cSs |jSr'rrQrrrrRs z:PassTimingsCollection.list_longest_first..T)rjreverse)sortedrr rrrlist_longest_firstsz(PassTimingsCollection.list_longest_firstcCs|j S)z rr rrris_emptyszPassTimingsCollection.is_emptyrlcCs|jrdSg}|j}|d|j|}|d|dt|jD]+\}}|d|d|j|j|d}|d|d d ||jj|d d q#d |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: rtz== #rmdz Percent: z.1f%)ryrzrw) rrnrre enumeraterrrr|rx)rryrqrs overall_timeirpercentrrrr|s zPassTimingsCollection.summarycCs |j|S)zGet the i-th timing record. Returns ------- res: (name, timings) A named tuple with two fields: - name: str - timings: ProcessedPassTimings r)rrrrr __getitem__s z!PassTimingsCollection.__getitem__cCr)z#Length of this collection. )lenrr rrr__len__s zPassTimingsCollection.__len__cCrOr')r|r rrr__str__szPassTimingsCollection.__str__N)rl)r r!r"r#r)rrrrrerpropertyrr|rrrrrrrr>s       r)rrDrg collectionsrcollections.abcr contextlibr functoolsr numba.corerllvmlite.bindingbindingr rr%r-rZrrrrrrrs(       -3