iF 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_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 n/Volumes/fast/ai/experiments/voice-extract-mac/.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_matchr2$s%00M''I =S^+"%m&:&:&N>N>P"Q h   +E<#4#4#> JWX|U+wx/GG,&--%%  ] *z]/J++#R.  . .,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_r3r4s rdecomposition_decoratorz7register_decomposition..decomposition_decoratorIs  *H'5::#8#89 9$w-I  ::* * #ScriptFunction__annotations__setr r r rr2r9r:rKrerNrSrZboolrbr(rrrqs  yy~~$$'<>T#uyy7778=!e3s8# T]t_ LQF?C# ZZ " "#tC!9!99:;#xB (2r6"223#F++, $#' ! ! $s) ! ! !  !-!H(()GvGGG*Gr