#include <cuml/experimental/fil/detail/raft_proto/device_id.hpp>
#include <cuml/experimental/fil/detail/raft_proto/device_type.hpp>
#include <cuml/experimental/fil/detail/raft_proto/gpu_support.hpp>
#include <type_traits>
Go to the source code of this file.
Namespaces | |
ML | |
ML::experimental | |
ML::experimental::fil | |
ML::experimental::fil::detail | |
ML::experimental::fil::detail::device_initialization | |
Functions | |
template<typename forest_t , raft_proto::device_type D> | |
std::enable_if_t< std::disjunction_v< std::bool_constant<!raft_proto::GPU_ENABLED >, std::bool_constant< D==raft_proto::device_type::cpu > >, void > | ML::experimental::fil::detail::device_initialization::initialize_device (raft_proto::device_id< D > device) |