fixed check in dbg mode (#1718)

uxlfoundation · Jun 17, 2021 · 47df50d · 47df50d
1 parent eaa800f
commit 47df50d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/...orithms/dtrees/forest/classification/oneapi/df_classification_predict_dense_oneapi_impl.i b/...orithms/dtrees/forest/classification/oneapi/df_classification_predict_dense_oneapi_impl.i
@@ -376,7 +376,7 @@ services::Status PredictKernelOneAPI<algorithmFPType, method>::predictByTreesWei
         DAAL_ASSERT_UNIVERSAL_BUFFER(featureIndexList, int32_t, maxTreeSize * nTrees);
         DAAL_ASSERT_UNIVERSAL_BUFFER(leftOrClassTypeList, int32_t, maxTreeSize * nTrees);
         DAAL_ASSERT_UNIVERSAL_BUFFER(featureValueList, algorithmFPType, maxTreeSize * nTrees);
-        DAAL_ASSERT_UNIVERSAL_BUFFER(classProba, double, maxTreeSize * nTrees * _nClasses);
+        DAAL_ASSERT_UNIVERSAL_BUFFER(classProba, algorithmFPType, maxTreeSize * nTrees * _nClasses);
         DAAL_ASSERT_UNIVERSAL_BUFFER(obsClassHist, algorithmFPType, nRows * _nClasses * _nTreeGroups);
 
         for (size_t procTrees = 0; procTrees < nTrees; procTrees += _nTreeGroups)