
    sg(                     8   d dl Z d dlZd dlmZ d dlmZmZmZ dgZdZ	dZ
d Zg Z ee
d      D ]E  Z ee
ez  d	      D ]4  Zeez  Z ee
ez  d
      D ]  Zeez  Zee	kD  sej%                  e         6 G ej'                           G d dej(                        Zy)    N)	benchmark)FuzzedParameterFuzzedTensorParameterAliasSpectralOpFuzzer   i @  c           	      l    fdt        t        t        j                  |             dz         D        S )Nc              3   (   K   | ]	  }|z    y w)N ).0ibases     \/var/www/html/venv/lib/python3.12/site-packages/torch/utils/benchmark/op_fuzzers/spectral.py	<genexpr>zpower_range.<locals>.<genexpr>   s     K!DAIKs      )rangeintmathlog)upper_boundr   s    `r   power_ranger      s(    KuS+t)D%E%IJKK             c                   J     e Zd Zej                  ddddededef fdZ xZ	S )r   F      ?)dtypecudaprobability_regularseedr   r    c                   t         |   t        dddddd      t        d      D cg c]  }t        d| t        t
        d	
       c}t        d      D cg c]5  }t        d| t        D ci c]  }|dt        t              z   c}      7 c}}t        d      D cg c]2  }t        d| t        d|       |t        d|       d|z
  id      4 c}t        d      D cg c]  }t        d| dddddd       c}gt        dddddddd||
      g|       y c c}w c c}w c c}}w c c}w c c}w )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   r7   	__class__s          r   rA   zSpectralOpFuzzer.__init__   s     5MVZ[   %Qx  $%aS\++%1	  %Qx	  $)!-P]%^dB]1C,C&C%^  %Qx	  $ W*Zs+;<>Q*VA3<8!>Q:Q&  $	   %Qx	  $$QC[),$4T%RM,\ +8+/!)!/)4"( y 	 =	
 &_	s)   !D'
D10D,

D1$7D7*D<,D1)
__name__
__module____qualname__torchfloat64r   boolfloatrA   __classcell__)rG   s   @r   r   r      s0    +0==##?
 ?
?
:??
 ?
r   )r   rK   torch.utilsr   torch.utils.benchmarkr   r   r   __all__rB   rC   r   rD   r   jijr,   ijkappendsortFuzzerr   r   r   r   <module>rY      s      ! O O 
L 	\1	% *A*A. *U\R/3 	*Aq&C\!$$S)	***    @
y'' @
r   