addderss Jack's feedback

Author: Krovatkin

test/cpp/test_aten_xla_tensor.cpp

@@ -285,6 +285,20 @@ TEST_F(AtenXlaTensorTest, TestSymSizes) {
     torch::Tensor xla_a = CopyToDevice(a, device);
     ASSERT_EQ(a.sym_sizes().at(0).expect_int(), 2);
     ASSERT_EQ(a.sym_sizes().at(0).is_symbolic(), false);
+
+    torch::Tensor b = torch::tensor({{0.0, 1.0}, {0.0, 0.0}},
+                                    torch::TensorOptions(torch::kFloat));
+    torch::Tensor xla_b = CopyToDevice(b, device);
+    xla_b = torch::nonzero(xla_b);
+    auto s0 = xla_b.sym_sizes().at(0);
+    ASSERT_EQ(s0.is_symbolic(), true);
+    auto sininode =
+        dynamic_cast<XLASymIntNodeImpl*>(s0.toSymIntNodeImpl().get());
+    auto snode =
+        std::dynamic_pointer_cast<torch_xla::SizeNode>(sininode->node());
+    ASSERT_TRUE(snode);
+    ASSERT_EQ(snode->getStaticValue(), 4);
+    ASSERT_EQ(snode->getDynamicValue(), 1);
   });
 }
 

torch_xla/csrc/tensor_impl.h

@@ -3,13 +3,8 @@
 #include <ATen/Tensor.h>
 #include <c10/core/Storage.h>
 #include <c10/core/TensorImpl.h>
-#include <torch/csrc/lazy/backend/backend_interface.h>
-#include <torch/csrc/lazy/core/config.h>
-#include <torch/csrc/lazy/core/ir.h>
-#include <torch/csrc/lazy/core/trie.h>
 
 #include "torch_xla/csrc/tensor.h"
-#include "torch_xla/csrc/xla_backend_impl.h"
 
 namespace torch_xla {