adding sum_sizes support

miladm · miladm · commit a79d062ec6bb · 2022-07-16T04:23:23.000Z
diff --git a/torch_xla/csrc/tensor_impl.cpp b/torch_xla/csrc/tensor_impl.cpp
@@ -126,7 +126,7 @@ at::IntArrayRef XLATensorImpl::sizes_custom() const {
 }
 
 c10::SymIntArrayRef XLATensorImpl::sym_sizes_custom() const {
-  if (false) {
+  if (true) {
     return c10::SymIntArrayRef(reinterpret_cast<const c10::SymInt*>(sym_sizes_.data()),
                                sym_sizes_.size());
   } else {

Original file line number	Diff line number	Diff line change
`@@ -126,7 +126,7 @@ at::IntArrayRef XLATensorImpl::sizes_custom() const {`
`126`	`126`	`}`
`127`	`127`
`128`	`128`	`c10::SymIntArrayRef XLATensorImpl::sym_sizes_custom() const {`
`129`		`- if (false) {`
	`129`	`+ if (true) {`
`130`	`130`	`return c10::SymIntArrayRef(reinterpret_cast<const c10::SymInt*>(sym_sizes_.data()),`
`131`	`131`	`sym_sizes_.size());`
`132`	`132`	`} else {`