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

[JVM-packages] Prevent memory leak. (#10307)

Browse Source
This commit is contained in:
Jiaming Yuan 2024-05-22 13:47:59 +08:00 committed by GitHub
parent 6a43a4b9d3
commit 1b25d23583
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 463 additions and 329 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

776
jvm-packages/xgboost4j/src/native/xgboost4j.cpp
View File

File diff suppressed because it is too large Load Diff

6
tests/cpp/collective/test_worker.h
View File

@ -13,8 +13,10 @@
#include <vector> // for vector #include <vector> // for vector
#include "../../../src/collective/comm.h" #include "../../../src/collective/comm.h"
#include "../../../src/collective/tracker.h" // for GetHostAddress #include "../../../src/collective/communicator-inl.h" // for Init, Finalize
#include "../helpers.h" // for FileExists #include "../../../src/collective/tracker.h" // for GetHostAddress
#include "../../../src/common/common.h" // for AllVisibleGPUs
#include "../helpers.h" // for FileExists
#if defined(XGBOOST_USE_FEDERATED) #if defined(XGBOOST_USE_FEDERATED)
#include "../plugin/federated/test_worker.h" #include "../plugin/federated/test_worker.h"

2
tests/cpp/helpers.cc
View File

@ -12,9 +12,9 @@
#include <xgboost/objective.h> #include <xgboost/objective.h>
#include <algorithm> #include <algorithm>
#include <cinttypes>
#include <random> #include <random>
#include "../../src/collective/communicator-inl.h" // for GetRank
#include "../../src/data/adapter.h" #include "../../src/data/adapter.h"
#include "../../src/data/iterative_dmatrix.h" #include "../../src/data/iterative_dmatrix.h"
#include "../../src/data/simple_dmatrix.h" #include "../../src/data/simple_dmatrix.h"

8
tests/cpp/helpers.h
View File

@ -19,9 +19,11 @@
#include <string> #include <string>
#include <vector> #include <vector>
#include "../../src/collective/communicator-inl.h" #if defined(__CUDACC__)
#include "../../src/common/common.h" #include "../../src/collective/communicator-inl.h" // for GetRank
#include "../../src/common/threading_utils.h" #include "../../src/common/common.h" // for AllVisibleGPUs
#endif // defined(__CUDACC__)
#include "filesystem.h" // dmlc::TemporaryDirectory #include "filesystem.h" // dmlc::TemporaryDirectory
#include "xgboost/linalg.h" #include "xgboost/linalg.h"
#if !defined(_OPENMP) #if !defined(_OPENMP)