Skip to content

Commit

Permalink
Py3 fix
Browse files Browse the repository at this point in the history
  • Loading branch information
inducer committed Jul 11, 2015
1 parent 247f5b0 commit 3919a38
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pycuda/gpuarray.py
Expand Up @@ -783,8 +783,8 @@ def transpose(self, axes=None):
axes = range(self.ndim-1, -1, -1)
if len(axes) != len(self.shape):
raise ValueError("axes don't match array")
new_shape = [self.shape[axes[i]] for i in xrange(len(axes))]
new_strides = [self.strides[axes[i]] for i in xrange(len(axes))]
new_shape = [self.shape[axes[i]] for i in range(len(axes))]
new_strides = [self.strides[axes[i]] for i in range(len(axes))]
return GPUArray(shape=tuple(new_shape),
dtype=self.dtype,
allocator=self.allocator,
Expand Down

0 comments on commit 3919a38

Please sign in to comment.