Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[MSLITE] add opencl gpu compile support for arm32

This commit is contained in:
Liu_Xuu 2021-07-08 14:30:18 +08:00
parent c44df2c9e3
commit a6b62ba502
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/lite/CMakeLists.txt
View File

@ -83,7 +83,13 @@ if(PLATFORM_ARM64)
message("invalid MSLITE_GPU_BACKEND value ${MSLITE_GPU_BACKEND} for arm64, MSLITE_GPU_BACKEND is set to off.")
set(MSLITE_GPU_BACKEND "off")
endif()
elseif(PLATFORM_ARM32 OR WIN32)
elseif(PLATFORM_ARM32)
if((NOT MSLITE_GPU_BACKEND STREQUAL "opencl") AND (NOT MSLITE_GPU_BACKEND STREQUAL "off") AND
(NOT MSLITE_GPU_BACKEND STREQUAL ""))
message("invalid MSLITE_GPU_BACKEND value ${MSLITE_GPU_BACKEND} for arm32, MSLITE_GPU_BACKEND is set to off.")
set(MSLITE_GPU_BACKEND "off")
endif()
elseif(WIN32)
set(MSLITE_GPU_BACKEND "off")
else()
# x86_64