a dB@sddlZddlZddlZddlZddlZddlZddlZddlZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZddlmZmZmZddlmZddlm 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/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5dd l6m7Z7dd l8m9Z9ej:j;<rRdd lm=Z=m>Z>dd lm?Z?dd l@mAZAGdddZBGdddZCddZDddZEddZFddZGddZHddZIddZJd d!ZKd"d#ZLd$d%ZMe9ejNd&d'd(ZOd)d*ZPd+d,ZQd-d.ZRgd/ZSdS)0N)ListTupleis_tupleis_listDictis_dictOptional is_optional_qualified_nameAnyFuture is_future_Awaitis_await is_ignored_fnUnionis_union)BroadcastingList1BroadcastingList2BroadcastingList3)_get_script_class) TensorType TupleType FloatTypeIntType ComplexTypeListType StringTypeDictTypeBoolType OptionalType InterfaceTypeAnyTypeNoneType DeviceObjType StreamObjType FutureType AwaitTypeEnumType UnionType NumberType)dedent)get_source_lines_and_file)Type)RRefis_rref)RRefType)OpOverloadPacketc@seZdZddZddZdS)ModulecCs||_||_dSN)namemembers)selfr6r7r9^/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/jit/annotations.py__init__szModule.__init__cCs<z |j|WSty6td|jd|dYn0dS)NzModule z has no member called )r7KeyError RuntimeErrorr6r8r6r9r9r: __getattr__"s  zModule.__getattr__N)__name__ __module__ __qualname__r;r?r9r9r9r:r4sr4c @sNeZdZeddejiejeddeieeee e e e d Z ddZdd Zd S) EvalEnvtorchTensortypingr) rDrErFrrrr rr AwaitcCs ||_tjjrt|jd<dS)Nr0)rcbrD distributedrpc is_availabler0env)r8rHr9r9r:r;7s zEvalEnv.__init__cCs4||jvr|j|S|jdur(||Stt|dSr5)rLrHgetattrbuiltinsr>r9r9r: __getitem__<s     zEvalEnv.__getitem__N)r@rArBr4rDrErrrr rr rrLr;rOr9r9r9r:rC)s  rCc Cst|trt|j|}n t||}|durJ|rJ|\}}|dd}||f}|durd\}}z"tdt|d}t|}WntyYn0|durt |||}|S)Nr)NNr) isinstancer3try_real_annotationsopr-joinr. get_type_line TypeErrorparse_type_line) fnrHlocZ is_method signatureZ param_types return_type type_linesourcer9r9r: get_signatureCs"       r^cCst|pt|Sr5)inspect isfunctionismethodZ the_callabler9r9r:is_function_or_method`srccCs8t|st|dr|j}t|r0t|jduSdSdS)N__call__F)rchasattrrdr_getfullargspecvarargsrbr9r9r: is_varargfs rhcCsnt|tr|j}t|s2t|dr2t|jr2|j}t|rXt|rLt|}t |j Sddt |DSdS)NrdcSsg|] }t|qSr9)str).0ir9r9r: z#get_param_names..) rQr3rSrcrerdrr_unwraprfargsrange)rXZn_argsr9r9r:get_param_namesrs   rqc Csztdt|d}Wnttfy2YdS0|dur@dSt|}t|jdkrt |jdtj rt j j |d|jdjdt|jdkst |jdtjst j j |ddS)NrPrrzCannot instantiate class 'z' in a script functionz$Expected a single top-level function)r-rTr.rVIOErrorastparselenbodyrQClassDefrDjitZfrontendZ FrontendErrorr6 FunctionDef)rXrYr]Zpy_astr9r9r:check_fns   rzcCsDt|ddd}t|D]}d|jvrtd|dqt|||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|\}}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|qSr9) ann_to_typerjannrYr9r:rlrmz#parse_type_line..) split_type_linerrC NameError SyntaxErrorr=rQtupler) r\rHrYZ arg_ann_strZ ret_ann_strZarg_anneZret_ann arg_typesr9rr:rWs    rWc 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|]\}}||fqSr9r9)rjline_numliner9r9r:rlrmz!get_type_line..cs |dvSNrr9r type_commentr9r:rmzget_type_line..z!# type:\ ignore(\[[a-zA-Z-]+\])?$cs|d Srsearchr) type_patternr9r:rrmrz&#[ ]*type[ ]*(?!: ignore(\[.*\])?$):cs|dSrrr)wrong_type_patternr9r:rrmzThe 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)rr9)rjrr9r9r:rlrmzl (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_typesz)get_type_line..get_parameter_typez, z...)split enumeratelistfilterrer~rur=rirappendrTmapreplace) r]linesZ type_linesZwrong_type_linesZ return_lineZparameter_type_linesrrrtypesZparameter_typesr9)rrrr:rUsT             rUcCsXtd}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)ruindex ValueErrorr=r)r\Z start_offsetZ arrow_posr9r9r:rs  rcszt|Wnty"YdS0jgddjD}tfdd|DrZdSfddjD}tj}||fS)z:Tries to use the Py3.5+ annotation syntax to get the type.NcSsg|] }|jqSr9) annotationrjpr9r9r:rlrmz(try_real_annotations..c3s|]}|juVqdSr5)emptyr)sigr9r: rmz'try_real_annotations..csg|]}t|jqSr9)rrrrr9r:rls)r_rZrreturn_annotation parametersvaluesallr)rXrYZ all_annotsrr[r9)rYrr:rRs   rR)rcsZt|}|std|jddd|D}fdd|D}tj|}|sVtS|S)NzNo enum values defined for: ''cSsh|]}t|jqSr9)typevalue)rjvr9r9r: #rmz&get_enum_value_type..csg|]}t|qSr9try_ann_to_type)rjtrr9r:rl$rmz'get_enum_value_type..)rr __class__rD_CZunify_type_listr$get)rrYZ enum_valuesrZir_typesresr9rr:get_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) issubclassrDrEZ LongTensorZ DoubleTensorZ FloatTensorZ IntTensorZ ShortTensorZ HalfTensorZ CharTensorZ ByteTensorZ BoolTensorwarningswarn)rr9r9r: 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 ttthkrtSg}|j D]T} | dur| tt| } d}| s J|t|t| | | qt!|St"j#j$%rNt&|rNt't|j dSt(|rlt)t|j dSt*|rt+|d rt|j dnt,} t-| S|turt.S|turt/S|turt0S|t1urt2S|t3urt4S|t5urt,S|tdur$tSt|rFt+|d rFt6|j7S|t"j8urZt9S|t"j:urnt;S|t"jrt?|durt"j@jAB|} | C} ntD|} tE| tF|tG|St|r"t?|}|dur|St"jHI|r"t"j@jAB|Sd d }t"jJK||S) Nrrr9csg|]}t|qSr9r)rjarr9r:rlJrmz#try_ann_to_type..Unknown type annotation: '' at zQUnsupported annotation {} could not be resolved because {} could not be resolved.__args____torch_script_interface__cSsdSr5r9)keyr9r9r:fake_rcbsz!try_ann_to_type..fake_rcb)Lr_ SignaturerrZ getInferredr%risclassrrrurrrrrrr highlightr r rrformatreprr"rsetintfloatcomplexr,rr+rDrIrJrKr1r2rr(rrer$r)rrrrirboolr!r r#rZdevicer&ZStreamr'ZdtypeenumEnumrrx_scriptZ_recursive_compile_classZqualified_namer r*rr _jit_internalZcan_compile_classrZ_resolve_type_from_object)rrYZ elem_typerrZ containedZ valid_typemsginnerrZ maybe_typeZ elementTypeZscripted_classr6Zmaybe_script_classrr9rr:r?s              $             rcCs2t||}|dur|Std|d|dS)Nrr)rrr)rrYZthe_typer9r9r:rs r)r rrrrrrrrrr rrrrrrrrr r$r4r^rzrqrWrUrrRrr)Trsrrr_rrNrDrrrrrrrrr r r r r rrrrrrrrr_staterZtorch._Crrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,textwrapr-Ztorch._sourcesr.rFr/rIrJrKr0r1r2Z torch._opsr3r4rCr^rcrhrqrzrrWrUrrRrrrrr__all__r9r9r9r:sFL X        >c