MCPcopy Create free account
hub / github.com/dmlc/xgboost / TransformDevice

Function TransformDevice

tests/cpp/common/test_transform_range.cc:17–23  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

15namespace xgboost::common {
16namespace {
17constexpr DeviceOrd TransformDevice() {
18#if defined(__CUDACC__)
19 return DeviceOrd::CUDA(0);
20#else
21 return DeviceOrd::CPU();
22#endif
23}
24} // namespace
25
26template <typename T>

Callers 1

TESTFunction · 0.85

Calls 1

CPUFunction · 0.85

Tested by

no test coverage detected