Add print statement

oleksandr-pavlyk · oleksandr-pavlyk · commit b9915580dfd7 · 2024-10-20T12:59:16.000-05:00
diff --git a/dpctl/tensor/_manipulation_functions.py b/dpctl/tensor/_manipulation_functions.py
@@ -378,6 +378,7 @@ def roll(x, /, shift, *, axis=None):
     res = dpt.empty(
         x.shape, dtype=x.dtype, usm_type=x.usm_type, sycl_queue=exec_q
     )
+    print(shifts)
     dep_evs = _manager.submitted_events
     ht_e, roll_ev = ti._copy_usm_ndarray_for_roll_nd(
         src=x, dst=res, shifts=shifts, sycl_queue=exec_q, depends=dep_evs

Original file line number	Diff line number	Diff line change
`@@ -378,6 +378,7 @@ def roll(x, /, shift, *, axis=None):`
`378`	`378`	`res = dpt.empty(`
`379`	`379`	`x.shape, dtype=x.dtype, usm_type=x.usm_type, sycl_queue=exec_q`
`380`	`380`	`)`
	`381`	`+ print(shifts)`
`381`	`382`	`dep_evs = _manager.submitted_events`
`382`	`383`	`ht_e, roll_ev = ti._copy_usm_ndarray_for_roll_nd(`
`383`	`384`	`src=x, dst=res, shifts=shifts, sycl_queue=exec_q, depends=dep_evs`