diff --git a/arraycontext/fake_numpy.py b/arraycontext/fake_numpy.py index 04f66e0e52ccd0a383a51abe3da90282b538f82e..28be07ee3b3a9c32b55633262b8568e71eeb7343 100644 --- a/arraycontext/fake_numpy.py +++ b/arraycontext/fake_numpy.py @@ -189,9 +189,6 @@ class BaseFakeNumpyLinalgNamespace: for _, subary in serialize_container(ary)], ord=ord) - if len(ary.shape) != 1: - raise NotImplementedError("only vector norms are implemented") - if ary.size == 0: return 0 diff --git a/test/test_arraycontext.py b/test/test_arraycontext.py index b632e208d0af7edc79baccda5cdaf87a2354f94b..19b427e0257010644c1a7b9be8c2900e717f2164 100644 --- a/test/test_arraycontext.py +++ b/test/test_arraycontext.py @@ -748,6 +748,9 @@ def test_norm_ord_none(actx_factory, ndim): norm_a_ref = np.linalg.norm(a, ord=None) norm_a = actx.np.linalg.norm(actx.from_numpy(a), ord=None) + if not np.isscalar(norm_a): + norm_a = actx.to_numpy(norm_a) + np.testing.assert_allclose(norm_a, norm_a_ref)