U
    §Ê¹jw  ã                   @   s0   d dl Z d dlmZmZ ee jjdœdd„ZdS )é    N)Ú
OpOverloadÚOpOverloadPacket©ÚopÚgraphc                 C   s:   t | tƒrtd| › �ƒ‚t | tƒs&t‚tj | j|¡ d S )Nz;Must pass specific op overload, not overload packet, found )Ú
isinstancer   ÚAssertionErrorr   ÚtorchÚ_CZ&_jit_register_decomposition_for_schemaZ_schemar   © r   ú]/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/jit/_decomposition_utils.pyÚ_register_decomposition   s     ÿþr   )r	   Z
torch._opsr   r   r
   ZGraphr   r   r   r   r   Ú<module>   s   