i vUddlZddlmZejjZddlZddlZddlmZddlm Z m Z ddl m Z ddl mZiZeeej$j&fed<eZeeed<e d Ze d Zdd Zd Z ddej6j8de eeej$j&fd eeeefgeeefffdZeej<j> ddede e e!de ede"d ef dZ#eej<jHddede"d efdZy)N)Tensor)Callable)OptionalTypeVar) ParamSpec)Numberdecomposition_tablefunction_name_set_T_Preturnc2tj}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 S/var/www/html/engine/venv/lib/python3.12/site-packages/torch/jit/_decompositions.py(check_decomposition_has_type_annotationsrsNNM   A C&&( =0 $UZZLqvvh G 0  M 1 ,QVVH5 1c|j}|j}t|t|k7ryt|j|jD]\}}tj }dD]I}|dk(r&|j dk(rtjddt||t||k7sHy|j}|j} ||k7s| |k7s|| k7sy|j|jk(S)NF)rrrselfz4PyTorch uses 'input' instead of 'self' on public api) stacklevel) 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#> JWX|U+wx/GG &--%%  ] *z]/J++.  . .,2P2P PPraten_opregistrycddtttfdtttfffd }|S)Nrr crtttjjsJ|j t vsJd|j t j|j tjj|}tjj|jtdD]T}tjj|jtjj|jV|t!j"<|S)NzDuplicated function name r )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_decoratorIs  *H'5::#8#8999zz!22 ' | 4 2 ajj) ((+  !!-"5"56q IA HH ' ' (;(; < HH 3 3M4G4G H I*7W__%&r)rr r )r2r3rHs`` rregister_decompositionrIEs08BF#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) r6tuplelistr"numelshapeatenmeansumfloatint RuntimeErrormax) rJrKrLrMdim_inrUsubsqrVdenoms rvar_decompositionr`fs {#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)NrOr)rL)r`)rJras rvarrcs UX FF1 FFr)r N)N)NNF)T)%r7ropsrTrr$collections.abcrtypingrrtyping_extensionsr torch.typesrr dictrDr<ScriptFunction__annotations__setr r r rr1r8r9rIrcrLrQrXboolr`r'rrros  yy~~$$'<>T#uyy7778=!e3s8# T]t_ QF?C# ZZ " "#tC!9!99:;#xB (2r6"223#B++, $#' ! ! $s) ! ! !  !-!H(()GvGGG*Gr