Browse Source

fix(mgb): fix megtee build errors

GitOrigin-RevId: b351dd3994
dev-support-lite-fork-debug-mode
Megvii Engine Team 2 years ago
parent
commit
116781ba9c
2 changed files with 3 additions and 1 deletions
  1. +2
    -1
      dnn/src/arm_common/conv_bias/int8/direct_kernels/dot_direct_nchw_large.h
  2. +1
    -0
      src/core/include/megbrain/common.h

+ 2
- 1
dnn/src/arm_common/conv_bias/int8/direct_kernels/dot_direct_nchw_large.h View File

@@ -1,4 +1,5 @@
#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(
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

+ 1
- 0
src/core/include/megbrain/common.h View File

@@ -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>


Loading…
Cancel
Save