΅ixNSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSK J r SSK J r SSKJrJr SSKJrJrJrJrJrJr SSKJr SSK Jr SSKJr \\\\ "S5\!\"\#\$\%4 r&\"S \\-\-\!-\"-\#-\$-\%-S-S 9r'\("5r)\ "S S 9"S S\\'55r*\(a:\)SSSSS4S\'\(-S\!S-S\!\"\!-S-S\!\"\!-S-S\ S-S\!S-S\'4Sjjr+O<\)SSSSS4S\'\(-S\!S-S\!\"\!-S-S\!\"\!-S-S\ S-S\!S-S\*\'4Sjjr+S\!S\\!-S-4Sjr,S\SS4Sjr-Sr.S\S\#\!4Sjr/\ "SS55r0"S S!\5r1"S"S#\Rd5r2"S$S%5r3S&\(S\!S'\(S\(4S(jr4S*S\!S\S\S-4S)jjr5g)+N)Callable) dataclass) FunctionType ModuleType)AnyGenericNoReturnOptional TYPE_CHECKINGTypeVar) deprecated)mock)justknobs_checkT)boundT)kw_onlyc\rSrSr%Sr\\-\S'Sr\ S-\S'Sr \ \ S-\S'Sr \ \ S-\S'Sr \S-\S'Sr\ S-\S 'SS jr\S \ \ \ -S-S \ \ S-4S j5rSrg)_Config aRepresents a config with richer behaviour than just a default value. :: i.e. foo = Config(justknob="//foo:bar", default=False) install_config_module(...) This configs must be installed with install_config_module to be used Precedence Order: alias: If set, the directly use the value of the alias. env_name_force: If set, this environment variable has precedence over everything after this. If multiple env variables are given, the precedence order is from left to right. user_override: If a user sets a value (i.e. foo.bar=True), that has precedence over everything after this. env_name_default: If set, this environment variable will override everything after this. If multiple env variables are given, the precedence order is from left to right. justknob: If this pytorch installation supports justknobs, that will override defaults, but will not override the user_override precedence. default: This value is the lowest precedence, and will be used if nothing is set. Environment Variables: These are interpreted to be either "0" or "1" to represent true and false. Arguments: justknob: the name of the feature / JK. In OSS this is unused. default: is the value to default this knob to in OSS. alias: The alias config to read instead. env_name_force: The environment variable, or list of, to read that is a FORCE environment variable. I.e. it overrides everything except for alias. env_name_default: The environment variable, or list of, to read that changes the default behaviour. I.e. user overrides take preference. defaultNjustknobenv_name_defaultenv_name_force value_typealiasreturnc:[RUR5Ul[RUR5UlURbFUR [ Ld'URcURc URb [S5egg)Nztif alias is set, none of {default, justknob, env_name_default and env_name_force} can be set) r string_or_list_of_string_to_listrrrr_UNSET_SENTINELrAssertionErrorselfs T/home/james-whalen/.local/lib/python3.13/site-packages/torch/utils/_config_module.py __post_init___Config.__post_init__Os ' H H  ! !! &FF     :: ! O3==,((4&&2$I3 "valcUcg[U[5(aU/$[U[5(d[S[ U535eU$)Nzval is not a list, got ) isinstancestrlistr type)r's r#r(_Config.string_or_list_of_string_to_listcsI ; c3  5L#t$$ #:49+!FG G r&)rrrN)__name__ __module__ __qualname____firstlineno____doc__robject__annotations__rr*rr+rrr,rr$ staticmethodr__static_attributes__r&r#rr s$LZHcDj)-d3i$&-'+NDI$+"Jt "E3:(  49_t #  cT   r&rrrrrrrrcgNr8rrrrrrs r#Configr<vs r&c [UUUUUUS9$)Nr;)rr;s r#r<r<s#-)!   r&namec`[RRU5nUS:XagUS:XagU$N1T0Fosenvironget)r>values r#_read_env_variablerHs- JJNN4 E | | Lr&modulec^^^"SS[5nS[[-S[[-S[SS4UUU4Sjjm0m[ T5nT"TTS 5 TTlUTlUTlS Tl STl g) z Converts a module-level config into a `ConfigModule()`. See _config_typing.pyi for instructions on how to get the converted module to typecheck. c(\rSrSr\"1Sk5rSrg)3install_config_module..ConfigModuleInstance> _is_dirty _hash_digestr5r8N)r/r0r1r2set _bypass_keysr7r8r&r#ConfigModuleInstancerLs KL r&rRsourcedestprefixrNc b>[R"U5n[URR 55GHupEUR S5(d[ U[[45(dk[US5(a0URS:XdJURR S5(d*[ U[5(a[U[5(aMUU3nURUS5n[ U[5(a*[![XWS95T U'UT La[#T U5 MM[ U[5(a;UbUR$cXul[!U5T U'UT La[#T U5 GMHGMK[ U[5(auURT R&:wa[)SUST R&35e[+T US 35nT "XXUS 35 UT La[-XU5 GMXRU'GM[)S US US [U5S 35e g)z?Walk the module structure and move everything to module._config__r0typingzcollections.abcN)rrzsubconfig class z must be defined in module .zUnhandled config =z ())inspectget_annotationsr+__dict__items startswithr)rrhasattrr0r, issubclassrrF CONFIG_TYPES _ConfigEntrydelattrrr/r SubConfigProxysetattr) rSrTrU type_hintskeyrGr>annotated_typeproxyconfigrIvisits r#rm$install_config_module..visits ,,V4 v4467JCt$$ej,%?@@E<00((H4 ++667HIIud++ 5'0J0JXcU#D'^^C6N%..+EE t 6>FC("E7++!-%2B2B2J'5$+E2t 6>FC("E4((##v6(*5'1LV__L]^'v$qz:etfAJ/6>Du-).MM#&$'8QugRU }TU%VWW[8r&T) ConfigModulerr,rfr*-get_assignments_with_compile_ignored_comments_config_compile_ignored_keys __class__rNrO)rIrRcompile_ignored_keysrlrms` @@r#install_config_modulervsM|M4XT!4X>)4X4X  4X4Xl')FHP &&"FN#7F +FFFr&z@compile_ignoredc>[R"U5n[5n[R"[R "UR S55R5nSnSnUGHnUR[R:XaRSnURR5n[U;a*US:wa[S[35eXvRS4nMrMtUR[R:XaU(dURnMMUR[R :XdMURS:XdM[US;a,USURSS- :XaUR#U5 SnSnGM US:wa[S[35eU$)Nutf-8)roroz unconsumed rrZ)r\ getsourcerPtokenizeioBytesIOencodereadliner,COMMENTstringstripCOMPILE_IGNORED_MARKERr startNAMEOPadd)rI source_code assignmentstokenscurrent_comment prev_nametoken maybe_currents r#rqrqs^##F+K%K  rzz+*<*'/!*<<#A&%++a.1*<< *"(I-.("{+A*BCDD r&c\rSrSr%\\S'\\S'\r\\S'Sr \ S-\S'\r \\S'\r \\S'S r \\S 'Sr\ S-\S 'S \S S4SjrSrg)rdirr user_overrideNrenv_value_forceenv_value_defaultFhiderrlrc4URUlURb URO[UR5UlURUlURUlUR b)UR Hn[ U5=ncMX0l O URb)URHn[ U5=ncMX0l O URbEURb8[UR[5(d[SURS35eURbgUR c URbLUR[[[[[[4;a[SURS35eggg)Nz!justknobs only support booleans, z is not a booleanzF  DDV  \\  " " ...!3C!88IE-6*/  ,,,!3C!88IE+4(- == $)AdllD11$7 ~EVW ?? &  # # /63H3H3T ' %RSWSbSbRccno 4U 'r&)rrrrrr)r/r0r1r2rr5r,rrrr*rrrrrrrr7r8r&r#rdrdskL)M3(HcDj*OS*,s,D$E3:#w#4#r&rdc ^\rSrSr%\\\4\S'\\\S'\\\S'\ \S'\ S-\S'S3S jr S \S \ SS4U4S jjr S \S\4S jrS \SS4SjrS\S\\\4S-4SjrS\S\4SjrS\S\SS4SjrS \S\ 4SjrS4S\\S-S\\S-S\ S\\\44SjjrS\S\4SjrS\ 4SjrSS.S\ S\\\44SjjrS\4S jrS\ 4S!jr\"S"\ S#9S\\\44S$j5r!\"S%\ S#9S\\\44S&j5r"S'\ \\\4-SS4S(jr#S\\\44S)jr$S\\\44S*jr%S5S+\\\\4-S-S,\S-\\\4SS.4S/jjr&S0\\\4S\4S1jr'S2r(U=r)$)6rpiTrrrQrsrNNrOrc([S[S35e)Nzuse z-.install_config_module(sys.modules[__name__]))NotImplementedErrorr/r!s r#rConfigModule.__init__`s!8*I J  r&r>rGch>XR;a[TU] X5 gXR;a[ UR SUS35eURUR b URURUU5 gX RUlSUl SURUl g)NrY does not existTF) rQsuper __setattr__rrAttributeErrorr/r_set_alias_valrrNr)r"r>rGrts r#rConfigModule.__setattr__es $$ $ G  ,  % DMM?!D6!IJ J \\$  % % 1    T 2E :/4LL  ,!DN&+DLL  #r&cURUnUR(a[URSUS35eUR U5nU[ LaU$UR [ La UR $UR[ La UR$UR[ La UR$URb[URURS9$[UR[[[45(a1[ R""UR5UlUR$UR$![$a!n[URSUS35UeSnAff=f)NrYr)r>r)rrrrr/_get_alias_valrrrrrrrr)r+rPdictcopydeepcopyKeyError)r"r>rl alias_vales r# __getattr__ConfigModule.__getattr__qs:! R\\$'F{{$ av_%MNN++F3I/  %%_<---##?:+++''>///*&FOOV^^TT &..4d*;<<'+}}V^^'D$+++>> ! R DMM?!D6!IJPQ Q Rs=AEE8EE6*E!AE< E E3E..E3cjSUl[URUlSURUlgNT)rNrrrrrr"r>s r# __delattr__ConfigModule.__delattr__s-,; T("& Tr&entrycURnUcgURSS5up4[R"U5nXT4$![an[ SUS35UeSnAff=f)NrYrzz config alias r)rrsplit importlib import_module ImportErrorr)r"rr module_name constant_namerIrs r#_get_alias_module_and_name'ConfigModule._get_alias_module_and_namess  =%*\\#q%9"  P,,[9F$$ P =!GHa O Ps? A AAcXURU5nUc[$Uup4[X45nU$r:)rrgetattr)r"rdatarIrconstant_values r#rConfigModule._get_alias_vals5..u5 <" " $ 7r&r'cbURU5nUc [S5eUupE[XEU5 g)Nz6alias data should not be None when setting alias value)rr rg)r"rr'rrIrs r#rConfigModule._set_alias_vals;..u5 < H !%s+r&cURUnUR[L=(d URUR:HnUR[L=(d URUR:HnUR [Ln[ UR[[[45(a"U=(d UR UR:HnU=(a U=(a U$)zo Returns true if the config is at its default value. configs overridden by the env are not considered default. ) rrrrrrrr)r+rPr)r"r> config_valnot_set_env_defaultnot_set_env_forceunsets r# _is_defaultConfigModule._is_defaults \\$'  ( (O ; B++z/A/AA   & &/ 9 @))Z-?-??  ((O; j((4d*; < <KZ559K9KKEB,B1BBr& ignored_keysignored_prefixes skip_defaultcR^0nURHmU(aTU;aMU(a[U4SjU55(aM5U(aURT5(aMTURTRbMp[R "[ UT55UT'M U$)agExport a dictionary of current configuration keys and values. This function is design to provide a single point which handles accessing config options and exporting them into a dictionary. This is used by a number of different user facing export methods which all have slightly different semantics re: how and what to skip. If a config is aliased, it skips this config. Arguments: ignored_keys are keys that should not be exported. ignored_prefixes are prefixes that if a key matches should not be exported skip_default does two things. One if a key has not been modified it skips it. c3F># UHnTRU5v M g7fr:)r`).0rUris r# )ConfigModule._get_dict..sM.get_module_names2//Kj( ;",s"  r&NcJ>T"US5nU(aTRU5 gg)NF)r)rrrimportss r# add_import/ConfigModule.codegen_config..add_imports#)$6K K(r&vc^>UVs/sHnT"US5UR3PM sn$s snfr)r/)ritemrs r#list_of_callables_to_string@ConfigModule.codegen_config..list_of_callables_to_string!s2PQRPQtT23DMM?CPQR RRs!*cd[U5=(a [US5=(a [US5$)Nr0r/)callablera)rs r#importable_callable8ConfigModule.codegen_config..importable_callable$s$A;V71l#;V:@V Vr&c>T"U5(a%T"U5 USUST"US5UR3$[U[[45(al[ U4SjU55(aRUH nT"U5 M T"U5n[U[5(a USUSU3$USUSSR U5S3$USUSU<3$)a Return a string version of the config line. Handle v when v is a callable, or a list/dict of callables. Add import statements for callables if necessary. We assume that the value of a single config won't be a mix of callables and non-callables. Example output: import logging import _warnings torch._dynamo.config.reorderable_logging_functions = { _warnings.warn, logging.warn, print } rYz = Tc34># UH nT"U5v M g7fr:r8)rrrs r#rGConfigModule.codegen_config..get_config_line..6s467d#D))asz = { z, z })r/r)r+rPalljoin) modkrrv_listrrrrs r#get_config_line4ConfigModule.codegen_config..get_config_line(s#1%%1 as#oa&>%? |LLAc{++467411Dt$4Q7a&&!U!A3c&22!U!A3fTYYv->,?sCCas#aU++r&rT)rrrzimport  ) rPrrr*r+rr/rrr_rinsertr) r"r linesrrr import_namerrrrrs @@@@@r#codegen_configConfigModule.codegen_config s % ( T c  )X )$ ) )  S4#: S$s) S W3 W4 W ,# , ,6mmNN '>T..6>>tDXL!&););)=">?N ' %%g.!fh  #DN   r&z`config.to_dict()` has been deprecated. It no longer changes the underlying config. use `config.get_config_copy()` instead if you just want a copy of the config, or config.load_config if you need mutable access)categoryc"UR5$r:get_config_copyr!s r#to_dictConfigModule.to_dictX##%%r&z`config.shallow_copy_dict()` has been deprecated. It no longer changes the underlying config. use `config.get_config_copy()` instead if you just want a copy of the config, or config.load_config if you need mutable accessc"UR5$r:rr!s r#shallow_copy_dictConfigModule.shallow_copy_dictar"r&maybe_pickled_configc[U[5(d[R"U5nOUnUR 5H7up4X0R ;a[ XU5 M"SSKJn U"SUSUS35 M9 g)zARestore from a prior call to save_config() or shallow_copy_dict()r) warn_oncezkey z with value z! is not understood by this configN) r)rrloadsr_rrrgtorch._dynamo.utilsr()r"r&rlrrr(s r# load_configConfigModule.load_configjse.55\\"67F)FLLNDALL #9D<s2STU #r&c"UR5$r:)rr!s r#rConfigModule.get_config_copyxs~~r&c6UR[US/5S9$)Nrr)rrr!s r#get_serializable_config_copy)ConfigModule.get_serializable_config_copy{s~~749NPR+S~TTr&arg1arg2kwargsContextDecoratorc |^^^Ub\Ub$[U[5(d [S5eX0mO"[U[5(d [S5eUmU(a [S5eOUmUb [S5e[T[5(d[S[ T535e0mUm"UUU4SjS[ 5nU"5$)a< Decorator and/or context manager to make temporary changes to a config. As a decorator: @config.patch("name", val) @config.patch(name1=val1, name2=val2) @config.patch({"name1": val1, "name2", val2}) def foo(...): ... As a context manager: with config.patch("name", val): ... zGfirst argument must be a string when passing 2 positional args to patchzKfirst argument must be a dict when passing a single positional arg to patchz:cannot pass both positional and keyword arguments to patchzLsecond positional argument is only valid when first argument is a key stringzexpected `dict` got cF>\rSrSrSU4SjjrSUU4SjjrUU4SjrSrg)'ConfigModule.patch..ConfigPatchiNc>TUlgr:changes)r"r;s r#r0ConfigModule.patch..ConfigPatch.__init__s & r&c>T(a [S5eURHnTRU5TU'M URR5Hup#TR X#5 M g)Nz/prior should be empty when entering ConfigPatch)r r;rr_r)r"rirrrlpriors r# __enter__1ConfigModule.patch..ConfigPatch.__enter__sc(I <TR5HupETRXE5 M TR5 gr:)r_rclear)r"exc_typeexc_valexc_tbrrrlr>s r#__exit__0ConfigModule.patch..ConfigPatch.__exit__s,!KKMDA&&q,* r&r:r.)r/r0r1r2rr?rFr7)r;rlr>sr# ConfigPatchr8s ' - -  r&rH)r)r*r rr,r5)r"r2r3r4rHr;rlr>s @@@r#patchConfigModule.patch~s.  !$,,(a ,!$--(e$P G$b'4(( #7W !GH H "  * (}r&r;c N^^^TRmS[/S44UUU4SjjnU$)z A lower-overhead version of patch() for things on the critical path. Usage: # do this off the critical path change_fn = config.make_closure_patcher(foo=True) ... revert = change_fn() try: ... finally: revert() rNc>^TVs0sHoTUR_M snmTR5HupUTRUlM SUU4SjjnU$s snf)Nc`>TR5HupUTRUlM gr:)r_rrr)rrr>r"s r#revertBConfigModule._make_closure_patcher..change..reverts%!KKMDA45DLLO1*r&r.)rr_rr)rrrNr>r;rlr"s @r#change2ConfigModule._make_closure_patcher..changes]9@AAq ///AE 01 Q-( 6 6MBsA)rrr)r"r;rPrls`` @r#_make_closure_patcher"ConfigModule._make_closure_patchers/$ T*   r&)rOrNr.)NNF)NN)*r/r0r1r2rr*rdr5rPrbytesrr4rrrrtuplerrrrrr+rr,rrrrrr FutureWarningr r$r+rr0rIrRr7 __classcell__rts@r#rprpTs #|# $$c(s8#O$, , ,F ,t ,"R"R"RH''' %! % z3 $ & %LS,L,s,t,CCC4*.-1" #3i$&#s)d*# # c3h #J4C4D4 U 159)-9 c3h9? ? B !% ! 8  &c3h&  & 8  &4S>&  & VS#X0F V4 V c3h Ud38nU -1JDcN"T)JJsCx. J  JXtCH~#r&rpc\^\rSrSrSrS SjrS\4SjrS\\ /\ 4S\ 4U4Sjjr Sr U=r $) r5izO Same as contextlib.ContextDecorator, but with support for `unittest.TestCase` rc[S5eNNYIrr!s r#r?ContextDecorator.__enter__ !%((r&c[S5er[r])r"rCrDrEs r#rFContextDecorator.__exit__r_r&rc>^[U[5(ab[U[R5(aC"U4SjSU5nUR UlUR UlURUlU$[TT]%U5$)NcV>^\rSrSr\SUU4Sjj5r\SUU4Sjj5rSrU=r$),ContextDecorator.__call__.._TestCaseic>TR5 [TU] 5 g![a TR SSS5 ef=fr:)r?r setUpClass ExceptionrFclsrtr"s r#rf7ContextDecorator.__call__.._TestCase.setUpClasss?NN$*,$ dD$7s "Acz>[TU]5 TRSSS5 g!TRSSS5 f=fr:)r tearDownClassrFrhs r#rl:ContextDecorator.__call__.._TestCase.tearDownClasss48-/ dD$7 dD$7s%:r8r.) r/r0r1r2 classmethodrfrlr7rW)rtr"s@r# _TestCaserds)88r&ro) r)r,rbunittestTestCaser/r1r0r__call__)r"rrorts` r#rrContextDecorator.__call__sl dD ! !jx7H7H&I&I 8D 8""&I %)%6%6I "#'??I  w%%r&r8r.) r/r0r1r2r3r?r rFrrrrr7rWrXs@r#r5r5s< ))X)&XseSj1&c&&r&r5cv^\rSrSrSrS\S\SS4U4SjjrS\S \SS4S jrS\S\ 4S jr S\SS4S jr S r U=r $)rfizb Shim to redirect to main config. `config.triton.cudagraphs` maps to _config["triton.cudagraphs"] rlrUrNcF>[TU]SU5 [TU]SU5 g)Nrr_prefix)rr)r"rlrUrts r#rSubConfigProxy.__init__s! Iv. Iv.r&r>rGcTURRURU-U5$r:)rrrrv)r"r>rGs r#rSubConfigProxy.__setattr__ s"||'' t(;UCCr&cRURRURU-5$r:)rrrrvrs r#rSubConfigProxy.__getattr__# ||'' t(;<rGs r# patch_objectr*s7#|$$yy%% ::  S ..r&c`[RRU5nUS:XagUS:XagU$r@rC)r>rrGs r#get_tristate_envr3s- JJNN4 E | | Nr&r:)6 contextlibrrrr\r}rDrr|rpcollections.abcr dataclassesrtypesrrrXrrr r r r typing_extensionsr rtorch._utils_internalrintfloatrr,r*r+rPrUrrcrr4rrr<rHrvrrqrdrpr5rfrrr8r&r#rs  $!*KK(1UD$t*c4eTJ  CsU{T)C/$6> >BU:Up &&z22&&R==*/f/C//6/3t r&