From 9d408bfa263b6ade3b36817ca6ced24d952a7d8a Mon Sep 17 00:00:00 2001
From: Rodrigo De Castro <rdcastro@meta.com>
Date: Fri, 30 Jan 2026 15:32:20 -0800
Subject: [PATCH 1/2] add spec for special_zeta.default

---
 facto/specdb/db.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/facto/specdb/db.py b/facto/specdb/db.py
index 6dda077..89236cf 100644
--- a/facto/specdb/db.py
+++ b/facto/specdb/db.py
@@ -5084,6 +5084,30 @@
             OutArg(ArgType.Tensor),
         ],
     ),
+    Spec(
+       op="special_zeta.default",  # (Tensor self, Tensor other) -> Tensor
+       inspec=[
+           InPosArg(
+               ArgType.Tensor,
+               name="self",
+               constraints=[
+                   cp.Dtype.In(lambda deps: [torch.float, torch.double]),
+               ],
+           ),
+           InPosArg(
+               ArgType.Tensor,
+               name="other",
+               deps=[0],
+               constraints=[
+                   cp.Dtype.Eq(lambda deps: deps[0].dtype),
+                   cp.Size.In(
+                       lambda deps, r, d: fn.broadcast_with(deps[0].shape, r, d)
+                   ),
+               ],
+           ),
+       ],
+       outspec=[OutArg(ArgType.Tensor)],
+    ),
     Spec(
         op="t_copy.default",  # (Tensor self) -> Tensor
         inspec=[

From e0b81a6c579f9ddf7715e5299f30a78bc2816a71 Mon Sep 17 00:00:00 2001
From: Rodrigo De Castro <rdcastro@meta.com>
Date: Fri, 6 Feb 2026 15:39:14 -0800
Subject: [PATCH 2/2] Fix lint formatting issues

---
 facto/specdb/db.py | 44 ++++++++++++++++++++++----------------------
 1 file changed, 22 insertions(+), 22 deletions(-)

diff --git a/facto/specdb/db.py b/facto/specdb/db.py
index 89236cf..9e0943f 100644
--- a/facto/specdb/db.py
+++ b/facto/specdb/db.py
@@ -5085,28 +5085,28 @@
         ],
     ),
     Spec(
-       op="special_zeta.default",  # (Tensor self, Tensor other) -> Tensor
-       inspec=[
-           InPosArg(
-               ArgType.Tensor,
-               name="self",
-               constraints=[
-                   cp.Dtype.In(lambda deps: [torch.float, torch.double]),
-               ],
-           ),
-           InPosArg(
-               ArgType.Tensor,
-               name="other",
-               deps=[0],
-               constraints=[
-                   cp.Dtype.Eq(lambda deps: deps[0].dtype),
-                   cp.Size.In(
-                       lambda deps, r, d: fn.broadcast_with(deps[0].shape, r, d)
-                   ),
-               ],
-           ),
-       ],
-       outspec=[OutArg(ArgType.Tensor)],
+        op="special_zeta.default",  # (Tensor self, Tensor other) -> Tensor
+        inspec=[
+            InPosArg(
+                ArgType.Tensor,
+                name="self",
+                constraints=[
+                    cp.Dtype.In(lambda deps: [torch.float, torch.double]),
+                ],
+            ),
+            InPosArg(
+                ArgType.Tensor,
+                name="other",
+                deps=[0],
+                constraints=[
+                    cp.Dtype.Eq(lambda deps: deps[0].dtype),
+                    cp.Size.In(
+                        lambda deps, r, d: fn.broadcast_with(deps[0].shape, r, d)
+                    ),
+                ],
+            ),
+        ],
+        outspec=[OutArg(ArgType.Tensor)],
     ),
     Spec(
         op="t_copy.default",  # (Tensor self) -> Tensor