Skip to content

Commit

Permalink
Skip auc_test.large due to it should run on cuda 12.3
Browse files Browse the repository at this point in the history
  • Loading branch information
EmmaQiaoCh authored and minseokl committed Dec 29, 2023
1 parent 5679966 commit 2364f7d
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion test/utest/metrics/auc_test.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -310,7 +310,8 @@ class MPIEnvironment : public ::testing::Environment {
} // namespace

::testing::Environment* const mpi_env = ::testing::AddGlobalTestEnvironment(new MPIEnvironment);
TEST(auc_test, large) { metric_test<float>({0}, 12000, 5000 * 12000, gen_random<float>, 1); }
// auc_test.large is supported on cuda 12.3.
// TEST(auc_test, large) { metric_test<float>({0}, 12000, 5000 * 12000, gen_random<float>, 1); }
TEST(auc_test, fp32_1gpu) { metric_test<float>({0}, 10, 200, gen_random<float>, 1); }
TEST(auc_test, fp32_1gpu_odd) { metric_test<float>({0}, 10, 182, gen_random<float>, 1); }
TEST(auc_test, fp32_2gpu) { metric_test<float>({0, 1}, 10, 440, gen_random<float>, 1); }
Expand Down

0 comments on commit 2364f7d

Please sign in to comment.