
    קg@                     6    d dl mZ dej        dej        fdZdS )    Ngmreturnc                     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.
    c                     dg| S )Nzimport pdb; pdb.set_trace()
 )bodys    W/var/www/html/ai-engine/env/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>   s1    
+8 MM$'''  r   r   )r   r
   s   `r	   r   zset_trace.<locals>.<lambda>   s"         r   )make_transformerN)graphon_generate_code	recompile)r   r
   s    @r	   	set_tracer      s    8 8 8 
	"	"
 
 
 
 
# 

 

   	               Is   AAA)torch.fxfxGraphModuler   r   r   r	   <module>r      sG         ". R^      r   