Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Clean up MGPU C++ tests (#9430)

Browse Source
This commit is contained in:
Rong Ou
2023-08-01 23:31:18 -07:00
committed by GitHub
parent a9da2e244a
commit c2b85ab68a
28 changed files with 200 additions and 194 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/cpp/linear/test_json_io.h
View File

@@ -12,7 +12,7 @@
namespace xgboost {
inline void TestUpdaterJsonIO(std::string updater_str) {
Context ctx{MakeCUDACtx(GPUIDX)};
Context ctx{MakeCUDACtx(GetGPUId())};
Json config_0 {Object() };
{

4
tests/cpp/linear/test_linear.cc
View File

@@ -17,7 +17,7 @@ TEST(Linear, Shotgun) {
auto p_fmat = xgboost::RandomDataGenerator(kRows, kCols, 0).GenerateDMatrix();
auto ctx = MakeCUDACtx(GPUIDX);
auto ctx = MakeCUDACtx(GetGPUId());
LearnerModelParam mparam{MakeMP(kCols, .5, 1)};
{
@@ -49,7 +49,7 @@ TEST(Linear, coordinate) {
auto p_fmat = xgboost::RandomDataGenerator(kRows, kCols, 0).GenerateDMatrix();
auto ctx = MakeCUDACtx(GPUIDX);
auto ctx = MakeCUDACtx(GetGPUId());
LearnerModelParam mparam{MakeMP(kCols, .5, 1)};
auto updater = std::unique_ptr<xgboost::LinearUpdater>(