U ‰dÐã@sfddlZddlmZddlmZmZGdd„dejƒZeƒZe dœdd„Z ee edd œd d „ƒZ dS) éN)Úcontextmanager)ÚOptionalÚIteratorc@seZdZUdZeeed<dS)ÚLocalsNÚ!use_const_ref_for_mutable_tensors)Ú__name__Ú __module__Ú __qualname__rrÚboolÚ__annotations__©r r ú2/tmp/pip-unpacked-wheel-ua33x9lu/torchgen/local.pyrs r)ÚreturncCstjdk stdƒ‚tjS)NzQneed to initialize local.use_const_ref_for_mutable_tensors with local.parametrize)Ú_localsrÚAssertionErrorr r r r rs ÿr)rrccs$tj}z|t_dVW5|t_XdS)N)rr)rZ%old_use_const_ref_for_mutable_tensorsr r r Ú parametrize!s  ÿr) Ú threadingÚ contextlibrÚtypingrrÚlocalrrr rrr r r r Ús