o h@s UddlZddlZddlmZddlmZddlZddlmZddl m Z ddl m Z ddl mZddlmZd d hZeeed <ed d GdddeZed d de defddZed d dejjddfddZed d deddfddZed d GdddZGdddeZdS)N)Iterable)Set) control_flow) OpOverload) FakeTensor) GraphModule) compatibilityval stack_tracePRESERVED_META_KEYSFis_backward_compatiblec@s eZdZdS)SpecViolationErrorN)__name__ __module__ __qualname__rrL/var/www/html/ai/venv/lib/python3.10/site-packages/torch/_export/verifier.pyrsropreturncCs |jj SN)_schema is_mutable)rrrr is_functionals rnodecs@fdd|jdd}|dus|std|jddS)Ncs2t|trdSt|trtfdd|DSdS)NTc3s|]}|VqdSrr).0x_check_is_fake_tensorrr $szH_check_has_fake_tensor.._check_is_fake_tensor..F) isinstancerrall)r rrrr s  z5_check_has_fake_tensor.._check_is_fake_tensorr z Node.meta z is missing val field.)metagetrname)rr rrr_check_has_fake_tensors r%gmcCsFt||D]\}}t|tjr |s td|dq dS)Nz,Tensors in Aten dialect must be contiguous, z is not contiguous) itertoolschainnamed_parameters named_buffersr torchTensor is_contiguousr)r&r$paramrrr_check_tensors_are_contiguous,s  r/c@szeZdZdeddfddZedddd Zeddd d Zedddeddfd d Zedddede fddZ dS)Verifierr&rNcCs||dSr) check_validselfr&rrr__call__9szVerifier.__call__Fr cCstjtjtjjgSr)operatorgetitemrcondr+opsmap_impl)r3rrrvalid_builtin_funcs<szVerifier.valid_builtin_funcscCsLt|dr|jn|j}t|tstd|dt|s$td|ddS)Nr$ Operator '' is not a registered Opz operator 'z' is not functional)hasattrr$rr rrrr3rop_namerrrcheck_valid_opDs   zVerifier.check_valid_opcCs|jjD]D}|jdvrtd|jd|jdkrHt||j|vr*||jt|jt rHt D]}|j |ddurGtd|d|q2qdS)N> call_method call_modulez'call_module is not valid: got a class 'z' call_functionznode z is missing metadata ) graphnodesrrtargetr%r:r@r rr r"r#)r3r&rr"rrrr1Ss$      zVerifier.check_validcCs&z||WdStyYdSw)NTF)r1rr2rrris_validks   zVerifier.is_valid) rrrrr4rr:r@r1boolrGrrrrr07s  r0csHeZdZeddd fdd Zedddeddffdd ZZS) ATenDialectVerifierFr rNcszt|t|dr|jn|j}t|tstd|dtj j |j vr9tj j |j vr;td|j d|jddSdS)Nr$r;r<z Operator .z is not Aten Canonical.)superr@r=r$rr rrr+Tagcoretags view_copyrr> __class__rrr@us   z"ATenDialectVerifier.check_valid_opr&cst|t|dSr)rKr1r/r2rPrrr1s  zATenDialectVerifier.check_valid)rN)rrrrr@rr1 __classcell__rrrPrrIts  rI)r'r5collections.abcrtypingrr+functorch.experimentalr torch._opsrtorch._subclasses.fake_tensorrtorch.fxrtorch.fx._compatibilityrr str__annotations__ ExceptionrrHrfxNoder%r/r0rIrrrrs0          <