add TensorAccessor related tests (#25)

Author: youge325

test/TensorAccessorTest.cpp

@@ -0,0 +1,99 @@
+#include <ATen/ATen.h>
+#include <ATen/core/Tensor.h>
+#include <ATen/ops/ones.h>
+#include <gtest/gtest.h>
+#include <torch/all.h>
+
+#include <string>
+#include <vector>
+
+#include "../src/file_manager.h"
+
+extern paddle_api_test::ThreadSafeParam g_custom_param;
+
+namespace at {
+namespace test {
+
+using paddle_api_test::FileManerger;
+using paddle_api_test::ThreadSafeParam;
+
+class TensorAccessorTest : public ::testing::Test {
+ protected:
+  void SetUp() override {
+    std::vector<int64_t> shape = {2, 3, 4};
+    tensor = at::ones(shape, at::kFloat);
+  }
+
+  at::Tensor tensor;
+};
+
+// 测试 packed_accessor32
+TEST_F(TensorAccessorTest, PackedAccessor32) {
+  auto file_name = g_custom_param.get();
+  FileManerger file(file_name);
+  file.createFile();
+  auto accessor = tensor.packed_accessor32<float, 3>();
+  file << std::to_string(accessor.size(0)) << " ";
+  file << std::to_string(accessor.size(1)) << " ";
+  file << std::to_string(accessor.size(2)) << " ";
+  file << std::to_string(accessor[0][0][0]) << " ";
+  file << std::to_string(accessor[1][2][3]) << " ";
+  file.saveFile();
+}
+
+// 测试 packed_accessor64
+TEST_F(TensorAccessorTest, PackedAccessor64) {
+  auto file_name = g_custom_param.get();
+  FileManerger file(file_name);
+  file.createFile();
+  auto accessor = tensor.packed_accessor64<float, 3>();
+  file << std::to_string(accessor.size(0)) << " ";
+  file << std::to_string(accessor.size(1)) << " ";
+  file << std::to_string(accessor.size(2)) << " ";
+  file << std::to_string(accessor[0][0][0]) << " ";
+  file << std::to_string(accessor[1][2][3]) << " ";
+  file.saveFile();
+}
+
+// 测试 generic_packed_accessor
+TEST_F(TensorAccessorTest, GenericPackedAccessor) {
+  auto file_name = g_custom_param.get();
+  FileManerger file(file_name);
+  file.createFile();
+  auto accessor = tensor.generic_packed_accessor<float, 3>();
+  file << std::to_string(accessor.size(0)) << " ";
+  file << std::to_string(accessor.size(1)) << " ";
+  file << std::to_string(accessor.size(2)) << " ";
+  file << std::to_string(accessor[0][0][0]) << " ";
+  file << std::to_string(accessor[1][2][3]) << " ";
+  file.saveFile();
+}
+
+// 测试 is_non_overlapping_and_dense
+TEST_F(TensorAccessorTest, IsNonOverlappingAndDense) {
+  auto file_name = g_custom_param.get();
+  FileManerger file(file_name);
+  file.createFile();
+  file << std::to_string(tensor.is_non_overlapping_and_dense()) << " ";
+
+  // 测试非连续的tensor
+  at::Tensor transposed = tensor.transpose(0, 2);
+  file << std::to_string(transposed.is_non_overlapping_and_dense()) << " ";
+
+  // 测试连续化后的tensor
+  at::Tensor contiguous = transposed.contiguous();
+  file << std::to_string(contiguous.is_non_overlapping_and_dense()) << " ";
+  file.saveFile();
+}
+
+// 测试 has_names
+TEST_F(TensorAccessorTest, HasNames) {
+  auto file_name = g_custom_param.get();
+  FileManerger file(file_name);
+  file.createFile();
+  file << std::to_string(tensor.has_names()) << " ";
+  file.saveFile();
+}
+
+}  // namespace test
+}  // namespace at