Скомпилируйте PyTorch с дополнительными опциями компоновщика

Question

У меня есть измененная версия библиотеки gloo. Я могу компилировать и запускать программы, которые используют эту библиотеку (аналогично тому, что вы можете найти в gloo/gloo/examples).

Теперь я хочу построить Pytorch с моей библиотекой. Я заменил папку third_party/gloo в PyTorch своей версией gloo и пытаюсь ее скомпилировать.

Однако моя версия gloo требует некоторых дополнительных библиотек и специальных опций компоновщика. Где эти опции компоновщика должны быть добавлены в систему сборки pytorch?

Без этих параметров компоновщика моя компиляция останавливается с ошибкой компоновщика:

/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).