From 714bb9ec3947d43b5b784324d3c2f01f90d9a044 Mon Sep 17 00:00:00 2001
From: "Gregory R. Lee" <gregory.lee@cchmc.org>
Date: Wed, 6 Dec 2017 17:38:14 -0500
Subject: [PATCH] fix: update signature of gpuarray.reshape to match the
 GPUArray method

---
 pycuda/gpuarray.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pycuda/gpuarray.py b/pycuda/gpuarray.py
index 99b63969..ba989521 100644
--- a/pycuda/gpuarray.py
+++ b/pycuda/gpuarray.py
@@ -667,7 +667,7 @@ class GPUArray(object):
 
         """
         return self._pow(other,new=False)
-        
+
 
     def reverse(self, stream=None):
         """Return this array in reversed order. The array is treated
@@ -1524,13 +1524,13 @@ def transpose(a, axes=None):
     return a.transpose(axes)
 
 
-def reshape(a, shape):
+def reshape(a, *shape, **kwargs):
     """Gives a new shape to an array without changing its data.
 
     .. versionadded:: 2015.2
     """
 
-    return a.reshape(shape)
+    return a.reshape(*shape, **kwargs)
 
 # }}}
 
-- 
GitLab