ó
    Këgj#  ã                   ó.  • S SK r S SKJr  S SKJrJr  S SKJr  \ R                  " SS V Vs/ s H  n S  H  n\ R                  SU 0US9PM     M!     snn S/S	.\D6\ R                  " S
/S9S\R                  S\R                  4S j5       5       rgs  snn f )é    N)ÚexpÚlog)Úautotune_cache_kwargs)é   é    é@   )é   é   é   ÚBT)Ú	num_warpsÚS)ÚconfigsÚkeyÚT)Údo_not_specializec                 óü  • [         R                  " S5      n[         R                  " SU5      n[         R                  " US S 2S 4   US S S 24   :¬  SS5      n[         R                  " U/[        S5      [         R                  S9n[         R                  " U/[         R                  S9n	[        [         R                  " X$5      5       GH°  n
X¤-  [         R                  " SU5      -   nX²:  nXU-  U-  -   US S 2S 4   U-  -   [         R                  " SU5      S S S 24   -   nXU-  U-  -   US S 2S 4   U-  -   [         R                  " SU5      S S S 24   -   n[         R                  " XÜS S 2S 4   SS9R                  [         R                  5      n[         R                  " US5      n[         R                  " UU5      nU	[        UU-
  5      -  n	[        UU-
  5      n[         R                  " XSS9U	-   n[         R                  " US5      nUnUn	[!        [         R                  " US:g  US	5      5      U-   n[         R"                  " UUR                  UR$                  R&                  5      US S 2S 4   S
9  GM³     g )Nr   g      ð?g        z-inf)Údtype)ÚmaskÚotherF)Ú
allow_tf32g#B’¡œÇ;)r   )ÚtlÚ
program_idÚarangeÚwhereÚfullÚfloatÚfloat32ÚzerosÚrangeÚcdivÚloadÚtoÚmaxÚmaximumr   Údotr   Ústorer   Ú
element_ty)ÚsÚzr   r   r   Úi_bhÚo_iÚm_sÚb_mpÚb_zpÚi_tÚo_tÚm_tÚp_sÚp_zÚb_sÚb_mcÚb_zÚb_zcs                      Ú¨C:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\fla/ops/utils/logcumsumexp.pyÚlogcumsumexp_fwd_kernelr:      s  € ô" �=Š=˜Ó€DÜ
�)Š)�A�rÓ
€CÜ
�(Š(�3’q˜$�w‘< 3 tªQ w¡<Ñ/°°RÓ
8€Cä�7Š7�A�3œ˜f›¬R¯Z©ZÑ8€DÜ�8Š8�Q�CœrŸz™zÑ*€DÜ”R—W’W˜Q“^×$ˆØ‰hœŸš 1 bÓ)Ñ)ˆØ‰gˆØ˜‘(˜1‘*‰n˜s¢1 d 7™|¨aÑ/Ñ/´"·)²)¸A¸q³/À$ÊÀ'Ñ2JÑJˆØ˜‘(˜1‘*‰n˜s¢1 d 7™|¨aÑ/Ñ/´"·)²)¸A¸q³/À$ÊÀ'Ñ2JÑJˆô �gŠg�c¢A t G¡°CÑ8×;Ñ;¼B¿J¹JÓGˆä�vŠv�c˜1‹~ˆÜ�zŠz˜$ Ó%ˆØ”c˜$ ™+Ó&Ñ&ˆä�#˜‘*‹oˆÜ�fŠf�S¨%Ñ0°4Ñ7ˆä�vŠv�c˜1‹~ˆØˆØˆô ”"—(’(˜3 !™8 S¨%Ó0Ó1°DÑ8ˆÜ
�Š��c—f‘f˜SŸY™Y×1Ñ1Ó2¸ºQÀ¸W¹ÕFò- %ó    © )ÚtritonÚtriton.languageÚlanguager   Úfla.ops.utils.opr   r   Ú	fla.utilsr   ÚautotuneÚConfigÚjitÚ	constexprr:   )r   r   s   00r9   Ú<module>rF      s»   ðó Ý ç %Ý +ð ‡‚ð ñ ôâˆBÜ"ˆIð 	�‰�t˜R�j¨IˆÓ6á"ñ 	7Ùòð
 
ˆñð ñð ‡‚˜s˜eÑ$ð#Gð 
‡|�|ð	#Gð
 	�‰ó#Gó %óñ#Gùós   «&B