issue/506 add test case for TensorMetaData Desconstructor

Signed-off-by: Ceng23333 <441651826@qq.com>

Author: Ceng23333

src/infinicore-test/README.md

@@ -64,7 +64,7 @@ xmake build infinicore-test
 ./infinicore-test --moore
 ./infinicore-test --iluvatar
 ./infinicore-test --kunlun
-./infinicore-test --sugon
+./infinicore-test --hygon
 ```
 
 ### Customize Test Parameters

src/infinicore-test/main.cc

@@ -1,4 +1,5 @@
 #include "memory_test.h"
+#include "test_tensor_destructor.h"
 #include <iostream>
 #include <memory>
 #include <spdlog/spdlog.h>
@@ -36,7 +37,7 @@ void printUsage() {
               << "  moore" << std::endl
               << "  iluvatar" << std::endl
               << "  kunlun" << std::endl
-              << "  sugon" << std::endl
+              << "  hygon" << std::endl
               << std::endl
               << "Available tests:" << std::endl
               << "  basic       - Basic memory allocation and deallocation tests" << std::endl
@@ -74,8 +75,8 @@ ParsedArgs parseArgs(int argc, char *argv[]) {
             args.device_type = INFINI_DEVICE_ILUVATAR;
         } else if (arg == "--kunlun") {
             args.device_type = INFINI_DEVICE_KUNLUN;
-        } else if (arg == "--sugon") {
-            args.device_type = INFINI_DEVICE_SUGON;
+        } else if (arg == "--hygon") {
+            args.device_type = INFINI_DEVICE_HYGON;
         } else if (arg == "--test") {
             if (i + 1 >= argc) {
                 std::cerr << "Error: --test requires a test name" << std::endl;
@@ -164,6 +165,10 @@ int main(int argc, char *argv[]) {
             spdlog::debug("Adding BasicMemoryTest");
             runner.addTest(std::make_unique<infinicore::test::BasicMemoryTest>());
             spdlog::debug("BasicMemoryTest added successfully");
+
+            spdlog::debug("Adding TensorDestructorTest");
+            runner.addTest(std::make_unique<infinicore::test::TensorDestructorTest>());
+            spdlog::debug("TensorDestructorTest added successfully");
         }
 
         if (args.run_concurrency) {

src/infinicore-test/memory_test.cc

@@ -30,24 +30,35 @@ TestResult BasicMemoryTest::run() {
             spdlog::debug("BasicMemoryTest: Testing memory access");
             // Test memory access
             std::byte *data = memory->data();
-            spdlog::debug("BasicMemoryTest: Got memory data pointer");
+            spdlog::debug("BasicMemoryTest: Got memory data pointer: {}", static_cast<void *>(data));
             if (!data) {
                 std::cerr << "Memory data pointer is null" << std::endl;
                 return false;
             }
             spdlog::debug("BasicMemoryTest: Memory data pointer is valid");
 
-            spdlog::debug("BasicMemoryTest: Testing memory write/read");
-            // Test memory write/read
-            std::memset(data, 0xAB, 1024);
-            spdlog::debug("BasicMemoryTest: Memory memset completed");
-            for (size_t i = 0; i < 1024; ++i) {
-                if (data[i] != static_cast<std::byte>(0xAB)) {
-                    std::cerr << "Memory write/read test failed at index " << i << std::endl;
-                    return false;
+            // Check if this is GPU memory that can't be accessed directly
+            Device current_device = context::getDevice();
+            spdlog::debug("BasicMemoryTest: Current device type: {}", static_cast<int>(current_device.getType()));
+            spdlog::debug("BasicMemoryTest: Memory is pinned: {}", memory->is_pinned());
+
+            // For GPU memory, we shouldn't try to access it directly with memset
+            if (current_device.getType() != Device::Type::CPU) {
+                spdlog::debug("BasicMemoryTest: Skipping direct memory access for GPU device");
+                spdlog::debug("BasicMemoryTest: GPU memory access test completed (skipped)");
+            } else {
+                spdlog::debug("BasicMemoryTest: Testing memory write/read");
+                // Test memory write/read
+                std::memset(data, 0xAB, 1024);
+                spdlog::debug("BasicMemoryTest: Memory memset completed");
+                for (size_t i = 0; i < 1024; ++i) {
+                    if (data[i] != static_cast<std::byte>(0xAB)) {
+                        std::cerr << "Memory write/read test failed at index " << i << std::endl;
+                        return false;
+                    }
                 }
+                spdlog::debug("BasicMemoryTest: Memory write/read test completed");
             }
-            spdlog::debug("BasicMemoryTest: Memory write/read test completed");
 
             spdlog::debug("BasicMemoryTest: Testing pinned memory allocation");
             // Test pinned memory allocation
@@ -60,8 +71,8 @@ TestResult BasicMemoryTest::run() {
 
             spdlog::debug("BasicMemoryTest: Checking pinned memory properties");
             // For CPU devices, pinned memory falls back to regular memory, so it may not be marked as pinned
-            Device current_device = context::getDevice();
-            if (current_device.getType() != Device::Type::CPU && !pinned_memory->is_pinned()) {
+            Device pinned_device = context::getDevice();
+            if (pinned_device.getType() != Device::Type::CPU && !pinned_memory->is_pinned()) {
                 std::cerr << "Pinned memory not marked as pinned" << std::endl;
                 return false;
             }