Ë
    ÌÓ^j0  ã                   ó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     €út/var/www/ramen.bs-engineer-server.com/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            	       óN   ‡ — e Zd Zej                  dddœdedededdfˆ fd	„Zˆ xZ	S )
r   Fç      ð?)ÚdtypeÚcudaÚprobability_regularÚseedr   r    ÚreturnNc                ó‚  •— 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   r8   Ú	__class__s          €r   rB   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ÚfloatrB   Ú__classcell__)rH   s   @r   r   r      s8   ø„ Ø+0¯=©=Ø#À#ò?
 ð ?
Øð?
Ø:?ð?
ØJN÷?
ñ ?
r   )r   rL   Útorch.utilsr   Útorch.utils.benchmarkr   r   r   Ú__all__rC   rD   r   rE   r   ÚjÚijr-   ÚijkÚappendÚsortÚFuzzerr   r   r   r   ú<module>rZ      sË   ðã ã Ý !ß OÑ Oð Ð
€à€Ø€òLð €Ù	�\ 1Ó	%ò *€AÙ˜¨Ñ*¨AÓ.ò *ˆØ�‰UˆÙ˜\¨RÑ/°Ó3ò 	*ˆAØ�q‘&ˆCØ�\Ó!Ø×$Ñ$ SÕ)ñ	*ñ*ð*ð × Ñ Ô ô@
�y×'Ñ'õ @
r   