fix

Author: Le-soleile

test/TensorTest.cpp

@@ -1,4 +1,6 @@
 #include <ATen/ATen.h>
+#include <ATen/core/Tensor.h>
+#include <ATen/ops/ones.h>
 #include <gtest/gtest.h>
 #include <torch/all.h>
 
@@ -272,21 +274,22 @@ TEST_F(TensorTest, PinMemoryResult) {
   }
   file << std::to_string(gpu_pin_ok) << " ";
   file.saveFile();
-// 测试 sym_size
-TEST_F(TensorTest, SymSize) {
-  // 获取符号化的单个维度大小
-  c10::SymInt sym_size_0 = tensor.sym_size(0);
-  c10::SymInt sym_size_1 = tensor.sym_size(1);
-  c10::SymInt sym_size_2 = tensor.sym_size(2);
-
-  // 验证符号化大小与实际大小一致
-  EXPECT_EQ(sym_size_0, 2);
-  EXPECT_EQ(sym_size_1, 3);
-  EXPECT_EQ(sym_size_2, 4);
-
-  // 测试负索引
-  c10::SymInt sym_size_neg1 = tensor.sym_size(-1);
-  EXPECT_EQ(sym_size_neg1, 4);
+  // 测试 sym_size
+  TEST_F(TensorTest, SymSize) {
+    // 获取符号化的单个维度大小
+    c10::SymInt sym_size_0 = tensor.sym_size(0);
+    c10::SymInt sym_size_1 = tensor.sym_size(1);
+    c10::SymInt sym_size_2 = tensor.sym_size(2);
+
+    // 验证符号化大小与实际大小一致
+    EXPECT_EQ(sym_size_0, 2);
+    EXPECT_EQ(sym_size_1, 3);
+    EXPECT_EQ(sym_size_2, 4);
+
+    // 测试负索引
+    c10::SymInt sym_size_neg1 = tensor.sym_size(-1);
+    EXPECT_EQ(sym_size_neg1, 4);
+  }
 }
 
 // 测试 sym_stride