Fix for Delta(int, Index)

firedrakeproject · Jan 22, 2025 · e77ed6f · e77ed6f
1 parent fdff8da
commit e77ed6f
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/gem/gem.py b/gem/gem.py
@@ -985,6 +985,12 @@ def __new__(cls, i, j, dtype=None):
         if isinstance(i, Integral) and isinstance(j, Integral):
             return one if i == j else Zero()
 
+        if isinstance(i, Integral):
+            return Indexed(Identity(j.extent), (i, j))
+
+        if isinstance(j, Integral):
+            return Indexed(Identity(i.extent), (i, j))
+
         self = super(Delta, cls).__new__(cls)
         self.i = i
         self.j = j