o
    hD                     @   s  d Z ddlZddlmZmZmZmZmZmZ ddl	m
Z
 edZe
dZe	ddeeef d	ed
eeef fddZe	dded d	ed
eeeef geeef f fddZ	ddeeeef  d	ed
eeeef eeeef geeef f f fddZdS )za
APIs related to torch.compile which lazily import torch._dynamo to avoid
circular dependencies.
    N)CallableLiteralOptionaloverloadTypeVarUnion)	ParamSpec_T_PTfn	recursivereturnc                 C      d S N r   r   r   r   b/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/_compile.py_disable_dynamo      r   c                 C   r   r   r   r   r   r   r   r      r   c                    sD    durt  dtjdtjdtf fdd}|S t jtdS )au  
    This API should be only used inside torch, external users should still use
    torch._dynamo.disable. The main goal of this API is to avoid circular
    imports issues that is common while using _dynamo.disable inside torch
    itself.

    This API avoids it by lazily importing torch._dynamo from the import time to
    the invocation of the decorated function.
    Nargskwargsr   c                     s>   t  dd }|d u rdd l}|j }| _|| i |S )N__dynamo_disabler   )getattrtorch._dynamo_dynamodisabler   )r   r   
disable_fntorchr   r   r   inner)   s   z_disable_dynamo.<locals>.inner)r   )	functoolswrapsr
   r   r   r	   partialr   )r   r   r   r   r   r   r      s
   ")T)NT)__doc__r   typingr   r   r   r   r   r   typing_extensionsr   r	   r
   boolr   r   r   r   r   <module>   s@     

,