File tree Expand file tree Collapse file tree 1 file changed +8
-6
lines changed
dpctl/tensor/libtensor/include/kernels Expand file tree Collapse file tree 1 file changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -1667,11 +1667,12 @@ struct SearchReduction
1667
1667
argT val = inp_[inp_offset];
1668
1668
if (val == local_red_val) {
1669
1669
if constexpr (!First) {
1670
- local_idx = std::min (local_idx, inds_[inp_offset]);
1670
+ local_idx =
1671
+ idx_reduction_op_ (local_idx, inds_[inp_offset]);
1671
1672
}
1672
1673
else {
1673
- local_idx = std::min (local_idx,
1674
- static_cast <outT>(arg_reduce_gid));
1674
+ local_idx = idx_reduction_op_ (
1675
+ local_idx, static_cast <outT>(arg_reduce_gid));
1675
1676
}
1676
1677
}
1677
1678
else {
@@ -1822,11 +1823,12 @@ struct CustomSearchReduction
1822
1823
argT val = inp_[inp_offset];
1823
1824
if (val == local_red_val) {
1824
1825
if constexpr (!First) {
1825
- local_idx = std::min (local_idx, inds_[inp_offset]);
1826
+ local_idx =
1827
+ idx_reduction_op_ (local_idx, inds_[inp_offset]);
1826
1828
}
1827
1829
else {
1828
- local_idx = std::min (local_idx,
1829
- static_cast <outT>(arg_reduce_gid));
1830
+ local_idx = idx_reduction_op_ (
1831
+ local_idx, static_cast <outT>(arg_reduce_gid));
1830
1832
}
1831
1833
}
1832
1834
else {
You can’t perform that action at this time.
0 commit comments