Add guards for CMSIS-NN tests

apache · Oct 21, 2021 · 9795a5b · 9795a5b
1 parent c347020
commit 9795a5b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/cmake/modules/contrib/CMSISNN.cmake b/cmake/modules/contrib/CMSISNN.cmake
@@ -16,6 +16,7 @@
 # under the License.
 
 if(USE_CMSISNN)
+  add_definitions(-DTVM_USE_CMSISNN)
   message(STATUS "Build with CMSIS-NN support")
   file(GLOB RELAY_CONTRIB_CMSISNN_SRCS src/relay/backend/contrib/cmsisnn/*.cc)
   list(APPEND COMPILER_SRCS ${RELAY_CONTRIB_CMSISNN_SRCS})

diff --git a/tests/cpp/relay/backend/contrib/cmsisnn/buffer_size_test.cc b/tests/cpp/relay/backend/contrib/cmsisnn/buffer_size_test.cc
@@ -17,6 +17,8 @@
  * under the License.
  */
 
+#ifdef TVM_USE_CMSISNN
+
 #include "../../../../../../src/relay/backend/contrib/cmsisnn/buffer_size.h"
 
 #include <gtest/gtest.h>
@@ -135,3 +137,5 @@ TEST(CMSISNNBufferSize, Default) {
 }  // namespace contrib
 }  // namespace relay
 }  // namespace tvm
+
+#endif
diff --git a/tests/cpp/relay/backend/contrib/cmsisnn/compiler_attrs_test.cc b/tests/cpp/relay/backend/contrib/cmsisnn/compiler_attrs_test.cc
@@ -17,6 +17,8 @@
  * under the License.
  */
 
+#ifdef TVM_USE_CMSISNN
+
 #include "../../../../../../src/relay/backend/contrib/cmsisnn/compiler_attrs.h"
 
 #include <gtest/gtest.h>
@@ -123,3 +125,5 @@ INSTANTIATE_TEST_CASE_P(CMSISNNFlags, CMSISNNFlagsNoExtensions, ::testing::Value
 }  // namespace contrib
 }  // namespace relay
 }  // namespace tvm
+
+#endif