Skip to content

v2.7.0-rc9

v2.7.0-rc9: Gracefully handle optree less than minimum version (#150977)
@pytorchbot pytorchbot tagged this 10 Apr 14:39 
Gracefully handle optree less than minimum version (#150956)

Summary:
- We are saying the minimum version of pytree that PyTorch can use is
  0.13.0
- If a user imports torch.utils._cxx_pytree, it will raise an
  ImportError if optree doesn't exist or exists and is less than the
  minimum version.

Fixes https://github.com/pytorch/pytorch/issues/150889. There are
actually two parts to that issue:
1. dtensor imports torch.utils._cxx_pytree, but the optree installed in
   the environment might be too old. Instead, raising ImportError in
   torch.utils._cxx_pytree solves the issue.
2. We emit an "optree too low version" warning. I've deleted the
   warning in favor of the more explicit ImportError.

Test Plan:
- code reading
Pull Request resolved: https://github.com/pytorch/pytorch/pull/150956
Approved by: https://github.com/albanD, https://github.com/atalman, https://github.com/XuehaiPan

(cherry picked from commit 061832bc7a6711daaaf2bca12c2140bd8dea7eb5)

Co-authored-by: rzou <zou3519@gmail.com>
Assets 2
Loading