complete test porting
This commit is contained in:
@@ -2,6 +2,9 @@
|
||||
* Copyright 2017-2023 XGBoost contributors
|
||||
*/
|
||||
#pragma once
|
||||
|
||||
#if defined(XGBOOST_USE_CUDA)
|
||||
|
||||
#include <thrust/binary_search.h> // thrust::upper_bound
|
||||
#include <thrust/device_malloc_allocator.h>
|
||||
#include <thrust/device_ptr.h>
|
||||
@@ -1381,3 +1384,7 @@ class LDGIterator {
|
||||
}
|
||||
};
|
||||
} // namespace dh
|
||||
|
||||
#elif defined(XGBOOST_USE_HIP)
|
||||
#include" device_helpers.hip.h"
|
||||
#endif
|
||||
|
||||
Reference in New Issue
Block a user