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

restore device helper

Browse Source
This commit is contained in:
amdsc21
2023-06-02 02:55:13 +02:00
parent 6ecd7903f2
commit 9ee1852d4e
13 changed files with 51 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/cpp/tree/gpu_hist/test_gradient_based_sampler.cu
View File

@@ -3,9 +3,13 @@
*/
#include <gtest/gtest.h>
#if defined(XGBOOST_USE_CUDA)
#include "../../../../src/data/ellpack_page.cuh"
#include "../../../../src/tree/gpu_hist/gradient_based_sampler.cuh"
#include "../../../../src/tree/param.h"
#elif defined(XGBOOST_USE_HIP)
#include "../../../../src/data/ellpack_page.hip.h"
#include "../../../../src/tree/gpu_hist/gradient_based_sampler.hip.h"
#endif
#include "../../../../src/tree/param.h" // TrainParam
#include "../../filesystem.h" // dmlc::TemporaryDirectory
#include "../../helpers.h"