ó
    "Eñiù  ã                   óX  • S SK r S SKrS SKJr  \ R                  S\4S j5       r\ R                  SS\\\4   S\\\4   4S jj5       r	\ R                  S\4S j5       r
\ R                  S\4S j5       r\ R                  S\4S	 j5       r\ R                  S\4S
 j5       r\ R                  S\4S j5       r\ R                  S\4S j5       r\ R                   " S5      S\4S j5       r\ R                  S\4S j5       r\ R                  S\4S j5       r\ R                  S\4S j5       rg)é    N)ÚAnyÚreturnc                  ó.   •  SS K n g! [         a     gf = f)Nr   TF)ÚtritonÚImportError)r   s    ÚP/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/utils/_triton.pyÚhas_triton_packager	      s   € ðÛàøÜó Ùðús   ‚ ‡
“Úfallbackc                 ó–   •  SS K n[        S UR                  R                  S5      S S  5       5      u  p#X#4$ ! [         a    U s $ f = f)Nr   c              3   ó8   #   • U  H  n[        U5      v •  M     g 7f©N)Úint)Ú.0Úvs     r   Ú	<genexpr>Ú%get_triton_version.<locals>.<genexpr>   s   é € ÐOÒ-N¨œS ŸV˜VÒ-Nùs   ‚Ú.é   )r   ÚtupleÚ__version__Úsplitr   )r
   r   ÚmajorÚminors       r   Úget_triton_versionr      sP   € ðÛäÑO¨V×-?Ñ-?×-EÑ-EÀcÓ-JÈ2ÈAÑ-NÓOÓO‰ˆØˆ~ÐøÜó ØŠðús   ‚69 ¹AÁAc                  óÊ   • SS K n U R                  R                  5       =(       a?    U R                  R                  5       S:¬  =(       a    U R                  R
                  (       + $ )Nr   ©é	   r   )ÚtorchÚcudaÚis_availableÚget_device_capabilityÚversionÚhip©r   s    r   Ú_device_supports_tmar%      sK   € ãð 	�
‰
×ÑÓ!÷ 	"Ø�J‰J×,Ñ,Ó.°&Ñ8÷	"à—‘×!Ñ!Ô!ðó    c                  ó®   • [        5       (       a&  [        5       (       a   SSKJn Jn   SSKJn  U" 5       $ g! [         a     gf = f! [         a     gf = f)Nr   )Úcreate_1d_tma_descriptorÚcreate_2d_tma_descriptor)Úenable_in_pytorchTF)r	   r%   Ú$triton.tools.experimental_descriptorr(   r)   r*   r   )r(   r)   r*   s      r   Ú has_triton_experimental_host_tmar,   &   s\   € ä×ÑÜ×!Ñ!ð÷ð
 ÝVá,Ó.Ð.ð øô #ó  Ùð ûäó Øàðús,    A ©7 ·
AÁA ÁAÁA Á
AÁAc                  óp   • [        5       (       a  [        5       (       a   SSKJn   gg! [         a     gf = f)Nr   ©ÚTensorDescriptorTF)r	   r%   Útriton.tools.tensor_descriptorr/   r   r.   s    r   Ú%has_triton_tensor_descriptor_host_tmar1   <   s?   € ä×ÑÜ×!Ñ!ðõð ð øô ó Øàðús    ( ¨
5´5c                  ó8   • [        5       =(       d
    [        5       $ r   )r1   r,   © r&   r   Úhas_triton_tmar4   L   s   € ä0Ó2×XÔ6VÓ6XÐXr&   c                  ó|  • [        5       (       a…  SS Kn U R                  R                  5       (       a9  U R                  R	                  5       S:¼  a  U R
                  R                  (       a  U R                  R                  5       (       a
   SSKJ	nJ
n  gg! [         a     Of = f SSKJn  g! [         a     gf = f)Nr   r   )Ú&experimental_device_tensormap_create1dÚ&experimental_device_tensormap_create2dT©Úmake_tensor_descriptorF)r	   r   r   r    r!   r"   r#   ÚxpuÚtriton.language.extra.cudar6   r7   r   Útriton.languager9   )r   r6   r7   r9   s       r   Úhas_triton_tma_devicer=   Q   s    € ä×ÑÛð �J‰J×#Ñ#×%Ñ%Ø—
‘
×0Ñ0Ó2°fÓ<Ø—M‘M×%×%Ø�Y‰Y×#Ñ#×%Ñ%ð÷ð
 ð øô ó ÙðúðÝBàøÜó Øàðús$   ÂB Â
B#Â"B#Â'B. Â.
B;Â:B;c                  ó.  • SS K n U R                  R                  5       (       ar  U R                  R                  5       S:¼  aT  U R                  R                  5       S:  a6  U R                  R
                  (       d  [        5       =(       a
    [        5       $ g)Nr   )é
   r   )é   r   F)r   r   r    r!   r"   r#   r=   r1   r$   s    r   Ú#has_datacenter_blackwell_tma_devicerA   q   sf   € ãð 	�
‰
×Ñ×!Ñ!Ø�J‰J×,Ñ,Ó.°'Ó9Ø�J‰J×,Ñ,Ó.°Ó8Ø—‘×!×!ä$Ó&×RÔ+PÓ+RÐRàr&   c                  óH  • [        5       (       aƒ  SS Kn U R                  R                  5       (       a9  U R                  R	                  5       S:¼  a  U R
                  R                  (       a  U R                  R                  5       (       a   SSKJ	n  gg! [         a     gf = f)Nr   r   r8   TF)r	   r   r   r    r!   r"   r#   r:   r<   r9   r   )r   r9   s     r   Úhas_triton_stable_tma_apirC   €   s   € ä×ÑÛð �J‰J×#Ñ#×%Ñ%Ø—
‘
×0Ñ0Ó2°fÓ<Ø—M‘M×%×%Ø�Y‰Y×#Ñ#×%Ñ%ðÝBàð øô ó ØØðús   ÂB Â
B!Â B!c                  óî   ^^• [        5       (       d  gSSKJn   U (       a  gSSKJm  S[
        S[        4S jnS[
        S[        4S jnS[
        S[        4S	 jnUUUUS
.mS[        4UU4S jjnU" 5       $ )NFr   )Útriton_disable_device_detection)Úget_interface_for_deviceÚdevice_interfacer   c                 óP   • U R                   R                  5       R                  S:¬  $ )Né   )ÚWorkerÚget_device_propertiesr   ©rG   s    r   Úcuda_extra_checkÚ$has_triton.<locals>.cuda_extra_checkŸ   s"   € Ø×&Ñ&×<Ñ<Ó>×DÑDÈÑIÐIr&   c                 ó<   • SS K nSUR                  R                  ;   $ )Nr   Úcpu)Útriton.backendsÚbackends)rG   r   s     r   Úcpu_extra_checkÚ#has_triton.<locals>.cpu_extra_check¢   s   € Ûà˜Ÿ™×0Ñ0Ñ0Ð0r&   c                 ó   • g)NTr3   rL   s    r   Ú_return_trueÚ has_triton.<locals>._return_true§   s   € Ør&   )r   r:   rP   Úmtiac                  ó”   >• TR                  5        H3  u  pT" U 5      nUR                  5       (       d  M$  U" U5      (       d  M3    g   g)NTF)Úitemsr    )ÚdeviceÚextra_checkrG   rF   Útriton_supported_devicess      €€r   Ú is_device_compatible_with_tritonÚ4has_triton.<locals>.is_device_compatible_with_triton±   sG   ø€ Ø#;×#AÑ#AÖ#CÑˆFÙ7¸Ó?ÐØ×,Ñ,×.Ó.±;Ð?O×3PÓ3PÙñ $Dð r&   )r	   Útorch._inductor.configrE   Útorch._dynamo.device_interfacerF   r   Úbool)rE   rM   rS   rV   r^   rF   r]   s        @@r   Ú
has_tritonrc   “   s‰   ù€ ä×ÑØåFæ&ØåGðJ¬3ð J´4ô Jð1¬#ð 1´$ô 1ð
¤sð ¬tô ð !ØØØñ	 Ðð¬d÷ ð ñ ,Ó-Ð-r&   c                  ó^   • SSK Jn   SSKJn  UR                  R                  5       nU " U5      $ )Nr   )Úmake_backend)Údriver)Útriton.compiler.compilerre   Útriton.runtime.driverrf   ÚactiveÚget_current_target)re   rf   Útargets      r   Útriton_backendrl   »   s%   € å5Ý,à�]‰]×-Ñ-Ó/€FÙ˜ÓÐr&   c                  óØ   • SSK Jn   [        5       nU " 5        SUR                  5        3n[        R
                  " UR                  S5      5      R                  5       R                  5       $ )Nr   )Ú
triton_keyÚ-zutf-8)	Ú%torch._inductor.runtime.triton_compatrn   rl   ÚhashÚhashlibÚsha256ÚencodeÚ	hexdigestÚupper)rn   ÚbackendÚkeys      r   Útriton_hash_with_backendry   Ä   sR   € å@äÓ€GÙ‹\ˆN˜!˜GŸL™L›NÐ+Ð
,€Cô �>Š>˜#Ÿ*™* WÓ-Ó.×8Ñ8Ó:×@Ñ@ÓBÐBr&   ))r   r   )Ú	functoolsrr   Útypingr   Úcacherb   r	   r   r   r   r%   r,   r1   r4   r=   rA   Ú	lru_cacherC   rc   rl   Ústrry   r3   r&   r   Ú<module>r      s§  ðÛ Û Ý ð ‡�ð˜Dó ó ðð ‡�ñ  s¨C x¡ð ¸eÀCÈÀH¹oô ó ðð ‡�ð˜dó ó ðð ‡�ð¨$ó ó ðð* ‡�ð¨tó ó ðð ‡�ðY˜ó Yó ðYð ‡�ð˜tó ó ðð> ‡�ð¨Tó ó ðð ×Ò�TÓð 4ó ó ðð$ ‡�ð$.�Dó $.ó ð$.ðN ‡�ð ˜ó  ó ð ð ‡�ðC #ó Có ñCr&   