
    sgJ                         d dl Zd dlZd dlmZmZmZmZ dZdZ	 e
d  e e ej                  e             e ej                  e	            dz         D              Z G d de      Zy)	    N)FuzzerFuzzedParameterParameterAliasFuzzedTensor   i   c              #   &   K   | ]	  }d |z    yw)   N ).0is     Y/var/www/html/venv/lib/python3.12/site-packages/torch/utils/benchmark/op_fuzzers/unary.py	<genexpr>r   
   s      !qAv s      c                   8     e Zd Zej                  df fd	Z xZS )UnaryOpFuzzerFc                    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]/  }t        d| t        d|       dt        d|       did      1 c}t        d      D cg c]  }t        d| dddddd       c}t        dddd
      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dimg333333?g?)r   r	      T)distributionstrictr   k_any_
loguniform)nameminvalmaxvalr   k_pow2_g      ?)r   r   kg?g?)r   r   r   x_step_gQ?g{Gz?)r   r	         r   random_valuer   l    uniformx)k0k1k2)x_step_0x_step_1x_step_2g      ?i   i   l        )
r   sizestepsprobability_contiguousmin_elementsmax_elementsmax_allocation_bytesdim_parameterdtypecuda)
parameterstensorsseed)
super__init__r   range_MIN_DIM_SIZE_MAX_DIM_SIZE_POW_TWO_SIZESlenr   r   )selfr5   r1   r2   r   r*   	__class__s         r   r7   zUnaryOpFuzzer.__init__   s     4LUYZ   %Qx  $%aS\,,%1	  %Qx	  $&qc]Q_%`dB^1D,D&D%`  %Qx	  $ W*VA3<8#*WQC=93&  $	   %Qx	  $&qc]),$4T%R  ^Ak`ij]/b +>+/!)!/)4"'  	 @	
 &a	s)   !D2
D<0D7

D<$4E'E7D<)__name__
__module____qualname__torchfloat32r7   __classcell__)r>   s   @r   r   r      s    #(==u A
 A
    r   )numpynprB   torch.utils.benchmarkr   r   r   r   r9   r:   tupler8   intlog2r;   r   r
   rE   r   <module>rL      sq      W W  u!#(  B
F B
rE   