From 28bb891a46d7c5169fedcb43c3bb7a86e543e775 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 11 Nov 2020 20:49:13 -0600
Subject: [PATCH] Fix dagrt-fusion for dofarray-not-ndarray-subclass
---
examples/dagrt-fusion.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/examples/dagrt-fusion.py b/examples/dagrt-fusion.py
index 2a285ad5..3cd6b71f 100755
--- a/examples/dagrt-fusion.py
+++ b/examples/dagrt-fusion.py
@@ -107,6 +107,8 @@ def dof_array_nbytes(ary: np.ndarray):
return sum(
dof_array_nbytes(ary[idx])
for idx in np.ndindex(ary.shape))
+ elif isinstance(ary, DOFArray):
+ return sum(dof_array_nbytes(ary_i) for ary_i in ary)
else:
return ary.nbytes
--
GitLab