a HSicp@@sddlZddlZddlZddlZddlZddlZddlZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZddlmZmZmZddlmZddlmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ddl0m1Z1dd l2m3Z3dd l4m5Z5ej6j78r>dd lm9Z9m:Z:dd lm;Z;dd lZ?Gddde>Z@ddZAddZBddZCddZDddZEddZFddZGd d!ZHd"d#ZIe5ejJd$d%d&ZKd'd(ZLd)d*ZMd+d,ZNgd-ZOdS).N)ListTupleis_tupleis_listDictis_dictOptional is_optional_qualified_nameAnyFuture is_future is_ignored_fnUnionis_union)BroadcastingList1BroadcastingList2BroadcastingList3)_get_script_class) TensorType TupleType FloatTypeIntType ComplexTypeListType StringTypeDictTypeBoolType OptionalType InterfaceTypeAnyTypeNoneType DeviceObjType StreamObjType FutureTypeEnumType UnionType NumberType)dedent)get_source_lines_and_file)Type)RRefis_rref)RRefType)OpOverloadPacketc@seZdZddZddZdS)ModulecCs||_||_dSN)namemembers)selfr3r4r6Q/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/jit/annotations.py__init__szModule.__init__cCs<z |j|WSty6td|jd|dYn0dS)NzModule z has no member called )r4KeyError RuntimeErrorr3r5r3r6r6r7 __getattr__!s  zModule.__getattr__N)__name__ __module__ __qualname__r8r<r6r6r6r7r1sr1c @sLeZdZeddejiejeddeieeee e e d Z ddZ dd Zd S) EvalEnvtorchTensortypingr) rArBrCrrrr rr cCs ||_tjjrt|jd<dS)Nr-)rcbrA distributedrpc is_availabler-env)r5rDr6r6r7r85s zEvalEnv.__init__cCs4||jvr|j|S|jdur(||Stt|dSr2)rHrDgetattrbuiltinsr;r6r6r7 __getitem__:s     zEvalEnv.__getitem__N)r=r>r?r1rArBrrrr rr rHr8rKr6r6r6r7r@(s  r@c Cst|trt|j|}n t||}|durJ|rJ|\}}|dd}||f}|durd\}}z"tdt|d}t|}WntyYn0|durt |||}|S)Nr)NNr) isinstancer0try_real_annotationsopr*joinr+ get_type_line TypeErrorparse_type_line) fnrDloc is_method signature param_types return_type type_linesourcer6r6r7 get_signatureAs"       r\cCst|pt|Sr2)inspect isfunctionismethodZ the_callabler6r6r7is_function_or_method^sracCs8t|st|dr|j}t|r0t|jduSdSdS)N__call__F)rahasattrrbr]getfullargspecvarargsr`r6r6r7 is_varargds rfcCsnt|tr|j}t|s2t|dr2t|jr2|j}t|rXt|rLt|}t |j Sddt |DSdS)NrbcSsg|] }t|qSr6)str).0ir6r6r7 z#get_param_names..) rMr0rOrarcrbrr]unwraprdargsrange)rTZn_argsr6r6r7get_param_namesps   roc Csztdt|d}Wnttfy2YdS0|dur@dSt|}t|jdkrt |jdtj rt j j |d|jdjdt|jdkst |jdtjst j j |ddS)NrLrrzCannot instantiate class 'z' in a script functionz$Expected a single top-level function)r*rPr+rRIOErrorastparselenbodyrMClassDefrAjitfrontend FrontendErrorr3 FunctionDef)rTrUr[py_astr6r6r7check_fns   r{c st|\}}zt|it|}Wn2ttfyR}ztd|WYd}~n d}~00t|tsd|f}zt|it|}Wn2ttfy}ztd|WYd}~n d}~00fdd|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 annotationcsg|]}t|qSr6) ann_to_typerhannrUr6r7rjrkz#parse_type_line..) split_type_lineevalr@ NameError SyntaxErrorr:rMtupler|) rZrDrUZ arg_ann_strZ ret_ann_strZarg_anneZret_ann arg_typesr6rr7rSs    rSc snd|d}ddt|D}ttfdd|}tdttfdd|}t|d krtd ttfd d|}t|d krtd t|d d d dddSt|dkr|d d Sd}g}|D]4\}}d|vr||f}qqވ|vr| |q|dur>tdd dd|Ddfdd}t ||} d | } |d d| S)zETries to find the line containing a comment with the type annotation.# type: cSsg|]\}}||fqSr6r6)rhline_numliner6r6r7rjrkz!get_type_line..cs |dvSNrr6r type_commentr6r7rkzget_type_line..z!# type:\ ignore(\[[a-zA-Z-]+\])?$cs|d Srsearchr) type_patternr6r7rrkrz&#[ ]*type[ ]*(?!: ignore(\[.*\])?$):cs|dSrrr)wrong_type_patternr6r7rrkzThe 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 examplesNrz# type: (...) -> z_Return type line '# type: (...) -> ...' not found on multiline type annotation for type lines: cSsg|] }|dqS)rr6)rhrr6r6r7rjrkzl (See PEP 484 https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)cs"||td}|Sr2)findrsstrip)r item_typerr6r7get_parameter_typesz)get_type_line..get_parameter_typez, z...)split enumeratelistfilterrecompilersr:rgrappendrPmapreplace) r[linesZ type_linesZwrong_type_linesZ return_lineZparameter_type_linesrrrtypesZparameter_typesr6)rrrr7rQsT             rQcCsXtd}z|d}Wnty2tddYn0|||||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)rsindex ValueErrorr:r)rZZ start_offsetZ arrow_posr6r6r7rs  rcszt|Wnty"YdS0jgddjD}tfdd|DrZdSfddjD}tj}||fS)z:Tries to use the Py3.5+ annotation syntax to get the type.NcSsg|] }|jqSr6) annotationrhpr6r6r7rjrkz(try_real_annotations..c3s|]}|juVqdSr2)emptyr})sigr6r7 rkz'try_real_annotations..csg|]}t|jqSr6)r|rrrr6r7rj s)r]rWrreturn_annotation parametersvaluesallr|)rTrUZ all_annotsrrYr6)rUrr7rNs   rN)rcsJt|}|std|jddd|D}fdd|D}tj|S)NzNo enum values defined for: ''cSsh|]}t|jqSr6)typevalue)rhvr6r6r7 rkz&get_enum_value_type..csg|]}t|qSr6try_ann_to_type)rhtrr6r7rjrkz'get_enum_value_type..)rr __class__rA_Cunify_type_list)rrUZ enum_valuesrZir_typesr6rr7get_enum_value_types rc CsPt|tjrdSt|tjtjtjtjtjtjtj tj tj f rLt 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) issubclassrArB LongTensor DoubleTensor FloatTensor IntTensor ShortTensor HalfTensor CharTensor ByteTensor BoolTensorwarningswarn)r~r6r6r7 is_tensor!s    rcs|tjjurtS|dur$tSt|r>t|r>tSt |rt |j dkrj|j ddkrjt gSt fdd|j DSt |rt|j d}|rt|St|r(t|j d}t|j d}|durtd|j dd|durtd|j ddt||St|rt|j dtdrT|j d}n |j d}t|}d}|sJ|t|t|t|St|r&t|j ttttgkrtSg}|j D]T} | dur| tt| } d}| sJ|t|t| | | qt!|St"j#j$%rRt&|rRt't|j dSt(|rpt)t|j dS|turt*S|turt+S|turt,S|t-urt.S|t/urt0S|t1urt2S|tdurtSt|rt3|d rt4|j5S|t"j6ur(t7S|t"j8urj?@|} | A} ntB|} tC| tD|tE|St|rt=|} | dur| St"jFG|rt"j>j?@|Sd d }t"jHI||S) Nrrr6csg|]}t|qSr6r)rharr6r7rj<rkz#try_ann_to_type..Unknown type annotation: '' at zQUnsupported annotation {} could not be resolved because {} could not be resolved.__torch_script_interface__cSsdSr2r6)keyr6r6r7fake_rcbsz!try_ann_to_type..fake_rcb)Jr] Signaturerr getInferredr#getisclassrrrs__args__rrrrrr highlightrr rrformatreprr rsetintfloatcomplexr)rr(rArErFrGr.r/rr&rrrrgrboolrr r"rcr!rdevicer$Streamr%dtypeenumEnumrrv_script_recursive_compile_classqualified_namer r'rr _jit_internalcan_compile_classr_resolve_type_from_object)r~rU elem_typerr containedZ valid_typemsginnerrZ maybe_typeZscripted_classr3Zmaybe_script_classrr6rr7r1s                          rcCs2t||}|dur|Std|d|dS)Nrr)rrr)r~rUthe_typer6r6r7r|s r|)r rrrrrrrrrr rrrrrrrrrr"r1r\r{rorSrQrrNrr|)Prqrr]rrJrArrrrrrrrr r r r r rrrrrrr_statertorch._Crrrrrrrrrr r!r"r#r$r%r&r'r(r)textwrapr*torch._sourcesr+rCr,rErFrGr-r.r/ torch._opsr0objectr1r@r\rarfror{rSrQrrNrrrrr|__all__r6r6r6r7sBD T       >`