From 9b315cf8124c700f79bdc3007aac103df3b831bf Mon Sep 17 00:00:00 2001 From: Samir Nasibli Date: Thu, 19 Sep 2024 23:32:38 +0200 Subject: [PATCH] FIX: `Kmeans` timeouts issue on GPU for sparse inputs (#2906) --- cpp/oneapi/dal/algo/kmeans/backend/gpu/kernels_csr_impl.hpp | 2 +- 1 file changed, 1 insertion(+), 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 index dba39bb9d01..228e9e01863 100644 --- 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