Skip to content

Commit

Permalink
try to bump up torch upperbound in tests (openvinotoolkit#22757)
Browse files Browse the repository at this point in the history
### Details:
- recently pytorch2.2 was
[released](https://github.com/pytorch/pytorch/releases)

### Tickets:
 - *ticket-id*
  • Loading branch information
akuporos committed Feb 15, 2024
1 parent 525a53a commit b1c699a
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion tests/constraints.txt
Original file line number Diff line number Diff line change
Expand Up @@ -24,4 +24,4 @@ pytest-html==4.1.1
pytest-timeout==2.2.0
jax<=0.4.14
jaxlib<=0.4.14
torch>=1.13,<2.2
torch>=1.13,<2.3
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ def use_torch_export():
from torch.fx.experimental.proxy_tensor import make_fx

em = export(model, tuple(torch_inputs))
if version.parse(torch.__version__) >= version.parse("2.2"):
if version.parse(torch.__version__) >= version.parse("2.3"):
em = em.run_decompositions()
print(em.graph_module.code)

Expand Down

0 comments on commit b1c699a

Please sign in to comment.