U <c @srddlmZmZmZmZddddZddZGdd d eZGd d d eZGd d d eZ GdddeZ dS))BinaryUfuncInfoOpInfoReductionOpInfoUnaryUfuncInfoNop_dbcCs>|dkrddlm}|D] }|j|kr|j|kr|SqdS)zH Finds the OpInfo with the given name that has no variant name. Nrr)Z2torch.testing._internal.common_methods_invocationsrnameZvariant_test_name)Zreferenced_nameZ variant_namerZopinfor G/tmp/pip-unpacked-wheel-gikjz4vx/torch/testing/_internal/opinfo/refs.py_find_referenced_opinfos  r cCs||ddddd}|}d|kr6||d|d=d|krD|d=d|krR|d=d|kr`|d=d|krn|d=||||d|d<d|d <d|d <d|d <d|d <|S) NF)ropaliasesZmethod_variantZinplace_variantZsupports_scriptingkwargsself __class__skipsZ decoratorsZsupports_autogradZsupports_gradgradZsupports_fwgrad_bwgradZsupports_inplace_autogradZsupports_forward_ad)copyupdate)rr inheritedZ overridesZ common_kwargsrr r r _inherit_constructor_args.s6   rcs.eZdZdZddddddfdd ZZS) PythonRefInfozM An OpInfo for a Python reference of an OpInfo base class operation. NT)r rtorch_opinfo_variant_namevalidate_view_consistencysupports_nvfuserc  sd||_||_t|||d|_||_||_t|jts8t|jj } t ||| |} t t |j f| dSNr)torch_opinfo_namerr torch_opinforr isinstancerAssertionErrorZ_original_opinfo_argsrsuperr__init__) rrr rrrrrrrukwargsrr r r!^s zPythonRefInfo.__init____name__ __module__ __qualname____doc__r! __classcell__r r r#r rYsrcs,eZdZdZdddddfdd ZZS)ReductionPythonRefInfoM An OpInfo for a Python reference of an elementwise unary operation. NrTr rrrc  s`||_||_t|||d|_||_t|jts2t|jj}t ||||} d|_ t j f| dS)NrF) rrr rrrrrZ_original_reduction_argsrrr r! rrr rrrrrrr"r#r r r!~s zReductionPythonRefInfo.__init__r$r r r#r r*ys r*cs,eZdZdZdddddfdd ZZS)ElementwiseUnaryPythonRefInfor+NrTr,c  s^||_||_t|||d|_||_t|jts2t|jj}t ||||} t t |j f| dSr) rrr rrrrrZ_original_unary_ufunc_argsrr r.r!r-r#r r r!s z&ElementwiseUnaryPythonRefInfo.__init__r$r r r#r r.s r.cs,eZdZdZdddddfdd ZZS)ElementwiseBinaryPythonRefInfozN An OpInfo for a Python reference of an elementwise binary operation. NrTr,c  s^||_||_t|||d|_||_t|jts2t|jj}t ||||} t t |j f| dSr) rrr rrrrrZ_original_binary_ufunc_argsrr r/r!r-r#r r r!s z'ElementwiseBinaryPythonRefInfo.__init__r$r r r#r r/s r/) Z#torch.testing._internal.opinfo.corerrrrr rrr*r.r/r r r r s + "