|h24SSKrSSKrSSKrSSKJr SSKJr SSKJr SSK J r SSK J r SSK Jr "SS5r"S S 5r\"S /S Q5rS r"SS5r\"SSS/5r"SS\5rg)N) namedtuple)Sequence)contextmanager)cached_property)configc0\rSrSrSrS/rSrSrSrSr g) RecordLLVMPassTimingsLegacy9A helper context manager to track LLVM pass timings. _datac2[R"S5 U$))Enables the pass timing in LLVM. T)llvmset_time_passesselfs V/home/james-whalen/.local/lib/python3.13/site-packages/numba/misc/llvm_pass_timings.py __enter__%RecordLLVMPassTimingsLegacy.__enter__s T" cd[R"5Ul[R"S5 g)2Reset timings and save report internally. FN)rreport_and_reset_timingsr rrexc_valexc_typeexc_tbs r__exit__$RecordLLVMPassTimingsLegacy.__exit__s%224  U#rc,[UR5$zTRetrieve timing data for processing. Returns ------- timings: ProcessedPassTimings ProcessedPassTimingsr rs rgetRecordLLVMPassTimingsLegacy.get!$DJJ//r)r N) __name__ __module__ __qualname____firstlineno____doc__ __slots__rrr$__static_attributes__rrr r s I 0rr c8\rSrSrSrSS/rSrSrSrSr S r g ) RecordLLVMPassTimings+r r _pbcXlSUlgN)r2r )rpbs r__init__RecordLLVMPassTimings.__init__1s rc:URR5 U$)r)r2start_pass_timingrs rrRecordLLVMPassTimings.__enter__5s ""$ rcBURR5Ulg)rN)r2finish_pass_timingr rs rrRecordLLVMPassTimings.__exit__;sXX002 rc,[UR5$r!r"rs rr$RecordLLVMPassTimings.getAr&r)r r2N) r'r(r)r*r+r,r6rrr$r-r.rrr0r0+s'% I  0rr0PassTimingRecord) user_time user_percent system_timesystem_percentuser_system_timeuser_system_percent wall_time wall_percent pass_name instructionc^^USmTRS:XdeU4SjnSVs/sH o!"U5PM snm[SU5nU4Sjn[[XC55$s snf)zAdjust timing records because of truncated information. Details: The percent information can be used to improve the timing information. Returns ------- res: List[PassTimingRecord] Totalc`>^^^US3mUS3m[R"T5mUUUU4SjnU$)N_time_percentc4>T"T5nXT-S-nX T'U$)z'Compute percent x total_time = adjustedg{Gz?r.)dtotaladjusted percent_attr time_attr time_getter total_recs radjust6_adjust_timings..make_adjuster..adjustns* *E.5H#iLHr)operator attrgetter)attrrYrUrVrWrXs @@@r make_adjuster&_adjust_timings..make_adjusteris;fEN x( )))4    r)usersystem user_systemwallc"UR5$r4)_asdictxs r!_adjust_timings..}s !))+rc<>TH nU"U5nM [S0UD6$)Nr.)r@)rRfnadj_fnss rchained _adjust_timings..chaineds&B1A $!$$r)rImaplist)recordsr^rgdictsrmrlrXs @@r_adjust_timingsrs\st I   ' )) )  #L"KQ a"KG %w /E% G# $$sAc\\rSrSrSrSrSrSrSrSr Sr SS jr \ S 5r S rS rg )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. cXlgr4 _raw_data)rraw_datas rr6ProcessedPassTimings.__init__s!rc,[UR5$r4)boolrxrs r__bool__ProcessedPassTimings.__bool__sDNN##rcUR$)z7Returns the raw string data. Returns ------- res: str rwrs r get_raw_data!ProcessedPassTimings.get_raw_datas~~rc<UR5SR$)zHCompute the total time spend in all passes. Returns ------- res: float rL) list_recordsrGrs rget_total_time#ProcessedPassTimings.get_total_times  "2&000rcUR$)z[Get the processed data for the timing report. Returns ------- res: List[PassTimingRecord] ) _processedrs rr!ProcessedPassTimings.list_recordssrcUR5n[R"S5n[R"XSSU5$)aGReturns 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. rGNrL)rr[r\heapqnlargest)rnrqkeys rlist_topProcessedPassTimings.list_tops9##%!!+.~~a"s33rc ^^/mSU-mUU4SjnU"SUR5SS35 U"S5 URU5H4nU"SURSSURS S UR35 M6 S R T5$) aJReturn 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  c0>TRTU35 gr4)append)argbufprefixs rap(ProcessedPassTimings.summary..aps JJ&#' (rzTotal .4fsz Top timings:z zs (5z%)  )rrrGrHrIjoin)rtopnindentrprrs @@rsummaryProcessedPassTimings.summarysv ) VD'')#.a 01 >t$A AKK$Cq'9Q[[MJ K%yy~rc"UR5$)zdA cached property for lazily processing the data and returning it. See ``_process()`` for details. )_processrs rrProcessedPassTimings._processeds }}rcTSn[U"UR55n[U5$)zParses the raw string data from LLVM timing report and attempts to improve the data by recomputing the times (See `_adjust_timings()``). c 3# UR5nSnSUS3n[U5nSSSSSS S .nUH[n[R"X65nU(dM"[R"S U5nUV s/sHoU R 5PM n n O W S S :Xde/n S n Sn U SS HCn U S:XaX- n MU R U S35 U R U S35 U SU SU S3- n ME 0n[RHn X;dM U S :wdMSX'M U S- n UHn[R"X5nUcM[UR55n[X5V Vs0sHupXb [U5OS_M nn nURU5 US n[SS U0UD6nUv URS:XdM O SRU5nSU;agU(a[!SU35egs sn fs snn f7f)z_A generator that parses the raw_data line-by-line to extract timing information for each pass. z [a-zA-Z+ ]+z(?:\s*-+z-+)+r`rarbrcrJrI)z User Timez System Timez User+Systemz Wall TimeInstrNamez[a-zA-Z][a-zA-Z+ ]+rLz\s*((?:[0-9]+\.)?[0-9]+)NrOrPz\s+(?:z\s*\(z%\)|-+)gz\s*(.*)rMrz Analysis execution timing reportz'unexpected text after parser finished: r.) splitlinesiterrematchfindallstriprr@_fieldsrpgroupszipfloatupdaterIr ValueError)rylines colheadermulticolheaders line_iter header_maplnm raw_headerskheadersattrsrpatmissingvdatarIrec remainings rparse,ProcessedPassTimings._process..parses?'')E&I!))D9OU I$',#&# J HH_11"$**-CR"HK>IJk!'')4kGJ  2;+- --E+ACSb\ %HCLLA3e-LLA3h0fQCuQCw77C "G%-->a;&6!$GJ. : CHHS%=#AHHJ/H(+E(<>(<=uQxcA(<>KK( ( I*"+/3CI}}/  ),I2Y> >ykJ[K6>s=A G:G:.G/ A;G: G:)G:>'G:%G48G:==G:)rprxrs)rrrqs rrProcessedPassTimings._processs) H VuT^^,-w''rrwN)r)r'r(r)r*r+r6r}rrrrrrrrr-r.rrr#r#sD "$14$6R(rr# NamedTimingsnametimingsc|\rSrSrSrSr\S5r\S5rSr Sr Sr \ S 5r SS jrS rS rS rSrg)PassTimingsCollectioni>z|A collection of pass timings. This class implements the ``Sequence`` protocol for accessing the individual timing records. cXl/Ulgr4_name_records)rrs rr6PassTimingsCollection.__init__Es  rc## [R(aA[5nSv SSS5 WR5nU(aUR X5 ggSv g!,(df  N==f7f)zRecord 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_legacy#PassTimingsCollection.record_legacyIsL  # #,.'/++-C T' /.s A/A7A/ A,(A/c## [R(aB[U5nSv SSS5 WR5nU(aUR X5 ggSv g!,(df  N==f7f)zRecord 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)rrr0r$r)rrr5rrs rrecordPassTimingsCollection.recordbsN  # #&r*g+++-C T' +*s!A0A7A0 A-)A0cLURR[X55 g)zAppend timing records Parameters ---------- name: str Name for the records. timings: ProcessedPassTimings the timing records. N)rrr)rrrs rrPassTimingsCollection._append{s \$89rc^UR(a[SUR55$g)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 c3T# UHoRR5v M g7fr4rr).0rs r 7PassTimingsCollection.get_total_time..sI=ayy//11=s&(N)rsumrs rr$PassTimingsCollection.get_total_times# ==I4==II Irc.[URSSS9$)zqReturns the timings in descending order of total time duration. Returns ------- res: List[ProcessedPassTimings] c6URR5$r4rrfs rrh:PassTimingsCollection.list_longest_first..sAII$<$<$>rT)rreverse)sortedrrs rlist_longest_first(PassTimingsCollection.list_longest_firstsdmm>"$ $rc$UR(+$)z rrs ris_emptyPassTimingsCollection.is_emptys==  rcUR(ag/nURnU"SUR35 UR5nU"SUS35 [ UR 5HjupVU"SUSUR 35 URR5U- S-nU"SUS S 35 U"URRUS S 95 Ml S RU5$)a1Return 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) rrrr enumeraterrrrr)rrrr overall_timeirpercents rrPassTimingsCollection.summarys ==2CB +DJJ<8 9..0L l3/0 1!$--0T!AaffX&'))224|CcIZ}A./199$$$q$9: 1 99S> !rc URU$)zGet the i-th timing record. Returns ------- res: (name, timings) A named tuple with two fields: - name: str - timings: ProcessedPassTimings r)rrs r __getitem__!PassTimingsCollection.__getitem__s}}Qrc,[UR5$)z#Length of this collection. )lenrrs r__len__PassTimingsCollection.__len__s4==!!rc"UR5$r4)rrs r__str__PassTimingsCollection.__str__s||~rrN)r)r'r(r)r*r+r6rrrrrrpropertyrrrrrr-r.rrrr>sk 00 :  $!! "<  " rr)rr[r collectionsrcollections.abcr contextlibr functoolsr numba.corerllvmlite.bindingbindingrr r0r@rsr#rrr.rrrs~  "$%%00:00@ "*%Zo(o(d.69*=> YHYr