o A&iG@sddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddl Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8dd l9m:Z:e j;j<=rdd l m>Z>dd l$m?Z?m@Z@dd lAmBZBGd ddZCGdddZDddZEddZFddZGddZHddZIddZJddZKdd ZLd!d"ZMd#d$ZNd%e ejOfd&d'ZPd(d)ZQd*d+ZRd1d,d-ZSd1d.d/ZTgd0ZUdS)2N)dedent)Type)_GeneratorTypeAnyType 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)selfr7r8r:@C:\wamp64\www\opt\env\Lib\site-packages\torch/jit/annotations.py__init__Gs zModule.__init__cCs4z|j|WStytd|jd|dw)NzModule z has no member called )r8KeyError RuntimeErrorr7r9r7r:r:r; __getattr__Ks  zModule.__getattr__N)__name__ __module__ __qualname__r<r@r:r:r:r;r5Fs r5c @sNeZdZeddejiejeddeieeee e e e d Z ddZdd Zd S) EvalEnvtorchTensortypingr-) rErFrGr-r+r!r,r.r"AwaitcCs$||_tjjrt|jd<dSdS)Nr3)rcbrE distributedrpc is_availabler3env)r9rIr:r:r;r<bs zEvalEnv.__init__cCs4||jvr |j|S|jdur||Stt|dSr6)rMrIgetattrbuiltinsr?r:r:r; __getitem__gs     zEvalEnv.__getitem__N)rArBrCr5rErFr-r+r!r,r.r"rrMr<rPr:r:r:r;rDTs  rDc Cst|tr t|j|}nt||}|dur%|r%|\}}|dd}||f}|durRd\}}ztdt|d}t|}Wn tyGYnw|durRt |||}|S)Nr/)NNr) isinstancer4try_real_annotationsoprjoinr get_type_line TypeErrorparse_type_line) fnrIlocZ is_method signatureZ param_types return_type type_linesourcer:r:r; get_signatureos$       r_cCst|p t|Sr6)inspect isfunctionismethodZ the_callabler:r:r;is_function_or_methodsrdcCs2t|s t|r |j}t|rt|jduSdS)NF)rdcallable__call__r`getfullargspecvarargsrcr:r:r; is_varargs ricCsht|tr|j}t|st|rt|jr|j}t|r+t|r%t|}t |j Sddt |DS)NcSsg|]}t|qSr:)str).0ir:r:r; z#get_param_names..) rRr4rTrdrerfr&r`unwraprgargsrange)rYZn_argsr:r:r;get_param_namess   rrc Csz tdt|d}Wn ttfyYdSw|dur dSt|}t|jdkrEt |jdtj rEt j j |d|jdjdt|jdksUt |jdtjs]t j j |ddS)NrQrr/zCannot instantiate class 'z' in a script functionz$Expected a single top-level function)rrUrOSErrorrWastparselenbodyrRClassDefrEjitZfrontendZ FrontendErrorr7 FunctionDef)rYrZr^Zpy_astr:r:r;check_fns$   r{cCsDt|ddd}t|D]}d|jvrtd|dq t|||S)zLEvaluate statement as long as it does not contain any method/function calls.rQeval)modeCALLz/Type annotation should not contain calls, but 'z' does)compiledisget_instructionsopnamer>r|)stmtglobrZbytecodeZinsnr:r:r; _eval_no_calls   rc st|\}}z t|it|}Wnttfy#}ztd|d}~wwt|ts,|f}z t|it|}WnttfyI}ztd|d}~wwfdd|D}|t|fS)zParse 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|qSr:) ann_to_typerkannrZr:r;rmz#parse_type_line..) split_type_linerrD NameError SyntaxErrorr>rRtupler) r]rIrZZ arg_ann_strZ ret_ann_strZarg_anneZret_ann arg_typesr:rr;rXs2  rXc s`d|d}tt|}ttfdd|}tdttfdd|}t|dkrXtdttfd d|}t|dkrVtd t|ddd d d dSt|dkrf|dd Sd}g}|D]\}}d|vrz||f}n |vr| |ql|durtdd dd|Ddfdd}t ||} d | } |d d| S)zCTry to find the line containing a comment with the type annotation.# type: cs |dvSNr/r:line type_commentr:r;s zget_type_line..z!# type:\ ignore(\[[a-zA-Z-]+\])?$cs|d Srsearchr) type_patternr:r;rsrz&#[ ]*type[ ]*(?!: ignore(\[.*\])?$):cs|dSrrr)wrong_type_patternr:r;r szThe 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/r:)rkrr:r:r;rm'rnz!get_type_line..zl (See PEP 484 https://www.python.org/dev/peps/pep-0484/#suggested-syntax-for-python-2-7-and-straddling-code)cs"||td}|Sr6)findrvstrip)rZ item_typerr:r;get_parameter_type+sz)get_type_line..get_parameter_typez, z...)splitlist enumeratefilterrerrvr>rjrappendrUmapreplace) r^linesZ type_linesZwrong_type_linesZ return_lineZparameter_type_linesline_numrrtypesZparameter_typesr:)rrrr;rVs^            rVcCsTtd}z|d}Wn tytddw|||||ddfS)aSplit 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)rvindex ValueErrorr>r)r]Z start_offsetZ arrow_posr:r:r;r5s  $rcszt|Wn tyYdSwjgddjD}tfdd|Dr-dSfddjD}tj}||fS)z8Try to use the Py3.5+ annotation syntax to get the type.NcSsg|]}|jqSr:) annotationrkpr:r:r;rmSsz(try_real_annotations..c3s|]}|juVqdSr6)emptyr)sigr:r; Vsz'try_real_annotations..csg|]}t|jqSr:)rrrrr:r;rmYs)r`r[rreturn_annotation parametersvaluesallr)rYrZZ all_annotsrr\r:)rZrr;rSIs   rSrcsZt|}|std|jddd|D}fdd|D}tj|}|s+tS|S)NzNo enum values defined for: ''cSsh|]}t|jqSr:)typevalue)rkvr:r:r; erz&get_enum_value_type..crr:try_ann_to_type)rktrr:r;rmfrz'get_enum_value_type..)rr __class__rE_CZunify_type_listrget)rrZZ enum_valuesrZir_typesresr:rr;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) issubclassrErFZ LongTensorZ DoubleTensorZ FloatTensorZ IntTensorZ ShortTensorZ HalfTensorZ CharTensorZ ByteTensorZ BoolTensorwarningswarn)rr:r:r; is_tensorrs( rcCsdSr6r:)inpr:r:r; _fake_rcbsrcsDt|}|tjjurtS|durtSt |r$t |r$tSt |rCt |dkr8|ddkr8t gSt fdd|DSt|rTt|d}|rTt|St|rt|d}t|d}|durxtd|dd|durtd|ddt||St|rt|dtdr|d}n|d}t|}d} |sJ| t|t|tt|St|r t|ttthkrt Sg} t|D]*} | dur| !tt| } d} | sJ| t|t| t| !| qt"| St#j$j%&rt'|rt(t|dSt)|r,t*t|dSt+|rC|r;t|dnt,} t-| S|turLt.S|turUt/S|tus`|t#j0urdt1S|t2urmt3S|t4urvt5S|t6urt,S|tdurtSt |rt7|d rt8|j9S|t#j:urt;S|t#jurt?S|t#j@urt1St |rt|tAjBrtC|durt#jDjEF|}|G}ntH|}tI|tJ|tK|St |rtC|}|dur|St#jLM|rt#jDjEF|S|durtN}t#jOP||S) Nr/rr:crr:r)rkarr:r;rmrz#try_ann_to_type..Unknown type annotation: '' at zWUnsupported annotation {} could not be resolved because {} could not be resolved. At {}__torch_script_interface__)QrGget_argsr` SignaturerrZ getInferredrrisclassrr)rvrr'rrr$r highlightr r(rrformatreprrr*setintfloatcomplexrrrrErJrKrLr2r1r%r r#rrr rZSymIntrrjrboolrrhasattrrrZdevicer GeneratorrZStreamrZdtypeenumEnumr0ry_scriptZ_recursive_compile_classZqualified_namerr rr _jit_internalZcan_compile_classrrZ_resolve_type_from_object)rrZrIZann_argsZ elem_typekeyr containedZ valid_typemsginnerrZ maybe_typeZ elementTypeZscripted_classr7Zmaybe_script_classr:rr;rs     "  "                 rcCs0t|||}|dur |Std|d|)Nrr)rrr)rrZrIZthe_typer:r:r;rs r)rr+rrr r-r)r'r!r$r(r*rrr rrrrr rr5r_r{rrrXrVrrSrrr6)VrtrOrrr`rrGrtextwraprrrEZtorch._Crrrrrr r r r r rrrrrrrrrrrZtorch._sourcesrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r._stater0rJrKrLr1r2r3Z torch._opsr4r5rDr_rdrirrr{rrXrVrrSrrrrrr__all__r:r:r:r;sJ  \ X      C  k