Vh;jUddlZddlmZddlmZddlmZddlZddlZddl m Z ddl m Z ddl mZmZmZmZe rddlZdZd Zd Zd Zd Zd ZdZdZdZdZdZdZdefdZ ddde!dedededede"edff dZ#e#d e r&e ejHjJjLndeee!e#d"e r&e ejHjJjNndeee!e#d#e r&e ejHjJjPndeee!e#d$e r&e ejHjJjRndeee!e#d%e r&e ejHjJjTndeee!e#d&e r&e ejHjJjVndeed'(eed'(e!e#d)e r&e ejHjJjXndeed*+eed*+eeejZd,d-d.f/e#d0e r&e ejHjJj\ndeed12eed12e!e#d3e r&e ejHjJj^ndeed4d5g6eed4d5g6e!e#d7e r&e ejHjJj`ndeed48eed48e!e#d9e r&e ejHjJjbndeee!g Z2e3ee4d:<y);N)partial)product)Callable)floating_types) TEST_SCIPY) DecorateInfo ErrorInputOpInfo SampleInputc /K|jddttddD]\}}t|g|||||d|yw)zBase function used to create sample inputs for windows. For additional required args you should use *args, as well as **kwargs for additional keyword arguments. include_conjugated_inputsN)TF)symdevicedtype requires_grad)poprranger )op_inforrrargskwargssizers a/home/dcms/DCMS/lib/python3.12/site-packages/torch/testing/_internal/opinfo/definitions/signal.pysample_inputs_windowrsj JJ*D1U1X}5  c   '      sAA c/Kt||||g|i|Ed{d}|D] }t|dt|d"y7,w)zReference inputs function to use for windows which have a common signature, i.e., window size and sym only. Implement other special functions for windows that have a specific signature. See exponential and gaussian windows for instance. N) @F)rTrr )rrrrrrcasesrs rreference_inputs_windowr$/sh$ 048> &E*$E**$D))* sAA-Ac+Kt||||fi|Ed{ddddfddddfddddfd d d dfd d ddfdddiff}|D]*\}}t|fddi|d|d<t|fddi|,y7Yw)Nr?)centertaurg@rgE@rg @r >cr!r) rFr(Tr"rrrrrr#rkws r#reference_inputs_exponential_windowr0As$GVUMTVTTT q %& # &' 4 () 3 '( B'( ubk  E0b$0E0R008 $/D/B//0UsA1A/AA1c+Kt||||fi|Ed{dddifdddifdddifddd ifd dd ifd dd iff}|D]%\}}t|fddi|t|fddi|'y7Ow)Nrstd皙?rg333333?rg@rg333333@r g@r!r-rFTr"r.s r reference_inputs_gaussian_windowr4Us#GVUMTVTTT UCL eS\ eS\ eS\ ucl ubk  E0b$0E0R00$/D/B//0UA'A%AA'c+Kt||||fi|Ed{dddifdddifdddifddd ifd dd ifd dd iff}|D]%\}}t|fddi|t|fddi|'y7Ow)Nrbetar rr#r gD@r!drFTr"r.s rreference_inputs_kaiser_windowr=fs#GVUMTVTTT VQK fb\ fb\ fb\ vtn vsm  E0b$0E0R00$/D/B//0Ur5c +Kt||||fi|Ed{ddddgifddddgifddgdifd ddgifd dgd ifd dgd iff}|D]%\}}t|fddi|t|fddi|'y7Zw)Nrar'rq= ףp?HzG?r)r@q= ףp?gףp= ?rr )r3皙?皙?rDr!)皙?rErErErErFTr"r.s r&reference_inputs_general_cosine_windowrFws$GVUMTVTTT S3*  cD$< ! c% &' cC5\ s*+, s-./  E0b$0E0R00$/D/B//0UsA2A0AA2c+Kt||||fi|Ed{dddifdddifdddifddd ifd dd ifd dd iff}|D]%\}}t|fddi|t|fddi|'y7Ow)NralpharArr'rrBrrCr g?r!rDrFTr"r.s r'reference_inputs_general_hamming_windowrIs$GVUMTVTTT WdO gs^ gt_ gs^ wn wo  E0b$0E0R00$/D/B//0Ur5c /Kttdg|tj|d|tdttdg|tj |tjd|tdttdg|tj |d|tdttdg|tj|d|td ttdg|tj|d|td yw) N)rrz-requires non-negative window length, got M=-1 error_type error_regex)layoutrrz>is implemented for strided tensors only, got: torch.sparse_cooz3expects float32 or float64 dtypes, got: torch.int64z6expects float32 or float64 dtypes, got: torch.bfloat16z5expects float32 or float64 dtypes, got: torch.float16) r r torchfloat32 ValueError sparse_coolongbfloat16float16)rrrrs rerror_inputs_windowrXs BLLU]]6LVLC     ##--    T   AHHEJJvHHI  ALLENN6LVLL  AKKEMM&KFKK sD D c +Kt||fi|Ed{ttd dtj|d|t dttdddtj|t d y7dw) NrOrK)r)rrz&Tau must be positive, got: -1 instead.rLT)r(rrrz)Center must be None for symmetric windowsrOrXr r rQrRrSrrrs rerror_inputs_exponential_windowr^sv"7F=f=== L2U]]6LVL<  AaTvN? >sA:A8A%A:c +Kt||fddi|Ed{ttddtj|d|t dy74w)Nr2r'rK)r2rrz5Standard deviation must be positive, got: -1 instead.rLr[r\r]s rerror_inputs_gaussian_windowr`sS"7FFFvFFF L2U]]6LVLK GA A 5A c +Kt||fddi|Ed{ttddtj|d|t dy74w)Nr7r9rK)r7rrz+beta must be non-negative, got: -1 instead.rLr[r\r]s rerror_inputs_kaiser_windowrcsS"7FFFvFFF MBemmFMfMA Grac +Kt||fdddgi|Ed{ttddtj|d|t dttdgtj|d|t dy7cw) Nr?rAr@)r?rrz!Coefficients must be a list/tuplerLzCoefficients cannot be emptyr[)rXr r rQrR TypeErrorrSr]s r"error_inputs_general_cosine_windowrfs"7FMtTlMfMMM LU]]6LVL7  J5==J6J2 NsA=A;A$A=fncVtjdtjddfd }|S)zWrapper for scipy signal window references. Discards keyword arguments for window reference functions that don't have a matching signature with torch, e.g., gaussian window. NF)rrrPrc2|i|j|S)z:The unused arguments are defined to disregard those values)astype)rrrPrrrrgs r_fnz$reference_signal_window.._fn s 4"6"))%00)numpyfloat64rQstrided)rgrks` rreference_signal_windowrps%mm}} 1 Jrl)skipsnamerefsample_inputs_funcreference_inputs_funcerror_inputs_funcrr.ct|tr|ndt|||ddttj ddttj ddtt j ddd tt j ddd tt j dd d tt j dd d tt j dd dtt j dddtt j dddtt j dddg | S)zFHelper function to create OpInfo objects related to different windows.NFTestOperatorSignatures(test_get_torch_func_signature_exhaustiveTestJittest_variant_consistency_jitzSkipped! TestCommontest_noncontiguous_samplestest_variant_consistency_eager TestMathBitstest_conj_viewtest_neg_conj_view test_neg_viewTestVmapOperatorsOpInfotest_vmap_exhaustivetest_op_has_batch_rule5Buggy on MPS for now (mistakenly promotes to float64)test_numpy_ref_mps) rsrtdtypesrurvrw supports_outsupports_autogradrr)r rrrunittestexpectedFailureskiprsrtrurvrwrrs rmake_signal_windows_opinforsE  C4-3+ (((:  (()5S   j)<9U   j)0  z2NDT U  j)>;O  z2NO T  j))&   j))(   UV$ M, VW, 6 6rlzsignal.windows.hamming)rsrtrurvrwzsignal.windows.hannzsignal.windows.bartlettzsignal.windows.blackmanzsignal.windows.cosinezsignal.windows.exponentialg= ףp=@)r)zsignal.windows.gaussiangQ?)r2rr}rrzsignal.windows.kaiserg(@)r7zsignal.windows.general_cosinerAr@)r?zsignal.windows.general_hamming)rHzsignal.windows.nuttallop_db)5r functoolsr itertoolsrtypingrrmrQ$torch.testing._internal.common_dtyper$torch.testing._internal.common_utilsr#torch.testing._internal.opinfo.corerr r r scipy.signalscipyrr$r0r4r=rFrIrXr^r`rcrfrpstrtuplersignalwindowshamminghannbartlettblackmancosine exponentialgaussianrkaisergeneral_cosinegeneral_hammingnuttallrlist__annotations__rqrlrrs ?; ,*$0(0"0"0&0&)X&  $8')@ @ @!@$ @  @ s" #@H %  $ELL$8$8$@$@ A /5- "BL #ELL$8$8$=$= >RV/5-  &  $ELL$8$8$A$A B /5- &  $ELL$8$8$A$A B /5- $  $ELL$8$8$?$? @ /5- )  $ELL$8$8$D$D E "#7TB%&ItT9 &  $ELL$8$8$A$A B "#7TB%&FDQ6  UV$    $  $ELL$8$8$?$? @ "#7dC%&D4P4 ,  $ELL$8$8$G$G H "#7D$<H% 2tTl =  -  $ELL$8$8$H$H I "#7tD% 34 .  %  $ELL$8$8$@$@ A /5-ImtF|mrl