chore: import upstream snapshot with attribution

2026-07-13 13:35:51 +08:00
commit c36a561cd8
2172 changed files with 455595 additions and 0 deletions
@@ -0,0 +1,52 @@
+/**
+ *  Copyright (c) 2017 by Contributors
+ * @file dgl/runtime/serializer.h
+ * @brief Serializer extension to support DGL data types
+ *  Include this file to enable serialization of DGLDataType, DGLContext
+ */
+#ifndef DGL_RUNTIME_SERIALIZER_H_
+#define DGL_RUNTIME_SERIALIZER_H_
+
+#include <dmlc/io.h>
+#include <dmlc/serializer.h>
+
+#include "c_runtime_api.h"
+#include "smart_ptr_serializer.h"
+
+namespace dmlc {
+namespace serializer {
+
+template <>
+struct Handler<DGLDataType> {
+  inline static void Write(Stream *strm, const DGLDataType &dtype) {
+    Handler<uint8_t>::Write(strm, dtype.code);
+    Handler<uint8_t>::Write(strm, dtype.bits);
+    Handler<uint16_t>::Write(strm, dtype.lanes);
+  }
+  inline static bool Read(Stream *strm, DGLDataType *dtype) {
+    if (!Handler<uint8_t>::Read(strm, &(dtype->code))) return false;
+    if (!Handler<uint8_t>::Read(strm, &(dtype->bits))) return false;
+    if (!Handler<uint16_t>::Read(strm, &(dtype->lanes))) return false;
+    return true;
+  }
+};
+
+template <>
+struct Handler<DGLContext> {
+  inline static void Write(Stream *strm, const DGLContext &ctx) {
+    int32_t device_type = static_cast<int32_t>(ctx.device_type);
+    Handler<int32_t>::Write(strm, device_type);
+    Handler<int32_t>::Write(strm, ctx.device_id);
+  }
+  inline static bool Read(Stream *strm, DGLContext *ctx) {
+    int32_t device_type = 0;
+    if (!Handler<int32_t>::Read(strm, &(device_type))) return false;
+    ctx->device_type = static_cast<DGLDeviceType>(device_type);
+    if (!Handler<int32_t>::Read(strm, &(ctx->device_id))) return false;
+    return true;
+  }
+};
+
+}  // namespace serializer
+}  // namespace dmlc
+#endif  // DGL_RUNTIME_SERIALIZER_H_