Bump torch from 2.2.0 to 2.2.1

[dependabot[bot]]

Bumps torch from 2.2.0 to 2.2.1.

Release notes

Sourced from torch's releases.

PyTorch 2.2.1 Release, bug fix release

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

• Fix missing OpenMP support on Apple Silicon binaries (pytorch/builder#1697)
• Fix crash when mixing lazy and non-lazy tensors in one operation (pytorch/pytorch#117653)
• Fix PyTorch performance regression on Linux aarch64 (pytorch/builder#1696)
• Fix silent correctness in DTensor _to_copy operation (pytorch/pytorch#116426)
• Fix properly assigning param.grad_fn for next forward (pytorch/pytorch#116792)
• Ensure gradient clear out pending AsyncCollectiveTensor in FSDP Extension (pytorch/pytorch#116122)
• Fix processing unflatten tensor on compute stream in FSDP Extension (pytorch/pytorch#116559)
• Fix FSDP AssertionError on tensor subclass when setting sync_module_states=True (pytorch/pytorch#117336)
• Fix DCP state_dict cannot correctly find FQN when the leaf module is wrapped by FSDP (pytorch/pytorch#115592)
• Fix OOM when when returning a AsyncCollectiveTensor by forcing _gather_state_dict() to be synchronous with respect to the mian stream. (pytorch/pytorch#118197) (pytorch/pytorch#119716)
• Fix Windows runtime torch.distributed.DistNetworkError: [WinError 32] The process cannot access the file because it is being used by another process (pytorch/pytorch#118860)
• Update supported python versions in package description (pytorch/pytorch#119743)
• Fix SIGILL crash during import torch on CPUs that do not support SSE4.1 (pytorch/pytorch#116623)
• Fix DCP RuntimeError in get_state_dict and set_state_dict (pytorch/pytorch#119573)
• Fixes for HSDP + TP integration with device_mesh (pytorch/pytorch#112435) (pytorch/pytorch#118620) (pytorch/pytorch#119064) (pytorch/pytorch#118638) (pytorch/pytorch#119481)
• Fix numerical error with mixedmm on NVIDIA V100 (pytorch/pytorch#118591)
• Fix RuntimeError when using SymInt input invariant when splitting graphs (pytorch/pytorch#117406)
• Fix compile DTensor.from_local in trace_rule_look up (pytorch/pytorch#119659)
• Improve torch.compile integration with CUDA-11.8 binaries (pytorch/pytorch#119750)

Release tracker pytorch/pytorch#119295 contains all relevant pull requests related to this release as well as links to related issues.

Commits

• 6c8c5ad [RelEng] Define BUILD_BUNDLE_PTXAS (#119750) (#119988)
• f00f0ab fix compile DTensor.from_local in trace_rule_look up (#119659) (#119941)
• 077791b Revert "Update state_dict.py to propagate cpu offload (#117453)" (#119995)
• 3eaaeeb Update state_dict.py to propagate cpu offload (#117453) (#119916)
• 0aa3fd3 HSDP + TP integration bug fixes (#119819)
• eef51a6 [Inductor] Skip triton templates for mixedmm on SM70- (#118591) (#119894)
• 940358f [dtensor] fix dtensor _to_copy op for mix precision (#116426) (#119687)
• 24e4751 [state_dict] Calls wait() for the DTensor to_local() result (#118197) (#119692)
• dcaeed3 [DCP][state_dict] Fix the issue that get_state_dict/set_state_dict ig… (#119807)
• 4f882a5 Properly preserve SymInt input invariant when splitting graphs (#117406) (#11...
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR: - `@dependabot rebase` will rebase this PR - `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it - `@dependabot merge` will merge this PR after your CI passes on it - `@dependabot squash and merge` will squash and merge this PR after your CI passes on it - `@dependabot cancel merge` will cancel a previously requested merge and block automerging - `@dependabot reopen` will reopen this PR if it is closed - `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually - `@dependabot show ignore conditions` will show all of the ignore conditions of the specified dependency - `@dependabot ignore this major version` will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this minor version` will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself) - `@dependabot ignore this dependency` will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 98.79%. Comparing base (1b25a18) to head (1f8bbf5).

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #234 +/- ## ======================================= Coverage 98.79% 98.79% ======================================= Files 33 33 Lines 2749 2749 ======================================= Hits 2716 2716 Misses 33 33 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.

[dependabot[bot]]

Looks like torch is up-to-date now, so this is no longer needed.