From 116781ba9c6b0128bc03975ece8420a106e8dd8e Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 15 Aug 2022 16:37:25 +0800 Subject: [PATCH] fix(mgb): fix megtee build errors GitOrigin-RevId: b351dd39943ef26f25e3e95f89d14f8005c49ab4 --- .../conv_bias/int8/direct_kernels/dot_direct_nchw_large.h | 3 ++- src/core/include/megbrain/common.h | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/dnn/src/arm_common/conv_bias/int8/direct_kernels/dot_direct_nchw_large.h b/dnn/src/arm_common/conv_bias/int8/direct_kernels/dot_direct_nchw_large.h index c689ae743..fc90ebee5 100644 --- a/dnn/src/arm_common/conv_bias/int8/direct_kernels/dot_direct_nchw_large.h +++ b/dnn/src/arm_common/conv_bias/int8/direct_kernels/dot_direct_nchw_large.h @@ -1,4 +1,5 @@ #pragma once +#include #include "megdnn/arch.h" #if MGB_ENABLE_DOT @@ -21,4 +22,4 @@ void megdnn_dot_nchw_large_chanwise_direct_conv_11x11s2_oh4_ow16( const int8_t* src, const int8_t* weight, int32_t bias, int8_t* dst, size_t oh, size_t ow, size_t OH, size_t OW, size_t pad_iw, const float scale, int8_t relu_val); -#endif \ No newline at end of file +#endif diff --git a/src/core/include/megbrain/common.h b/src/core/include/megbrain/common.h index e76fd81b4..033a6a6cc 100644 --- a/src/core/include/megbrain/common.h +++ b/src/core/include/megbrain/common.h @@ -9,6 +9,7 @@ #include #include +#include #include #include #include -- GitLab