Skip to content
Snippets Groups Projects
Commit 28bb891a authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Fix dagrt-fusion for dofarray-not-ndarray-subclass

parent 507fe163
No related branches found
No related tags found
No related merge requests found
...@@ -107,6 +107,8 @@ def dof_array_nbytes(ary: np.ndarray): ...@@ -107,6 +107,8 @@ def dof_array_nbytes(ary: np.ndarray):
return sum( return sum(
dof_array_nbytes(ary[idx]) dof_array_nbytes(ary[idx])
for idx in np.ndindex(ary.shape)) for idx in np.ndindex(ary.shape))
elif isinstance(ary, DOFArray):
return sum(dof_array_nbytes(ary_i) for ary_i in ary)
else: else:
return ary.nbytes return ary.nbytes
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment