tj3 PUddlZddlmZejjZddlZddlZddlmZddlm Z ddl m Z ddl m Z iZeeejjfed<eZeeed<e d Ze d Zdd Zd Z ddejjdeeejjfdzd eeeefgeeefffdZeejj ddedee dzde dzde!d ef dZ"eejj#ddede!d efdZdS)N)Tensor)Callable)TypeVar) ParamSpec)Numberdecomposition_tablefunction_name_set_T_Preturnctj}tj|}|jD],}|j|krt d|jd|j-|j|krt d|jdS)NzNo signature on param z for function z"No return annotation for function ) inspect_empty signature parametersvalues annotationAssertionErrornamereturn_annotation)f inspect_emptysigparams W/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/jit/_decompositions.py(check_decomposition_has_type_annotationsrsNM  A  C&&((  } , , KKK16KK  -   --J!&JJKKK.-c|j}|j}t|t|krdSt||D]\}}tj}dD]O}|dkr!|jdkrtjddt||t||krdSP|j }|j } ||kr| |kr || krdS|j |j kS)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$s0%0M'I =S^^++u"%m&:&:&<&N>N>P>P"Q"Q h   +  E<#4#>#> JWX|U++wx/G/GGGuuuH&-%  ] * *z]/J/J++uu  .,2P PPraten_opregistryc|dtttfdtttfffd }|S)Nrr ctttjjst dt |jtvrt d|jt |jtj |}tj |jtdD]J}tj |jtj |jK|t%j<|S)Nz'Expected aten_op to be OpOverload, got zDuplicated function name r )r isinstancetorch_ops OpOverloadrtype__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:#899  I$w--II  :* * * !IQZ!I!IJJ Jaj))) ((++  !!-"5666q I IA H ' ' (; < < < H 3 3M4G H H H H)6W_%%&r)rr r )r2r3rIs`` rregister_decompositionrJEsO8BF#3R8H2 #"rFinputdim correctionkeepdimcV|g}|}t|ttfr(t|dkr|}nd}|D]}||j|z}t ||d}||z }||z}t |||} |t|dz } n_t|trt||z } n7t|trt||z } ntd| td| z S)NrTzcorrection must be int or float) r6tuplelistr"numelshapeatenmeansumfloatint RuntimeErrormax) rKrLrMrNdim_inrVsubsqrWdenoms rvar_decompositionrahs/ {#t}%%$#c((a-- KKMM  $ $E U# #AA 99UC & &D $,C sB ((2sG $ $Ca!e  j# & & B!j.))EE  E * * B!HHz)EE@AA A Q rTunbiasedc,t||rdndS)NrPr)rM)ra)rKrbs rvarrds UX0D1 F F FFr)r N)N)NNF)T)$r7ropsrUrr$collections.abcrtypingrtyping_extensionsr torch.typesrrdictrEr=ScriptFunction__annotations__setr r r rr1r8r9rJrdrMrRrYboolrar'rrrpsW  y~$$$$$$''''''<>T#uy778===!cee3s8### WT]]Yt__ L L L LQQQF<@## Z "#3 001D8#xB (2r6"223####F+,,! $ !! ! cT ! ! !  !!!-,!H())GGvGGGGG*)GGGr