使用附加的链接器选项编译PyTorch

Question

我有一个改良版的环球图书馆。我能够编译和运行使用这个库的程序(类似于您在gloo/gloo/examples中可以找到的)。

现在，我想用我的图书馆造火把。我用我的gloo版本替换了PyTorch中的PyTorch文件夹，我正在尝试编译它。

但是，我的gloo版本需要一些额外的库和特殊的链接器选项。应在哪里添加这些链接器选项？

如果没有这些链接器选项，我的编译将停止，链接器错误如下：

/pytorch/build/lib/libcaffe2_gpu.so: undefined reference to <my code>
/pytorch/build/lib/libcaffe2.so: undefined reference to <my code>

Answer 1

应将附加的链接器选项添加到：

• 使用命令：Caffe2_DEPENDENCY_LIBS在pytorch/caffe2/CMakeLists.txt中的list(APPEND Caffe2_DEPENDENCY_LIBS <linker_options>)变量
• 使用命令：C10D_LIBS在pytorch/torch/lib/c10d/CMakeLists.txt中的list(APPEND C10D_LIBS <linker_options>)变量

附加库应该有独立位置的代码(必须使用-fPIC标志进行编译)。