Phq>ddlZddlZddlZddlZddlmZddlmZddl m Z m Z gdZ e Zejej ej"ej"ej$ej&e ej"ej$ffZej*eefZedGd d ej.ZedGd d eZej4ej*e ej6j8fej:j<gefZ edd e defdZ!eddedefdZ"eddedefdZ#edGddZ$dej:j<dejJfdZ&y)N) compatibility)TensorMetadata)get_node_targetCALLABLE_NODE_OPS)OperatorSupportBaseOperatorSupportcreate_op_supportchain OpSupports any_chainF)is_backward_compatibleceZdZdZej dejee jjfde jjdefdZy)rz@Interface for determining if a fx.Node is supported by a backend submodulesnodereturnctN)NotImplementedError)selfrrs kC:\Users\daisl\Desktop\realtime-object-detection\venv\Lib\site-packages\torch/fx/passes/operator_support.pyis_node_supportedz%OperatorSupportBase.is_node_supporteds "##N)__name__ __module__ __qualname____doc__abcabstractmethodtMappingstrtorchnnModulefxNodeboolrrrrrsSJ$))C$89$AF$ $$rrceZdZUdZeed< d dejefdZdeje e jjfde jjdefd Zy) r a `_support_dict` maps node.target typename to supported inputs dtypes. node.target typename is retrieved using helper function `get_node_target()` If supported inputs dtypes is None, it means any dtype is supported, else we should see a tuple like (([dtypes], ...), {"name":[dtypes], ...}). The first tuple ([dtypes], ...) indicates what dtypes are supported for inputs in node.args and the second dict {"name": [dtypes], ...} indicates what dtypes are supported for inputs in node.kwargs. For inputs in args, if we don't want to check it, we can put None there, e.g. (None, [torch.float]) indicates that we don't care about the type of the first input in args. And for inputs in kwargs, if not listed, will not be checked. _support_dictN support_dictc|xsi|_yr)r+)rr,s r__init__zOperatorSupport.__init__;s*/Rrrrrc|jtvryt||}||jvry|j|y|j|\}}t |D]r\}}t |j |krnU|#t|j |tjjsUt|j |}||vsry|jD]d\} }| |jvrt|j| tjjsGt|j| } | |vsdyy)aO Args: `submodules`: mapping from module name to the module. This can be retrieved by calling model.named_modules(). `node`: a Fx node that we want to determine whether it's supported. Returns: `is_supported`: whether the arg `node` is supported. TF)oprrr+ enumeratelenargs isinstancer#r&r'_get_arg_dtypeitemskwargs) rrrtarget args_dtypes kwargs_dtypesidtypes arg_dtypek kwarg_dtypes rrz!OperatorSupport.is_node_supportedAs5 77+ + T2 ++ +   f % -%)%7%7%?" ]#;/IAv499~"~diilEHHMM:&tyy|4I&0"',,.IAv #dkk!nehhmm<(Q8K&(/rr)rrrr SupportDict__annotations__r Optionalr.r!r"r#r$r%r&r'r(rr)rrr r %se$150jj-0 :))C$89:AF: :rr rrc4Gfddt}|S)zWraps a `IsNodeSupported` function into an `OperatorSupportBase` instance `IsNodeSupported` has the same call signature as `OperatorSupportBase.is_node_supported` ceZdZdejeejjfdejjde ffd Z y)4create_op_support..FunctionalOperatorSupportrrrc||Srr))rrrrs rrzFcreate_op_support..FunctionalOperatorSupport.is_node_supporteds%Z6 6rN) rrrr r!r"r#r$r%r&r'r(rrsrFunctionalOperatorSupportrEs> 7"#))C,@"A 7IN 7  7rrH)r)rrHs` rr r s7$77 % &&r op_supportc2dtffd }t|S)zCombines a sequence of `OperatorSupportBase` instances to form a single `OperatorSupportBase` instance by evaluating each input `OperatorSupportBase` instance, and returns False if any of it reports False. rc2tfdDS)Nc3BK|]}|jywrrG.0xrsubmodss r z(chain.._chain..&     .)allrPrrIs``r_chainzchain.._chain    rr(r )rIrVs` rr r s   V $$rc2dtffd }t|S)zCombines a sequence of `OperatorSupportBase` instances to form a single `OperatorSupportBase` instance by evaluating each input `OperatorSupportBase` instance, and returns True if any of it reports True. rc2tfdDS)Nc3BK|]}|jywrrGrMs rrQz0any_chain.._any_chain..rRrS)anyrUs``r _any_chainzany_chain.._any_chainrWrrX)rIr]s` rr r s  T Z ((rcreZdZdZedej defdZede je defdZ y)r zA set of atomic `OperatorSupportBase` instances that can be combined together to form more complex operator support logic. dtypercdtjttjj fdtj jdtffd }t|S)zCReport a node as non-supported, if any of its arguments is of dtyperrrcl|jD]$}|jdk(rt|}|k(s$yy)Nget_attrFT)all_input_nodesr0r5)rrargr=r_s r_decline_if_input_dtypezBOpSupports.decline_if_input_dtype.._decline_if_input_dtypes>++66Z'*3/ % ,r r r!r"r#r$r%r&r'r(r )clsr_res ` rdecline_if_input_dtypez!OpSupports.decline_if_input_dtypesN  #uxx"67 ((--  !!899r disallow_setcdtjttjj fdtj jdtffd }t|S)za If a node has a name that is in the disallow set, reported it as non-supported. rrrc$|jvryy)NFT)name)rrris r_decline_if_node_in_nameszFOpSupports.decline_if_node_in_names.._decline_if_node_in_namessyyL(rrf)rgrirms ` rdecline_if_node_in_namesz#OpSupports.decline_if_node_in_namessN   #uxx"67 ((--  !!:;;rN) rrrr classmethodr#r_rrhr Setr"rnr)rrr r sY:5;;:;N::" rs  1&< n**GG 1::ekk*+ sAJJu{{++, .ii(??@ e,$#''$-$e,U)U-Uz**aiiUXX__(<=uxx}}MtSTe, ' '=P '- 'e, %* %/B %- %e, ). )3F )- )e,#<#<-#