
    [Th+                     \    S SK Jr  S SKJr  S/rS\R
                  S\R
                  4S jrg)    )SequenceN	set_tracegmreturnc                    ^ S[         [           S[        [           4S jmU R                  R	                  U4S jS9   U R                  5         SSS5        U $ ! , (       d  f       U $ = f)z
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                     S/U Q$ )Nzimport pdb; pdb.set_trace()
 )r   s    S/var/www/auris/envauris/lib/python3.13/site-packages/torch/fx/experimental/debug.py
insert_pdbset_trace.<locals>.insert_pdb   s    /7$77    c                    >^  U U4S j$ )Nc                 :   > T" T(       a  T" U 5      5      $ U 5      $ )Nr
   )r   cur_transformr   s    r   <lambda>-set_trace.<locals>.<lambda>.<locals>.<lambda>   s    *M]4%8TtTr   r
   )r   r   s   `r   r   set_trace.<locals>.<lambda>   s    Ur   )make_transformerN)r   strlistgraphon_generate_code	recompile)r   r   s    @r   r   r   	   se    8# 849 8 
	"	"
 
# 
 	
 I
 
 Is   A
A()collections.abcr   torch.fxfx__all__GraphModuler   r
   r   r   <module>r       s/    $  -".. R^^ r   