fixes

oneapi-src · Oct 1, 2024 · 8a01218 · 8a01218
1 parent a6d6e8b
commit 8a01218
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/...gorithms/k_nearest_neighbors/kdtree_knn_classification_predict_dense_default_batch_impl.i b/...gorithms/k_nearest_neighbors/kdtree_knn_classification_predict_dense_default_batch_impl.i
@@ -24,8 +24,8 @@
 #ifndef __KDTREE_KNN_CLASSIFICATION_PREDICT_DENSE_DEFAULT_BATCH_IMPL_I__
 #define __KDTREE_KNN_CLASSIFICATION_PREDICT_DENSE_DEFAULT_BATCH_IMPL_I__
 
-#include "src/threading/threading.h"
 #include "services/daal_defines.h"
+#include "src/threading/threading.h"
 #include "src/services/service_utils.h"
 #include "algorithms/algorithm.h"
 #include "services/daal_atomic_int.h"
@@ -208,6 +208,8 @@ Status KNNClassificationPredictKernel<algorithmFpType, defaultDense, cpu>::compu
 
     services::internal::TArrayScalable<algorithmFpType *, cpu> soa_arrays;
     bool isHomogenSOA = checkHomogenSOA<algorithmFpType, cpu>(data, soa_arrays);
+
+    services::Environment::getInstance()->setNumberOfThreads(1);
     daal::threader_for(blockCount, blockCount, [&](int iBlock) {
         Local * const local = localTLS.local();
         DAAL_CHECK_MALLOC_THR(local);
@@ -270,7 +272,7 @@ Status KNNClassificationPredictKernel<algorithmFpType, defaultDense, cpu>::compu
 
     status = safeStat.detach();
     if (!status) return status;
-
+    services::Environment::getInstance()->setNumberOfThreads(nThreads);
     localTLS.reduce([&](Local * ptr) -> void {
         if (ptr)
         {