From c912885b4e277e2337383b25eb2494b5448d5035 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 6 Nov 2014 14:17:06 -0600
Subject: [PATCH] Allow join_fields for nD arrays

---
 pytools/obj_array.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pytools/obj_array.py b/pytools/obj_array.py
index f4a74e0..6dfd52b 100644
--- a/pytools/obj_array.py
+++ b/pytools/obj_array.py
@@ -85,7 +85,7 @@ def join_fields(*args):
             if log_shape(arg) == ():
                 res_list.append(arg)
             else:
-                res_list.extend(arg)
+                res_list.extend(arg.flat)
         else:
             res_list.append(arg)
 
-- 
GitLab