Support debugging console scripts from launch.json

[ringohoffman]

Related:

• StackOverflow: VS Code - Run Debug Configuration for a console_script in module mode
• StackOverflow: How can I debug Python console_script command line apps with the VSCode debugger?

I am trying to debug torchrun, which exists as a console script in pytorch. Today, in order to debug it I have to go searching thru the pytorch source code to find what I should run as a module, which is involved and probably not obvious to most users. I am also fortunate that the entrypoint to torchrun is the same function that is called from if __name__ == "__main__":. It would be extremely intuitive and useful if launch.json had first class support for debugging console scripts.

[Kin-Zhang]

In case, other users found this issue and still not sure how to run debug with torchrun, here is the command:

python -m debugpy --listen 5678 --wait-for-client -m torch.distributed.run --nproc_per_node 1 --nnodes 1 xxx.py

the .vscode/launch.json setting is:

{
    // kill -9 $(pgrep -f "python3 -m debugpy" | xargs echo)
    // python -m debugpy --listen 5678 --wait-for-client xxx.py
    // python -m debugpy --listen 5678 --wait-for-client -m torch.distributed.run --nproc_per_node 1 --nnodes 1 xxx.py
    "version": "0.2.0",
    "configurations": [
        {
            "name": "LDebug",
            "type": "python",
            "request": "attach",
            "connect": {
            "host": "localhost",
            "port": 5678
            }
        }
    ]
}