File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -843,7 +843,7 @@ class SketchContainerImpl {
843
843
dmlc::OMPException exc;
844
844
size_t ridx_block_size = batch.Size () / n_threads_ + (batch.Size () % n_threads_ > 0 );
845
845
size_t min_ridx_block_size = 1024 ;
846
- if ((n_features < n_threads_) && (ridx_block_size > min_ridx_block_size)) {
846
+ if ((n_features < size_t ( n_threads_) ) && (ridx_block_size > min_ridx_block_size)) {
847
847
/* Row-wise parallelisation.
848
848
*/
849
849
std::vector<std::set<float >> categories_buff (n_threads_ * n_features);
@@ -897,7 +897,7 @@ class SketchContainerImpl {
897
897
size_t fidx_begin = tid * fidx_block_size;
898
898
size_t fidx_end = std::min (fidx_begin + fidx_block_size, n_features);
899
899
for (size_t ii = fidx_begin; ii < fidx_end; ++ii) {
900
- for (size_t th = 0 ; th < n_threads_; ++th) {
900
+ for (int th = 0 ; th < n_threads_; ++th) {
901
901
if (IsCat (feature_types_, ii)) {
902
902
categories_[ii].merge (categories_buff[th * n_features + ii]);
903
903
} else {
You can’t perform that action at this time.
0 commit comments