Fixed WrapIndex class returning negative indices

ndgrigorian · ndgrigorian · commit 1e6794308b76 · 2023-03-02T09:21:47.000-08:00
diff --git a/dpctl/tensor/libtensor/include/kernels/integer_advanced_indexing.hpp b/dpctl/tensor/libtensor/include/kernels/integer_advanced_indexing.hpp
@@ -68,7 +68,7 @@ class WrapIndex
     void operator()(py::ssize_t max_item, py::ssize_t &ind) const
     {
         max_item = std::max<py::ssize_t>(max_item, 1);
-        ind = ind % max_item;
+        ind = (ind < 0) ? ind % max_item + max_item : ind % max_item;
         return;
     }
 };

Original file line number	Diff line number	Diff line change
`@@ -68,7 +68,7 @@ class WrapIndex`
`68`	`68`	`void operator()(py::ssize_t max_item, py::ssize_t &ind) const`
`69`	`69`	`{`
`70`	`70`	`max_item = std::max<py::ssize_t>(max_item, 1);`
`71`		`- ind = ind % max_item;`
	`71`	`+ ind = (ind < 0) ? ind % max_item + max_item : ind % max_item;`
`72`	`72`	`return;`
`73`	`73`	`}`
`74`	`74`	`};`