Support shift tuple when axis=None for roll

Author: vlad-perevezentsev

dpctl/tensor/_manipulation_functions.py

@@ -426,6 +426,9 @@ def roll(X, shift, axis=None):
     if not isinstance(X, dpt.usm_ndarray):
         raise TypeError(f"Expected usm_ndarray type, got {type(X)}.")
     if axis is None:
+        # get the combined shift value for all axes
+        if type(shift) is tuple:
+            shift = sum(shift)
         res = dpt.empty(
             X.shape, dtype=X.dtype, usm_type=X.usm_type, sycl_queue=X.sycl_queue
         )

dpctl/tests/test_usm_ndarray_manipulation.py

@@ -590,6 +590,8 @@ def test_roll_empty():
     "data",
     [
         [2, None],
+        [(0, 1), None],
+        [(-1, 0), None],
         [-2, None],
         [2, 0],
         [-2, 0],
@@ -617,6 +619,8 @@ def test_roll_1d(data):
     "data",
     [
         [1, None],
+        [(2, 1), None],
+        [(-1, 2), None],
         [1, 0],
         [1, 1],
         [1, ()],