+
    ړLj                         ^ 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 ^RIHtHtHt ]P"                  R R l4       t]P"                  R R	 l4       tR
 tR t ! R R]P,                  4      tR R ltR# )    )Any)protonN)
MockTensor)
exit_stateenter_stateCOMPUTE_METADATA_SCOPE_NAMEc                x    V ^8  d   QhR\         P                  R\         P                  R\         P                  /# )   size	metric_idmetric_value_sizetluint64)formats   "B/app/.local/lib/python3.14/site-packages/triton/profiler/metric.py__annotate__r      s1     / /bii /TVT]T] /,.II/    c                 T   R p\         P                  ! V4      p\         P                  ! W,           V4       V^,           V,          p\         P                  ! WV4      p\         P                  ! ^ V4      p	\         P
                  ! ^ V4       F`  p
WV,          ,           pW8  p\         P                  ! WK,           VR7      p\         P                  ! WV,           V,          ,           WR7       Kb  	  \         P                  ! 4        Wu,           V,          p\         P                  ! W4       R# )   )maskN)r   loadstorecdivarangerangedebug_barrier)
device_ptrdevice_offset_ptrr   r   metric_value_ptrr   
BLOCK_SIZEdevice_offset	num_itersoffsetsicur_offsetsr   metric_values   &&&&&&        r   tensor_metric_kernelr(   
   s      #JGG-.MHHZ'3"Q&$.M)6Iii:&GXXa#J..ww/=DI
{:dBBL\	 $
 "6$>MHH.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     / /bii /TVT]T] /momvmv /r   c                 <   \         P                  ! V4      p\         P                  ! W,           V4       V^,           V,          p\         P                  ! W,           V4       V^,           V,          p\         P                  ! 4        \         P                  ! W4       R# )   N)r   r   r   r   )r   r   r   r   r'   r"   s   &&&&& r   scalar_metric_kernelr,      sj    GG-.MHHZ'3"Q&$.MHHZ'6"Q&$.MHH.r   c                 b    V P                   ! VR RR^/ pVP                  4        VP                  # )grid	num_warps)r+   )warmup_init_handlesfunction)	kernel_fnargskernels   &* r   _get_kernelr6   (   s2    t=%=1=F
??r   c            	      >   \        \        P                  4      p ^ p^p^p\        \        V V VVV V4      p\        \
        V V VVV4      p\        P                  P                  4       p\        P                  P                  V4      p\        P                  ! WEV4       R# )r   N)r   r   r   r6   r(   r,   driveractiveget_current_deviceget_current_stream	libprotonset_metric_kernels)mock_ptrmock_metric_id	mock_sizemock_metric_value_sizetensor_metric_kernel_fnscalar_metric_kernel_fndevicestreams           r   r=   r=   .   s    "))$HNI) * ]]--/F]]--f5F  !8SYZr   c                   2   a a ] tR t^It 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   rJ   _TensorMetric.__init__K   s%    )+<kkmLr   )rM   )__name__
__module____qualname____firstlineno__rJ   __static_attributes____classdictcell____classcell__)rQ   __classdict__s   @@r   rG   rG   I   s      r   rG   c                    V ^8  d   QhR\         \        \        3,          R\        \         \        \        3,          \         \        \        P
                  3,          3,          /# )r
   metricsreturn)dictstrr   tupler<   TensorMetric)r   s   "r   r   r   Q   sF     * *d38n *tCH~tTWYbYoYoToOp?p9q *r   c                 L   / p/ pV P                  4        EF	  w  r4\        VR 4      '       d   VP                  P                  R8X  d   WBV&   K9  \	        \
        4       VP                  4       '       dH   VP                  4       pVP                  4       ^8  d   \        P                  pMW\        P                  pMFVP                  4       pVP                  4       ^8  d   \        P                  pM\        P                  p\        4        \!        WE4      W&   EK  WBV&   EK  	  W!3# )rK   cpu)itemshasattrrD   typer   r   is_floating_pointdoublerL   r<   metric_type_vector_double_indexmetric_type_double_indexlongmetric_type_vector_int64_indexmetric_type_int64_indexr   rG   )r\   tensor_metricsscalar_metricskeyrO   metric_indexs   &     r   transform_tensor_metricsrr   Q   s    N%'Nmmo
5*%%||  E)&+s#78**,,!LLNE{{}q('0'P'P'0'I'I!JJLE{{}q('0'O'O'0'H'H&3E&H#"'3- &. ))r   )typingr   triton._C.libprotonr   r<   triton.runtime.driverruntimer8   triton.languagelanguager   tritonr   stater   r   r   jitr(   r,   r6   r=   ra   rG   rr    r   r   <module>r}      sr     3 & &    G G / /$ / /[6I** *r   