[catboost] better exception message
[catboost] better exception message
([arc::mergeid] b43632c0-9340f90c-cdaf9c8-f)
#include "pairwise_optimization_subsets.h"
#include "blocked_histogram_helper.h"
+
#include <catboost/cuda/cuda_lib/cuda_buffer.h>
#include <catboost/cuda/cuda_lib/cuda_manager.h>
#include <catboost/cuda/cuda_lib/cuda_buffer_helpers/reduce_scatter.h>
#include <catboost/cuda/methods/pointwise_kernels.h>
#include <catboost/cuda/methods/histograms_helper.h>
+#include <util/stream/labeled.h>
+
namespace NCatboostCuda {
struct TBinaryFeatureSplitResults {
TStripeBuffer<TCBinFeature> BinFeatures;
@@ -43,7 +46,7 @@ namespace NCatboostCuda {
.SetReadSlice((TSlice(idx, (idx + 1))))
.Read(*matrixDiagonal);
- CB_ENSURE(result.size() == rowSize);
+ CB_ENSURE(result.size() == rowSize, LabeledOutput(result.size(), rowSize));
}
};
