forked from xuos/xiuos
add makefiles to adapt nnom and cmsis-nn for xizi
This commit is contained in:
parent
10c39d930c
commit
9b3c321150
|
@ -1,3 +1,15 @@
|
|||
SRC_DIR := mnist
|
||||
SRC_DIR :=
|
||||
|
||||
ifeq ($(CONFIG_APP_MNIST),y)
|
||||
SRC_DIR += mnist
|
||||
endif
|
||||
|
||||
ifeq ($(CONFIG_USING_NNOM_DEMOAPP),y)
|
||||
SRC_DIR += nnom_demo
|
||||
endif
|
||||
|
||||
ifeq ($(CONFIG_USING_CMSIS_5_DEMOAPP),y)
|
||||
SRC_DIR += cmsis_5_demo
|
||||
endif
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -0,0 +1,11 @@
|
|||
SRC_DIR :=
|
||||
|
||||
ifeq ($(CONFIG_USING_CMSIS_5_NN_DEMOAPP),y)
|
||||
SRC_DIR += cmsisnn-cifar10
|
||||
endif
|
||||
|
||||
ifeq ($(CONFIG_USING_CMSIS_5_NN_DEMOAPP_VEG_CLASSIFY),y)
|
||||
SRC_DIR += cmsisnn_vegetable_classify
|
||||
endif
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -0,0 +1,3 @@
|
|||
SRC_FILES := demo/cmsisnn_demo.c demo/tjpgd.c model/m4/nn.c
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -0,0 +1,3 @@
|
|||
SRC_FILES := cmsisnn_vegetable_classify.c
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -1,7 +1,9 @@
|
|||
SRC_FILES :=
|
||||
ifeq ($(CONFIG_APP_MNIST),y)
|
||||
SRC_FILES := \
|
||||
main.cpp \
|
||||
mnistmain.c
|
||||
CPPPATHS += -I.
|
||||
|
||||
endif
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -26,5 +26,3 @@ void mnist_app(void);
|
|||
int tfmnist(void) {
|
||||
mnist_app();
|
||||
}
|
||||
|
||||
PRIV_SHELL_CMD_FUNCTION(tfmnist, a mnist test sample, PRIV_SHELL_CMD_MAIN_ATTR);
|
||||
|
|
|
@ -0,0 +1,5 @@
|
|||
ifeq ($(CONFIG_USING_NNOM_DEMOAPP),y)
|
||||
SRC_DIR := mnist_nnom
|
||||
endif
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -0,0 +1,3 @@
|
|||
SRC_FILES := main.c
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -71,7 +71,9 @@ void mnist_nnom(int argc, char **argv)
|
|||
printf("Predicted label: %d\n", predic_label);
|
||||
printf("Probability: %d%%\n", (int)(prob * 100));
|
||||
}
|
||||
|
||||
#ifdef ADD_XIZI_FETURES
|
||||
PRIV_SHELL_CMD_FUNCTION(mnist_nnom, a mnist_nnom test sample, PRIV_SHELL_CMD_MAIN_ATTR);
|
||||
#endif
|
||||
#ifdef __RT_THREAD_H__
|
||||
MSH_CMD_EXPORT(mnist_nnom, nnom mnist demo and image number should be followed);
|
||||
#endif
|
||||
|
|
|
@ -0,0 +1,8 @@
|
|||
ifeq ($(CONFIG_USING_CMSIS_5),y)
|
||||
SRC_FILES := $(wildcard NN/Source/ActivationFunctions/*.c) $(wildcard NN/Source/BasicMathFunctions/*.c) $(wildcard NN/Source/ConcatenationFunctions/*.c)\
|
||||
$(wildcard NN/Source/ConvolutionFunctions/*.c) $(wildcard NN/Source/FullyConnectedFunctions/*.c) $(wildcard NN/Source/NNSupportFunctions/*.c) \
|
||||
$(wildcard NN/Source/PoolingFunctions/*.c) $(wildcard NN/Source/ReshapeFunctions/*.c) $(wildcard NN/Source/SoftmaxFunctions/*.c) $(wildcard NN/Source/SVDFunctions/*.c)
|
||||
endif
|
||||
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -0,0 +1,6 @@
|
|||
SRC_DIR :=
|
||||
ifeq ($(CONFIG_USING_NNOM),y)
|
||||
SRC_FILES := $(wildcard src/backends/*.c) $(wildcard src/core/*.c) $(wildcard src/layers/*.c)
|
||||
endif
|
||||
|
||||
include $(KERNEL_ROOT)/compiler.mk
|
|
@ -45,7 +45,7 @@ nnom_layer_t *reshape_s(const nnom_reshape_config_t *config)
|
|||
|
||||
// config
|
||||
//nnom_memcpy(layer->dim, config->dim, config->num_dim * sizeof(nnom_shape_data_t));
|
||||
layer->super.config = config;
|
||||
layer->super.config = (void*)config;
|
||||
layer->dim = config->dim; // temporary use the config directly. (not preferable.)
|
||||
layer->num_dim = config->num_dim;
|
||||
|
||||
|
|
Loading…
Reference in New Issue