tnt
tnt copied to clipboard
Updated in broadcast_str to use correct tensor size
Summary: Tensor size from source rank was [] before the fix while on other ranks tensor size was [1]. Broadcasting from [] to [1] should be illegal usage. The bug heppened to not cause any failures.
Differential Revision: D77901586
This pull request was exported from Phabricator. Differential Revision: D77901586