Ph:ddlZddlZddlZddlmZddlmZmZmZm Z m Z m Z m Z ddl Z ddlmZmZddlmZddlmZddlmZmZmZmZddlmZdd lmZmZmZGd d e Z!d ed e"fdZ#de jHjJd dfdZ&de jHjJd dfdZ'Gdde(Z)Gdde)Z*ddZ+de,d e e e*fdZ-y)N)Iterable)AnyDictfinalListOptionalTupleType)HigherOrderOperator OpOverload) FakeTensor)ExportedProgram)ExportGraphSignature InputKindSymIntArgumentTensorArgument) GraphModule)SymBoolSymFloatSymIntc eZdZy)SpecViolationErrorN)__name__ __module__ __qualname__aC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/_export/verifier.pyrrsrropreturnc0|jj SN)_schema is_mutablers r is_functionalr&szz$$ $$rnodect|Sr") _check_val)r's r_check_has_fake_tensorr*s d rc fdd}d|jvr;|jdk(r||jrytd|jd|jd}|std|jd|y)Nc|yt|ttttfryt|t j t jt jt jfryt|tt jfryt|tttfryt|trt!fd|DSy)NTc3.K|] }|ywr"r).0x_check_correct_vals r z9_check_val.._check_correct_val../s:c)!,csF) isinstanceintboolstrfloattorch memory_formatdtypedevicelayoutr Tensorrrrrall)valr0s rr0z&_check_val.._check_correct_val#s ; c4e4 5 e115;; ell[ \ j%,,7 8 fh8 9 X &:c:: :rcht|tsyt|jjdk(S)NFr)r2r lenr#returnsr%s r _no_returnsz_check_val.._no_returns2s)"j)2::%%&!++rr> call_functionz Node.meta z is missing val field.z has invalid val field )metartargetrname)r'rBr>r0s @rr)r)"s ,  DII 77o %+dkk*B  :dii[8N!OPP ))E C c " :dii[8OPSu!UVV #rc2eZdZUiZeeedfed<dZy) _VerifierMetaVerifier _registryc|r!d|vsd|vr tdd|vr|ddk7sJd|vsJd|vsJ|ddk(sJt|dtsJtj ||||}||j |d<|S)Ncheck_check_graph_modulez'Overriding method check is not allowed.dialectATEN) SyntaxErrorr2r5type__new__rJ)metaclsrFbasesattrsrets rrRz_VerifierMeta.__new__Ds %#8E#A!"KLL%% *:f*D DDe# ##(E1 11#v- --% *C000ll7D%7.1% *+ rN) rrrrJrr5r __annotations__rRrrrrHrHAs -/ItCj))*/ rrHceZdZdZdefdZdeeedffdZ deeedffdZ dZ de dd fd Z ed edd fd Zedej$jdd fd Zy )rIrOr c|tjtjtjtjtj tj tjtjtjtjtjtjtjtjtjtj tj"tj$tj&t(j*t(j,gSr")operatorgetitemaddmulsubtruedivgelegtlteqnefloordivmodand_or_not_pownegabsmathceilfloorselfs rallowed_builtin_opszVerifier.allowed_builtin_opsWs    LL LL LL    KK KK KK KK KK KK    LL MM LL MM LL LL LL II JJ+  r.cttfSr")r r rqs rallowed_op_typeszVerifier.allowed_op_typesps/00rc8tjjfSr")r7fxrrqs rallowed_getattr_typeszVerifier.allowed_getattr_typesss$$&&rcyr"r)rrrs rcheck_valid_opzVerifier.check_valid_opvs rgmNcy)zG Additional checks that are specific to some dialects. Nr)rrr{s rcheck_additionalzVerifier.check_additionalys repct|jtsy|j|j t |y#t $r}|jdk(rn|Yd}~yd}~wwxYw)NEDGE)r2graph_signaturerrM graph_module"_verify_exported_program_signaturerrN)rrr~es rrLzVerifier.checks^",,.BC    1  .r 2! zzV# s A A) A$$A)c  dtttdfffd }dfd }|jD]:}t |t j js)|jj|jjD]}|jdvrtd|jd|jdk(rt|||jX|jd k(r[t |jts!td t!|jt#||j t t j$j&r fd }t! j(d k(r|d tr(|dt*r|dt,rt/ dr/t# d d}t# dd}t# dd} tdt! dt!|dt!|dt!| t |rtdt! d||jdk(st|=j1|y)Nr .cPj}td|DrJ|S)Nc3,K|] }|tuywr"objectr.ts rr1zOVerifier._check_graph_module.._allowed_getattr_types..s411;)rxanyrVrrs r_allowed_getattr_typesz._allowed_getattr_typess*,,.C444 44Jrc dtffd }dtttdfffd }tj j jf}t||s.||vr%||vr!td|d|d|d|t|trt|std |d j|y) Nr cPj}td|DsJ|S)Nc3FK|]}tj|ywr")inspect isbuiltin)r.rs rr1zfVerifier._check_graph_module.._check_valid_op.._allowed_builtin_ops..s?3R7,,R03s!)rsr=rs r_allowed_builtin_opszSVerifier._check_graph_module.._check_valid_op.._allowed_builtin_opss*..0?3???? r.cPj}td|DrJ|S)Nc3,K|] }|tuywr"rrs rr1zcVerifier._check_graph_module.._check_valid_op.._allowed_op_types..s8CqqF{Cr)rurrs r_allowed_op_typeszPVerifier._check_graph_module.._check_valid_op.._allowed_op_typess*++-8C8888 rz Operator 'z#' is not an allowed operator type: z Valid builtin ops: zValid torch functions: z operator 'z' is not functional) rr r rr7autograd grad_modeset_grad_enabledr2rr r&rz)rrr_allowed_torch_functionsrrs r_check_valid_opz5Verifier._check_graph_module.._check_valid_ops $   uT#Y^'<  ).(@(@(Q(Q'S $b"3"56133BZ8Z,$RD(KL]L_K`a..B.D-E12J1KM "j)$R(,$RD(;<    #r> call_method call_modulez'call_module is not valid: got a class 'z' rCget_attrz/Expected get_attr target to be string, but got c2tt|d|Sr")r2getattr)rFtyattrs r_is_typez.Verifier._check_graph_module.._is_types#-gdD$.G#LLrLoweredBackendModule backend_idprocessed_bytes compile_specsoriginal_modulezInvalid get_attr type z2. LoweredBackendModule fields: backend_id(str) : z, processed_bytes(bytes) : z, compile_specs(list) : z. Valid get_attr types: placeholderr N)r r rmodulesr2r7rwrgraphlintnodesrrrEr)r5rQrnnModulerbyteslisthasattrr}) rrr{rrrgr'rrrrrs ` @rrMzVerifier._check_graph_modulesA d3in(=   $<::|jj|k7r%td|jjd||jtj k(r|jtj"k(rt|jts&td|jd|jd |j$td |jd |j$}||j&vrtd|d t|j&|t(j*j,ratd |d|jtj.k(rt|jts&td|jd|jd |j$td |jd |j$}||j&vstd|d |jtj0k(rt|jts&td|jd|jd |j$td |jd |j$}||j2vstd|dtd|jdt5|jjd}|jdk(sJ|j6dD cgc]} | j} } t | t |j8k7rDtdt | dt |j:dt |j<dt |j:t |j>z} | d| } | | | t |j<z} | D]}||j:vrE|j:||j@vs-td|d|j:d|j@d||j>vrE|j>||jvstd|d |j>d!|jdtd"|d#|j:d$|j>t| |j<D]&\}}||k7s td%|d&|j<d'ycc}wcc} w)(NrzNumber of graph inputs (z:) does not match number of inputs in the graph signature ()zInput spec name z does not match node name z Parameter z! is not a tensor argument. Found z instead.zInputSpec for z has no target.z is not in the state dict.zState dict entry for parameter z* is not an instance of torch.nn.Parameter.zBuffer zConstant tensor z+ is not in the tensor constants dictionary.zUnknown InputKind .outputrzNumber of output nodes zg is different Than the number of outputs specified by the graph signature: Number of mutated buffers: z. Number of user outputs: z. zBuffer output zV does not point to a buffer that exists. Dict of buffers that are mutated, in order: z Buffer nodes available: z zUser input output z^ does not point to a user input that exists. Dict of user inputs that are mutated, in order: z User input nodes available: zMutation node z: is neither a buffer nor a user input. Buffers to mutate: z, User inputs to mutate: z User output zY is not in the correct order or is not found in the exported program's user_output list: z. )!rrrrrFr@ input_specsr user_inputszipr2argrrkindr USER_INPUT PARAMETERrE state_dictr7r ParameterBUFFERCONSTANT_TENSORtensor_constantsrargs output_specsbuffers_to_mutate user_outputsuser_inputs_to_mutatebuffers)exported_programgsr'input_node_names input_specparambuffer tensor_const output_noder output_nodesend mutate_nodesuser_output_nodes mutation_nodeuser_output_nodeuser_output_names rrrs  ) )B/?.D.D.J.Jg.JddggYfNf .Jg BNN 33 &s+;'<&=>GGJ2>>GZF[[\ ^   0@A D jnn~~&F G~~""d*(&z~~':':&;;UVZU[\ ??i22 2  __ 3 3 3jnnn=(  11RS]SaSaRbbkl  (($Z__$5_E%%E,777( 'AB.99%@%((BTBTU(5eW>X %% %(3(8(8(;<(;CHH(;L< <C00 %c,&7%89**-b.B.B*C)DE''*2??';&>2B#GHH % 0&&(&:&:%;;TUWUmUmTnp &(/22CR__.U** / /$/0188:7HL /V_hZ=sV0V0V5rNcp|dk(rtjj|Stj|S)NrO)rHrJget)rNs r load_verifierrms2&&&**733  " "7 ++rr).rrnrZcollections.abcrtypingrrrrrr r r7 torch._opsr r torch._subclasses.fake_tensorr torch.export.exported_programrtorch.export.graph_signaturerrrrtorch.fxr%torch.fx.experimental.symbolic_shapesrrr Exceptionrr4r&rwNoder*r)rQrHrIrr5rrrrrs $@@@ 649 !KK  %j%T%4 WUXX]]WtW>D&Y"Y"xzz,3,8DN#;,r