`i UddlZddlZddlZddlmZmZddlmZddlm Z m Z ddl m Z ddl Z iiiiiiidiiiiiiiiddZeeeee ffed <Gd d eZee d fZd ZdefdZedGddZiZeeefed<eZee ed<iZeeeefed<iZeeeefed<deddfdZdZdZ dZ!dZ"dS)N) dataclassfield)Enum)AnyOptional) ModuleType)condz dynamic-shapez escape-hatchmapz dynamic-valueoperatormutation)assertbuiltinclosurezcontext-managerz control-flowzdata-structurezstandard-libraryz object-model)torchpython_TAGSceZdZdZdZdZdS) SupportLevelz[ Indicates at what stage the feature used in the example is handled in export. rN)__name__ __module__ __qualname____doc__ SUPPORTEDNOT_SUPPORTED_YETi/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/_export/db/case.pyrr#s' Irr.cHt|tstdt|t|tstdt||D]6}t|t stdt|7dS)Nz(Expecting args type to be a tuple, got: z)Expecting kwargs type to be a dict, got: z+Expecting kwargs keys to be a string, got: ) isinstancetuple ValueErrortypedictstr)argskwargskeys rcheck_inputs_typer)0s dE " "  CtDzz C C    fd # #  FV F F   #s## Id3iiII  rtagc|d}t}|D][}t|ttjdzks Jd|||vr ||}It d|ddS)N.-z!Tag contains invalid characters: zTag z! is not found in registered tags.)splitrsetstringascii_lowercaser")r*partstparts r _validate_tagr5?s IIcNNE ALL4yyC  "S (     5t 5 5    199$AAJCJJJKK KLLrT)frozenceZdZUeed<eed<ejjed<eed<e e Z e ee fed<dZ eeed<e eZeeed <ejZeed <dZee ee fed <d ZdS) ExportCase example_args descriptionmodelname)default_factoryexample_kwargsN extra_argstags support_leveldynamic_shapesc8t|j|j|jt|ji|jD]}t |t |jtrt|jdkrtd|jddS)NrzInvalid description: "") r)r9r>r?r@r5r r:r%lenr")selfr*s r __post_init__zExportCase.__post_init__Ys$+T-@AAA ? & dor 2 2 29  C #    $*C00 KC8H4I4IQ4N4NId6FIIIJJ J5O4Nr)rrrArgsType__annotations__r%rnnModulerr$r>rr?rr/r@rrrArBrGrrrr8r8Ls 8? III%*U4%@%@%@NDcN@@@%)J")))U3///D#c(///"."8M<888/3NHT#s(^,333 K K K K Krr8_EXAMPLE_CASES_MODULES_EXAMPLE_CONFLICT_CASES_EXAMPLE_REWRITE_CASEScasereturnc|jtvrU|jtvr t|jgt|j<t|j|dS|t|j<dS)zA Registers a user provided ExportCase into example bank. N)r<rLrNappend)rPs rregister_db_caserTksh yN"" 93 3 32@2K1L #DI . *11$777 $N49rc~tjdd|}tjdd|S)Nz(.)([A-Z][a-z]+)z\1_\2z([a-z0-9])([A-Z]))resublower)r<s r to_snake_caserYxs7 6$h 5 5D 6%x 6 6 < < > >>rct|tjjst dd|vr!|j Jd|i|d|ji}t dii|||dS)Nz.Export case class should be a torch.nn.Module.r:z9Could not find description or docstring for export case: )r;r<r)r rrJrK TypeErrorrr8)mr<configss r_make_export_caser^}s a ) )JHIIIG## I ! ! Jq J J " ! !7W7mQY77  > >=7=Q=== > >>rc fd}|S)zK Decorator for registering a user provided case into example bank. c }tj|}|tvrtd|Jt||jdd}t|||}t||S)Nz6export_case should only be used once per example file.r,) inspect getmodulerM RuntimeErroraddrr.r^rT)r\r]module module_namerPr's rwrapperzexport_case..wrappers"1%% X  WXX X!!! Vo++C004  K99 rrr'rhs` r export_caserjs#      Nrc fd}|S)Nc2}|d}t|tsJ|j}|tvr gt|<|j|d<t |t|j|}t| ||S)Nparentr9) popr r8r<rOr9r^rYrrS)r\r]rmr(rPr's rrhz$export_rewrite_case..wrappersX&&&*-----k , , ,*, "3 '"("5 M!*$=$=wGGs#**4000 rrris` rexport_rewrite_caseros#      Nr)#rbrVr0 dataclassesrrenumrtypingrrtypesrrrr$r%rIrr!rHr)r5r8rLr/rMrNlistrOrTrYr^rjrorrrrus ((((((((    $$tCc3h  .4 c?    Ls L L L L $KKKKKKKK0)+S*_%***CEE#j/!!!79c4 #33499968S$z"223888 %: %$ % % % %??? ? ? ?*r