diff --git a/sumpy/tools.py b/sumpy/tools.py index bf57b0fb0c83021d662a59e774d5154d9554c0f1..4a5ab86a6c3b0c4bd64a3892e9512f63f15d8883 100644 --- a/sumpy/tools.py +++ b/sumpy/tools.py @@ -449,7 +449,8 @@ class MatrixBlockIndex(object): raise ValueError("CL `Array`s are not supported." "Use MatrixBlockIndex.get() and then view into matrices.") - return x[np.ix_(*self.block_indices)] + irow, icol = self.block_indices(i) + return x[np.ix_(irow, icol)] def block_take(self, x, i): """Retrieve a block from a linear representation of the matrix blocks.