Open5

PyTorchのbuild_variables.bzlについて

nariaki3551nariaki3551

ちなみにlist(APPEND A B)はCMakeにおけるリスト操作で、Bに含まれるすべてのファイルを TORCH_SRCS に追加する

nariaki3551nariaki3551
  • libtorch_distributed_[base,extra]_sourcesのファイルはTORCH_SRCS変数に格納される
    • TORCH_SRCSはtorch_cpuライブラリのコンパイルに使用される
  • libtorch_cuda_distributed_[base,extra]_sourcesのファイルはCaffe2_GPU_SRCS変数に格納される
    • Caffe2_GPU_SRCSはtorch_cudaライブラリのコンパイルに使用される