Skip to content

Commit addb341

Browse files
Fixed blunder in work-item id to data_id computation
gid-lane_id is already a multiple of sg_size.
1 parent bff71fc commit addb341

File tree

1 file changed

+1
-2
lines changed
  • dpctl/tensor/libtensor/include/kernels/sorting

1 file changed

+1
-2
lines changed

dpctl/tensor/libtensor/include/kernels/sorting/topk.hpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,7 @@ sycl::event write_out_impl(sycl::queue &exec_q,
101101
const std::uint32_t lane_id = sg.get_local_id()[0];
102102
const std::uint32_t sg_size = sg.get_max_local_range()[0];
103103

104-
const std::size_t start_id =
105-
(gid - lane_id) * sg_size * n_wi + lane_id;
104+
const std::size_t start_id = (gid - lane_id) * n_wi + lane_id;
106105

107106
#pragma unroll
108107
for (std::uint32_t i = 0; i < n_wi; ++i) {

0 commit comments

Comments
 (0)