llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

build: manually update PyTorch version (#3568) 

Set PyTorch and TorchVision version to nightly release 2024-08-04.

Signed-Off By: Vivek Khandelwal <vivekkhandelwal1424@gmail.com>
pull/3601/head
Vivek Khandelwal 2024-08-06 21:36:39 +05:30 committed by GitHub
parent b1a232222f
commit 78d0fa8998
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/torch_mlir/extras/fx_importer.py
View File

@ -265,6 +265,8 @@ PY_BUILTIN_TO_TORCH_OP = {
"ge": torch.ops.aten.ge, "ge": torch.ops.aten.ge,
"ne": torch.ops.aten.ne, "ne": torch.ops.aten.ne,
"gt": torch.ops.aten.gt, "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", # torch with cuda has a __version__ that looks like "2.1.0+cu113",

2
pytorch-hash.txt
View File

@ -1 +1 @@
5147aeb49a367b4a338d446b604be4b65eed83f5 d6ea1eb2bc8ba770fd5a689a30e234837df27384

2
pytorch-requirements.txt
View File

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/nightly/cpu/torch/ -f https://download.pytorch.org/whl/nightly/cpu/torch/
--pre --pre
torch==2.5.0.dev20240718 torch==2.5.0.dev20240804

2
torchvision-requirements.txt
View File

@ -1,3 +1,3 @@
-f https://download.pytorch.org/whl/nightly/cpu/torchvision/ -f https://download.pytorch.org/whl/nightly/cpu/torchvision/
--pre --pre
torchvision==0.20.0.dev20240718 torchvision==0.20.0.dev20240804