o iHE@sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z ddl!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7dd l8m9Z9e j:j;<rdd l m=Z=dd l#m>Z>m?Z?dd l@mAZAGd ddZBGdddZCddZDddZEddZFddZGddZHddZIddZJdd ZKd!d"ZLd#d$ZMd%e ejNfd&d'ZOd(d)ZPd*d+ZQd1d,d-ZRd1d.d/ZSgd0ZTdS)2N)dedent)Type)AnyType AwaitTypeBoolType ComplexType DeviceObjTypeDictTypeEnumType FloatType FutureType InterfaceTypeIntTypeListTypeNoneType NumberType OptionalType StreamObjType StringType TensorType TupleType UnionType)get_source_lines_and_file)_Await_qualified_nameAnyBroadcastingList1BroadcastingList2BroadcastingList3DictFutureis_awaitis_dict is_future is_ignored_fnis_list is_optionalis_tupleis_unionListOptionalTupleUnion)_get_script_class)RRefType)is_rrefRRef)OpOverloadPacketc@seZdZddZddZdS)ModulecCs||_||_dSN)namemembers)selfr6r7r9d/var/www/html/eduruby.in/lip-sync/lip-sync-env/lib/python3.10/site-packages/torch/jit/annotations.py__init__Es zModule.__init__cCs4z|j|WStytd|jd|dw)NzModule z has no member called )r7KeyError RuntimeErrorr6r8r6r9r9r: __getattr__Is  zModule.__getattr__N)__name__ __module__ __qualname__r;r?r9r9r9r:r4Ds r4c @sNeZdZeddejiejeddeieeee e e e d Z ddZdd Zd S) EvalEnvtorchTensortypingr,) rDrErFr,r*r r+r-r!AwaitcCs$||_tjjrt|jd<dSdS)Nr2)rcbrD distributedrpc is_availabler2env)r8rHr9r9r:r;`s zEvalEnv.__init__cCs4||jvr |j|S|jdur||Stt|dSr5)rLrHgetattrbuiltinsr>r9r9r: __getitem__es     zEvalEnv.__getitem__N)r@rArBr4rDrEr,r*r r+r-r!rrLr;rOr9r9r9r:rCRs  rCc Cst|tr t|j|}nt||}|dur%|r%|\}}|dd}||f}|durRd\}}ztdt|d}t|}Wn tyGYnw|durRt |||}|S)Nr.)NNr) isinstancer3try_real_annotationsoprjoinr get_type_line TypeErrorparse_type_line) fnrHlocZ is_method signatureZ param_types return_type type_linesourcer9r9r: get_signaturems$       r^cCst|p t|Sr5)inspect isfunctionismethodZ the_callabler9r9r:is_function_or_methodsrccCs2t|s t|r |j}t|rt|jduSdS)NF)rccallable__call__r_getfullargspecvarargsrbr9r9r: is_varargs rhcCsht|tr|j}t|st|rt|jr|j}t|r+t|r%t|}t |j Sddt |DS)NcSsg|]}t|qSr9)str).0ir9r9r: z#get_param_names..) rQr3rSrcrdrer%r_unwraprfargsrange)rXZn_argsr9r9r:get_param_namess   rqc Csz tdt|d}Wn ttfyYdSw|dur dSt|}t|jdkrEt |jdtj rEt j j |d|jdjdt|jdksUt |jdtjs]t j j |ddS)NrPrr.zCannot instantiate class 'z' in a script functionz$Expected a single top-level function)rrTrOSErrorrVastparselenbodyrQClassDefrDjitZfrontendZ FrontendErrorr6 FunctionDef)rXrYr]Zpy_astr9r9r:check_fns$   rzcCsDt|ddd}t|D]}d|jvrtd|dq t|||S)zKEvaluate statement as long as it does not contain any method/function callsrPeval)modeCALLz/Type annotation should not contain calls, but 'z' does)compiledisget_instructionsopnamer=r{)stmtglobrYbytecodeZinsnr9r9r: _eval_no_calls   rc st|\}}z t|it|}Wnttfy#}ztd|d}~wwt|ts,|f}z t|it|}WnttfyI}ztd|d}~wwfdd|D}|t|fS)zParses a type annotation specified as a comment. Example inputs: # type: (Tensor, torch.Tensor) -> Tuple[Tensor] # type: (Tensor, Tuple[Tensor, Tensor]) -> Tensor z6Failed to parse the argument list of a type annotationNz4Failed to parse the return type of a type annotationcg|]}t|qSr9) ann_to_typerjannrYr9r:rlz#parse_type_line..) split_type_linerrC NameError SyntaxErrorr=rQtupler) r\rHrYZ arg_ann_strZ ret_ann_strZarg_anneZret_ann arg_typesr9rr:rWs2  rWc sfd|d}ddt|D}ttfdd|}tdttfdd|}t|d kr[td ttfd d|}t|d krYtd t|d d d dddSt|dkri|d d Sd}g}|D]\}}d|vr}||f}n |vr| |qo|durtdd dd|Ddfdd}t ||} d | } |d d| S)zETries to find the line containing a comment with the type annotation.# type: cSsg|]\}}||fqSr9r9)rjline_numliner9r9r:rlz!get_type_line..cs |dvSNr.r9r type_commentr9r:s zget_type_line..z!# type:\ ignore(\[[a-zA-Z-]+\])?$cs|d Srsearchr) type_patternr9r:rsrz&#[ ]*type[ ]*(?!: ignore(\[.*\])?$):cs|dSrrr)wrong_type_patternr9r:rszThe annotation prefix in line z* is probably invalid. It must be '# type:'zl See PEP 484 (https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)z for examplesNr.z# type: (...) -> z_Return type line '# type: (...) -> ...' not found on multiline type annotation for type lines: cSsg|]}|dqS)r.r9)rjrr9r9r:rl%rmzl (See PEP 484 https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)cs"||td}|Sr5)findrustrip)rZ item_typerr9r:get_parameter_type)sz)get_type_line..get_parameter_typez, z...)split enumeratelistfilterrer~rur=rirappendrTmapreplace) r]linesZ type_linesZwrong_type_linesZ return_lineZparameter_type_linesrrrtypesZparameter_typesr9)rrrr:rUs^           rUcCsTtd}z|d}Wn tytddw|||||ddfS)aSplits the comment with the type annotation into parts for argument and return types. For example, for an input of: # type: (Tensor, torch.Tensor) -> Tuple[Tensor, Tensor] This function will return: ("(Tensor, torch.Tensor)", "Tuple[Tensor, Tensor]") rz->z3Syntax error in type annotation (cound't find `->`)Nr)ruindex ValueErrorr=r)r\ start_offsetZ arrow_posr9r9r:r3s  $rcszt|Wn tyYdSwjgddjD}tfdd|Dr-dSfddjD}tj}||fS)z:Tries to use the Py3.5+ annotation syntax to get the type.NcSsg|]}|jqSr9) annotationrjpr9r9r:rlQsz(try_real_annotations..c3s|]}|juVqdSr5)emptyr)sigr9r: Tsz'try_real_annotations..csg|]}t|jqSr9)rrrrr9r:rlWr)r_rZrreturn_annotation parametersvaluesallr)rXrYZ all_annotsrr[r9)rYrr:rRGs   rRrcsZt|}|std|jddd|D}fdd|D}tj|}|s+tS|S)NzNo enum values defined for: ''cSsh|]}t|jqSr9)typevalue)rjvr9r9r: crz&get_enum_value_type..crr9try_ann_to_type)rjtrr9r:rldrz'get_enum_value_type..)rr __class__rD_CZunify_type_listrget)rrYZ enum_valuesrZir_typesresr9rr:get_enum_value_type^s rc CsPt|tjrdSt|tjtjtjtjtjtjtj tj tj f r&t ddSdS)NTzTorchScript will treat type annotations of Tensor dtype-specific subtypes as if they are normal Tensors. dtype constraints are not enforced in compilation either.F) issubclassrDrEZ LongTensorZ DoubleTensorZ FloatTensorZ IntTensorZ ShortTensorZ HalfTensorZ CharTensorZ ByteTensorZ BoolTensorwarningswarn)rr9r9r: is_tensorps( rcCsdSr5r9)inpr9r9r: _fake_rcbsrcs<|tjjur tS|durtSt|rt|rtSt |rAt |j dkr5|j ddkr5t gSt fdd|j DSt |rSt|j d}|rSt|St|rt|j d}t|j d}|durztd|j dd|durtd|j ddt||St|rt|j dtdr|j d}n|j d}t|}d}|sJ|t|t|tt|St|rt|j ttthkrtSg} |j D]+} | dur| tt| } d}| sJ|t|t| t| | qt!| St"j#j$%r%t&|r%t't|j dSt(|r4t)t|j dSt*|rOt+|d rGt|j dnt,} t-| S|turXt.S|turat/S|turjt0S|t1urst2S|t3ur|t4S|t5urt,S|tdurtSt|rt+|d rt6|j7S|t"j8urt9S|t"j:urt;S|t"jrt?|durt"j@jAB|} | C}ntD|}tE|tF|tG|St|rt?|}|dur|St"jHI|rt"j@jAB|S|durtJ}t"jKL||S) Nr.rr9crr9r)rjarr9r:rlrz#try_ann_to_type..Unknown type annotation: '' at zWUnsupported annotation {} could not be resolved because {} could not be resolved. At {}__args____torch_script_interface__)Mr_ SignaturerrZ getInferredrrisclassrr(rurrr&rrr#r highlightr r'rrformatreprrr)setintfloatcomplexrrrrDrIrJrKr1r0r$r r"hasattrrrr rrrirboolrrr rZdevicerZStreamrZdtypeenumEnumr/rx_scriptZ_recursive_compile_classZqualified_namerr rr _jit_internalZcan_compile_classrrZ_resolve_type_from_object)rrYrHZ elem_typekeyr containedZ valid_typemsginnerrZ maybe_typeZ elementTypeZscripted_classr6Zmaybe_script_classr9rr:rs     "   $                 rcCs0t|||}|dur |Std|d|)Nrr)rrr)rrYrHZthe_typer9r9r:rs r)rr*rrrr,r(r&r r#r'r)rrr rrrrr rr4r^rzrqrWrUrrRrrr5)UrsrNrrr_rrtextwraprrFrrDZtorch._Crrrrrr r r r r rrrrrrrrrrZtorch._sourcesrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-_stater/rIrJrKr0r1r2Z torch._opsr3r4rCr^rcrhrqrzrrWrUrrRrrrrrr__all__r9r9r9r:sH  X X      C  k