From 78d0fa8998c93514737f7075f4f2a3f4a9ff7b57 Mon Sep 17 00:00:00 2001 From: Vivek Khandelwal Date: Tue, 6 Aug 2024 21:36:39 +0530 Subject: [PATCH] build: manually update PyTorch version (#3568) Set PyTorch and TorchVision version to nightly release 2024-08-04. Signed-Off By: Vivek Khandelwal --- python/torch_mlir/extras/fx_importer.py | 2 ++ pytorch-hash.txt | 2 +- pytorch-requirements.txt | 2 +- torchvision-requirements.txt | 2 +- 4 files changed, 5 insertions(+), 3 deletions(-) diff --git a/python/torch_mlir/extras/fx_importer.py b/python/torch_mlir/extras/fx_importer.py index 91d81de01..3cb0d86aa 100644 --- a/python/torch_mlir/extras/fx_importer.py +++ b/python/torch_mlir/extras/fx_importer.py @@ -265,6 +265,8 @@ PY_BUILTIN_TO_TORCH_OP = { "ge": torch.ops.aten.ge, "ne": torch.ops.aten.ne, "gt": torch.ops.aten.gt, + "mod": torch.ops.aten.fmod, + "eq": torch.ops.aten.eq, } # torch with cuda has a __version__ that looks like "2.1.0+cu113", diff --git a/pytorch-hash.txt b/pytorch-hash.txt index d41426301..39eddc4a8 100644 --- a/pytorch-hash.txt +++ b/pytorch-hash.txt @@ -1 +1 @@ -5147aeb49a367b4a338d446b604be4b65eed83f5 +d6ea1eb2bc8ba770fd5a689a30e234837df27384 diff --git a/pytorch-requirements.txt b/pytorch-requirements.txt index 2dc08ff86..a17516b9b 100644 --- a/pytorch-requirements.txt +++ b/pytorch-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/nightly/cpu/torch/ --pre -torch==2.5.0.dev20240718 +torch==2.5.0.dev20240804 diff --git a/torchvision-requirements.txt b/torchvision-requirements.txt index 96bed200c..3b7e41b43 100644 --- a/torchvision-requirements.txt +++ b/torchvision-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/nightly/cpu/torchvision/ --pre -torchvision==0.20.0.dev20240718 +torchvision==0.20.0.dev20240804