55ee272ea8
* Extend array interface to handle ndarray. The `ArrayInterface` class is extended to support multi-dim array inputs. Previously this class handles only 2-dim (vector is also matrix). This PR specifies the expected dimension at compile-time and the array interface can perform various checks automatically for input data. Also, adapters like CSR are more rigorous about their input. Lastly, row vector and column vector are handled without intervention from the caller.
57 lines
1.5 KiB
Plaintext
57 lines
1.5 KiB
Plaintext
/*!
|
|
* Copyright 2021 by Contributors
|
|
*/
|
|
#include "../common/common.h"
|
|
#include "array_interface.h"
|
|
|
|
namespace xgboost {
|
|
void ArrayInterfaceHandler::SyncCudaStream(int64_t stream) {
|
|
switch (stream) {
|
|
case 0:
|
|
/**
|
|
* disallowed by the `__cuda_array_interface__`. Quote:
|
|
*
|
|
* This is disallowed as it would be ambiguous between None and the default
|
|
* stream, and also between the legacy and per-thread default streams. Any use
|
|
* case where 0 might be given should either use None, 1, or 2 instead for
|
|
* clarity.
|
|
*/
|
|
LOG(FATAL) << "Invalid stream ID in array interface: " << stream;
|
|
case 1:
|
|
// default legacy stream
|
|
break;
|
|
case 2:
|
|
// default per-thread stream
|
|
default:
|
|
dh::safe_cuda(cudaStreamSynchronize(reinterpret_cast<cudaStream_t>(stream)));
|
|
}
|
|
}
|
|
|
|
bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {
|
|
if (!ptr) {
|
|
return false;
|
|
}
|
|
cudaPointerAttributes attr;
|
|
auto err = cudaPointerGetAttributes(&attr, ptr);
|
|
// reset error
|
|
CHECK_EQ(err, cudaGetLastError());
|
|
if (err == cudaErrorInvalidValue) {
|
|
// CUDA < 11
|
|
return false;
|
|
} else if (err == cudaSuccess) {
|
|
// CUDA >= 11
|
|
switch (attr.type) {
|
|
case cudaMemoryTypeUnregistered:
|
|
case cudaMemoryTypeHost:
|
|
return false;
|
|
default:
|
|
return true;
|
|
}
|
|
return true;
|
|
} else {
|
|
// other errors, `cudaErrorNoDevice`, `cudaErrorInsufficientDriver` etc.
|
|
return false;
|
|
}
|
|
}
|
|
} // namespace xgboost
|