From 1934c6c2ff612bc4ddd46ce5937ed46a793d95c2 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Tue, 5 Nov 2019 19:43:26 +0100
Subject: [PATCH] Add called function to FunctionCall.args
---
src/pystencils_autodiff/framework_integration/astnodes.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/pystencils_autodiff/framework_integration/astnodes.py b/src/pystencils_autodiff/framework_integration/astnodes.py
index fbb2a64..6dee37e 100644
--- a/src/pystencils_autodiff/framework_integration/astnodes.py
+++ b/src/pystencils_autodiff/framework_integration/astnodes.py
@@ -102,7 +102,7 @@ class FunctionCall(Node):
@property
def args(self):
- return [p.symbol for p in self.kernel_function.get_parameters()]
+ return [p.symbol for p in self.kernel_function.get_parameters()] + [self.kernel_function]
@property
def symbols_defined(self) -> Set[sp.Symbol]:
--
GitLab