Ë
    [^(h(  ã                   ó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     €úg/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torch/utils/benchmark/op_fuzzers/spectral.pyú	<genexpr>zpower_range.<locals>.<genexpr>   s   øè ø€ ÒK˜!ˆD�A�IÑKùs   ƒé   )ÚrangeÚintÚmathÚlog)Úupper_boundr   s    `r   Úpower_ranger      s(   ø€ ÛKœu¤S¬¯©°+¸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{®Gáz¤?)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–  ø€ ä‰Ñô   ¸ÀÈÑ5MÐVZÔ[ô   % Q›xöð ô $Ø% a S˜\Ü+Ü+Ø%1ö	òô  % Q›x÷	ð ô $Ø)¨!¨Ð-ÜP]Ö%^È d¨B´´]Ó1CÑ,CÑ&CÒ%^öóô  % Q›xö	ð ô $Ø   ˜Wä*¨Z¸°sÐ+;Ó<Ð>QÜ*¨V°A°3¨<Ó8¸!Ð>QÑ:Qð&ð  $öò	ô   % Q›xö	ð ô $Ø$ Q C˜[Ø),°¸$À4ÈTÑ%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   