提交 116781ba 编写于 作者: M Megvii Engine Team

fix(mgb): fix megtee build errors

GitOrigin-RevId: b351dd39943ef26f25e3e95f89d14f8005c49ab4
上级 54b5db17
#pragma once #pragma once
#include <cstddef>
#include "megdnn/arch.h" #include "megdnn/arch.h"
#if MGB_ENABLE_DOT #if MGB_ENABLE_DOT
...@@ -21,4 +22,4 @@ void megdnn_dot_nchw_large_chanwise_direct_conv_11x11s2_oh4_ow16( ...@@ -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, 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, size_t ow, size_t OH, size_t OW, size_t pad_iw, const float scale,
int8_t relu_val); int8_t relu_val);
#endif #endif
\ No newline at end of file
...@@ -9,6 +9,7 @@ ...@@ -9,6 +9,7 @@
#include <mutex> #include <mutex>
#include <string> #include <string>
#include <cerrno>
#include <cstdarg> #include <cstdarg>
#include <cstddef> #include <cstddef>
#include <cstdint> #include <cstdint>
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册