v2.2.1
版本发布时间: 2024-02-23 05:15:00
pytorch/pytorch最新发布版本:v2.5.1(2024-10-30 01:58:24)
This release is meant to fix the following issues (regressions / silent correctness):
- Fix missing OpenMP support on Apple Silicon binaries (https://github.com/pytorch/builder/pull/1697)
- Fix crash when mixing lazy and non-lazy tensors in one operation (https://github.com/pytorch/pytorch/pull/117653)
- Fix PyTorch performance regression on Linux aarch64 (https://github.com/pytorch/builder/pull/1696)
- Fix silent correctness in DTensor
_to_copy
operation (https://github.com/pytorch/pytorch/pull/116426) - Fix properly assigning
param.grad_fn
for next forward (https://github.com/pytorch/pytorch/pull/116792) - Ensure gradient clear out pending
AsyncCollectiveTensor
in FSDP Extension (https://github.com/pytorch/pytorch/pull/116122) - Fix processing unflatten tensor on compute stream in FSDP Extension (https://github.com/pytorch/pytorch/pull/116559)
- Fix FSDP
AssertionError
on tensor subclass when settingsync_module_states=True
(https://github.com/pytorch/pytorch/pull/117336) - Fix DCP state_dict cannot correctly find FQN when the leaf module is wrapped by FSDP (https://github.com/pytorch/pytorch/pull/115592)
- Fix OOM when when returning a AsyncCollectiveTensor by forcing
_gather_state_dict()
to be synchronous with respect to the mian stream. (https://github.com/pytorch/pytorch/pull/118197) (https://github.com/pytorch/pytorch/pull/119716) - Fix Windows runtime
torch.distributed.DistNetworkError
: [WinError 32] The process cannot access the file because it is being used by another process (https://github.com/pytorch/pytorch/pull/118860) - Update supported python versions in package description (https://github.com/pytorch/pytorch/pull/119743)
- Fix SIGILL crash during
import torch
on CPUs that do not support SSE4.1 (https://github.com/pytorch/pytorch/issues/116623) - Fix DCP RuntimeError in
get_state_dict
andset_state_dict
(https://github.com/pytorch/pytorch/pull/119573) - Fixes for HSDP + TP integration with device_mesh (https://github.com/pytorch/pytorch/pull/112435) (https://github.com/pytorch/pytorch/pull/118620) (https://github.com/pytorch/pytorch/pull/119064) (https://github.com/pytorch/pytorch/pull/118638) (https://github.com/pytorch/pytorch/pull/119481)
- Fix numerical error with
mixedmm
on NVIDIA V100 (https://github.com/pytorch/pytorch/pull/118591) - Fix RuntimeError when using SymInt input invariant when splitting graphs (https://github.com/pytorch/pytorch/pull/117406)
- Fix compile
DTensor.from_local
in trace_rule_look up (https://github.com/pytorch/pytorch/pull/119659) - Improve torch.compile integration with CUDA-11.8 binaries (https://github.com/pytorch/pytorch/pull/119750)
Release tracker https://github.com/pytorch/pytorch/issues/119295 contains all relevant pull requests related to this release as well as links to related issues.
1、 pytorch-v2.2.1.tar.gz 274.14MB