diff --git a/dnn/test/CMakeLists.txt b/dnn/test/CMakeLists.txt index 6d42880dfaff813ef1069fd355316716a68cabc8..07bd780381bc0856cf954a6d5efed5bd6d825eda 100644 --- a/dnn/test/CMakeLists.txt +++ b/dnn/test/CMakeLists.txt @@ -11,7 +11,7 @@ if(NOT ${MGE_ARCH} STREQUAL "naive") list(APPEND SOURCES ${SOURCES_}) if(${MGE_ARCH} STREQUAL "fallback") message(WARNING "build only with fallback") - elseif(${MGE_ARCH} STREQUAL "x86") + elseif(${MGE_ARCH} STREQUAL "x86_64" OR ${MGE_ARCH} STREQUAL "i386") file(GLOB_RECURSE SOURCES_ x86/*.cpp) list(APPEND SOURCES ${SOURCES_}) endif()