MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / TEST_F

Function TEST_F

dnn/test/arm_common/conv_bias_multi_thread_conv1x1.cpp:13–43  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

11
12#if MGB_ENABLE_DOT
13TEST_F(ARM_COMMON_MULTI_THREADS, CONV_BIAS_CONV1x1_QUANTIZEDSYM_MK4_DOT) {
14 UniformIntRNG rng{-50, 50};
15
16#define cb(name) \
17 checker_conv_bias_common( \
18 get_nchw44_conv_bias_args( \
19 {1}, QUAN_NLMODE, ONLY_BR_BIASMODE, 1, true, false, true), \
20 handle(), &rng, epsilon, dtype::QuantizedS8(2.5f), \
21 dtype::QuantizedS8(2.5f), dtype::QuantizedS32(6.25f), \
22 dtype::QuantizedS8(60.25f), name); \
23 checker_conv_bias_common( \
24 get_nchw44_conv_bias_args( \
25 {1}, ONLY_IDENTITY_NLMODE, ONLY_NO_BIASMODE, 1, true, false, \
26 true), \
27 handle(), &rng, epsilon, dtype::QuantizedS8(2.5f), \
28 dtype::QuantizedS8(2.5f), dtype::QuantizedS32(6.25f), {}, name); \
29 checker_conv_bias_common( \
30 get_nchw44_conv_bias_args( \
31 {1}, ONLY_IDENTITY_NLMODE, ONLY_NO_BIASMODE, 1, true, false, \
32 true), \
33 handle(), &rng, epsilon, dtype::Int8(), dtype::Int8(), dtype::Int32(), {}, \
34 name);
35
36 float epsilon = 0.001;
37#if MEGDNN_AARCH64
38 cb("CONV1x1:AARCH64_INT8X8X32_MK4_8X12X4_DOTPROD");
39#elif MEGDNN_ARMV7
40 cb("CONV1x1:AARCH32_INT8_MK4_8X4X4_DOTPROD");
41#endif
42#undef cb
43}
44#endif
45
46// clang-format on

Callers

nothing calls this directly

Calls 8

get_conv_bias_1x1_argsFunction · 0.85
check_conv_biasFunction · 0.85
checker_conv_bias_commonFunction · 0.85
emplace_backMethod · 0.80
cbFunction · 0.50
push_backMethod · 0.45

Tested by

no test coverage detected