
    VhE                         U d dl Z d dl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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     Q/home/dcms/DCMS/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-   listr.   r%   r*   r/   r   r   r1   r1   (   s    9tCy 97r   r1   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__2   s    r   c                 8    | j                   j                  |      S r   )r=   __getattribute__)r"   attrs     r   __getattr__zQuantizedEngine.__getattr__6   s    vv&&t,,r   )
r+   r,   r-   r<   rC   r   enginer1   supported_engines__classcell__)r?   s   @r   r9   r9   1   s    - ^F.0r   r9   rD   rE   )systypesr   r.   intr   r   r   r1   
ModuleTyper9   modulesr+   __annotations__r7   r/   r   r   <module>rM      s    
  S S $2c 2c 2
4 47 7	1e&& 	1 (H(=xHH 9 r   