feat(python_ffi): 添加一个字符型的 trace 格式

Signed-off-by: YdrMaster <[email protected]>

Author: YdrMaster

src/09python_ffi/src/executor.cc

@@ -86,6 +86,60 @@ namespace refactor::python_ffi {
         os.write(ptr, size);
     }
 
+    static void writeText(std::ofstream os, char const *ptr, size_t size,
+                          DataType dataType, computation::Shape const &shape) {
+        if (shape.empty()) {
+            os << dataType.name() << "<>" << std::endl;
+            return;
+        } else {
+            auto iter = shape.begin();
+            os << dataType.name() << '<' << *iter++;
+            while (iter != shape.end()) { os << 'x' << *iter++; }
+            os << '>' << std::endl;
+        };
+
+#define CASE(T)                                        \
+    case DataType::T: {                                \
+        using T_ = primitive<DataType::T>::type;       \
+        auto ptr_ = reinterpret_cast<T_ const *>(ptr); \
+        for (auto i : range0_(size / sizeof(T_))) {    \
+            os << ptr_[i] << '\t';                     \
+        }                                              \
+    } break
+
+        switch (dataType) {
+            case DataType::U8: {
+                auto ptr_ = reinterpret_cast<uint8_t const *>(ptr);
+                for (auto i : range0_(size)) {
+                    os << static_cast<int>(ptr_[i]) << '\t';
+                }
+            } break;
+            case DataType::I8: {
+                auto ptr_ = reinterpret_cast<int8_t const *>(ptr);
+                for (auto i : range0_(size)) {
+                    os << static_cast<int>(ptr_[i]) << '\t';
+                }
+            } break;
+            case DataType::Bool: {
+                auto ptr_ = reinterpret_cast<bool const *>(ptr);
+                for (auto i : range0_(size)) {
+                    os << (ptr_[i] ? "true " : "false") << '\t';
+                }
+            } break;
+                CASE(F32);
+                CASE(U16);
+                CASE(I16);
+                CASE(I32);
+                CASE(I64);
+                CASE(F64);
+                CASE(U32);
+                CASE(U64);
+            default:
+                UNREACHABLE();
+                break;
+        }
+    }
+
     static void writeNpy(std::ofstream os, char const *ptr, size_t size,
                          DataType dataType, computation::Shape const &shape) {
         std::stringstream ss;
@@ -136,7 +190,6 @@ namespace refactor::python_ffi {
         fs::create_directories(path);
         ASSERT(fs::is_directory(path), "Failed to create \"{}\"", path.c_str());
 
-        auto const npy = format == "npy";
         size_t dataIdx = 0;
 
         auto const &graph = _graph.internal().contiguous();
@@ -164,9 +217,12 @@ namespace refactor::python_ffi {
                 auto file = path / fmt::format("data{:06}.{}", dataIdx++, format);
                 fs::remove(file);
                 std::ofstream os(file, std::ios::binary);
-                if (npy) {
+                if (format == "npy") {
                     writeNpy(std::move(os), buffer.data(), size,
                              edge.tensor->dataType, edge.tensor->shape);
+                } else if (format == "text") {
+                    writeText(std::move(os), buffer.data(), size,
+                              edge.tensor->dataType, edge.tensor->shape);
                 } else {
                     writeBin(std::move(os), buffer.data(), size);
                 }