a
    ¢º”hw  ã                   @  sŠ   d dl mZ d dlZd dlmZ d dlmZ er<d dlmZ G dd„ dej	ƒZ
e
ƒ Zdd	œd
d„Zdd	œdd„Zeddddœdd„ƒZdS )é    )ÚannotationsN)Úcontextmanager)ÚTYPE_CHECKING)ÚIteratorc                   @  s&   e Zd ZU dZded< dZded< dS )ÚLocalsNzbool | NoneÚ!use_const_ref_for_mutable_tensorsÚuse_ilistref_for_tensor_lists)Ú__name__Ú
__module__Ú__qualname__r   Ú__annotations__r   © r   r   ú</var/www/auris/lib/python3.9/site-packages/torchgen/local.pyr      s   
r   Úbool)Úreturnc                   C  s   t jd usJ dƒ‚t jS )NzQneed to initialize local.use_const_ref_for_mutable_tensors with local.parametrize)Ú_localsr   r   r   r   r   r   !   s    ÿr   c                   C  s   t jd usJ dƒ‚t jS )NzMneed to initialize local.use_ilistref_for_tensor_lists with local.parametrize)r   r   r   r   r   r   r   )   s    ÿr   zIterator[None])r   r   r   c                 c  sB   t j}t j}z"| t _|t _d V  W |t _|t _n|t _|t _0 d S )N)r   r   r   )r   r   Z%old_use_const_ref_for_mutable_tensorsZ!old_use_ilistref_for_tensor_listsr   r   r   Úparametrize0   s    ÿþÿr   )Ú
__future__r   Ú	threadingÚ
contextlibr   Útypingr   Úcollections.abcr   Úlocalr   r   r   r   r   r   r   r   r   Ú<module>   s   