From f888eae3b414c544324a4fd0c78d850aa51e0c96 Mon Sep 17 00:00:00 2001 From: Alexandru Fikl Date: Fri, 22 Oct 2021 16:03:56 -0500 Subject: [PATCH] allow dir_vec to have offsets --- sumpy/kernel.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/sumpy/kernel.py b/sumpy/kernel.py index 859b2dbd..6d84c5aa 100644 --- a/sumpy/kernel.py +++ b/sumpy/kernel.py @@ -1158,7 +1158,8 @@ class DirectionalTargetDerivative(DirectionalDerivative): self.dir_vec_name, None, shape=(self.dim, "ntargets"), - dim_tags="sep,C"), + dim_tags="sep,C", + offset=lp.auto), ) ] + self.inner_kernel.get_source_args() @@ -1203,7 +1204,8 @@ class DirectionalSourceDerivative(DirectionalDerivative): self.dir_vec_name, None, shape=(self.dim, "nsources"), - dim_tags="sep,C"), + dim_tags="sep,C", + offset=lp.auto), ) ] + self.inner_kernel.get_source_args() -- GitLab