Skip to content

Commit

Permalink
CPU build.
Browse files Browse the repository at this point in the history
  • Loading branch information
trivialfis committed Jul 31, 2024
1 parent 70ac561 commit bee5a5b
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions tests/cpp/common/test_stats.cc
Original file line number Diff line number Diff line change
Expand Up @@ -176,20 +176,21 @@ TEST(Stats, SampleMean) {
TestSampleMean(&ctx);
}

#if defined(XGBOOST_USE_CUDA)
TEST(Stats, GpuSampleMean) {
auto ctx = MakeCUDACtx(0);
TestSampleMean(&ctx);
}
#endif // defined(XGBOOST_USE_CUDA)

TEST(Stats, WeightedSampleMean) {
Context ctx;
TestWeightedSampleMean(&ctx);
}

#if defined(XGBOOST_USE_CUDA)
TEST(Stats, GpuSampleMean) {
auto ctx = MakeCUDACtx(0);
TestSampleMean(&ctx);
}

TEST(Stats, GpuWeightedSampleMean) {
auto ctx = MakeCUDACtx(0);
TestWeightedSampleMean(&ctx);
}
#endif // defined(XGBOOST_USE_CUDA)
} // namespace xgboost::common

0 comments on commit bee5a5b

Please sign in to comment.