From a96f3fed21941a6d5c1bf8d0ebc8d03d5dcc7995 Mon Sep 17 00:00:00 2001 From: Matthias Diener Date: Thu, 17 Jun 2021 16:16:03 -0500 Subject: [PATCH] check that all subary are cl.Arrays --- arraycontext/fake_numpy.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py index a117faf..ee714f2 100644 --- a/arraycontext/fake_numpy.py +++ b/arraycontext/fake_numpy.py @@ -190,7 +190,8 @@ class BaseFakeNumpyLinalgNamespace: # mimics numpy's norm computation return self.norm(_flatten_cl_array(ary), ord=2) - if _is_meshmode_dofarray(ary): + if (_is_meshmode_dofarray(ary) and all([_is_pyopencl_array(subary) + for _, subary in serialize_container(ary)])): from arraycontext.impl import _flatten_cl_array from warnings import warn -- GitLab