From 0c6d4a603335ae2cba2771e939eff1ddeb98fbe3 Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Fri, 11 Mar 2022 10:45:08 +0900 Subject: [PATCH] add public, fix include path convention --- src/relay/backend/te_compiler_cache.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/relay/backend/te_compiler_cache.cc b/src/relay/backend/te_compiler_cache.cc index fc3a3ab335f4..276c7f9f017e 100644 --- a/src/relay/backend/te_compiler_cache.cc +++ b/src/relay/backend/te_compiler_cache.cc @@ -28,11 +28,13 @@ #include #include #include +#include #include #include #include #include #include +#include #include #include @@ -45,8 +47,6 @@ #include "../../te/operation/create_primfunc.h" #include "../op/memory/memory.h" #include "../transforms/pass_utils.h" -#include "tvm/relay/op_strategy.h" -#include "tvm/tir/function.h" #include "utils.h" namespace tvm { @@ -138,6 +138,7 @@ class LowerToTECompute : public backend::MemoizedExprTranslator outputs = this->VisitExpr(relay_func->body); candidate_name_ = readable_name_stream_.str(); + constexpr static size_t kMaxFuncNameLength = 80; // WARNING: Please make sure to also update TVM_CRT_MAX_STRLEN_FUNCTION_NAME // whenever the value of kMaxFuncNameLength changes @@ -291,7 +292,7 @@ class LowerToTECompute : public backend::MemoizedExprTranslator