mirror of https://github.com/llvm/torch-mlir
Replace asIntArrayRefSlow with macro (#1724)
* Replace asIntArrayRefSlow with macro * Update pytorch requirements.txtpull/1727/head
parent
211cf8fc36
commit
a2a93891ea
|
@ -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))
|
||||
|
|
|
@ -1 +1 @@
|
|||
971c769d42f384e46cc96a6f10a7bffcad6b79ae
|
||||
1902ca9d533da6d24b1fb9d0c345349d51ae26ad
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
-f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
|
||||
--pre
|
||||
torch==2.0.0.dev20221214
|
||||
torchvision==0.15.0.dev20221214
|
||||
torch==2.0.0.dev20221215
|
||||
torchvision==0.15.0.dev20221215
|
||||
|
|
Loading…
Reference in New Issue