o H&i/ @sddlZddlZddlZddlmZmZmZddlZejGdddZ GdddZ de de fd d Z d e dee e ffd d Zd e dejjfddZdedefddZdS)N)AnyCallableTuplec@s*eZdZUdZeed<eed<ddZdS)Kernelz$Models a (function, source location)funcsourcecOs|j|i|SN)r)selfargskwargsr ?C:\wamp64\www\opt\env\Lib\site-packages\torch/_library/utils.py__call__szKernel.__call__N)__name__ __module__ __qualname____doc__r__annotations__strrr r r r r s  rc@s(eZdZdZdefddZd ddZdS) RegistrationHandlez2Does something when someone calls .destroy() on it on_destroycCs ||_dSrZ _on_destroy)r rr r r __init__s zRegistrationHandle.__init__returnNcCs |dSrr)r r r r destroys zRegistrationHandle.destroy)rN)rrrrrrrr r r r rsr stacklevelrcCs&tt|}|jd|j}|S)zGet a string that represents the caller. Example: "/path/to/foo.py:42" Use stacklevel=1 to get the caller's source Use stacklevel=2 to get the caller's caller's source etc. :)inspect getframeinfosys _getframefilenamelineno)rframerr r r get_sources r$qualnamecCs6|d}t|dkrtd|d|d|dfS)Nz::zAExpected `qualname` to be of the form "namespace::name", but got zf. The qualname passed to the torch.library APIs must consist of a namespace and a name, e.g. aten::sinr)splitlen ValueError)r%Zsplitsr r r parse_namespace,s  r+cCsHt|\}}d|vr|d\}}nd}ttj|}t||}t||S)N.default)r+r(getattrtorchops)r% namespacenameoverloadnspacketr r r lookup_op8s    r6schemacCsddlm}m}t|t|fsJt|tr||}||jkr$dS|j}t |dko5t dd|D}|r:dS|js?dSdS)zCheck if the schema is functional. An operator is functional if: - it does not mutate any of its inputs - it does not return a view on any of its inputs - it has at least one return r)FunctionSchema SchemaKindFcss$|] }|jduo |jj VqdSr) annotationZis_write).0rr r r Vs z'is_functional_schema..T) Ztorchgen.modelr8r9 isinstancerparsekindZ functionalreturnsr)any)r7r8r9ZretsZis_non_mutating_viewr r r is_functional_schemaCs   rC) dataclassesrrtypingrrrr/ dataclassrrintrr$r+Z_opsZOpOverloadPacketr6boolrCr r r r s