+
    דLj
                     r    ^ RI HtHt ^ RIHtHtHtHt  ! R R]4      t ! R R]R7      t	 ! R R	]	4      t
R
# )    )ABCMetaabstractmethod)CallableListProtocolSequencec                   2   a  ] tR t^t o V 3R lR ltRtV tR# )Benchmarkerc                L   < V ^8  d   QhRS[ RS[S[,          RS[S[,          /# )   kernel_call	quantilesreturn)r   r   floatr   )format__classdict__s   "B/app/.local/lib/python3.14/site-packages/triton/backends/driver.py__annotate__Benchmarker.__annotate__   s-      H DK V^_dVe     c                   R # N )selfr   r   kwargss   &&$,r   __call__Benchmarker.__call__       r   r   N)__name__
__module____qualname____firstlineno__r   __static_attributes____classdictcell__r   s   @r   r
   r
      s      r   r
   c                      a  ] tR t^t o ]]R 4       4       t]V 3R lR l4       t]R 4       t]R 4       t	]V 3R lR l4       t
V 3R lR	 ltR
tV tR# )
DriverBasec                    R # r   r   r   s   &r   	is_activeDriverBase.is_active   s     	r   c                &   < V ^8  d   QhRS[ RS[ /# )r   tyr   )str)r   r   s   "r   r   DriverBase.__annotate__   s     
 
 
 
r   c                    R# )z
Converts a Triton type string to its corresponding C++ type string for this backend.

Args:
    ty (str): The Triton type string. e.g., 'i32', '*fp16', 'fp32'.

Returns:
    str: The C++ type string.
Nr   )r   r-   s   &&r   map_python_to_cpp_type!DriverBase.map_python_to_cpp_type   s     	r   c                    R # r   r   r)   s   &r   get_current_targetDriverBase.get_current_target       r   c                    R # r   r   r)   s   &r   get_active_torch_device"DriverBase.get_active_torch_device#   r6   r   c                    < V ^8  d   QhRS[ /# )r   r   )r
   )r   r   s   "r   r   r/   (   s     " " "r   c                    \         h)zK
Return the benchmarking function that this backend should use by default.
)NotImplementedErrorr)   s   &r   get_benchmarkerDriverBase.get_benchmarker'   s
    
 "!r   c                   < V ^8  d   QhRR/# )r   r   Nr   )r   r   s   "r   r   r/   .   s      $ r   c                    R # r   r   r)   s   &r   __init__DriverBase.__init__.   r   r   r   N)r   r    r!   r"   classmethodr   r*   r1   r4   r8   r=   rA   r#   r$   r%   s   @r   r'   r'      sy        
 
     " " r   r'   )	metaclassc                   ,   a  ] tR t^2t o R tR tRtV tR# )	GPUDriverc                  a ^ RI oSP                  P                  V n         ^ RIHp Wn        SP                  P                  V n        SP                  P                  V n
        R#   \         d    T3R lT n         LPi ; i)r   N)_cuda_getCurrentRawStreamc                 N   < SP                   P                  V 4      P                  # r   )cudacurrent_streamcuda_stream)idxtorchs   &r   <lambda>$GPUDriver.__init__.<locals>.<lambda><   s    %**2K2KC2P2\2\r   )rN   rJ   get_device_capabilitytorch._CrH   get_current_streamImportErrorcurrent_deviceget_current_device
set_deviceset_current_device)r   rH   rN   s   & @r   rA   GPUDriver.__init__4   sh    %*ZZ%E%E"	]:&?# #(**";";"'**"7"7  	]&\D#	]s   A& &A?>A?c                    V# r   r   )r   tensormaps_infoargss   &&&r   assemble_tensormap_to_arg#GPUDriver.assemble_tensormap_to_argA   s    r   )rV   rS   rQ   rX   N)r   r    r!   r"   rA   r]   r#   r$   r%   s   @r   rF   rF   2   s     
8 r   rF   N)abcr   r   typingr   r   r   r   r
   r'   rF   r   r   r   <module>ra      s3    ' 5 5( $7 $N
 r   