fix to handle output dist when only some ranks have tables sharded onto them

[faran928]

Differential Revision: D80360860

[meta-codesync[bot]]

@faran928 has exported this pull request. If you are a Meta employee, you can view the originating Diff in D80360860.