
    sh(                     H   S SK r S SKrS SKJr  S SKJrJrJr  S/rSr	Sr
S r/ r\" \
S5       HL  r\" \
\-  S	5       H9  r\\-  r\" \
\-  S
5       H!  r\\-  r\\	:  d  M  \R%                  \5        M#     M;     MN     \R'                  5          " S S\R(                  5      rg)    N)	benchmark)FuzzedParameterFuzzedTensorParameterAliasSpectralOpFuzzer   i @  c           	      p   ^ U4S j[        [        [        R                  " U T5      5      S-   5       5       $ )Nc              3   .   >#    U  H
  nTU-  v   M     g 7f)N ).0ibases     }/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/utils/benchmark/op_fuzzers/spectral.py	<genexpr>power_range.<locals>.<genexpr>   s     KJ!DAIJs      )rangeintmathlog)upper_boundr   s    `r   power_ranger      s(    KuS+t)D%E%IJKK             c                   X   ^  \ rS rSr\R
                  SSS.S\S\S\4U 4S jjjr	S	r
U =r$ )
r      F      ?)dtypecudaprobability_regularseedr!   r"   c                t  > [         TU ]  [        SSSSS.SS9[        S5       Vs/ s H  n[        SU 3[        [
        S	S
9PM     sn[        S5       VVs/ s H5  n[        SU 3[         Vs0 s H  ofS[        [        5      -  _M     snS9PM7     snn[        S5       Vs/ s H0  n[        SU 3[        SU 35      U[        SU 35      SU-
  0SS9PM2     sn[        S5       Vs/ s H  n[        SU 3SSSSSS.S9PM     sn/[        SSSSSSSSUUS9
/US9  g s  snf s  snf s  snnf s  snf s  snf )Nndimg333333?g?)r   r   r   T)distributionstrictr   k_any_
loguniform)nameminvalmaxvalr&   
k_regular_r   )r*   r&   kr   )r*   r&   r'   step_g?gQ?g{Gz?)r   r         r   x)k0k1k2)step_0step_1step_2g      ?i   i   l        )
r*   sizestepsprobability_contiguousmin_elementsmax_elementsmax_allocation_bytesdim_parameterr    r!   )
parameterstensorsr#   )
super__init__r   r   MIN_DIM_SIZEMAX_DIM_SIZEREGULAR_SIZESlenr   r   )selfr#   r    r!   r"   r   r9   	__class__s          r   rC   SpectralOpFuzzer.__init__   s     5MVZ[   %Qx  (! $%aS\++%1	
  (  %Qx	  (! $)!-P]%^P]B]1C,C&CP]%^  (	  %Qx	  (! $ W*Zs+;<>Q*VA3<8!>Q:Q&  $  (	   %Qx	  (! $$QC[),$4T%R  (	M,\ +8+/!)!/)4"( y 	 =	
 &_	s)   !D 
D*.D%
	D*#7D0)D5%D*r   )__name__
__module____qualname____firstlineno__torchfloat64r   boolfloatrC   __static_attributes____classcell__)rI   s   @r   r   r      s0    +0==##?
 ?
?
:??
 ?
r   )r   rO   torch.utilsr   torch.utils.benchmarkr   r   r   __all__rD   rE   r   rF   r   jijr.   ijkappendsortFuzzerr   r   r   r   <module>r^      s      ! O O 
L 	\1	%A*A.U\R/3Aq&C\!$$S) 4 / 
&    @
y'' @
r   