set INCX by strides

Author: jessegrabowski

pytensor/link/numba/dispatch/linalg/dot/banded.py

@@ -69,7 +69,7 @@ def impl(A: np.ndarray, x: np.ndarray, kl: int, ku: int) -> np.ndarray:
         KU = val_to_int_ptr(ku)
 
         ALPHA = np.array(1.0, dtype=dtype)
-        INCX = val_to_int_ptr(1)
+        INCX = val_to_int_ptr(x.strides[0] // x.itemsize)
         BETA = np.array(0.0, dtype=dtype)
         Y = np.empty(m, dtype=dtype)
         INCY = val_to_int_ptr(1)

tests/link/numba/test_slinalg.py

@@ -743,3 +743,15 @@ def test_banded_dot():
         numba_mode=numba_inplace_mode,
         eval_obj_mode=False,
     )
+
+    # Test non-contiguous x input
+    x_val = rng.normal(size=(20,))[::2]
+
+    compare_numba_and_py(
+        [A, x],
+        output,
+        test_inputs=[A_val, x_val],
+        inplace=True,
+        numba_mode=numba_inplace_mode,
+        eval_obj_mode=False,
+    )