From 0f9760ddf4ce06d3936b548aa08fb0e981a4669e Mon Sep 17 00:00:00 2001
From: xiaoliangstd <2303335747@qq.com>
Date: Wed, 15 Mar 2023 16:04:24 +0800
Subject: [PATCH] update CMakeLists.txt

---
 CMakeLists.txt | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index c6303d7..9b82dd9 100755
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -4,5 +4,12 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -O3")
 
 include_directories(include)
 link_directories(lib)
+
+if(CMAKE_HOST_SYSTEM_PROCESSOR MATCHES "aarch64") 
+    set(EXTRA_LIBS libUnitreeMotorSDK_M80106_Arm64.so)
+else()
+    set(EXTRA_LIBS libUnitreeMotorSDK_M80106_Linux64.so)
+endif()
+
 add_executable(motorctrl example/main.cpp)
-target_link_libraries(motorctrl libUnitreeMotorSDK_M80106_Linux64.so)
+target_link_libraries(motorctrl ${EXTRA_LIBS})