Skip to content

Commit 1c225e6

Browse files
committed
Git rid of ascontiguousarray in numba reshape
1 parent 72c6a81 commit 1c225e6

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

pytensor/link/numba/dispatch/basic.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -549,11 +549,7 @@ def reshape(x, shape):
549549

550550
@numba_njit
551551
def reshape(x, shape):
552-
# TODO: Use this until https://github.com/numba/numba/issues/7353 is closed.
553-
return np.reshape(
554-
np.ascontiguousarray(np.asarray(x)),
555-
numba_ndarray.to_fixed_tuple(shape, ndim),
556-
)
552+
return np.reshape(x, numba_ndarray.to_fixed_tuple(shape, ndim))
557553

558554
return reshape
559555

0 commit comments

Comments
 (0)