
    %VjiE                         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<   dS )    Nqenginereturnc                     | dk    s| dk    s| d}n8| dk    rd}n/| dk    rd}n&| dk    rd	}n| d
k    rd}nd}t          |  d          |S )Nnone r   fbgemm   qnnpack   onednn   x86   z* is not a valid value for quantized engineRuntimeError)r   rets     [/root/voice-cloning/.venv/lib/python3.11/site-packages/torch/backends/quantized/__init__.py_get_qengine_idr   	   s    &GrMMW_	H			I			H			E		gQQQRRRJ    c                 >    dddddd}|                     | d          S )Nr   r   r
   r   r   )r   r	   r   r   r   z
*undefined)get)r   all_enginess     r   _get_qengine_strr      s)    iHOOK??7L111r   c                   *    e Zd ZdefdZdeddfdZdS )_QEnginePropr   c                 X    t          t          j                                                  S N)r   torch_C_get_qengine)selfobjobjtypes      r   __get__z_QEngineProp.__get__!   s     5 5 7 7888r   valNc                 ^    t           j                            t          |                     d S r   )r   r    _set_qenginer   r"   r#   r&   s      r   __set__z_QEngineProp.__set__$   s&    oc2233333r   )__name__
__module____qualname__strr%   r*    r   r   r   r       sQ        9s 9 9 9 94 4 4 4 4 4 4 4r   r   c                   .    e Zd Zdee         fdZddZdS )_SupportedQEnginesPropr   c                 V    t           j                                        }d |D             S )Nc                 ,    g | ]}t          |          S r/   )r   ).0qes     r   
<listcomp>z2_SupportedQEnginesProp.__get__.<locals>.<listcomp>+   s!    888 $$888r   )r   r    _supported_qengines)r"   r#   r$   qenginess       r   r%   z_SupportedQEnginesProp.__get__)   s)    8//1188x8888r   Nc                      t          d          )NzAssignment not supportedr   r)   s      r   r*   z_SupportedQEnginesProp.__set__-   s    5666r   )r   N)r+   r,   r-   listr.   r%   r*   r/   r   r   r1   r1   (   sG        9tCy 9 9 9 97 7 7 7 7 7r   r1   c                   L     e Zd Z fdZd Z e            Z e            Z xZ	S )QuantizedEnginec                 X    t                                          |           || _        d S r   )super__init__m)r"   r@   name	__class__s      r   r?   zQuantizedEngine.__init__2   s&    r   c                 6    | j                             |          S r   )r@   __getattribute__)r"   attrs     r   __getattr__zQuantizedEngine.__getattr__6   s    v&&t,,,r   )
r+   r,   r-   r?   rF   r   enginer1   supported_engines__classcell__)rB   s   @r   r<   r<   1   s[            - - - \^^F..00r   r<   rG   rH   )systypesr   r.   intr   r   r   r1   
ModuleTyper<   modulesr+   __annotations__r:   r/   r   r   <module>rP      s#   




  S S    $2c 2c 2 2 2 2
4 4 4 4 4 4 4 47 7 7 7 7 7 7 7	1 	1 	1 	1 	1e& 	1 	1 	1 (H(=xHHH 9     r   