diff --git a/torch_xla/csrc/aten_xla_type.cpp b/torch_xla/csrc/aten_xla_type.cpp index 48a400335d4..87461759409 100644 --- a/torch_xla/csrc/aten_xla_type.cpp +++ b/torch_xla/csrc/aten_xla_type.cpp @@ -636,9 +636,9 @@ at::Tensor AtenXlaType::as_strided(const at::Tensor& self, at::IntArrayRef size, XlaHelpers::I64Optional(storage_offset))); } -at::Tensor& AtenXlaType::as_strided_(at::Tensor& self, at::IntArrayRef size, - at::IntArrayRef stride, - c10::optional storage_offset) { +const at::Tensor& AtenXlaType::as_strided_( + const at::Tensor& self, at::IntArrayRef size, at::IntArrayRef stride, + c10::optional storage_offset) { XLA_FN_COUNTER("xla::"); XLATensor self_tensor = bridge::GetXlaTensor(self); auto xsize = XlaHelpers::I64List(size); diff --git a/torch_xla/csrc/aten_xla_type.h b/torch_xla/csrc/aten_xla_type.h index 79362ecf6cb..9bd781a418e 100644 --- a/torch_xla/csrc/aten_xla_type.h +++ b/torch_xla/csrc/aten_xla_type.h @@ -153,9 +153,10 @@ class AtenXlaType { at::IntArrayRef stride, c10::optional storage_offset); - static at::Tensor& as_strided_(at::Tensor& self, at::IntArrayRef size, - at::IntArrayRef stride, - c10::optional storage_offset); + static const at::Tensor& as_strided_(const at::Tensor& self, + at::IntArrayRef size, + at::IntArrayRef stride, + c10::optional storage_offset); static at::Tensor asin(const at::Tensor& self);