sg' xUddlZddlmZejjZddlZddlZddlmZmZm Z m Z m Z m Z ddl mZddlmZiZeeej&j(fed<eZe eed<e dZed Zd Zd Z dd ej8j:d e eeej&j(fdeeeefgeeefffdZeej>j@ ddede e e!de ede"def dZ#eej>jHddede"defdZy)N)Tensor)CallableDictListOptionalSetTypeVar) ParamSpec)Numberdecomposition_tablefunction_name_set_T_Pc2tj}tj|}|jj D]2}|j |k7rJd|j d|j |j|k7sJd|j y)NzNo signature on param z for function z"No return annotation for function )inspect_empty signature parametersvalues annotationnamereturn_annotation)f inspect_emptysigparams L/var/www/html/venv/lib/python3.12/site-packages/torch/jit/_decompositions.py(check_decomposition_has_type_annotationsrsNNM   A C&&(G    - G #EJJ<~affX F G -G .5 +AFF845.c|j}|j}t|t|k7ryt|j|jD]\}}tj }dD]G}|dk(r$|j dk(rtjdt||t||k7sFy|j}|j} ||k7s| |k7s|| k7sy|j|jk(S)NF)rrrselfz4PyTorch uses 'input' instead of 'self' on public api) rlenziprrrrwarningswarngetattrdefaultr) decomposition_sig torch_op_sig decomp_params op_params decomp_paramop_paramrfielddecomp_default op_defaults rsignatures_matchr1#s%00M''I =S^+"%m&:&:&N>N>P"Q h   + E<#4#4#> TU|U+wx/GG  &--%%  ] *z]/J+'*  . .,2P2P PPraten_opregistryreturncddtttfdtttfffd }|S)Nrr4crtttjjsJ|j t vsJd|j t j|j tjj|}tjj|jtdD]T}tjj|jtjj|jV|t!j"<|S)NzDuplicated function name )r isinstancetorch_ops OpOverload__name__r addjitscript_C_jit_pass_inlinegraphrange_jit_pass_peephole_jit_pass_constant_propagationstr_schema)r scripted_func_r2r3s rdecomposition_decoratorz7register_decomposition..decomposition_decoratorFs  *H'5::#8#8999 JJ/ / 4 &qzzl 3 4 /ajj) ((+  !!-"5"56q IA HH ' ' (;(; < HH 3 3M4G4G H I*7W__%&r)rrr)r2r3rJs`` rregister_decompositionrKBs08BF#3R8H. #"rinputdim correctionkeepdimc|g}|}t|ttfrt|dk(r|j }nd}|D]}||j |z}t j||d}||z }||z}t j|||} |t|dz } nIt|trt||z } n*t|trt||z } n td| td| z S)NrTzcorrection must be int or float) r8tuplelistr"numelshapeatenmeansumfloatint RuntimeErrormax) rLrMrNrOdim_inrWsubsqrXdenoms rvar_decompositionrbcs {#t}%#c(a- KKM  $E U# #A $ 99UC &D $,C sB ((2sG $Ca!e  j# &!j.)E  E *!Hz)E@A A Q runbiasedc.t||rdSdS)NrQr)rN)rb)rLrcs rvarres UX FF1 FFr)N)NNF)T)%r9ropsrVrr$typingrrrrrr typing_extensionsr torch.typesr r rFr>ScriptFunction__annotations__setr rrrr1r:r;rKrerNrZboolrbr'rrros  yy~~??'<>T#uyy7778=!e3s8# T]t_ 5QB?C# ZZ " "#tC!9!99:;#xB (2r6"223#B++, $#'    $s)       - F(()GvGGG*Gr