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})