o iE@sddlZddlZddlZddlZddlZddlZddlZddlZddlm Z ddlm Z ddl Z ddl m Z mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z 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.m/Z/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6ddl7m8Z8ddl9m:Z:e j;j<=rdd l m>Z>dd l"m?Z?m@Z@dd lAmBZBGd d d ZCGdddZDddZEddZFddZGddZHddZIddZJddZKddZLd d!ZMd"d#ZNd$e ejOfd%d&ZPd'd(ZQd)d*ZRd0d+d,ZSd0d-d.ZTgd/ZUdS)1N)dedent)Type)_GeneratorTypeAnyType AwaitTypeBoolType ComplexType DeviceObjTypeDictTypeEnumType FloatType FutureType InterfaceTypeIntTypeListTypeNoneType NumberType OptionalType StreamObjType StringType TensorType TupleType UnionType)_Await_qualified_nameAnyBroadcastingList1BroadcastingList2BroadcastingList3DictFutureis_awaitis_dict is_future is_ignored_fnis_list is_optionalis_tupleis_unionListOptionalTupleUnion)get_source_lines_and_file)_get_script_class)RRefType)is_rrefRRef)OpOverloadPacketc@seZdZddZddZdS)ModulecCs||_||_dSN)namemembers)selfr6r7r9S/home/ubuntu/transcripts/venv/lib/python3.10/site-packages/torch/jit/annotations.py__init__Hs zModule.__init__cCs4z|j|WStytd|jd|dw)NzModule z has no member called )r7KeyError RuntimeErrorr6r8r6r9r9r: __getattr__Ls  zModule.__getattr__N)__name__ __module__ __qualname__r;r?r9r9r9r:r4Gs r4c @sNeZdZeddejiejeddeieeee e e e d Z ddZdd Zd S) EvalEnvtorchTensortypingr+) rDrErFr+r)rr*r,r AwaitcCs$||_tjjrt|jd<dSdS)Nr2)rcbrD distributedrpc is_availabler2env)r8rHr9r9r:r;cs zEvalEnv.__init__cCs4||jvr |j|S|jdur||Stt|dSr5)rLrHgetattrbuiltinsr>r9r9r: __getitem__hs     zEvalEnv.__getitem__N)r@rArBr4rDrEr+r)rr*r,r rrLr;rOr9r9r9r:rCUs  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) fnrHloc is_method signature param_types return_type type_linesourcer9r9r: get_signatureps$       r`cCst|p t|Sr5)inspect isfunctionismethod the_callabler9r9r:is_function_or_methodsrfcCs2t|s t|r |j}t|rt|jduSdS)NF)rfcallable__call__ragetfullargspecvarargsrdr9r9r: is_varargs rkcCsht|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..) rQr3rSrfrgrhr$raunwrapriargsrange)rXn_argsr9r9r:get_param_namess   ruc 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)rrTr-OSErrorrVastparselenbodyrQClassDefrDjitfrontend FrontendErrorr6 FunctionDef)rXrYr_py_astr9r9r:check_fns$   rcCsDt|ddd}t|D]}d|jvrtd|dq t|||S)zLEvaluate statement as long as it does not contain any method/function calls.rPeval)modeCALLz/Type annotation should not contain calls, but 'z' does)compiledisget_instructionsopnamer=r)stmtglobrYbytecodeinsnr9r9r: _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|qSr9) ann_to_typermannrYr9r:roz#parse_type_line..) split_type_linerrC NameError SyntaxErrorr=rQtupler) r^rHrY arg_ann_str ret_ann_strarg_anneret_ann arg_typesr9rr:rWs2  rWc 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.r9line 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: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.r9)rmrr9r9r:ro(rpz!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}|Sr5)findrystrip)r item_typerr9r:get_parameter_type,sz)get_type_line..get_parameter_typez, z...)splitlist enumeratefilterrerryr=rlrappendrTmapreplace) r_lines type_lineswrong_type_lines return_lineparameter_type_linesline_numrrtypesparameter_typesr9)rrrr:rUs^            rUcCsTtd}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->z4Syntax error in type annotation (couldn't find `->`)N)ryindex ValueErrorr=r)r^ start_offset arrow_posr9r9r:r6s  $rcszt|Wn tyYdSwjgddjD}tfdd|Dr-dSfddjD}tj}||fS)z8Try to use the Py3.5+ annotation syntax to get the type.NcSsg|]}|jqSr9) annotationrmpr9r9r:roTsz(try_real_annotations..c3s|]}|juVqdSr5)emptyr)sigr9r: Wsz'try_real_annotations..csg|]}t|jqSr9)rrrrr9r:roZs)rar[rreturn_annotation parametersvaluesallr)rXrY all_annotsrr]r9)rYrr:rRJs   rRrcsZt|}|std|jddd|D}fdd|D}tj|}|s+tS|S)NzNo enum values defined for: ''cSsh|]}t|jqSr9)typevalue)rmvr9r9r: frz&get_enum_value_type..crr9try_ann_to_type)rmtrr9r:rogrz'get_enum_value_type..)rr __class__rD_Cunify_type_listrget)rrY enum_valuesrir_typesresr9rr:get_enum_value_typeas 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) issubclassrDrE LongTensor DoubleTensor FloatTensor IntTensor ShortTensor HalfTensor CharTensor ByteTensor BoolTensorwarningswarn)rr9r9r: is_tensorss( rcCsdSr5r9)inpr9r9r: _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.rr9crr9r)rmarr9r:rorz#try_ann_to_type..Unknown type annotation: '' at zWUnsupported annotation {} could not be resolved because {} could not be resolved. At {}__torch_script_interface__)QrFget_argsra Signaturerr getInferredrrisclassrr'ryrr%rrr"r highlightr r&rrformatreprrr(setintfloatcomplexrrrrDrIrJrKr1r0r#r r!rrr rSymIntrrlrboolrrhasattrrrdevicer GeneratorrStreamrdtypeenumEnumr/r|_script_recursive_compile_classqualified_namerr rr _jit_internalcan_compile_classrr_resolve_type_from_object)rrYrHann_args elem_typekeyr contained valid_typemsginnerr maybe_type elementTypescripted_classr6maybe_script_classr9rr:rs     "  "                 rcCs0t|||}|dur |Std|d|)Nrr)rrr)rrYrHthe_typer9r9r:rs r)rr)rrrr+r'r%rr"r&r(rrr rrrrr rr4r`rrurWrUrrRrrr5)VrwrNrrrarrFrtextwraprrrDtorch._Crrrrrr r r r r rrrrrrrrrrrtorch._jit_internalrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,torch._sourcesr-_stater/rIrJrKr0r1r2 torch._opsr3r4rCr`rfrkrurrrWrUrrRrrrrrr__all__r9r9r9r:sJ  \X       C  k