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 c689ae743dd6dc00a9aff79147d711990b3f2dfe..fc90ebee590de7b78f121577bd1617b6d63df0c3 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 e76fd81b475178e48bc5ab379cfad6e6d87f4018..033a6a6cc1875b208ed4152c3f20e4addd33f79f 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