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

add hip.h

Browse Source
This commit is contained in:
amdsc21
2023-05-20 01:25:33 +02:00
parent 7663d47383
commit b22644fc10
44 changed files with 249 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -7,7 +7,11 @@
#include "../helpers.h"
#include <thrust/device_vector.h>
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/data/device_adapter.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/data/device_adapter.hip.h"
#endif
#include "test_array_interface.h"
using namespace xgboost; // NOLINT

5
tests/cpp/data/test_iterative_dmatrix.cu
View File

@@ -3,8 +3,13 @@
*/
#include <gtest/gtest.h>
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/data/device_adapter.cuh"
#include "../../../src/data/ellpack_page.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/data/device_adapter.hip.h"
#include "../../../src/data/ellpack_page.hip.h"
#endif
#include "../../../src/data/iterative_dmatrix.h"
#include "../../../src/tree/param.h" // TrainParam
#include "../helpers.h"

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

@@ -7,7 +7,11 @@
#include <any> // for any_cast
#include <memory>
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/data/device_adapter.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/data/device_adapter.hip.h"
#endif
#include "../../../src/data/proxy_dmatrix.h"
#include "../helpers.h"

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

@@ -4,7 +4,11 @@
#include <thrust/sequence.h>
#if defined(XGBOOST_USE_CUDA)
#include "../../../src/data/device_adapter.cuh"
#elif defined(XGBOOST_USE_HIP)
#include "../../../src/data/device_adapter.hip.h"
#endif
#include "../helpers.h"
#include "test_array_interface.h"
#include "../../../src/data/array_interface.h"