added support for loongarch64 (#3094)

Co-authored-by: tsuibin <tsuibin@loongson.cn>
Tencent · Jul 20, 2021 · 1699aaa · 1699aaa
1 parent f153366
commit 1699aaa
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -179,6 +179,8 @@ elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^(riscv)")
     endif()
 elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^(powerpc)")
     set(NCNN_TARGET_ARCH powerpc)
+elseif(CMAKE_SYSTEM_PROCESSOR MATCHES "^(loongarch)")
+    set(NCNN_TARGET_ARCH loongarch)
 else()
     set(NCNN_TARGET_ARCH x86)
     option(NCNN_SSE2 "optimize x86 platform with sse2" ON)

diff --git a/toolchains/loongarch64-linux-gnu.toolchain.cmake b/toolchains/loongarch64-linux-gnu.toolchain.cmake
@@ -0,0 +1,16 @@
+set(CMAKE_SYSTEM_NAME Linux)
+set(CMAKE_SYSTEM_PROCESSOR loongarch64)
+
+set(CMAKE_C_COMPILER "loongarch64-linux-gnu-gcc")
+set(CMAKE_CXX_COMPILER "loongarch64-linux-gnu-g++")
+
+set(CMAKE_FIND_ROOT_PATH_MODE_PROGRAM NEVER)
+set(CMAKE_FIND_ROOT_PATH_MODE_LIBRARY ONLY)
+set(CMAKE_FIND_ROOT_PATH_MODE_INCLUDE ONLY)
+
+set(CMAKE_C_FLAGS "-march=loongarch64")
+set(CMAKE_CXX_FLAGS "-march=loongarch64")
+
+# cache flags
+set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS}" CACHE STRING "c flags")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}" CACHE STRING "c++ flags")