fixed the custom call site

miladm · miladm · commit 535f35465217 · 2022-08-11T17:52:22.000Z
diff --git a/torch_xla/csrc/tensor_impl.cpp b/torch_xla/csrc/tensor_impl.cpp
@@ -123,7 +123,7 @@ at::IntArrayRef XLATensorImpl::sizes_custom() const {
 }
 
 c10::SymIntArrayRef XLATensorImpl::sym_sizes_custom() const {
-  auto sizes = sizes_default();
+  auto sizes = sizes_custom();
   return c10::SymIntArrayRef(reinterpret_cast<const c10::SymInt*>(sizes.data()),
                              sizes.size());
 }

Original file line number	Diff line number	Diff line change
`@@ -123,7 +123,7 @@ at::IntArrayRef XLATensorImpl::sizes_custom() const {`
`123`	`123`	`}`
`124`	`124`
`125`	`125`	`c10::SymIntArrayRef XLATensorImpl::sym_sizes_custom() const {`
`126`		`- auto sizes = sizes_default();`
	`126`	`+ auto sizes = sizes_custom();`
`127`	`127`	`return c10::SymIntArrayRef(reinterpret_cast<const c10::SymInt*>(sizes.data()),`
`128`	`128`	`sizes.size());`
`129`	`129`	`}`