
    wg                     F    d dl mZmZmZ  G d de      Z G d de      Zy)    )ABCMetaabstractmethodabstractclassmethodc                   4    e Zd Zed        Zed        ZddZy)
DriverBasec                      y N selfs    [/home/mcse/projects/flask/flask-venv/lib/python3.12/site-packages/triton/backends/driver.py	is_activezDriverBase.is_active           c                      y r	   r
   r   s    r   get_current_targetzDriverBase.get_current_target
   r   r   Nc                      y r	   r
   r   s    r   __init__zDriverBase.__init__   s    r   )returnN)__name__
__module____qualname__r   r   r   r   r   r
   r   r   r   r      s-       r   r   )	metaclassc                       e Zd Zd Zd Zy)	GPUDriverc                     dd l j                  j                  | _        	 ddlm} || _        j                  j                  | _        j                  j                  | _
        y # t        $ r fd| _        Y Lw xY w)Nr   )_cuda_getCurrentRawStreamc                 N    j                   j                  |       j                  S r	   )cudacurrent_streamcuda_stream)idxtorchs    r   <lambda>z$GPUDriver.__init__.<locals>.<lambda>   s    %**2K2KC2P2\2\ r   )r#   r   get_device_capabilitytorch._Cr   get_current_streamImportErrorcurrent_deviceget_current_device
set_deviceset_current_device)r   r   r#   s     @r   r   zGPUDriver.__init__   sj    %*ZZ%E%E"	]:&?D# #(**";";"'**"7"7  	]&\D#	]s   A& &A<;A<c                     |S r	   r
   )r   tensormaps_infoargss      r   assemble_tensormap_to_argz#GPUDriver.assemble_tensormap_to_arg!   s    r   N)r   r   r   r   r0   r
   r   r   r   r      s    
8r   r   N)abcr   r   r   r   r   r
   r   r   <module>r2      s#    < <7 
 r   