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

4
tests/cpp/data/test_array_interface.h
View File

@@ -6,7 +6,11 @@
#include <memory>
#include "../../../src/common/bitfield.h"
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/common/device_helpers.hip.h"
#endif
namespace xgboost {

4
tests/cpp/data/test_ellpack_page.cu
View File

@@ -7,7 +7,11 @@
#include "../../../src/common/categorical.h"
#include "../../../src/common/hist_util.h"
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/data/ellpack_page.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/data/ellpack_page.hip.h"
#endif
#include "../../../src/tree/param.h" // TrainParam
#include "../helpers.h"
#include "../histogram_helpers.h"

4
tests/cpp/data/test_ellpack_page_raw_format.cu
View File

@@ -4,7 +4,11 @@
#include <gtest/gtest.h>
#include <xgboost/data.h>
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/data/ellpack_page.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/data/ellpack_page.hip.h"
#endif
#include "../../../src/data/sparse_page_source.h"
#include "../../../src/tree/param.h" // TrainParam
#include "../filesystem.h" // dmlc::TemporaryDirectory

4
tests/cpp/data/test_metainfo.cu
View File

@@ -6,7 +6,11 @@
#include <xgboost/data.h>
#include <xgboost/json.h>
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/common/device_helpers.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/common/device_helpers.hip.h"
#endif
#include "test_array_interface.h"
#include "test_metainfo.h"

4
tests/cpp/data/test_sparse_page_dmatrix.cu
View File

@@ -4,7 +4,11 @@
#include <xgboost/data.h> // for DMatrix
#include "../../../src/common/compressed_iterator.h"
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/data/ellpack_page.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/data/ellpack_page.hip.h"
#endif
#include "../../../src/data/sparse_page_dmatrix.h"
#include "../../../src/tree/param.h" // TrainParam
#include "../filesystem.h" // dmlc::TemporaryDirectory