
    }Yh+                     H    d dl mZ d dlmZ dgZdej        dej        fdZdS )    )SequenceN	set_tracegmreturnc                     dt           t                   dt          t                   fd| j                            fd          5  |                                  ddd           n# 1 swxY w Y   | S )a  
    Sets a breakpoint in `gm`'s generated python code. It drops into pdb when
    `gm` gets run.

    Args:
        gm: graph module to insert breakpoint. It is then recompiled for it to
            take effect.

    Returns:
        the `gm` with breakpoint inserted.
    bodyr   c                     dg| S )Nzimport pdb; pdb.set_trace()
 )r   s    m/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/fx/experimental/debug.py
insert_pdbzset_trace.<locals>.insert_pdb   s    /7$77    c                       fdS )Nc                 4     r |           n|           S )Nr
   )r   cur_transformr   s    r   <lambda>z-set_trace.<locals>.<lambda>.<locals>.<lambda>   s%    **M%S]]4%8%8%8tTT r   r
   )r   r   s   `r   r   zset_trace.<locals>.<lambda>   s    UUUUU r   )make_transformerN)r   strlistgraphon_generate_code	recompile)r   r   s    @r   r   r   	   s    8# 849 8 8 8 8 
	"	"
 
 
 
 
# 
 
   	               Is   A))A-0A-)collections.abcr   torch.fxfx__all__GraphModuler   r
   r   r   <module>r      sb    $ $ $ $ $ $       -". R^      r   