restore device helper

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

View File

@@ -2,9 +2,6 @@
* 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>
@@ -1385,7 +1382,3 @@ class LDGIterator {
}
};
} // namespace dh
#elif defined(XGBOOST_USE_HIP)
#include "device_helpers.hip.h"
#endif