'}h`bddlZddlZddlZddlmZmZmZddlZejGddZ GddZ de de fd Z d e dee e ffd Zd e dejj fd Zd ejj$defdZdedefdZd ejj$fdZdZy)N)AnyCallableTuplec,eZdZUdZeed<eed<dZy)Kernelz$Models a (function, source location)funcsourcec&|j|i|SN)r)selfargskwargss S/var/www/html/test/engine/venv/lib/python3.12/site-packages/torch/_library/utils.py__call__zKernel.__call__styy$)&))N)__name__ __module__ __qualname____doc__r__annotations__strrrrrr s. N K*rrc$eZdZdZdefdZddZy)RegistrationHandlez2Does something when someone calls .destroy() on it on_destroyc||_yr  _on_destroy)r rs r__init__zRegistrationHandle.__init__s %rNc$|jyr r)r s rdestroyzRegistrationHandle.destroys r)returnN)rrrrrrr!rrrrrs<&8&rr stacklevelr"ctjtj|}|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)r#framer s r get_sourcer-s;  z!: ;Eq /F Mrqualnamecr|jd}t|dk7rtd|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.splitss rparse_namespacer6,sR ^^D !F 6{a**248 9  !9fQi rct|\}}d|vr|jd\}}nd}ttj|}t||}t||S)N.default)r6r2getattrtorchops)r. namespacenameoverloadnspackets r lookup_oprB8sV%h/OIt d{Ch I &B R F 68 $$ropcjt|tjjsJ|jdvS)N>atenprimprims) isinstancer;_ops OpOverloadr=)rCs r is_builtinrKCs, b%**// 00 0 <<4 44rschemac6ddlm}m}t|t|fsJt|tr|j |}|j |jk7ry|j}t|dkDxrtd|D}|ry|jsyy)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 SchemaKindFc3jK|]+}|jduxr|jj -ywr ) annotationis_write).0rs r z'is_functional_schema..[s31CD D >)>)>%>>1s13T) torchgen.modelrNrOrHrparsekind functionalreturnsr3any)rLrNrOretsis_non_mutating_views ris_functional_schemar^Hs: fsN3 44 4&#%%f- {{} --- >>Dt9q=S1HL1. >> rc|jdk7ry|j}t|jdk(sy|jdjy|jdjj }t|dk7ryt t|}t|jdkry|jd}|jy|jjsy|jj }t|dk7ry|t t|k7ry|jddD]}|jyy)aNCheck if an op is an inplace aten op, i.e. it mutates and returns the first arg. TODO: torchgen/model.py's FunctionSchema.parse is the source of truth for this, but not all PyTorch builds have torchgen (due to the yaml dependency being weird). Figure this out. Example: add_(Tensor(a!) x, Tensor y) -> Tensor(a) rEFr1rNT) r=_schemar3rZ alias_info after_setnextiter argumentsrR)rCrL alias_setloc first_argargs rmutates_and_returns_first_argrjes( ||v ZZF v~~ ! # ~~a##+q!,,66I 9~ tI C 6  q   #I#    ( ($$..I 9~ d4 ?### >> % rc#ZKt|jt|t|zk\sJtt|jD]W}|j|}|jr"|j|vr|||jf@|t|k\rO|||fYyw)zzips schema.arguments and (args, kwargs) together. Assumes that (args, kwargs) were the inputs to some torch._ops.OpOverload: that is, kwargs must be keyword-only arguments and default values may be omitted. N)r3rerange kwarg_onlyr>)rLr riinfos r zip_schemarps v CIF $; ;; ; 3v''( ) " ??yyF"F499---  D > DGm  sB)B+) dataclassesr&r(typingrrrr; dataclassrrintrr-r6rIOpOverloadPacketrBrJboolrKr^rjrprrrrws ''  *** 3 3   c  eCHo  %% ; ;%55::((5T5 :#ejj&;&;#L r