Skip to content

Commit 7322f7b

Browse files
committed
修正了CMSISNN兼容性问题
1 parent 3b0570c commit 7322f7b

4 files changed

Lines changed: 39 additions & 38 deletions

File tree

tensorflow/lite/micro/SConscript

Lines changed: 33 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -2,39 +2,38 @@ from building import *
22
import os
33

44
src = Split('''
5-
kernels/activations.cc
6-
kernels/arg_min_max.cc
7-
kernels/ceil.cc
8-
kernels/circular_buffer.cc
9-
kernels/comparisons.cc
10-
kernels/concatenation.cc
11-
kernels/dequantize.cc
12-
kernels/elementwise.cc
13-
kernels/ethosu.cc
14-
kernels/floor.cc
15-
kernels/hard_swish.cc
16-
kernels/kernel_runner.cc
17-
kernels/kernel_util.cc
18-
kernels/l2norm.cc
19-
kernels/logical.cc
20-
kernels/logistic.cc
21-
kernels/maximum_minimum.cc
22-
kernels/neg.cc
23-
kernels/pack.cc
24-
kernels/pad.cc
25-
kernels/prelu.cc
26-
kernels/quantize.cc
27-
kernels/reduce.cc
28-
kernels/reshape.cc
29-
kernels/resize_nearest_neighbor.cc
30-
kernels/round.cc
31-
kernels/split.cc
32-
kernels/strided_slice.cc
33-
kernels/sub.cc
34-
kernels/svdf.cc
35-
kernels/tanh.cc
36-
kernels/unpack.cc
37-
''') + Glob('*.c') + Glob('*.cc') + Glob('benchmarks/*.cc') \
5+
kernels/activations.cc
6+
kernels/arg_min_max.cc
7+
kernels/ceil.cc
8+
kernels/circular_buffer.cc
9+
kernels/comparisons.cc
10+
kernels/concatenation.cc
11+
kernels/dequantize.cc
12+
kernels/elementwise.cc
13+
kernels/ethosu.cc
14+
kernels/floor.cc
15+
kernels/hard_swish.cc
16+
kernels/kernel_runner.cc
17+
kernels/kernel_util.cc
18+
kernels/l2norm.cc
19+
kernels/logical.cc
20+
kernels/logistic.cc
21+
kernels/maximum_minimum.cc
22+
kernels/neg.cc
23+
kernels/pack.cc
24+
kernels/pad.cc
25+
kernels/prelu.cc
26+
kernels/quantize.cc
27+
kernels/reduce.cc
28+
kernels/reshape.cc
29+
kernels/resize_nearest_neighbor.cc
30+
kernels/round.cc
31+
kernels/split.cc
32+
kernels/strided_slice.cc
33+
kernels/sub.cc
34+
kernels/svdf.cc
35+
kernels/tanh.cc
36+
kernels/unpack.cc''') + Glob('*.c') + Glob('*.cc') + Glob('benchmarks/*.cc') \
3837
+ Glob('examples/micro_speech/*.cc') + Glob('examples/micro_speech/micro_features/*.cc') \
3938
+ Glob('memory_planner/*.cc') + Glob('testing/*.cc')
4039

@@ -48,6 +47,7 @@ if GetDepend(['PKG_USING_TENSORFLOWLITEMICRO_REFERENCE']):
4847
kernels/pooling.cc
4948
kernels/softmax.cc
5049
''')
50+
5151
if GetDepend(['PKG_USING_TENSORFLOWLITEMICRO_CMSISNN']):
5252
src += Split('''
5353
kernels/cmsis-nn/add.cc

tensorflow/lite/micro/tools/make/downloads/cmsis/CMSIS/NN/Source/NNSupportFunctions/arm_nn_mat_mult_nt_t_s8.c

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,11 +33,12 @@
3333
#include "cmsis/CMSIS/NN/Include/arm_nnsupportfunctions.h"
3434

3535
// Work around for https://github.com/ARMmbed/mbed-os/issues/12568
36-
__STATIC_FORCEINLINE uint32_t __patched_SXTB16_RORn(uint32_t op1, uint32_t rotate) {
37-
uint32_t result;
38-
__ASM ("sxtb16 %0, %1, ROR %2" : "=r" (result) : "r" (op1), "i" (rotate) );
39-
return result;
40-
}
36+
#define __patched_SXTB16_RORn(op1, rotate) \
37+
({ \
38+
uint32_t result; \
39+
__ASM ("sxtb16 %0, %1, ROR %2" : "=r" (result) : "r" (op1), "i" (rotate) ); \
40+
result;\
41+
})
4142

4243
/**
4344
* @ingroup groupSupport

0 commit comments

Comments
 (0)