Fix for Delta(int, Index)

firedrakeproject · Jan 22, 2025 · fdff8da · fdff8da
1 parent 27cf83d
commit fdff8da
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 12 deletions.
diff --git a/gem/gem.py b/gem/gem.py
@@ -982,19 +982,9 @@ def __new__(cls, i, j, dtype=None):
             return one
 
         # Fixed indices
-        if isinstance(i, int) and isinstance(j, int):
+        if isinstance(i, Integral) and isinstance(j, Integral):
             return one if i == j else Zero()
 
-        if isinstance(i, int):
-            expr = numpy.full((j.extent), Zero(), dtype=object)
-            expr[i] = one
-            return Indexed(ListTensor(expr), (j,))
-
-        if isinstance(j, int):
-            expr = numpy.full((i.extent), Zero(), dtype=object)
-            expr[j] = one
-            return Indexed(ListTensor(expr), (i,))
-
         self = super(Delta, cls).__new__(cls)
         self.i = i
         self.j = j

diff --git a/gem/optimise.py b/gem/optimise.py
@@ -623,7 +623,7 @@ def _replace_delta_delta(node, self):
         return Indexed(Identity(size), (i, j))
     else:
         def expression(index):
-            if isinstance(index, int):
+            if isinstance(index, Integral):
                 return Literal(index)
             elif isinstance(index, VariableIndex):
                 return index.expression