U ʹj@sUddlZddlZddlmZddlmZddlZddlmZddl m Z ddl m Z ddl mZddlmZd d hZeeed <ed d GdddeZed d e edddZed d ejjddddZed d eddddZed d GdddZGdddeZdS)N)Iterable)Set) control_flow) OpOverload) FakeTensor) GraphModule) compatibilityvalZ stack_tracePRESERVED_META_KEYSFZis_backward_compatiblec@s eZdZdS)SpecViolationErrorN)__name__ __module__ __qualname__rrU/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/_export/verifier.pyr sr )opreturncCs |jj SN)Z_schemaZ is_mutable)rrrr is_functionalsr)nodercs@fdd|jdd}|dks*|s$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.)metagetr name)rr rrr_check_has_fake_tensors r!gmrcCsFt||D],\}}t|tjr|std|dqdS)Nz,Tensors in Aten dialect must be contiguous, z is not contiguous) itertoolschainZnamed_parametersZ named_buffersrtorchZTensorZ is_contiguousr )r#r paramrrr_check_tensors_are_contiguous,s   r(c@steZdZeddddZeddddZeddd d Zeddeddd d Zeddee dd dZ dS)VerifierNr"cCs||dSr) check_validselfr#rrr__call__9szVerifier.__call__Fr cCstjtjtjjgSr)operatorgetitemrZcondr&ZopsZmap_impl)r,rrrvalid_builtin_funcs<szVerifier.valid_builtin_funcscCsLt|dr|jn|j}t|ts0td|dt|sHtd|ddS)Nr Operator '' is not a registered Opz operator 'z' is not functional)hasattrr r rrr rr,rZop_namerrrcheck_valid_opDs   zVerifier.check_valid_opcCs|jjD]}|jdkr(td|jd|jdkrt||j|krT||jt|jt rt D]*}|j |ddkrdtd|d|qdqdS)N> call_module call_methodz'call_module is not valid: got a class 'z' Z call_functionznode z is missing metadata ) graphZnodesrr targetr!r0r5rrr rr)r,r#rrrrrr*Ss      zVerifier.check_validcCs,z||WdStk r&YdSXdS)NTF)r*r r+rrris_validks  zVerifier.is_valid) r rrrr-rr0r5r*boolr:rrrrr)7s  r)csJeZdZeddddfdd Zeddeddfdd ZZS) ATenDialectVerifierFr N)rcsvt|t|dr|jn|j}t|tsr*r(r+rBrrr*s zATenDialectVerifier.check_valid)r rrrr5rr* __classcell__rrrBrr<tsr<)r$r.collections.abcrtypingrr&Zfunctorch.experimentalrZ torch._opsrZtorch._subclasses.fake_tensorrZtorch.fxrZtorch.fx._compatibilityrr str__annotations__ Exceptionr r;rZfxNoder!r(r)r<rrrrs0         <