
    shJ                         S SK rS SKrS SKJrJrJrJr  SrSr	\
" S \" \" \R                  " \5      5      \" \R                  " \	5      5      S-   5       5       5      r " S S\5      rg)	    N)FuzzerFuzzedParameterParameterAliasFuzzedTensor   i   c              #   ,   #    U  H
  nS U-  v   M     g7f)   N ).0is     z/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/torch/utils/benchmark/op_fuzzers/unary.py	<genexpr>r   
   s       (!qAv (s      c                   D   ^  \ rS rSr\R
                  S4U 4S jjrSrU =r$ )UnaryOpFuzzerr   Fc                   > [         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  oUS[        [        5      -  _M     snS9PM7     snn[        S5       Vs/ s H-  n[        SU 3[        SU 35      S[        SU 35      S0SS9PM/     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
9/[        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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   UnaryOpFuzzer.__init__   s     4LUYZ   %Qx  (! $%aS\,,%1	
  (  %Qx	  (! $&qc]Q_%`Q_B^1D,D&DQ_%`  (	  %Qx	  (! $ W*VA3<8#*WQC=93&  $  (	   %Qx	  (! $&qc]),$4T%R  (	  ^Ak`ij]/b +>+/!)!/)4"'  	 @	
 &a	s)   !D(
D2.D-
	D2#4D8&D=-D2r
   )	__name__
__module____qualname____firstlineno__torchfloat32r7   __static_attributes____classcell__)r>   s   @r   r   r      s    #(==u A
 A
    r   )numpynprD   torch.utils.benchmarkr   r   r   r   r9   r:   tupler8   intlog2r;   r   r
   rH   r   <module>rO      sm      W W  u!#(  B
F B
rH   