#include <cuml/fil/detail/cpu_introspection.hpp>
#include <cuml/fil/detail/evaluate_tree.hpp>
#include <cuml/fil/detail/index_type.hpp>
#include <cuml/fil/detail/postprocessor.hpp>
#include <cuml/fil/detail/raft_proto/ceildiv.hpp>
#include <cuml/fil/exceptions.hpp>
#include <cuml/fil/infer_kind.hpp>
#include <raft/core/error.hpp>
#include <algorithm>
#include <cinttypes>
#include <cstddef>
#include <iostream>
#include <new>
#include <numeric>
#include <string>
#include <vector>
Go to the source code of this file.
|
| template<bool has_categorical_nodes, bool predict_leaf, typename forest_t , typename vector_output_t = std::nullptr_t, typename categorical_data_t = std::nullptr_t> |
| void | ML::fil::detail::infer_kernel_cpu (forest_t const &forest, postprocessor< typename forest_t::io_type > const &postproc, typename forest_t::io_type *output, typename forest_t::io_type const *input, index_type row_count, index_type col_count, index_type num_outputs, index_type chunk_size=hardware_constructive_interference_size, index_type grove_size=hardware_constructive_interference_size, vector_output_t vector_output_p=nullptr, categorical_data_t categorical_data=nullptr, infer_kind infer_type=infer_kind::default_kind) |
| |
◆ omp_get_max_threads
| #define omp_get_max_threads |
( |
| ) |
1 |