v2.3.1

版本发布时间: 2024-06-06 03:16:07

pytorch/pytorch最新发布版本:v2.5.1(2024-10-30 01:58:24)

This release is meant to fix the following issues (regressions / silent correctness):

Fix format_utils executable, which was causing it to run as a no-op (https://github.com/pytorch/pytorch/pull/123407)
Fix regression with device_mesh in 2.3.0 during initialization causing memory spikes (https://github.com/pytorch/pytorch/pull/124780)
Fix crash of FSDP + DTensor with ShardingStrategy.SHARD_GRAD_OP (https://github.com/pytorch/pytorch/pull/123617)
Fix failure with distributed checkpointing + FSDP if at least 1 forward/backward pass has not been run. (https://github.com/pytorch/pytorch/pull/121544) (https://github.com/pytorch/pytorch/pull/127069)
Fix error with distributed checkpointing + FSDP, and with use_orig_params = False and activation checkpointing (https://github.com/pytorch/pytorch/pull/124698) (https://github.com/pytorch/pytorch/pull/126935)
Fix set_model_state_dict errors on compiled module with non-persistent buffer with distributed checkpointing (https://github.com/pytorch/pytorch/pull/125336) (https://github.com/pytorch/pytorch/pull/125337)

Fix UTF-8 encoding on Windows .pyi files (https://github.com/pytorch/pytorch/pull/124932)
Fix import torch failure when wheel is installed for a single user on Windows(https://github.com/pytorch/pytorch/pull/125684)
Fix compatibility with torchdata 0.7.1 (https://github.com/pytorch/pytorch/pull/122616)
Fix aarch64 docker publishing to https://ghcr.io (https://github.com/pytorch/pytorch/pull/125617)
Fix performance regression an aarch64 linux (https://github.com/pytorch/builder/pull/1803)

Fix DeepSpeed transformer extension build on ROCm (https://github.com/pytorch/pytorch/pull/121030)
Fix kernel crash on tensor.dtype.to_complex() after ~100 calls in ipython kernel (https://github.com/pytorch/pytorch/pull/125154)

Release tracker https://github.com/pytorch/pytorch/issues/125425 contains all relevant pull requests related to this release as well as links to related issues.