更新到 Visual Studio 17.4.0 会产生与 TLS 相关的 linker 错误

Question

编辑：只是为了关闭，这个问题原来是由于linker 中的一个错误。 Microsoft 在版本 17.4.3 中修复了它

我刚刚将我的 Visual Studio 实例从 17.3.6 更新到 17.4.0。 然后我尝试了一个干净的解决方案。 突然我的一个项目给我 linker 错误

8>pch.obj : error LNK2001: unresolved external symbol __imp___tls_index_?init@?1??lazy_init_num_threads@internal@at@@YAXXZ@4_NA
8>pch.obj : error LNK2001: unresolved external symbol __imp___tls_offset_?init@?1??lazy_init_num_threads@internal@at@@YAXXZ@4_NA
8>C:\Users\jmole\Documents\Dev\Main\Solutions\..\Mobile\x64\Debug\net6.0-windows\mld_v143.dll : fatal error LNK1120: 2 unresolved externals

这完全让我感到困惑。 当我打开详细链接时，我看到它在 MSVCRTD.lib 中找到各种相似的符号。 例如。

2>      Found _tls_index
2>      Found __dyn_tls_init

还有其他人遇到这个吗？

Answer 1

在您的 Python 环境中转到\Lib\site-packages\torch\include\ATen\Parallel.h下的文件并通过删除at::internal::lazy_init_num_threads()的内联实现来编辑它，只留下声明。 这将强制您的构建使用非内联导入。 问题很可能在于 VS C++ 处理具有静态变量的内联导出的方式——在本例thread_local变量。

完美的解决方案当然是与您的扩展一起重建 pytorch（就像任何 C++ DLL“导出”类的情况一样，而没有真正关心以安全的 ABI 兼容方式进行操作）但是修补包含文件在这里也可以，它只会阻止编译器内联 API 并创建对本地静态thread_local变量的引用，该引用会导致稍后的链接步骤失败。 然后，编译器将使用来自 pytorch DLL 的非内联变体，链接器错误将被消除。

Answer 2

名称 mangling 似乎指向at::internal::lazy_init_num_threads ，这是一个 PyTorch 函数（有点奇怪，但它很可能使用线程本地存储）。 您可能需要使用相同的工具链重建 PyTorch

Answer 3

Visual Studio 2022 版本 17.4.3 解决了这个问题。

• 修复了 thread_local 变量导致 linker 发出“致命错误 LNK1161：导出规范无效”错误的错误。

Answer 4

更新到 17.4 后，请在 Platform Toolset 中切换到 LLVM for Visual Studio 2022，以避免链接时出现此类错误。

PS：适用于 Visual Studio 2017、2019 和 2022 的 LLVM： Visual-Studio-llvm-utils