Replaced std::min with idx_reduction_op_

ndgrigorian · ndgrigorian · commit 1217550f63e5 · 2023-09-26T20:48:59.000-07:00
diff --git a/dpctl/tensor/libtensor/include/kernels/reductions.hpp b/dpctl/tensor/libtensor/include/kernels/reductions.hpp
@@ -1672,11 +1672,12 @@ struct SearchReduction
                 argT val = inp_[inp_offset];
                 if (val == local_red_val) {
                     if constexpr (!First) {
-                        local_idx = std::min(local_idx, inds_[inp_offset]);
+                        local_idx =
+                            idx_reduction_op_(local_idx, inds_[inp_offset]);
                     }
                     else {
-                        local_idx = std::min(local_idx,
-                                             static_cast<outT>(arg_reduce_gid));
+                        local_idx = idx_reduction_op_(
+                            local_idx, static_cast<outT>(arg_reduce_gid));
                     }
                 }
                 else {
@@ -1827,11 +1828,12 @@ struct CustomSearchReduction
                 argT val = inp_[inp_offset];
                 if (val == local_red_val) {
                     if constexpr (!First) {
-                        local_idx = std::min(local_idx, inds_[inp_offset]);
+                        local_idx =
+                            idx_reduction_op_(local_idx, inds_[inp_offset]);
                     }
                     else {
-                        local_idx = std::min(local_idx,
-                                             static_cast<outT>(arg_reduce_gid));
+                        local_idx = idx_reduction_op_(
+                            local_idx, static_cast<outT>(arg_reduce_gid));
                     }
                 }
                 else {

Original file line number	Diff line number	Diff line change
`@@ -1672,11 +1672,12 @@ struct SearchReduction`
`1672`	`1672`	`argT val = inp_[inp_offset];`
`1673`	`1673`	`if (val == local_red_val) {`
`1674`	`1674`	`if constexpr (!First) {`
`1675`		`- local_idx = std::min(local_idx, inds_[inp_offset]);`
	`1675`	`+ local_idx =`
	`1676`	`+ idx_reduction_op_(local_idx, inds_[inp_offset]);`
`1676`	`1677`	`}`
`1677`	`1678`	`else {`
`1678`		`- local_idx = std::min(local_idx,`
`1679`		`- static_cast<outT>(arg_reduce_gid));`
	`1679`	`+ local_idx = idx_reduction_op_(`
	`1680`	`+ local_idx, static_cast<outT>(arg_reduce_gid));`
`1680`	`1681`	`}`
`1681`	`1682`	`}`
`1682`	`1683`	`else {`
`@@ -1827,11 +1828,12 @@ struct CustomSearchReduction`
`1827`	`1828`	`argT val = inp_[inp_offset];`
`1828`	`1829`	`if (val == local_red_val) {`
`1829`	`1830`	`if constexpr (!First) {`
`1830`		`- local_idx = std::min(local_idx, inds_[inp_offset]);`
	`1831`	`+ local_idx =`
	`1832`	`+ idx_reduction_op_(local_idx, inds_[inp_offset]);`
`1831`	`1833`	`}`
`1832`	`1834`	`else {`
`1833`		`- local_idx = std::min(local_idx,`
`1834`		`- static_cast<outT>(arg_reduce_gid));`
	`1835`	`+ local_idx = idx_reduction_op_(`
	`1836`	`+ local_idx, static_cast<outT>(arg_reduce_gid));`
`1835`	`1837`	`}`
`1836`	`1838`	`}`
`1837`	`1839`	`else {`