Fixes according to the issue oneapi-src#1093

KulikovNikita · Jan 5, 2023 · 88bdc89 · 88bdc89
1 parent c423627
commit 88bdc89
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cpp/daal/src/algorithms/k_nearest_neighbors/bf_knn_impl.i b/cpp/daal/src/algorithms/k_nearest_neighbors/bf_knn_impl.i
@@ -106,13 +106,13 @@ public:
  dist->init();
 
  const size_t outBlockSize = 128;
- const size_t inBlockSize = 128;
+ const size_t inBlockSize = 256;
  const size_t nOuterBlocks = nTest / outBlockSize + !!(nTest % outBlockSize);
 
  TlsMem<FPType, cpu> tlsDistances(inBlockSize * outBlockSize);
  TlsMem<int, cpu> tlsIdx(outBlockSize);
- TlsMem<FPType, cpu> tlsKDistances(inBlockSize * k);
- TlsMem<int, cpu> tlsKIndexes(inBlockSize * k);
+ TlsMem<FPType, cpu> tlsKDistances(outBlockSize * k);
+ TlsMem<int, cpu> tlsKIndexes(outBlockSize * k);
  TlsMem<FPType, cpu> tlsVoting(nClasses);
 
  SafeStatus safeStat;