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 c689ae74..fc90ebee 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 e76fd81b..033a6a6c 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