we don't need shuffle data if there are no queries
@@ -147,12 +147,14 @@ namespace NCatboostCuda {
const ui32 loadBalancingPermutationId = FromString<ui32>(GetEnv("CB_LOAD_BALANCE_PERMUTATION", "42"));
LearnDocPerDevicesSplit = new TDocParallelSplit(*DataProvider,
- GetPermutation(dataProvider,
- loadBalancingPermutationId));
+ DataProvider->HasQueries()
+ ? GetPermutation(dataProvider, loadBalancingPermutationId)
+ : GetPermutation(dataProvider, TDataPermutation::IdentityPermutationId()));
if (TestDataProvider) {
TestDocPerDevicesSplit = new TDocParallelSplit(*TestDataProvider,
- GetPermutation(*TestDataProvider,
- loadBalancingPermutationId));
+ DataProvider->HasQueries()
+ ? GetPermutation(*TestDataProvider, loadBalancingPermutationId)
+ : GetPermutation(*TestDataProvider, TDataPermutation::IdentityPermutationId()));
}
}
