3jzSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK r SSK J r SSK J r SSKJr SSKJrJr SSKJrJrJrJrJrJr SSKJr SSKJr \\\ \!"S5\"\#\$\%\&4 r'\\\ \!"S5\"\%4r(\"S \\-\ -\"-\#-\$-\%-\&-S-S 9r)\*"5r+\"S S 9"S S\\)55r,\(aC\+SSSSSSS4S\)\*-S\"S-S\"\#\"-S-S\"\#\"-S-S\!S-S\"S-S\ S\"S-S\)4Sjjr-OE\+SSSSSSS4S\)\*-S\"S-S\"\#\"-S-S\"\#\"-S-S\!S-S\"S-S\ S\"S-S\,\)4Sjjr-S\"S\ \"-S-4Sjr.S\SS4Sjr/Sr0S\S\$\"4S jr1\"S!S"55r2"S#S$\5r3"S%S&\Rh5r4"S'S(5r5S+S\"S\S\ S-4S)jjr6S*r7g),N)Callable) ContextVar) dataclass) FunctionType ModuleType)AnyGenericNoReturnOptional TYPE_CHECKINGTypeVar) deprecated)justknobs_checkT)boundT)kw_onlyc\rSrSr%Sr\\-\S'Sr\ S-\S'Sr \ \ S-\S'Sr \ \ S-\S'Sr \S-\S'Sr\ S-\S 'S r\\S 'Sr\ S-\S 'SSjr\S\ \ \ -S-S \ \ S-4Sj5rSrg)_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. User overrides are thread-local. 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_typealiasFrdeprecation_messagereturnc:[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/wildlama/miniconda3/lib/python3.13/site-packages/torch/utils/_config_module.py __post_init___Config.__post_init__Vs ' 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_listjsI ; c3  5L#t$$ #:49+!FG G r')rrrN)__name__ __module__ __qualname____firstlineno____doc__robject__annotations__rr+rr,rrr-rrboolrr% staticmethodr__static_attributes__r'r$rr$s$LZHcDj)-d3i$&-'+NDI$+"Jt "E3:J&*t*(  49_t #  cT   r'rFrrrrrrrrrcgNr:rrrrrrrrs r$Configr>}s r'c "[UUUUUUUUS9$)Nr=)rr=s r$r>r>s)-)!! 3  r'namec`[RRU5nUS:XagUS:XagU$N1T0Fosenvironget)r@values r$_read_env_variablerJs- 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_digestr6r:N)r0r1r2r3set _bypass_keysr9r:r'r$ConfigModuleInstancerNs KL r'rTsourcedestprefixrNc d>[R"U5n[URR 55GHupEUR S5(d[ U[[45(dk[US5(a0URS:XdJURR S5(d*[ U[5(a[U[5(aMUU3nURUS5n[ U[5(a+[![XWS9U5T U'UT La[#T U5 MM[ U[5(a;UbUR$cXul[!XV5T U'UT La[#T U5 GMIGML[ 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__r1typingzcollections.abcN)rrzsubconfig class z must be defined in module .zUnhandled config =z ())inspectget_annotationsr,__dict__items startswithr*rrhasattrr1r- issubclassrrH CONFIG_TYPES _ConfigEntrydelattrrr0r!SubConfigProxysetattr) rUrVrW type_hintskeyrIr@annotated_typeproxyconfigrKvisits r$ro$install_config_module..visits ,,V4 v4467JCt$$ej,%?@@E<00((H4 ++667HIIud++ 5'0J0JXcU#D'^^C6N%..+EEt t 6>FC("E7++!-%2B2B2J'5$+E8t 6>FC("E4((##v6(*5'1LV__L]^'v$qz:etfAJ/6>Du-).MM#&$'8QugRU }TU%VWW[8r'T) ConfigModulerr-rhr+-get_assignments_with_compile_ignored_comments_config_compile_ignored_keys __class__rPrQ)rKrTcompile_ignored_keysrnros` @@r$install_config_modulerxsM|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)rqrqz unconsumed rr\)r^ getsourcerRtokenizeioBytesIOencodereadliner-COMMENTstringstripCOMPILE_IGNORED_MARKERr!startNAMEOPadd)rK source_code assignmentstokenscurrent_comment prev_nametoken maybe_currents r$rsrss^##F+K%K  rzz+*<*'/!*<<#A&%++a.1*<< *"(I-.("{+A*BCDD r'c\rSrSr%\\S'\\S'\\\S'Sr \ S-\S'\ r \\S'\ r \\S'S r\\S 'Sr\ S-\S 'S r\\S 'Sr\ S-\S 'S r\\S'S\S\ SS4SjrSrg)rfi"rr user_overrideNrenv_value_forceenv_value_defaultFhiderrr_deprecation_warnedrnr@rcURUlURb URO[UR5UlURUlURUlUR UlUR UlSUl[U[S9Ul URb)URHn[U5=ncMX@l O URb)URHn[U5=ncMX@l O URbEURb8[!UR["5(d[%SURS35eURbgURc URbLUR["[&[(["[([&4;a[%SURS35eggg)NFrz!justknobs only support booleans, z is not a booleanzF  DDV  \\  ++#)#=#= #( 'oF  " " ...!3C!88IE-6*/  ,,,!3C!88IE+4(- == $)AdllD11$7 ~EVW ?? &  # # /63H3H3T ' %RSWSbSbRccno 4U 'r') rrrrrrrrrr)r0r1r2r3rr6r-rr5rr+r rrrr7rrrrrrr9r:r'r$rfrf"sLf%%HcDj*OS*,s,D$E3:J&*t* %%)w)c)d)r'rfc ^\rSrSr%\\\4\S'\\\S'\\\S'\ \S'\ S-\S'S5S jr S \S \SS4S jr S \S \ SS4U4SjjrS \S\4SjrS \SS4SjrS\S\\\4S-4SjrS\S\4SjrS\S\SS4SjrS \S\ 4SjrS6S\\S-S\\S-S\ S\\\44SjjrS\S\4SjrS\ 4SjrSS.S \ S\\\44S!jjrS\4S"jrS\ 4S#jr\ "S$\!S%9S\\\44S&j5r"\ "S'\!S%9S\\\44S(j5r#S)\ \\\4-SS4S*jr$S\\\44S+jr%S\\\44S,jr&S7S-\\\\4-S-S.\S/\\\4SS04S1jjr'S2\\\4S\4S3jr(S4r)U=r*$)8rrinrtrSrurPNrQrc([S[S35e)Nzuse z-.install_config_module(sys.modules[__name__]))NotImplementedErrorr0r"s r$rConfigModule.__init__zs!8*I J  r'r@rncUR(amUR(d[SSKnURSUS3nUR(aUSUR3- nUS- nUR U[ SS9 S Ulggg) z;Issue deprecation warning for config if not already warned.rNr[z is deprecatedz and z4. It will be removed in a future version of PyTorch.) stacklevelT)rrwarningsr0rwarn FutureWarning)r#r@rnrmsgs r$_warn_if_deprecated ConfigModule._warn_if_deprecateds{   V%?%? ]]O1TF.9C))v99:;; I IC MM#}M ;)-F &&@ r'rIch>XR;a[TU] X5 gXR;a[ UR SUS35eURUnUR X5 URbURX25 gURRU5 SUl SUl g)Nr[ does not existTF) rSsuper __setattr__rtAttributeErrorr0rr_set_alias_valrrRrPr)r#r@rIrnrvs r$rConfigModule.__setattr__s $$ $ G  ,  % DMM?!D6!IJ J\\$'F  $ $T 2||'##F2$$((/!%# r'cHURUnUR(a[URSUS35eUR X5 UR U5nU[ LaU$UR[ La UR$URR5nU[ LaU$UR[ La UR$URb[URURS9$[UR[5(dSURR!["R$"UR55 URR5$UR$![&a!n[URSUS35UeSnAff=f)Nr[rr@r)rtrrr0r_get_alias_valr rrrHrrrrr*_IMMUTABLE_CONFIG_TYPESrRcopydeepcopyKeyError)r#r@rn alias_valres r$ __getattr__ConfigModule.__getattr__sa$ R\\$'F{{$ av_%MNN  $ $T 2++F3I/  %%_<---"00446MO3$$''>///*&FOOV^^TTfnn.EFF$$((v~~)FG++//11>> ! R DMM?!D6!IJPQ Q Rs=A'E6*E6 $E6.E6 *E68A1E6* E66 F!FF!cSUlURURR[5 SURUlgNT)rPrtrrRr rr#r@s r$ __delattr__ConfigModule.__delattr__s9 T((,,_="& Tr'entrycURnUcgURSS5up4[R"U5nXT4$![an[ SUS35UeSnAff=f)Nr[r|z config alias r)rrsplit importlib import_module ImportErrorr)r#rr module_name constant_namerKrs 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<)rr getattr)r#rdatarKrconstant_values r$rConfigModule._get_alias_vals5..u5 <" " $ 7r'r(cbURU5nUc [S5eUupE[XEU5 g)Nz6alias data should not be None when setting alias value)rr!ri)r#rr(rrKrs r$rConfigModule._set_alias_vals;..u5 < H !%s+r'cURUnUR[L=(d URUR:HnUR[L=(d URUR:HnUR R 5[Ln[UR[5(d0U=(d' UR R 5UR: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. ) rtrr rrrrHr*r)r#r@ config_valnot_set_env_defaultnot_set_env_forceunsets r$ _is_defaultConfigModule._is_defaults \\$'  ( (O ; B++z/A/AA   & &/ 9 @))Z-?-??  ((,,./A*,,.EFFQZ5599;z?Q?QQEB,B1BBr' ignored_keysignored_prefixes skip_defaultc^0nURR5GH4umnURbMU(aTU;aM%U(a[U4SjU55(aMHU(aUR T5(aMgUR R 5nUR[La URnOdU[LaUnOXUR[La URnO8URb[URURS9nO URn[U[5(d[R "U5nXtT'GM7 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<)rb).0rWrks r$ )ConfigModule._get_dict..sM>5==Qmmc#:;;mmC(3K5/8 r' config_namec4URUR$r<)rtr)r#rs r$get_typeConfigModule.get_type5s||K(333r'c`[US/5n[R"URUS9SS9$)z Convert config to a pickled blob_save_config_ignorer)protocol)rpickledumpsr)r#rs r$ save_configConfigModule.save_config8s3t%:B? || NN N 5  r'T)ignore_private_configsrcT/nU(aURS5 UR[US/55 URUS9n[US/5HQnXC;dM X4cMX4"5n[ US5(aUR 5X4'M?[ SUSX4S35e U$) z!Convert config to portable format__cache_config_ignore_prefix)r_cache_config_factory_keysuuidzConfig 'z ' is set to zO which does not implement uuid(). Implement uuid() for cache key participation.)appendextendrrrcr RuntimeError)r#rprefixesrnrkinstances r$save_config_portable!ConfigModule.save_config_portable@s ! OOC &CRHI:4!=rBC}!8!;=8V,,"*--/FK&"3%|FK=A)* C r'c^^^ ^ ^ [5m S[S[S[4SjmS[SS4UU 4SjjmS[[-S[[4U4Sjjm S[ S[4S jm S[4UUU U 4S jjn/nUR nUR[US /5S S 9R5HupEURU"X4U55 M T HnURSSU35 M SRU5$)zConvert config to Python statements that replicate current config. This does NOT include config settings that are at default values. funcadd_dotrcRURnUS:XaSnU(a US:waUS- nU$)Nbuiltinsrqr[)r1)rrrs r$get_module_name4ConfigModule.codegen_config..get_module_name^s2//Kj( ;",s"  r'NcJ>T"US5nU(aTRU5 gg)NF)r)rrrimportss r$ add_import/ConfigModule.codegen_config..add_importfs#)$6K K(r'vc^>UVs/sHnT"US5UR3PM sn$s snfr)r0)r itemrs r$list_of_callables_to_string@ConfigModule.codegen_config..list_of_callables_to_stringks2PQRPQtT23DMM?CPQR RRs!*cd[U5=(a [US5=(a [US5$)Nr1r0)callablerc)r s r$importable_callable8ConfigModule.codegen_config..importable_callablens$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 } r[z = Tc34># UH nT"U5v M g7fr<r:)rr rs r$rGConfigModule.codegen_config..get_config_line..s467d#D))asz = { z, z })r0r*r,rRalljoin) modkr r v_listrrrr s r$get_config_line4ConfigModule.codegen_config..get_config_liners#1%%1 as#oa&>%? |LLAc{++467411Dt$4Q7a&&!U!A3c&22!U!A3fTYYv->,?sCCas#aU++r'rT)rrrzimport  ) rRrr7r+r,rr0rrrarinsertr) r#rlinesrrr  import_namerrrrr s @@@@@r$codegen_configConfigModule.codegen_configVs % ( 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_dict##%%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<r/r"s r$shallow_copy_dictConfigModule.shallow_copy_dictr3r'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*dictrloadsrartritorch._dynamo.utilsr9)r#r7rnrr r9s r$ load_configConfigModule.load_configse.55\\"67F)FLLNDALL #9D<s2STU #r'c"UR5$r<)rr"s r$r0ConfigModule.get_config_copys~~r'c6UR[US/5S9$)Nrr)rrr"s r$get_serializable_config_copy)ConfigModule.get_serializable_config_copys~~749NPR+S~TTr'arg1arg2kwargsContextDecoratorc t^^Ub\Ub$[U[5(d [S5eX0mO"[U[5(d [S5eUmU(a [S5eOUmUb [S5e[T[5(d[S[ T535eUm"UU4SjS[ 5nU"5$)aj Decorator and/or context manager to make temporary changes to a config. Note that patched settings are thread-local. 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 cD>\rSrSrSUU4SjjrSU4SjjrU4SjrSrg)'ConfigModule.patch..ConfigPatchiNcd>TUl[TRS[U5S3SS9Ulg)Nz .ConfigPatch[]r:r)changesrr0id_prior)r#rMrns r$r0ConfigModule.patch..ConfigPatch.__init__s3& FP'}RXJa@G r'c>0nURHnTRU5X'M URR5nURR /UQUP75 URR 5HupETR XE5 M g![a URR U5 ef=fr<)rMrrOrHrRrar Exception)r#priorrk prior_stackrr rns r$ __enter__1ConfigModule.patch..ConfigPatch.__enter__s(*<URR5nU(d [S5eUSnURRUSS5 UR 5HupgTR Xg5 M g)Nz2prior should not be empty when exiting ConfigPatchr{)rOrHr!rRrar) r#exc_typeexc_valexc_tbrTrSrr rns r$__exit__0ConfigModule.patch..ConfigPatch.__exit__sh"kkoo/ "(L$B  CR 01!KKMDA&&q,*r')rOrMr/)r0r1r2r3rrUr[r9)rMrnsr$ ConfigPatchrJs    - -r'r])r*r+r!r:r-rG)r#rDrErFr]rMrns @@r$patchConfigModule.patchs.  !$,,(a ,!$--(e$P G$b'4(( #7W !GH H - -* -B}r'rMc 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>^TVs0sH oTURR5_M" snmTR5H-upTRURR U5 M/ SUU4SjjnU$s snf)Nc>TR5H-upTRURRU5 M/ gr<)rartrrR)rr rSr#s r$revertBConfigModule._make_closure_patcher..change..revert9s1!KKMDALLO1155a8*r'r/)rrHrartrR)rr rcrSrMrnr#s @r$change2ConfigModule._make_closure_patcher..change4sr?FGw!q //3355wGE  Q--11!4( 9 9MHs'A=)rtr)r#rMrerns`` @r$_make_closure_patcher"ConfigModule._make_closure_patcher s/$ T*   r')rQrPr/)NNF)NN)+r0r1r2r3r:r+rfr6rRr7bytesrrr5rrrrtuplerrrrrr,rr-rrrr r+rrr1r5r=r0rBr^rgr9 __classcell__rvs@r$rrrrns #|# $$c(s8#O$, . .\ .d .$$F$t$"%R%R%RN''' %! % z3 $ & %LS,L,s,t,CCC4*.-1" 33i$&3s)d*3 3 c3h 3j4C4D4 U 15)- c3h,? ? B !% ! 8  &c3h&  & 8  &4S>&  & VS#X0F V4 V c3h Ud38nU -1VDcN"T)VVsCx. V  VptCH~#r'rrc\^\rSrSrSrS SjrS\4SjrS\\ /\ 4S\ 4U4Sjjr Sr U=r $) rGiBzO Same as contextlib.ContextDecorator, but with support for `unittest.TestCase` rc[S5eNNYIrr"s r$rUContextDecorator.__enter__H !%((r'c[S5erorq)r#rXrYrZs r$r[ContextDecorator.__exit__Krsr'rc>^[U[5(ab[U[R5(aC"U4SjSU5nUR UlUR UlURUlU$[TT]%U5$)NcV>^\rSrSr\SUU4Sjj5r\SUU4Sjj5rSrU=r$),ContextDecorator.__call__.._TestCaseiQc>TR5 [TU] 5 g![a TR SSS5 ef=fr<)rUr setUpClassrRr[clsrvr#s r$rz7ContextDecorator.__call__.._TestCase.setUpClassRs?NN$*,$ dD$7s "Acz>[TU]5 TRSSS5 g!TRSSS5 f=fr<)r tearDownClassr[r{s r$r:ContextDecorator.__call__.._TestCase.tearDownClass[s48-/ dD$7 dD$7s%:r:r/) r0r1r2r3 classmethodrzrr9rk)rvr#s@r$ _TestCaserxQs)88r'r) r*r-rdunittestTestCaser0r2r1r__call__)r#rrrvs` r$rContextDecorator.__call__Nsl dD ! !jx7H7H&I&I 8D 8""&I %)%6%6I "#'??I  w%%r'r:r/) r0r1r2r3r4rUr r[rrrr9rkrls@r$rGrGBs< ))X)&XseSj1&c&&r'rGcv^\rSrSrSrS\S\SS4U4SjjrS\S \SS4S jrS\S\ 4S jr S\SS4S jr S r U=r $)rhikzb Shim to redirect to main config. `config.triton.cudagraphs` maps to _config["triton.cudagraphs"] rnrWrNcF>[TU]SU5 [TU]SU5 g)Nrt_prefix)rr)r#rnrWrvs r$rSubConfigProxy.__init__qs! Iv. Iv.r'r@rIcTURRURU-U5$r<)rtrr)r#r@rIs r$rSubConfigProxy.__setattr__vs"||'' t(;UCCr'cRURRURU-5$r<)rtrrrs r$rSubConfigProxy.__getattr__y ||'' t(;<TRR5H:upURS5(aMXR;dM.[XU5 M< U$)Nr)r`rarbri) child_clsrr  parent_clss r$wrapper$inherit_fields_from..wrappersJ''--/DA<<$$2D2D)D a(0r'r:)rrs` r$inherit_fields_fromrs Nr'r<)8 contextlibrr&rr^rrFrr~rcollections.abcr contextvarsr dataclassesrtypesrrrZrr r r r r typing_extensionsrtorch._utils_internalrintfloatr7r-r+r,rRrjr:rerrr5r rr>rJrxrrsrfrrrGrhrrr:r'r$rs  $"!*KK(1UD$t*c4eTJ tT$ZeD CsU{T)C/$6