trunk/31527a43dbf6adf4df2e6eb5c7b38094fec6b6f6: Widen FSDP param group stream type annotations to torch.Stream (#196563)
- PyTorch 近 90 天出现 786 次
- PyTorch 近 90 天第 776 次发版
- 上一次：同一天稍早 · trunk/6e700ca9014c0c7ba112f66d02a2e81b94cc1f20: Handle non-UTF-8 subprocess output in minifier isolate_fails (#196620)
发生了什么
Summary Widen the type annotations of _all_reduce_hook_stream and all_reduce_stream in FSDPParamGroup from torch.cuda.Stream to torch.Stream , allowing out-of-tree accelerator backends to pass their own Stream subtypes through FSDP's all-reduce hook path. Changes torch/distributed/fsdp/_fully_shard/_fsdp_param_group.py : L208 self._all_reduce_hook_stream: torch.cuda.Stream | None → torch.Stream | None torch/distribu…
摘要按规则整理自下方来源原文