+
    ‹°³j÷  ã                   óê  € ^ RI Ht ^ RIHt ^ RIHu Ht ^ RIH	t
 ^ RIt^ RIHt ]P                  R R l4       t]P                  R R l4       tR	 tR
 t ! R R]P$                  4      tRt/ R]]P,                  ]P.                  3/C] U u/ uF   p RV  2]]P,                  ]P.                  3bK"  	  up Ct/ ]CR]]P4                  ]P6                  3/CtR R ltR R ltR R ltR R lt R# u up i )é    )ÚAny)ÚprotonN)Ú
MockTensorc                óx   € V ^8„  d   QhR\         P                  R\         P                  R\         P                  /# )é   ÚsizeÚseq_idÚmetric_value_size©ÚtlÚuint64)Úformats   "ÚW/home/luca/screen-checking/.venv/lib/python3.14/site-packages/triton/profiler/metric.pyÚ__annotate__r   
   s6   € ÷ ]ñ ]¼b¿i¹ið ]ÔQS×QZÑQZð ]Ü,.¯I©Iñ]ó    c                 ó  € ^€pV^,           p\         P                  ! WRR7      V,          p\         P                  ! W,           V4       V^,           V,          p\         P                  ! WV4      p	\         P                  ! ^ V4      p
\         P
                  ! ^ V	4       F`  pW«V,          ,           pWÅ8  p\         P                  ! WL,           VR7      p\         P                  ! WV,           V,          ,           WíR7       Kb  	  R# )é€   Úrelaxed©Úsem)ÚmaskN)r   Ú
atomic_addÚstoreÚcdivÚarangeÚrangeÚload)Ú
device_ptrÚdevice_offset_ptrr   r	   Úmetric_value_ptrr
   Ú
BLOCK_SIZEÚrecord_sizeÚdevice_offsetÚ	num_itersÚoffsetsÚiÚcur_offsetsr   Úmetric_values   &&&&&&         r   Útensor_metric_kernelr)   	   sÃ   € ð  #€JØ# aÕ'€Kô —M’MÐ"3ÀiÔPÐSWÕW€MÜ‡H‚HˆZÕ'¨Ô0Ø" QÕ&¨$Õ.€MÜ—’Ð)Ó6€IÜ�iŠi˜˜:Ó&€GÜ�XŠX�a˜Ö#ˆØ J¥Õ.ˆØÑ.ˆÜ—w’wÐ/Õ=ÀDÔIˆÜ
�Š�¨{Õ:¸dÕBÕBÀL×\ó	 $r   c                óx   € V ^8„  d   QhR\         P                  R\         P                  R\         P                  /# )r   r   r	   r(   r   )r   s   "r   r   r      s3   € ÷ 7ñ 7¼b¿i¹ið 7ÔQS×QZÑQZð 7Ôjl×jsÑjsñ 7r   c                 óÚ   € \         P                  ! V^RR7      V,          p\         P                  ! W,           V4       V^,           V,          p\         P                  ! W,           V4       R# )r   r   r   N)r   r   r   )r   r   r   r	   r(   r#   s   &&&&& r   Úscalar_metric_kernelr,      sM   € ô —M’MÐ"3°Q¸IÔFÈÕM€MÜ‡H‚HˆZÕ'¨Ô0Ø" QÕ&¨$Õ.€MÜ‡H‚HˆZÕ'¨Ö6r   c                 óh  € V P                   ! VR RR^/ pVP                  4        \        VP                  RR4      p\        VRR4      pVf)   \        P
                  P                  4       P                  pVP                  P                  V,          pVP                  WRP                  P                  3# )ÚgridÚ	num_warpsÚtargetNÚ	warp_size)é   )ÚwarmupÚ_init_handlesÚgetattrÚmetadataÚdriverÚactiveÚget_current_targetr1   r/   ÚfunctionÚshared)Ú	kernel_fnÚargsÚkernelr0   r1   Únum_threadss   &*    r   Ú_get_kernelr@   &   s“   € Ø×Ò˜tÐ=¨%Ð=¸1Ñ=€FØ
×ÑÔÜ�V—_‘_ h°Ó5€FÜ˜ ¨TÓ2€IØÒÜ—M‘M×4Ñ4Ó6×@Ñ@ˆ	Ø—/‘/×+Ñ+¨iÕ7€KØ�?‰?˜K¯©×)?Ñ)?Ð?Ð?r   c            	      óT  € \        \        P                  4      p ^ p^p^p\        \        V V VVV V4      w  rEp\        \
        V V VVV4      w  rxp	\        P                  P                  4       p
\        P                  P                  V
4      p\        P                  ! VVVVVVV	4       R# )r   N)r   r   r   r@   r)   r,   r7   r8   Úget_current_deviceÚget_current_streamÚ	libprotonÚset_metric_kernels)Úmock_ptrÚmock_seq_idÚ	mock_sizeÚmock_metric_value_sizeÚtensor_metric_kernel_fnÚ tensor_metric_kernel_num_threadsÚtensor_metric_kernel_sharedÚscalar_metric_kernel_fnÚ scalar_metric_kernel_num_threadsÚscalar_metric_kernel_sharedÚdeviceÚstreams               r   rE   rE   1   sº   € Üœ"Ÿ)™)Ó$€HØ€KØ€IØÐÜ]hÜØØØØØØó^ÑZÐÐ?Zô ^iÜØØØØØó^ÑZÐÐ?Zô �]‰]×-Ñ-Ó/€FÜ�]‰]×-Ñ-¨fÓ5€FÜ× Ò ØØØØ(Ø#Ø(Ø#ör   c                   ó2   a a€ ] tR t^Tt oV 3R ltRtVtV ;t# )Ú_TensorMetricc                ól   <€ \         SV `  VP                  4       W!P                  4       4       Wn        R # ©N)ÚsuperÚ__init__Údata_ptrÚnumelÚ_value)ÚselfÚvalueÚmetric_type_indexÚ	__class__s   &&&€r   rW   Ú_TensorMetric.__init__V   s%   ø€ Ü‰Ñ˜Ÿ™Ó)Ð+<¿k¹k»mÔLàŽr   )rZ   )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__rW   Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r^   Ú__classdict__s   @@r   rS   rS   T   s   ù‡ € ÷õ r   rS   ÚflopsÚbytesc                ó<   € V ^8„  d   QhR\         R\        R\         /# )r   r\   ÚconvertÚreturn)r   Útype)r   s   "r   r   r   j   s!   € ÷ ñ ¤3ð ´ð ¼#ñ r   c                 ó  € \        V R 4      '       d6   V P                  4       ^8”  d   V P                  4       MV P                  4       p \	        V \
        \        34      '       d   V  Uu. uF
  q!! V4      NK  	  up# V! V 4      # u upi )rX   )ÚhasattrrY   ÚtolistÚitemÚ
isinstanceÚlistÚtuple)r\   rk   Úvs   && r   Ú_scalar_or_vector_valuerv   j   sh   € Üˆu�j×!Ò!Ø"'§+¡+£-°!Ô"3�—‘”¸¿¹»ˆÜ�%œ$¤˜×'Ò'Ù$)Ó*¡E˜q�˜–
¡EÑ*Ð*Ù�5‹>Ðùò +s   Á(Bc                ó<   € V ^8„  d   QhR\         R\        R\        /# ©r   Úkeyr\   rl   )Ústrr   )r   s   "r   r   r   r   s!   € ÷ ñ œcð ¬#ð ´#ñ r   c                 óT   € V \         9   d   \         V ,          w  p p\        W4      # V# rU   )ÚROOFLINE_DTYPESrv   )ry   r\   rk   Ú_s   &&  r   Ú_scalar_metric_valuer~   r   s,   € ð ŒoÔÜ'¨Õ,‰ˆ��AÜ& uÓ6Ð6Ø€Lr   c                ó^   € V ^8„  d   QhR\         R\        R\        \        \        3,          /# rx   )rz   r   rt   Úint)r   s   "r   r   r   {   s)   € ÷ 4ñ 4¬ð 4´Cð 4¼EÄ#ÄsÀ(½Oñ 4r   c                 ó  € V \         9   dS   \         V ,          w  r#pV R 8X  d   VP                  4       MVP                  4       pYP                  4       ^8”  d   V3# T3# VP	                  4       '       dJ   VP                  4       pVP                  4       ^8”  d   V\
        P                  3# V\
        P                  3# VP                  4       pVP                  4       ^8”  d   V\
        P                  3# V\
        P                  3# )ri   )
r|   ÚlongÚdoublerY   Úis_floating_pointrD   Úmetric_type_vector_double_indexÚmetric_type_double_indexÚmetric_type_vector_int64_indexÚmetric_type_int64_index)ry   r\   r}   Úscalar_indexÚvector_indexs   &&   r   Ú_tensor_metric_value_and_indexr‹   {   s×   € Ø
ŒoÔÜ(7¸Õ(<Ñ%ˆ˜Ø # w¤�—
‘
”°E·L±L³NˆØ§k¡k£m°aÔ&7�lÐIÐI¸\ÐIÐIð ×Ñ× Ò Ø—‘“ˆØ�;‰;‹=˜1ÔØœ)×CÑCÐCÐCØ”i×8Ñ8Ð8Ð8à�J‰J‹L€EØ‡{�{ƒ}�qÔØ”i×>Ñ>Ð>Ð>Ø”)×3Ñ3Ð3Ð3r   c                óÌ   € V ^8„  d   QhR\         \        \        3,          R\        \         \        \        3,          \         \        \        P
                  3,          3,          /# )r   Úmetricsrl   )Údictrz   r   rt   rD   ÚTensorMetric)r   s   "r   r   r   Ž   sF   € ÷ *ñ *¤d¬3´¨8¥nð *¼¼tÄCÌÀH½~ÌtÔTWÔYb×YoÑYoÐToÕOpÐ?pÕ9qñ *r   c                 ó  € / p/ pV P                  4        Fj  w  r4\        VR 4      '       dG   VP                  P                  R8X  d   \	        W44      W#&   KA  \        W44      w  rE\        WE4      W&   K]  \	        W44      W#&   Kl  	  W!3# )rX   Úcpu)Úitemsro   rP   rm   r~   r‹   rS   )r�   Útensor_metricsÚscalar_metricsry   r\   Úmetric_indexs   &     r   Útransform_tensor_metricsr–   Ž   s|   € Ø€NØ%'€NØ—m‘m–o‰
ˆÜ�5˜*×%Ò%Ø�|‰|× Ñ  EÔ)Ü&:¸3Ó&F�Ó#ä&DÀSÓ&PÑ#�Ü&3°EÓ&H�Ó#ä"6°sÓ"BˆNÓñ &ð Ð)Ð)r   )é   é   é    é@   )!Útypingr   Útriton._C.libprotonr   rD   Útriton.runtime.driverÚruntimer7   Útriton.languageÚlanguager   Útritonr   Újitr)   r,   r@   rE   r�   rS   ÚFLOPS_WIDTHSÚfloatr†   r…   ÚFLOPS_DTYPESr€   rˆ   r‡   r|   rv   r~   r‹   r–   )Úwidths   0r   Ú<module>r§      s  ðÝ Ý 3ß &Ð &Ý Û Ý ð ‡�ô]ó ð]ð& ‡�ô7ó ð7ò@ò ôF�I×*Ñ*ô ð €ðØ�˜	×:Ñ:¸I×<eÑ<eÐfÐgðñ "óá!ˆEð �ˆwˆ˜% ×!CÑ!CÀY×EnÑEnÐoÒoÙ!ñð€ðØðàˆc�9×4Ñ4°i×6^Ñ6^Ð_ñ€õõõ4÷&*ùò_s   Â&C0