Fix export.cpp compilation with MSVC (#1737)

2025-09-18 01:50:16 +08:00 · 2024-12-29 23:56:30 +09:00
parent 4ba0c24a8f
commit 8ecdfb718b
2 changed files with 8 additions and 12 deletions
--- a/mlx/export.cpp
+++ b/mlx/export.cpp
@@ -9,17 +9,13 @@
 #define TOSTRING(x) STRINGIFY(x)

 // clang-format off
-#define SERIALIZE_PRIMITIVE(primitive, keys...)          \
-  {                                                      \
-    #primitive, {                                        \
-      [](Writer& os, const Primitive& p) {               \
-        serialize_primitive<primitive>(os, p);           \
-      },                                                 \
-      [](Reader& is, Stream s) {                         \
-        return deserialize_primitive<primitive>(is, s);  \
-      },                                                 \
-      {keys}                                             \
-    }                                                    \
+#define SERIALIZE_PRIMITIVE(primitive, ...)  \
+  {                                          \
+    #primitive, {                            \
+      serialize_primitive<primitive>,        \
+      deserialize_primitive<primitive>,      \
+      {__VA_ARGS__}                          \
+    }                                        \
  }
 // clang-format on