
    sg]                         U d dl Z d dlZd dlmZ d dlZdedefdZdedefdZ G d d      Z	 G d	 d
      Z
 G d dej                        Z ee j                  e   e      e j                  e<   eed<   ee   ed<   y)    N)Listqenginereturnc                     | dk(  s| dk(  s| d}|S | dk(  rd}|S | dk(  rd}|S | dk(  rd	}|S | d
k(  rd}|S d}t        |  d      )Nnone r   fbgemm   qnnpack   onednn   x86   z* is not a valid value for quantized engineRuntimeError)r   rets     T/var/www/html/venv/lib/python3.12/site-packages/torch/backends/quantized/__init__.py_get_qengine_idr   
   s    &GrMW_ J 
H	 J 
I	 J 
H	 J 
E	 J gY&PQRR    c                 6    dddddd}|j                  | d      S )Nr   r	   r   r   r   )r   r
   r   r   r   z
*undefined)get)r   all_enginess     r   _get_qengine_strr      s#    iHOK??7L11r   c                   (    e Zd ZdefdZdeddfdZy)_QEnginePropr   c                 P    t        t        j                  j                               S N)r   torch_C_get_qengine)selfobjobjtypes      r   __get__z_QEngineProp.__get__"   s     5 5 788r   valNc                 T    t         j                  j                  t        |             y r   )r    r!   _set_qenginer   r#   r$   r'   s      r   __set__z_QEngineProp.__set__%   s    oc23r   )__name__
__module____qualname__strr&   r+    r   r   r   r   !   s!    9s 94 4 4r   r   c                   &    e Zd Zdee   fdZddZy)_SupportedQEnginesPropr   c                 z    t         j                  j                         }|D cg c]  }t        |       c}S c c}w r   )r    r!   _supported_qenginesr   )r#   r$   r%   qenginesqes        r   r&   z_SupportedQEnginesProp.__get__*   s.    88//1/78 $888s   8Nc                     t        d      )NzAssignment not supportedr   r*   s      r   r+   z_SupportedQEnginesProp.__set__.   s    566r   )r   N)r,   r-   r.   r   r/   r&   r+   r0   r   r   r2   r2   )   s    9tCy 97r   r2   c                   @     e Zd Z fdZd Z e       Z e       Z xZ	S )QuantizedEnginec                 2    t         |   |       || _        y r   )super__init__m)r#   r=   name	__class__s      r   r<   zQuantizedEngine.__init__3   s    r   c                 8    | j                   j                  |      S r   )r=   __getattribute__)r#   attrs     r   __getattr__zQuantizedEngine.__getattr__7   s    vv&&t,,r   )
r,   r-   r.   r<   rC   r   enginer2   supported_engines__classcell__)r?   s   @r   r9   r9   2   s    - ^F.0r   r9   rD   rE   )systypestypingr   r    r/   intr   r   r   r2   
ModuleTyper9   modulesr,   __annotations__r0   r   r   <module>rN      s    
   S S $2c 2c 2
4 47 7	1e&& 	1 (H(=xHH 9 r   