We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bff71fc commit addb341Copy full SHA for addb341
dpctl/tensor/libtensor/include/kernels/sorting/topk.hpp
@@ -101,8 +101,7 @@ sycl::event write_out_impl(sycl::queue &exec_q,
101
const std::uint32_t lane_id = sg.get_local_id()[0];
102
const std::uint32_t sg_size = sg.get_max_local_range()[0];
103
104
- const std::size_t start_id =
105
- (gid - lane_id) * sg_size * n_wi + lane_id;
+ const std::size_t start_id = (gid - lane_id) * n_wi + lane_id;
106
107
#pragma unroll
108
for (std::uint32_t i = 0; i < n_wi; ++i) {
0 commit comments