PiPPy
PiPPy copied to clipboard
Make RemoteInterpreter use the full implementation of `Interpreter.run`
https://github.com/jamesr66a/PiPPy/blob/527af1fd8123d35bd81b9fe304a8d0ed29c9fd8d/pippy/PipelineDriver.py#L565
I wrote run_until as a hack, it should probably be a copy-paste of Interpreter.run with some termination branch inside (or we should refactor Interpreter.run to make implementing like this easier
More logic: https://github.com/pytorch/pytorch/blob/80f6d2e9e66a0d133a24863e6338f4c3a493e2fc/torch/fx/interpreter.py#L129
Hi, is this still open? I am a first time good contributor