U hd- @sddlZddlZddlZddlmZddlmZddlmZddl m Z ddl m Z ddl mZGdddZed d d d d ddddddg ZddZGdddZedddgZGdddeZdS)N) namedtuple)Sequence)contextmanager)cached_property)configc@s.eZdZdZdgZddZddZddZd S) RecordLLVMPassTimingsz9A helper context manager to track LLVM pass timings. _datacCstd|S)z)Enables the pass timing in LLVM. T)llvmset_time_passesselfr U/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/numba/misc/llvm_pass_timings.py __enter__s zRecordLLVMPassTimings.__enter__cCst|_tddS)z2Reset timings and save report internally. FN)r Zreport_and_reset_timingsrr )r exc_valexc_typeexc_tbr r r__exit__s  zRecordLLVMPassTimings.__exit__cCs t|jS)ztRetrieve timing data for processing. Returns ------- timings: ProcessedPassTimings )ProcessedPassTimingsrr r r rget!szRecordLLVMPassTimings.getN)__name__ __module__ __qualname____doc__ __slots__rrrr r r rrs rPassTimingRecordZ user_timeZ user_percentZ system_timeZsystem_percentZuser_system_timeZuser_system_percent wall_time wall_percent pass_name instructioncs\|djdkstfddfdddDtdd |}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_recr radjustNsz6_adjust_timings..make_adjuster..adjust)operator attrgetter)attrr+)r*)r'r(r)r make_adjusterIs    z&_adjust_timings..make_adjustercsg|] }|qSr r ).0x)r/r r Xsz#_adjust_timings..)usersystem user_systemwallcSs|SN)_asdictr1r r r]z!_adjust_timings..csD] }||}qtf|Sr7)r)r$fn)adj_fnsr rchained_s z _adjust_timings..chained)rAssertionErrormaplist)recordsZdictsr>r )r=r/r*r_adjust_timings<s    rCc@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 ||_dSr7 _raw_data)r raw_datar r r__init__pszProcessedPassTimings.__init__cCs t|jSr7)boolrEr r r r__bool__sszProcessedPassTimings.__bool__cCs|jS)zWReturns the raw string data. Returns ------- res: str rDr r r r get_raw_datavsz!ProcessedPassTimings.get_raw_datacCs|djS)zhCompute the total time spend in all passes. Returns ------- res: float r ) list_recordsrr r r rget_total_timesz#ProcessedPassTimings.get_total_timecCs|jS)z{Get the processed data for the timing report. Returns ------- res: List[PassTimingRecord] ) _processedr r r rrKsz!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. rNr )rKr,r-heapqnlargest)r nrBkeyr r rlist_tops zProcessedPassTimings.list_toprcsvgd|fdd}|d|dd|d||D](}|d|jdd |jd d |jqBd 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|dSr7)append)argbufprefixr rapsz(ProcessedPassTimings.summary..apzTotal .4fsz Top timings:z zs (5z%)  )rLrRrrrjoin)r topnindentrZpr rWrsummarys&zProcessedPassTimings.summarycCs|S)ztA cached property for lazily processing the data and returning it. See ``_process()`` for details. )_processr r r rrMszProcessedPassTimings._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]6}t||}|r6td |}fd d |D}qnq6|dd ks~tg} d} d} |ddD]L} | dkr| | 7} q| | d| | d| d| d| d7} qi} tjD]} | | kr| d krd| | <q| d7} |D]x}t| |}|dk rt | }ddt | |D}| | |d}tfd |i|}|V|j dkrqqd|}|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-+)+r3r4r5r6rr)z User Timez System Timez User+Systemz Wall TimeZInstrNamez[a-zA-Z][a-zA-Z+ ]+csg|]}|qSr )strip)r0kZ header_mapr rr2sz@ProcessedPassTimings._process..parse..r z\s*((?:[0-9]+\.)?[0-9]+)Nr"r#z\s+(?:z\s*\(z%\)|-+)z\s*(.*)cSs&i|]\}}||dk rt|ndqS)Nrj)float)r0rgvr r r sz@ProcessedPassTimings._process..parse..r!r^z'unexpected text after parser finished: ) splitlinesiterrematchfindallr?rUr_fieldsrAgroupszipupdaterr_ ValueError)rFlinesZ colheaderZmulticolheadersZ line_iterlnmZ raw_headersheadersattrsrPpatrgmissingdatarrec remainingr rhrparsesl             z,ProcessedPassTimings._process..parse)rArErC)r rrBr r rrdsFzProcessedPassTimings._processN)rSr)rrrrrGrIrJrLrKrRrcrrMrdr r r rris     r NamedTimingsnametimingsc@sjeZdZdZddZeddZddZdd Zd d Z e d d Z dddZ ddZ ddZddZdS)PassTimingsCollectionzA collection of pass timings. This class implements the ``Sequence`` protocol for accessing the individual timing records. cCs||_g|_dSr7)_name_records)r rr r rrG szPassTimingsCollection.__init__c csBtjr8t }dVW5QRX|}|r>|||ndVdS)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)rZLLVM_PASS_TIMINGSrr_append)r rrrr r rrecord$s zPassTimingsCollection.recordcCs|jt||dS)zAppend timing records Parameters ---------- name: str Name for the records. timings: ProcessedPassTimings the timing records. N)rrUr)r rrr r rr=s zPassTimingsCollection._appendcCs"|jrtdd|jDSdSdS)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|]}|jVqdSr7rrL)r0rr r r Ssz7PassTimingsCollection.get_total_time..N)rsumr r r rrLIs z$PassTimingsCollection.get_total_timecCst|jddddS)zReturns the timings in descending order of total time duration. Returns ------- res: List[ProcessedPassTimings] cSs |jSr7rr9r r rr:_r;z:PassTimingsCollection.list_longest_first..T)rQreverse)sortedrr r r rlist_longest_firstWsz(PassTimingsCollection.list_longest_firstcCs|j S)z rr r r ris_emptybszPassTimingsCollection.is_emptyrScCs|jr dSg}|j}|d|j|}|d|dt|jD]V\}}|d|d|j|j|d}|d|d d ||jj|d d qFd |SdS)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: r[z== #rTdz Percent: z.1f%)r`rar^N) rrUrrL enumeraterrrrcr_)r r`rXrZZ overall_timeirpercentr r rrchszPassTimingsCollection.summarycCs |j|S)zGet the i-th timing record. Returns ------- res: (name, timings) A named tuple with two fields: - name: str - timings: ProcessedPassTimings r)r rr r r __getitem__s z!PassTimingsCollection.__getitem__cCs t|jS)z#Length of this collection. )lenrr r r r__len__szPassTimingsCollection.__len__cCs|Sr7)rcr r r r__str__szPassTimingsCollection.__str__N)rS)rrrrrGrrrrLrpropertyrrcrrrr r r rrs      r)rpr,rN collectionsrcollections.abcr contextlibr functoolsrZ numba.corerZllvmlite.bindingZbindingr rrrCrrrr r r rs8      -.