U ʹj%@s$ddlmZejejdddZdS)N)gmreturnc s6dd|jjfddd|W5QRX|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. cSsd|S)Nimport pdb; pdb.set_trace() )rbodyrrZ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/fx/experimental/debug.py insert_pdbszset_trace..insert_pdbcsfddS)Ncsr|n|S)Nrr) cur_transformr rrs z-set_trace....rr r r rr szset_trace..)Zmake_transformer)graphZon_generate_codeZ recompile)rrr r set_traces   r)Ztorch.fxZfxZ GraphModulerrrrrs