pytorch

b4a6730c - [DataPipe] Refactor 'mux' to have buffer as an instance variable

[DataPipe] Refactor 'mux' to have buffer as an instance variable Pull Request resolved: https://github.com/pytorch/pytorch/pull/77775 Approved by: https://github.com/ejguan