24.02.1
|
Go to the documentation of this file.
24 #ifndef SRC_RUNTIME_CL_MLGO_MLGO_HEURISTICS_H
25 #define SRC_RUNTIME_CL_MLGO_MLGO_HEURISTICS_H
138 static constexpr
size_t _max_num_trees{100};
142 std::map<HeuristicTree::TreeID, HeuristicTree::Index> _indices;
143 std::map<HeuristicTree::Index, HeuristicTree> _trees;
144 std::map<HeuristicTree::TreeID, bool> _tree_valid;
150 #endif //SRC_RUNTIME_CL_MLGO_MLGO_HEURISTICS_H
bool reload_from_stream(std::istream &istream)
(Re)Load the heuristics from reading an input stream
bool reload_from_file(const std::string &filename)
(Re)Load the heuristics from reading a dotmlgo file
unsigned int m
Number of rows for the lhs matrix.
std::pair< bool, GEMMType > query_gemm_type(const Query &query) const
Query the gemm type.
MLGOHeuristics()
Constructor.
unsigned int b
Batch size.
MLGOHeuristics & operator=(const MLGOHeuristics &)=delete
Prevent Copy Assignment.
~MLGOHeuristics()=default
Default Destructor.
bool check_heuristic_tree(HeuristicTree::TreeID id)
Check the validity of the heuristic tree.
GEMM Configuration for Reshaped Only RHS kernel.
bool check_all() const
Check the overall validity of the heuristics.
unsigned int n
Number of columns for the rhs matrix.
A binary decision tree based heuristic.
std::pair< bool, HeuristicTree * > get_heuristic_tree(HeuristicTree::TreeID id)
Get the heuristic tree from tree id.
GEMM Configuration for Reshaped kernel.
std::pair< bool, GEMMConfigReshapedOnlyRHS > query_gemm_config_reshaped_only_rhs(const Query &query) const
Query the gemm configuration for reshaped only rhs kernel.
MLGOHeuristics mlgo(TokenStream &in, bool &valid)
bool add_heuristic_tree(HeuristicTree &&t)
Add a heuristic tree.
std::string ip_target
The name of the IP target.
unsigned int k
Number of rows for the rhs matrix.
Copyright (c) 2017-2024 Arm Limited.
std::pair< bool, GEMMConfigReshaped > query_gemm_config_reshaped(const Query &query) const
Query the gemm configuration for reshaped kernel.
DataType data_type
Data type.
GEMM Configuration for Native kernel.
std::pair< bool, GEMMConfigNative > query_gemm_config_native(const Query &query) const
Query the gemm configuration for native kernel.
MLGOHeuristics for configuring GEMM kernels.
bool operator==(const GEMMConfigNative &lhs, const GEMMConfigNative &rhs)
DataType
Available data types.