From 5fb07c9964c9be33dbd4089478ae24e1b24ed52a Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Mon, 6 Jul 2020 15:55:36 +0800
Subject: [PATCH] fix(dnn/x86): fix cmake error for build x86 gtest

GitOrigin-RevId: 76131907331dff004b0ce04bd56f0ad1f7e6ab65
---
 dnn/test/CMakeLists.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/dnn/test/CMakeLists.txt b/dnn/test/CMakeLists.txt
index 6d42880d..07bd7803 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()