hstack: fix allocator

inducer · Aug 9, 2022 · 3a96299 · 3a96299
1 parent c7718d2
commit 3a96299
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyopencl/array.py b/pyopencl/array.py
@@ -2749,7 +2749,8 @@ def hstack(arrays, queue=None):
                  "an instance of the type of arrays[0]",
                  stacklevel=2)
 
-    result = arrays[0].__class__(queue, lead_shape+(w,), arrays[0].dtype)
+    result = arrays[0].__class__(queue, lead_shape+(w,), arrays[0].dtype,
+                                 allocator=arrays[0].allocator)
     index = 0
     for ary in arrays:
         result[..., index:index+ary.shape[-1]] = ary