LogCumSumExp (#2069)

2025-12-16 01:49:05 +08:00 · 2025-04-13 11:27:29 +03:00
parent 7275ac7523
commit e9e268336b
15 changed files with 209 additions and 3 deletions
--- a/mlx/export.cpp
+++ b/mlx/export.cpp
@@ -298,7 +298,13 @@ struct PrimitiveFactory {
      SERIALIZE_PRIMITIVE(Reshape),
      SERIALIZE_PRIMITIVE(Reduce, "And", "Or", "Sum", "Prod", "Min", "Max"),
      SERIALIZE_PRIMITIVE(Round),
-      SERIALIZE_PRIMITIVE(Scan, "CumSum", "CumProd", "CumMin", "CumMax"),
+      SERIALIZE_PRIMITIVE(
+          Scan,
+          "CumSum",
+          "CumProd",
+          "CumMin",
+          "CumMax",
+          "CumLogaddexp"),
      SERIALIZE_PRIMITIVE(Scatter),
      SERIALIZE_PRIMITIVE(Select),
      SERIALIZE_PRIMITIVE(Sigmoid),