<Цid@SSKJr SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK J r J r SSK J r Jr SSKJr SSKJrJrJrJrJrJrJr SSKJr SSKJr \(aSS KJr SS KJ r J!r!J"r" \"\#5RI5RJRJr&"S S \5r'S r(S#Sjr)\"S5r*\"S5r+S$Sjr,S%Sjr-\R\S&Sj5r/S'Sjr0\RbS(Sj5r2S)Sjr3"SS5r4S*S+Sjjr5S,S-Sjjr6S.S/Sjjr7S0Sjr8S1Sjr9S2Sjr:"SS 5r;"S!S"\\*5rNs 7s  c## Sv g![acnU"5n[R"US5nUR(aURSSU3OUnU4URSS-UleSnAff=f7f)Nz r ) Exceptiontextwrapindentargs)msg_fnemsgs r&contextrIVsn hooc4('(vv 2cU#3!&&*$ s%A9 A9 A6AA11A66A9cD[S[U5R35e)NzUnhandled type: )AssertionErrortyper)r:s r& assert_neverrMfs +DG,<,<+=> ??r%c.[R"U5$r7)r from_file) template_fns r&_read_templaterQjs  ! !+ ..r%c[R"URS55R5n[R USS9$)Nlatin1little) byteorder)hashlibsha1encodedigestint from_bytes)srWs r&string_stable_hashr]ps5 <<* + 2 2 4D >>$(> 33r%c\rSrSr%S\S'S\S'S\S'S\S'SS jrSS jrSS jrSS jrSS jr SS.SSjjr SSjr SSjr Sr g) FileManagerxstr install_dir template_dirbooldry_runset[str] filenamescFXlX l[5UlX0lgr7)rbrcsetrgre)selfrbrcres r&__init__FileManager.__init__~s&( r%ct[U5nUR5nSSS5 UW:waY[R"[R R U5SS9 [US5nURU5 SSS5 gg!,(df  Nn=f![a SnNf=f!,(df  g=f)NT)exist_okw)openreadOSErrorosmakedirspathdirnamewrite)rjfilenamecontentsf old_contentss r&_write_if_changedFileManager._write_if_changeds h1 vvx   | # KK1D Ah$!%$ $   L %$s9 BBB*B) BBB B&%B&) B7c[RRURU5nU"5n[ U[ 5(aSU;a[ S- S- n[[RSR=(d U5R5nUR[ 5R5n0UESSSUSU3-0En[%U5nUR'U5$[ U[(5(aU$[+U5 g![[4a UR5nNf=f![ a UR"nNf=f)Ngenerated_commenttorchgenzgen.py__main__@z generated by z from )rsrujoinrc isinstancedict REPO_ROOTrsysmodules__file__absoluteKeyErrorAttributeError relative_toas_posix ValueErrorr0rQ substituterarM) rjrP env_callable template_pathenvgenerator_default generatorgenerator_pathtemplates r&substitute_with_template$FileManager.substitute_with_templates: T%6%6 D n c4 "#-$- $:X$E!= $ J/88MUR5H!up#UT;d SU35eX==U- ss'M# g)Nzundeclared sharded key )items)intofrom_kv sharded_keyss r& merge_env,FileManager.write_sharded..merge_envs; L(G,CA3*GG(1 &r%r.c>T$r7r)shardsr&+FileManager.write_sharded..s5r%r)rdict[str, list[str]]rrreturnNone) rangeupdaterlistcopyrer]rfindlenrrgdiscardrb)rjrxrkey_fnr num_shardsrr everythingishards all_shardskeyritemsidrdot_pos base_filename extensionrrs ` @r& write_shardedFileManager.write_shardeds'1,%? +0+<( +,FileManager.write_outputs..sN7Mt#*s*7MsN)formatrsortedrgr|)rj variable_namerxcontents r& write_outputsFileManager.write_outputs sB$**  MMNvdnn7MN N  x1r%c[RRUR[RR [ 55$)z This needs to be deterministic. The template dir is an absolute path that varies across builds. So, just use the path relative to this file, which will point to the codegen source but will be stable. )rsrurelpathrcrvrrjs r&template_dir_for_comments%FileManager.template_dir_for_commentss, wwt00"''//(2KLLr%)rergrbrc)rbrarcrarerdrr)rxraryrarr)rPrar"Callable[[], str | dict[str, Any]]rra)rxrarPrarrrr)rxrarrrr)rxrar Iterable[T]rzCallable[[T], str]rz#Callable[[T], dict[str, list[str]]]rrZrzdict[str, Any] | Nonerrfrr)rrarxrarrrra)rrrr__annotations__rkr|rrrwrrrr$rr%r&r_r_xs M ".P BOOO9 O  O CC9C  C+/? ? ? # ? : ? ? (? ?  ? B2Mr%r_c[RRURS5nU(aUO URn[ XUR S9$)N templates)rbrcre)rsrur source_pathrbr_re)optionsrbrcs r&make_file_managerrsC77<< 3 3[AL!,+'2E2EK GOO r%c\[RS:aSSKJn U"XU5$[ XUS9$)N) r)pformat)rDwidth)r version_infopprintr_pformat)objrDrrs r&dataclass_reprr)s/  7""sE** Ce 44r%c[U5(dS[U535eURRnU[ U5S-- n[ U5Vs/sH6oUR (dMUR[XR54PM8 nn/nUHupU[ U5-S-n [U 5(a [XX*5n OZ[U [5(a [XX*5n O8[U [[[45(a [!XX*5n O [ U 5n UR#USU 35 M US-n SU 3R%U5n USU S3$s snf)Nz%obj should be a dataclass, received: rA= z, ())rrL __class__rrrreprr0getattrrrr _format_dictrrituple _format_listappendr)rrDr curr_indent class_namerz fields_list fields_strr0attr _curr_indentstr_repr indent_strbodys r&rr7sM   Q Ed3i[QQ ''J3z?Q&&K;A#;Q;a&&1AFFGC01;KQJ! #SY.2   eBH d # ##D%FH tS%0 1 1#D%FHDzHTF!H:./"s"J   " ": .D\4& "")Rs E#0%E#c  X1S-- n/nUR5HYupV[U5n[U5(a[XaX#[ U5-5O [U5nUR USU35 M[ [ XAX#SS5$)Nrz: {})rrrrrr_format) rrDrr dict_reprrrk_reprv_strs r&rrZs A:KI aA QS['@ Aa  F82eW-. 9e#s CCr%c X1S-- nUVs/sH*n[U5(a [XAX#5O [U5PM, nn[U[5(aSOSupg[ XQX#Xg5$s snf)NrA)[])rr)rrrrrr )rrDrrl list_reprstartends r&rrnsv A:KA4@??E/QO *$55:JE 9e% EE s1A'cSupg[[U55U:aSnSU-nSU-nSUU3RU5n UUU U3$)N)rr@rr()rrr) rrDrrrr delimitercurr_indent_strrrs r&r r }si"(I 4 %  +vJ  {?+ , 1 1* =DWZLse ,,r%c\rSrSrSrS S Sjjr\S SSjj5r\SSj5r \SSj5r \SSj5r SSSjjr S r g )NamespaceHelperizA helper for constructing the namespace open and close strings for a nested set of namespaces. e.g. for namespace_str torch::lazy, prologue: namespace torch { namespace lazy { epilogue: } // namespace lazy } // namespace torch cTURS5n[U5U::dSUSUS35eXlSRUVs/sH nSUS3PM sn5UlSR[ U5Vs/sHnSU3PM sn5UlX@lX lgs snfs snf) N::z"Codegen doesn't support more than z# level(s) of custom namespace. Got rr@z namespace z {z} // namespace ) r-rcpp_namespace_r prologue_reversed epilogue_ namespaces_ entity_name_)rj namespace_str entity_name max_levelcpp_namespacesns r&rkNamespaceHelper.__init__s',,T2  9 , o / {:]^k]llm n o ,+#PAj3$7#PQ-5n-E F-Es #-E F *' $Q Fs B 8B%cjURS5nUSnSRUSS5n[XCUS9$)za Generate helper from nested namespaces as long as class/function name. E.g.: "torch::lazy::add" rrN)r"r#r$)r-rr)namespaced_entityr$namesr#r"s r&from_namespaced_entity&NamespaceHelper.from_namespaced_entitysC"''-Bi  %*- 'I  r%cUR$r7)rrs r&prologueNamespaceHelper.prologue ~~r%cUR$r7)rrs r&epilogueNamespaceHelper.epiloguer0r%cUR$r7)r!rs r&r#NamespaceHelper.entity_names   r%c@UR(a UR$U$)z Return the namespace string from joining all the namespaces by "::" (hence no leading "::"). Return default if namespace string is empty. )r)rjdefaults r&get_cpp_namespace!NamespaceHelper.get_cpp_namespaces '+&9&9t""FwFr%)rr!rr rN)r)r"rar#rar$rZrr)r:)r)rar$rZrrr)r)r7rarra)rrrr__doc__rk staticmethodr+propertyr.r2r#r8r$rr%r&rrs KL( (/2(DG( ( 12    +.      !!GGr%rc\rSrSr%S\S'SSSjjrSSjrSSjrSSjrSS jr SS jr \ SS j5r SS jr SS jrSSjrSrg) OrderedSetizdict[T, Literal[None]]storageNcNUc0Ulg[RU5Ulgr7)r@rfromkeys)rjiterables r&rkOrderedSet.__init__s  DL==2DLr%cXR;$r7r@rjrs r& __contains__OrderedSet.__contains__s||##r%cH[URR55$r7)iterr@keysrs r&__iter__OrderedSet.__iter__sDLL%%'((r%cNURRUR5 gr7)r@r)rjrs r&rOrderedSet.updates EMM*r%c"SURU'gr7rFrGs r&rOrderedSet.adds! Tr%cX[5nURR5UlU$r7)r?r@r)rjrets r&rOrderedSet.copys"'\ll'')  r%cfUSR5nUSSHnURU5 M U$)NrrA)rr)rErTr\s r&unionOrderedSet.unions11gllnabA JJqM r%c,[RX5$r7)r?rWrjothers r&__or__OrderedSet.__or__s,,r%c(URU5 U$r7)rrZs r&__ior__OrderedSet.__ior__s E r%c[U[5(aURUR:H$[URR 55U:H$r7)rr?r@rirLrZs r&__eq__OrderedSet.__eq__s? eZ ( (<<5==0 0t||((*+u4 4r%rFr7)rCzIterable[T] | Nonerr)rr4rrd)rz Iterator[T])r OrderedSet[T]rr)rr4rr)rrd)rErdrrd)r[rdrrd)r[rdrr)r[objectrrd)rrrrrrkrHrMrrrr<rWr\r_rbr$rr%r&r?r?sI ##3 $)+"  -5r%r?)r.rarztuple[str, list[str]])r8zCallable[[T], S | None]r9rr Iterator[S])r8zCallable[[T], Sequence[S]]r9rrrf)rFzCallable[[], str]rzIterator[None])r:r rr )rPrarr)r\rarrZr7)rrrbz str | Nonerr_)rP)rr rDrZrrZrra)r) rr rDrZrrZrrZrra) rzdict[Any, Any]rDrZrrZrrZrra) rz&list[Any] | set[Any] | tuple[Any, ...]rDrZrrZrrZrra)rz list[str]rDrZrrZrrZrrarrarra)= __future__r contextlib functoolsrVrsr)rrC dataclassesrrenumrrpathlibrtypingr r r r r rrtyping_extensionsrtorchgen.code_templaterargparsercollections.abcrrrrrparentrr IDENT_REGEXr3r4r5r<r>contextmanagerrIrMcacherQr]r_rrrrrr rr?rr%r&rwsG" ,TTT"/"<< N # # % , , 3 3  $T $ $ CL CL   @ // 4bMbMN37 %/ 5  5  5  5 5$ #  #  #  # #  #FD D D D D  D( F 0 F  F  F F  F-- - - -  - - -&>G>GB,5,5r%