24 #ifndef SRC_RUNTIME_CL_MLGO_MLGO_HEURISTICS_H 25 #define SRC_RUNTIME_CL_MLGO_MLGO_HEURISTICS_H 75 std::pair<bool, GEMMType> query_gemm_type(
const Query &query)
const;
82 std::pair<bool, GEMMConfigNative> query_gemm_config_native(
const Query &query)
const;
89 std::pair<bool, GEMMConfigReshapedOnlyRHS> query_gemm_config_reshaped_only_rhs(
const Query &query)
const;
96 std::pair<bool, GEMMConfigReshaped> query_gemm_config_reshaped(
const Query &query)
const;
103 bool reload_from_file(
const std::string &filename);
110 bool reload_from_stream(std::istream &istream);
135 bool check_all()
const;
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 unsigned int b
Batch size.
bool operator==(const GEMMConfigNative &lhs, const GEMMConfigNative &rhs)
GEMM Configuration for Reshaped kernel.
Copyright (c) 2017-2021 Arm Limited.
std::string ip_target
The name of the IP target.
GEMM Configuration for Reshaped Only RHS kernel.
MLGOHeuristics for configuring GEMM kernels.
DataType data_type
Data type.
unsigned int n
Number of columns for the rhs matrix.
A binary decision tree based heuristic.
MLGOHeuristics mlgo(TokenStream &in, bool &valid)
GEMM Configuration for Native kernel.
DataType
Available data types.
unsigned int m
Number of rows for the lhs matrix.
unsigned int k
Number of rows for the rhs matrix.