From 2cebdaf562d205c9269e46418228246484f7f177 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andreas=20Kl=C3=B6ckner?= <inform@tiker.net>
Date: Thu, 1 Oct 2020 00:15:18 -0500
Subject: [PATCH] diff_batch matrix construction: use field dtype
---
grudge/execution.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/grudge/execution.py b/grudge/execution.py
index 3afe569b..546ef165 100644
--- a/grudge/execution.py
+++ b/grudge/execution.py
@@ -527,13 +527,13 @@ class ExecutionMapper(mappers.Evaluator,
# Cache operator
cache_key = "diff_batch", in_grp, out_grp, tuple(insn.operators),\
- field.dtype
+ field.entry_dtype
try:
matrices_ary_dev = self.bound_op.operator_data_cache[cache_key]
except KeyError:
matrices = repr_op.matrices(out_grp, in_grp)
matrices_ary = np.empty((
- noperators, out_grp.nunit_dofs, in_grp.nunit_dofs))
+ noperators, out_grp.nunit_dofs, in_grp.nunit_dofs), dtype=field.entry_dtype)
for i, op in enumerate(insn.operators):
matrices_ary[i] = matrices[op.rst_axis]
matrices_ary_dev = self.array_context.from_numpy(matrices_ary)
--
GitLab