o
    ãÊi  ã                   @   s4   d dl Z dd„ Zdd„ Zdd„ Zdd	„ Zd
d„ ZdS )é    Nc                  O   ó   t jjj| i |¤ŽS )z±
    Wrapper around torch.ops.quantized_decomposed.quantize_per_channel_group to mitigate
    availability issue until it can be supplanted by new quantize_affine function.
    )ÚtorchÚopsÚquantized_decomposedÚquantize_per_channel_group©ÚargsÚkwargs© r
   úK/home/ubuntu/.local/lib/python3.10/site-packages/torchao/_executorch_ops.pyÚ8_quantized_decomposed_quantize_per_channel_group_wrapper   ó   r   c                  O   r   )zÀ
    Wrapper around torch.ops.quantized_decomposed.choose_qparams_per_token_asymmetric to mitigate
    availability issue until it can be supplanted by new choose_qparams_affine function.
    )r   r   r   Ú#choose_qparams_per_token_asymmetricr   r
   r
   r   ÚA_quantized_decomposed_choose_qparams_per_token_asymmetric_wrapper   s
   ÿÿr   c                  O   r   )z¹
    Wrapper around torch.ops.quantized_decomposed.dequantize_per_channel_group to mitigate
    availability issue until it can be supplanted by new choose_qparams_affine function.
    )r   r   r   Údequantize_per_channel_groupr   r
   r
   r   Ú:_quantized_decomposed_dequantize_per_channel_group_wrapper   r   r   c                  O   r   )z¯
    Wrapper around torch.ops.quantized_decomposed.quantize_per_token to mitigate
    availability issue until it can be supplanted by new choose_qparams_affine function.
    )r   r   r   Úquantize_per_tokenr   r
   r
   r   Ú0_quantized_decomposed_quantize_per_token_wrapper%   r   r   c                  O   r   )z±
    Wrapper around torch.ops.quantized_decomposed.dequantize_per_token to mitigate
    availability issue until it can be supplanted by new choose_qparams_affine function.
    )r   r   r   Údequantize_per_tokenr   r
   r
   r   Ú2_quantized_decomposed_dequantize_per_token_wrapper-   r   r   )r   r   r   r   r   r   r
   r
   r
   r   Ú<module>   s   
