Skip to content

Commit

Permalink
FIX: Kmeans timeouts issue on GPU for sparse inputs (#2906) (#2907)
Browse files Browse the repository at this point in the history
(cherry picked from commit 9b315cf)

Co-authored-by: Samir Nasibli <samir.nasibli@intel.com>
  • Loading branch information
mergify[bot] and samir-nasibli authored Sep 20, 2024
1 parent 00dacc9 commit 6f26c85
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -371,7 +371,7 @@ sycl::event handle_empty_clusters(const dal::backend::context_gpu& ctx,
auto event = queue.submit([&](sycl::handler& cgh) {
cgh.depends_on(deps);
cgh.parallel_for(range, [=](auto it) {
const auto local_id = it.get_local_id(1);
const auto local_id = it.get_local_id()[1];
for (std::int64_t cluster_id = rank; cluster_id < num_clusters;
cluster_id += rank_count) {
// no need to handle non-empty clusters
Expand Down

0 comments on commit 6f26c85

Please sign in to comment.