9j3 rUddlZddlmZejjZddlZddlZddlmZddlm Z ddl m Z ddl m Z iZeeej"j$fed<eZeeed<e d Ze d Zdd Zd Z ddej4j6deeej"j$fdzd eeeefgeeefffdZeej:j< ddedee dzde dzde!d ef dZ"eej:jFddede!d efdZy)N)Tensor)Callable)TypeVar) ParamSpec)Numberdecomposition_tablefunction_name_set_T_PreturncBtj}tj|}|jj D]6}|j |k(st d|jd|j|j|k(rt d|jy)NzNo signature on param z for function z"No return annotation for function ) inspect_empty signature parametersvalues annotationAssertionErrornamereturn_annotation)f inspect_emptysigparams Y/media/conek/DATA/Code/OCR/venv/lib/python3.12/site-packages/torch/jit/_decompositions.py(check_decomposition_has_type_annotationsrsNNM   A C&&(   } , ( N166(K    -A!&&JKK.c|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 ctttjjst dt |jtvrt d|jtj|jtjj|}tjj|jtdD]T}tjj!|jtjj#|jV|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::#8#89 9$w-I  ::* * #rps  yy~~$'<>T#uyy7778=!e3s8# T]t_ LQF<@# ZZ " "#3 0001D8#xB (2r6"223#F++,! $ ! ! cT ! ! !  !-!H(()GvGGG*Gr