From c57d80126089dd4ce112a1ae2ed547eb9bcbf065 Mon Sep 17 00:00:00 2001 From: "Jae Hoon (Antonio) Kim" <17433012+antoniojkim@users.noreply.github.com> Date: Wed, 5 Oct 2022 10:23:48 -0400 Subject: [PATCH] Fix functionalize_aten_op calls for symint ops (#1459) * Fix functionalize_aten_op calls for symint ops * Update PyTorch version --- .../csrc/base_lazy_backend/mlir_native_functions.cpp | 6 +++--- pytorch-requirements.txt | 2 +- pytorch-version.txt | 2 +- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/python/torch_mlir/csrc/base_lazy_backend/mlir_native_functions.cpp b/python/torch_mlir/csrc/base_lazy_backend/mlir_native_functions.cpp index aaad8a0b7..207c77961 100644 --- a/python/torch_mlir/csrc/base_lazy_backend/mlir_native_functions.cpp +++ b/python/torch_mlir/csrc/base_lazy_backend/mlir_native_functions.cpp @@ -391,7 +391,7 @@ at::Tensor LazyNativeFunctions::new_empty_strided_symint( c10::optional device, c10::optional pin_memory) { return at::functionalization:: - functionalize_aten_op::call( + functionalize_aten_op_symint::call( self, size, stride, dtype, layout, device, pin_memory); } @@ -400,7 +400,7 @@ at::Tensor LazyNativeFunctions::narrow_copy_symint( int64_t dim, c10::SymInt start, c10::SymInt length) { - return at::functionalization::functionalize_aten_op::call(self, dim, start, length); } at::Tensor LazyNativeFunctions::pixel_shuffle( @@ -426,7 +426,7 @@ at::Tensor LazyNativeFunctions::slice_backward_symint( c10::SymInt start, c10::SymInt end, c10::SymInt step) { - return at::functionalization::functionalize_aten_op::call(grad_output, input_sizes, dim, start, end, step); } at::Tensor LazyNativeFunctions::diagonal_backward( diff --git a/pytorch-requirements.txt b/pytorch-requirements.txt index 9a87971eb..84230deb9 100644 --- a/pytorch-requirements.txt +++ b/pytorch-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html --pre -torch==1.13.0.dev20221003 +torch==1.13.0.dev20221004 diff --git a/pytorch-version.txt b/pytorch-version.txt index b7778b370..591132c78 100644 --- a/pytorch-version.txt +++ b/pytorch-version.txt @@ -1 +1 @@ -57d0543a3fbfb27d1b365d3515e2a2ba86b44878 +9f3d8fec5747fde5191618eb895fbec2d50edf93