When creating a single parition skip the output nodes, but process possible nodes after it. (#60370)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60370
When creating a single parition skip the output nodes, but process possible nodes after it.
Test Plan: Run all CI tests.
Reviewed By: jfix71
Differential Revision: D29265278
fbshipit-source-id: 2242009973a54498d8027cce5a294558a1206fdf