From 9da2da5befa33e29db70225c2e75522b9ad0d3e5 Mon Sep 17 00:00:00 2001 From: Christopher Sidebottom Date: Wed, 9 Feb 2022 14:38:32 +0000 Subject: [PATCH] [CMSIS-NN] Fix extension detection for CPUs (#10200) --- src/relay/backend/contrib/cmsisnn/compiler_attrs.cc | 2 +- .../cpp/relay/backend/contrib/cmsisnn/compiler_attrs_test.cc | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/relay/backend/contrib/cmsisnn/compiler_attrs.cc b/src/relay/backend/contrib/cmsisnn/compiler_attrs.cc index c6fb8f32e7a9..6ec61ad6c42d 100644 --- a/src/relay/backend/contrib/cmsisnn/compiler_attrs.cc +++ b/src/relay/backend/contrib/cmsisnn/compiler_attrs.cc @@ -29,7 +29,7 @@ namespace contrib { namespace cmsisnn { static const char* mveCPUs[] = {"cortex-m55"}; -static const char* dspCPUs[] = {"cortex-m7", "cortex-m33", "cortex-m35p"}; +static const char* dspCPUs[] = {"cortex-m4", "cortex-m7", "cortex-m33", "cortex-m35p"}; TVM_REGISTER_NODE_TYPE(CMSISNNCompilerConfigNode); TVM_REGISTER_PASS_CONFIG_OPTION("relay.ext.cmsisnn.options", CMSISNNCompilerConfig); diff --git a/tests/cpp/relay/backend/contrib/cmsisnn/compiler_attrs_test.cc b/tests/cpp/relay/backend/contrib/cmsisnn/compiler_attrs_test.cc index 0eb980f14af0..8cf08ebd05ad 100644 --- a/tests/cpp/relay/backend/contrib/cmsisnn/compiler_attrs_test.cc +++ b/tests/cpp/relay/backend/contrib/cmsisnn/compiler_attrs_test.cc @@ -33,8 +33,8 @@ namespace contrib { namespace cmsisnn { static const char* mveCPUs[] = {"cortex-m55"}; -static const char* dspCPUs[] = {"cortex-m7", "cortex-m33", "cortex-m35p"}; -static const char* noExtensions[] = {"cortex-m0", "cortex-m3", "cortex-m4"}; +static const char* dspCPUs[] = {"cortex-m4", "cortex-m7", "cortex-m33", "cortex-m35p"}; +static const char* noExtensions[] = {"cortex-m0", "cortex-m3"}; class CMSISNNFlagsMVECPUs : public testing::TestWithParam {}; class CMSISNNFlagsDSPCPUs : public testing::TestWithParam {};