a ==ic%@s$ddlmZejejdddZdS)N)gmreturncsJdd|jjfddd|Wdn1s<0Y|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. cSs dg|S)Nzimport pdb; pdb.set_trace() bodyrrl/home/droni/.local/share/virtualenvs/DPS-5Je3_V2c/lib/python3.9/site-packages/torch/fx/experimental/debug.py insert_pdbszset_trace..insert_pdbcsfddS)Ncsr|n|S)Nrr) cur_transformrrrs z-set_trace....rr rr rr szset_trace..)Zmake_transformerN)graphZon_generate_codeZ recompile)rrr r set_traces   &r)Ztorch.fxZfxZ GraphModulerrrrrs