From 4c24729a07eaa82bee588399f618fd117bd2671c Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 14 Jan 2020 16:27:27 +0100
Subject: [PATCH] Enable Diff wrt to itself
---
pystencils/interpolation_astnodes.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/pystencils/interpolation_astnodes.py b/pystencils/interpolation_astnodes.py
index 71bb325b2..b67a06e71 100644
--- a/pystencils/interpolation_astnodes.py
+++ b/pystencils/interpolation_astnodes.py
@@ -207,6 +207,8 @@ class InterpolatorAccess(TypedSymbol):
for i in range(len(self.offsets))])
def diff(self, *symbols, **kwargs):
+ if symbols == (self,):
+ return 1
rtn = self._diff_interpolation_vec.T * sp.Matrix(self.offsets).diff(*symbols, **kwargs)
if rtn.shape == (1, 1):
rtn = rtn[0, 0]
--
GitLab