#include <cuml/experimental/fil/detail/raft_proto/device_id.hpp>#include <cuml/experimental/fil/detail/raft_proto/device_setter.hpp>#include <cuml/experimental/fil/detail/raft_proto/device_type.hpp>#include <cuml/experimental/fil/detail/raft_proto/gpu_support.hpp>#include <type_traits>
Include dependency graph for gpu.hpp:
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::conjunction_v< std::bool_constant< raft_proto::GPU_ENABLED >, std::bool_constant< D==raft_proto::device_type::gpu > >, void > | ML::experimental::fil::detail::device_initialization::initialize_device (raft_proto::device_id< D > device) |
