如何解决 GTest 和 LibTorch 联动冲突

Posted

技术标签:

【中文标题】如何解决 GTest 和 LibTorch 联动冲突【英文标题】:How to solve GTest and LibTorch linkage conflict 【发布时间】:2020-10-22 20:50:20 【问题描述】:

这个问题跟随my precedent one。

我正在使用 OpenCV、Torch 和 NumCpp 编写 C++ 程序。该程序现在可以编译并且运行良好,但我需要编写单元测试。

我已经关注google's tutorial 在我的项目中构建 GTest 和 GMock,但它失败了。当我不链接 Torch 库时,它可以工作。

链接 GTest + Torch 时出错:

/usr/bin/ld: CMakeFiles/TryGTest_test.dir/test/boxTest.cpp.o: in function `testing::AssertionResult testing::internal::CmpHelperEQFailure<int, int>(char const*, char const*, int const&, int const&)':
/tmp/tmp.Z1zXnMtLsD/cmake-build-debug-ubuntu_2/googletest-src/googletest/include/gtest/gtest.h:1511: undefined reference to `testing::internal::EqFailure(char const*, char const*, std::string const&, std::string const&, bool)'
collect2: error: ld returned 1 exit status

我想它来自 Libtorch 定义的与 Gtest 同名的宏,Google 提出了一个workaround 来解决此类问题,但我需要找出哪个宏失败了。

希望有人能帮忙!

这是我的 CMakeLists.txt

cmake_minimum_required(VERSION 2.6)
project(TryGtest)

set(CMAKE_CXX_STANDARD 14) # C14 required to compile Torch
set(CMAKE_CXX_STANDARD_REQUIRED TRUE)

# Specifying we are using pthread for UNIX systems.
set(CMAKE_CXX_FLAGS "$CMAKE_CXX_FLAGS $TORCH_CXX_FLAGS -pthread -Wall")

find_package(OpenCV REQUIRED)
find_package(Torch REQUIRED)

if(NOT Torch_FOUND)
    message(FATAL_ERROR "Pytorch Not Found!")
endif(NOT Torch_FOUND)

message(STATUS "Pytorch status :")
message(STATUS "    libraries: $TORCH_LIBRARIES")
message(STATUS "    Torch Flags: $TORCH_CXX_FLAGS")

message(STATUS "OpenCV library status :")
message(STATUS "    version: $OpenCV_VERSION")
message(STATUS "    libraries: $OpenCV_LIBS")
message(STATUS "    include path: $OpenCV_INCLUDE_DIRS")

# -------- GOOGLE TEST ----------
# Download and unpack googletest at configure time
configure_file(CMakeLists.txt.in googletest-download/CMakeLists.txt)
execute_process(COMMAND $CMAKE_COMMAND -G "$CMAKE_GENERATOR" .
        RESULT_VARIABLE result
        WORKING_DIRECTORY $CMAKE_CURRENT_BINARY_DIR/googletest-download )
if(result)
    message(FATAL_ERROR "CMake step for googletest failed: $result")
endif()
execute_process(COMMAND $CMAKE_COMMAND --build .
        RESULT_VARIABLE result
        WORKING_DIRECTORY $CMAKE_CURRENT_BINARY_DIR/googletest-download )
if(result)
    message(FATAL_ERROR "Build step for googletest failed: $result")
endif()

# Prevent overriding the parent project's compiler/linker
# settings on Windows
set(gtest_force_shared_crt ON CACHE BOOL "" FORCE)

# Add googletest directly to our build. This defines
# the gtest and gtest_main targets.
add_subdirectory($CMAKE_CURRENT_BINARY_DIR/googletest-src
        $CMAKE_CURRENT_BINARY_DIR/googletest-build
        EXCLUDE_FROM_ALL)

# The gtest/gtest_main targets carry header search path
# dependencies automatically when using CMake 2.8.11 or
# later. Otherwise we have to add them here ourselves.
if (CMAKE_VERSION VERSION_LESS 2.8.11)
    include_directories("$gtest_SOURCE_DIR/include")
endif()
# -------------------------------------------------------------------------
enable_testing()
include_directories("$gtest_SOURCE_DIR/include")


# Program executable
add_executable(TryGTest src/main.cpp src/box.cpp include/util.h)

# Test executable
add_executable(TryGTest_test test/main.cpp src/box.cpp test/boxTest.cpp include/util.h)

target_link_libraries(TryGTest PRIVATE pthread dl util $TORCH_LIBRARIES $OpenCV_LIBS )
target_link_libraries (TryGTest_test PRIVATE pthread dl util $TORCH_LIBRARIES $OpenCV_LIBS gtest gmock)

和 CMakeLists.txt.in

cmake_minimum_required(VERSION 2.8.2)

project(googletest-download NONE)

include(ExternalProject)
ExternalProject_Add(googletest
        GIT_REPOSITORY    https://github.com/google/googletest.git
        GIT_TAG           release-1.10.0
        SOURCE_DIR        "$CMAKE_CURRENT_BINARY_DIR/googletest-src"
        BINARY_DIR        "$CMAKE_CURRENT_BINARY_DIR/googletest-build"
        CONFIGURE_COMMAND ""
        BUILD_COMMAND     ""
        INSTALL_COMMAND   ""
        TEST_COMMAND      ""
        )

【问题讨论】:

只有三个宏可以使用此解决方法取消定义:FAIL、SUCCEED 和 TEST。我怀疑其中任何一个都会有所帮助。你如何调用链接器?你能显示完整的命令吗? @n.'pronouns'm。我已经用我的 CMakeLists 编辑了这篇文章 @n.'pronouns'm。你确定我们只能重新定义这些词吗?我在编译选项中添加了-DGTEST_DONT_DEFINE_ASSERT_EQ=1,并且不得不将我的测试代码中的 ASSERT_EQ 重命名为 GTEST_ASSERT_EQ,证明这可能有效。 CMakeLists 离动作太远了。如果不实际构建您的项目,我无法说出它的作用。这就是为什么我要求链接器调用命令,而不是生成它的配置脚本。 显然所有的 ASSERT_??宏也可以是未定义的,尽管没有记录。 【参考方案1】:

PyTorch 使用 -D_GLIBCXX_USE_CXX11_ABI=0 编译标志(这在 2020 IMNSHO 中应该是刑事犯罪)。您需要使用此标志编译所有代码,包括 gtest。

这个标志在TORCH_CXX_FLAGS 中,但是 gtest 使用它自己的 CMakeLists.txt 和它自己的一组标志。您应该手动添加它。最简单的方法可能是在 CMakeLists.txt 顶部附近的某处使用 add_compile_definitions(_GLIBCXX_USE_CXX11_ABI=0)

【讨论】:

你。是。天才。十分感谢你的帮助 !你可以编辑答案吗? -D 不能在 add_compile_definitions 命令中,它是自动添加的。感谢您为我花费的时间! 您能否解释一下或向我发送有关为什么不能使用此 ABI 标志的文档? 该标志将标准库 ABI 降级为 gcc5.1 之前的版本,这是一个古老的版本,无法完全支持 c++11(这是首先更改 ABI 的原因)。该更改是向后不兼容的,因此不可能将使用旧 ABI 和新 ABI 的库链接在一起。查找“dual abi”了解更多信息。

以上是关于如何解决 GTest 和 LibTorch 联动冲突的主要内容,如果未能解决你的问题,请参考以下文章

libtorch 报错与解决

libtorch 报错与解决

如何从 libtorch 输出中删除乘数并显示最终结果?

如何在 Maskcrnn libtorch 中获取元组对象返回的值

libtorch (pytorch c++) 教程

libtorch (pytorch c++) 教程