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

Unrevert #1724 (#1737) 

* Unrevert #1724

* Update pytorch requirements.txt
pull/1680/head snapshot-20221221.694
Jae Hoon (Antonio) Kim 2022-12-20 11:17:21 -05:00 committed by GitHub
parent 66d7a412cb
commit 1d695239ff
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/torch_mlir/csrc/base_lazy_backend/mlir_native_functions.cpp
View File

@ -309,7 +309,7 @@ at::Tensor LazyNativeFunctions::empty_symint(
c10::optional<bool> pin_memory,
c10::optional<at::MemoryFormat> memory_format) {
// TODO: support this directly
auto size = c10::asIntArrayRefSlow(sym_size);
auto size = C10_AS_INTARRAYREF_SLOW(sym_size);
const auto device_type = torch::lazy::getBackend()->EagerFallbackDeviceType();
at::TensorOptions options = at::TensorOptions()
.device(c10::Device(device_type))

2
pytorch-hash.txt
View File

@ -1 +1 @@
87269961258692d20a59a8f54116a85aa441cdbc
8b450f69345d14fef8e8e3fbec7da6527820b9aa

4
pytorch-requirements.txt
View File

@ -1,4 +1,4 @@
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
--pre
torch==2.0.0.dev20221217
torchvision==0.15.0.dev20221217
torch==2.0.0.dev20221220
torchvision==0.15.0.dev20221220