FIX: Kmeans timeouts issue on GPU for sparse inputs (#2906)

oneapi-src · Sep 19, 2024 · 9b315cf · 9b315cf
1 parent b6b6a65
commit 9b315cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/oneapi/dal/algo/kmeans/backend/gpu/kernels_csr_impl.hpp b/cpp/oneapi/dal/algo/kmeans/backend/gpu/kernels_csr_impl.hpp
@@ -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