diff options
Diffstat (limited to 'src/runtime/CL/mlgo/Utils.h')
-rw-r--r-- | src/runtime/CL/mlgo/Utils.h | 19 |
1 files changed, 13 insertions, 6 deletions
diff --git a/src/runtime/CL/mlgo/Utils.h b/src/runtime/CL/mlgo/Utils.h index 2e324dd439..c634a887e9 100644 --- a/src/runtime/CL/mlgo/Utils.h +++ b/src/runtime/CL/mlgo/Utils.h @@ -21,14 +21,16 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#ifndef SRC_MLGO_UTILS_H -#define SRC_MLGO_UTILS_H +#ifndef SRC_RUNTIME_CL_MLGO_UTILS_H +#define SRC_RUNTIME_CL_MLGO_UTILS_H #include "src/runtime/CL/mlgo/Common.h" #include "src/runtime/CL/mlgo/HeuristicTree.h" +#include "src/runtime/CL/mlgo/MLGOHeuristics.h" #include "src/runtime/CL/mlgo/MLGOParser.h" #include <ostream> +#include <string> namespace arm_compute { @@ -37,14 +39,19 @@ namespace mlgo std::ostream &operator<<(std::ostream &os, const GEMMConfigNative &config); std::ostream &operator<<(std::ostream &os, const GEMMConfigReshapedOnlyRHS &config); std::ostream &operator<<(std::ostream &os, const GEMMConfigReshaped &config); -std::ostream &operator<<(std::ostream &os, const HeuristicType &ht); -std::ostream &operator<<(std::ostream &os, const DataType &dt); +std::ostream &operator<<(std::ostream &os, HeuristicType ht); +std::ostream &operator<<(std::ostream &os, DataType dt); std::ostream &operator<<(std::ostream &os, const HeuristicTree::Index &index); +std::ostream &operator<<(std::ostream &os, const Query &query); +std::string to_string(const GEMMConfigNative &config); +std::string to_string(const GEMMConfigReshapedOnlyRHS &config); +std::string to_string(const GEMMConfigReshaped &config); +std::string to_string(const Query &query); namespace parser { -std::ostream &operator<<(std::ostream &os, CharPosition tok); +std::ostream &operator<<(std::ostream &os, const CharPosition &pos); } // namespace parser } // namespace mlgo } // namespace arm_compute -#endif //SRC_MLGO_UTILS_H
\ No newline at end of file +#endif //SRC_RUNTIME_CL_MLGO_UTILS_H
\ No newline at end of file |